Open-ILS/src/sql/Pg/upgrade/0576.fix_maintain_901_quoting.sql

   1 -- Evergreen DB patch 0576.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('0576', :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;