upgrade script for bugfix for lp#809540
[Evergreen.git] / Open-ILS / src / sql / Pg / upgrade / XXXX.fix_maintain_901_quoting.sql
1 -- Evergreen DB patch XXXX.fix_maintain_901_quoting.sql
2 --
3 -- Fix for bug LP#809540 - fixes crash when inserting or updating
4 -- bib whose tcn_value contains regex metacharacters.
5 --
6 BEGIN;
7
8 -- check whether patch can be applied
9 SELECT evergreen.upgrade_deps_block_check('XXXX', :eg_version);
10
11 CREATE OR REPLACE FUNCTION evergreen.maintain_901 () RETURNS TRIGGER AS $func$
12 DECLARE
13     use_id_for_tcn BOOLEAN;
14 BEGIN
15     -- Remove any existing 901 fields before we insert the authoritative one
16     NEW.marc := REGEXP_REPLACE(NEW.marc, E'<datafield[^>]*?tag="901".+?</datafield>', '', 'g');
17
18     IF TG_TABLE_SCHEMA = 'biblio' THEN
19         -- Set TCN value to record ID?
20         SELECT enabled FROM config.global_flag INTO use_id_for_tcn
21             WHERE name = 'cat.bib.use_id_for_tcn';
22
23         IF use_id_for_tcn = 't' THEN
24             NEW.tcn_value := NEW.id;
25         END IF;
26
27         NEW.marc := REGEXP_REPLACE(
28             NEW.marc,
29             E'(</(?:[^:]*?:)?record>)',
30             E'<datafield tag="901" ind1=" " ind2=" ">' ||
31                 '<subfield code="a">' || REPLACE(evergreen.xml_escape(NEW.tcn_value), E'\\', E'\\\\') || E'</subfield>' ||
32                 '<subfield code="b">' || REPLACE(evergreen.xml_escape(NEW.tcn_source), E'\\', E'\\\\') || E'</subfield>' ||
33                 '<subfield code="c">' || NEW.id || E'</subfield>' ||
34                 '<subfield code="t">' || TG_TABLE_SCHEMA || E'</subfield>' ||
35                 CASE WHEN NEW.owner IS NOT NULL THEN '<subfield code="o">' || NEW.owner || E'</subfield>' ELSE '' END ||
36                 CASE WHEN NEW.share_depth IS NOT NULL THEN '<subfield code="d">' || NEW.share_depth || E'</subfield>' ELSE '' END ||
37              E'</datafield>\\1'
38         );
39     ELSIF TG_TABLE_SCHEMA = 'authority' THEN
40         NEW.marc := REGEXP_REPLACE(
41             NEW.marc,
42             E'(</(?:[^:]*?:)?record>)',
43             E'<datafield tag="901" ind1=" " ind2=" ">' ||
44                 '<subfield code="c">' || NEW.id || E'</subfield>' ||
45                 '<subfield code="t">' || TG_TABLE_SCHEMA || E'</subfield>' ||
46              E'</datafield>\\1'
47         );
48     ELSIF TG_TABLE_SCHEMA = 'serial' THEN
49         NEW.marc := REGEXP_REPLACE(
50             NEW.marc,
51             E'(</(?:[^:]*?:)?record>)',
52             E'<datafield tag="901" ind1=" " ind2=" ">' ||
53                 '<subfield code="c">' || NEW.id || E'</subfield>' ||
54                 '<subfield code="t">' || TG_TABLE_SCHEMA || E'</subfield>' ||
55                 '<subfield code="o">' || NEW.owning_lib || E'</subfield>' ||
56                 CASE WHEN NEW.record IS NOT NULL THEN '<subfield code="r">' || NEW.record || E'</subfield>' ELSE '' END ||
57              E'</datafield>\\1'
58         );
59     ELSE
60         NEW.marc := REGEXP_REPLACE(
61             NEW.marc,
62             E'(</(?:[^:]*?:)?record>)',
63             E'<datafield tag="901" ind1=" " ind2=" ">' ||
64                 '<subfield code="c">' || NEW.id || E'</subfield>' ||
65                 '<subfield code="t">' || TG_TABLE_SCHEMA || E'</subfield>' ||
66              E'</datafield>\\1'
67         );
68     END IF;
69
70     RETURN NEW;
71 END;
72 $func$ LANGUAGE PLPGSQL;
73
74 COMMIT;