pg-triggers 0.4.2 → 0.4.3

package/lib/pg-triggers.d.ts

@@ -1,5 +1,5 @@
 declare module "pg-triggers"{
 
-function table_name_max_id_trg(tableName:string, idName:string): Promise<string>;
+function dumpMaxIdTrigger(tableName:string, idName:string, opts:{firstId:number, grouping:string[]}): Promise<string>;
 
 }

package/lib/pg-triggers.js

@@ -5,9 +5,22 @@ var fs = require("fs/promises");
 
 var pgTriggers = {}
 
-pgTriggers.dumpMaxIdTrigger = async function dumpMaxIdTrigger (tableName, idName) {
+function q(str) {
+    return '"' + str.replace(/"/g, '""') + '"';
+}
+
+pgTriggers.dumpMaxIdTrigger = async function dumpMaxIdTrigger (tableName, idName, opts) {
+    opts = opts || {};
+    opts.firstId = opts.firstId || '1';
+    opts.grouping = opts.grouping || [];
     var content = await fs.readFile(__dirname + "/table-max-id-trg.sql", "utf8");
-    var sql = content.replace(/table_name/g, tableName).replace(/id_name/g, idName);
+    var sql = content
+        .replace(/table_name/g, tableName.replace(/"/g, '""')) 
+        .replace(/"id_name"/g, q(idName)) 
+        .replace(/v_first_id/g, opts.firstId)
+        .replace(/where true/g, opts.grouping.length ? `where ${
+            opts.grouping.map(g => `x.${q(g)} = new.${q(g)}`).join(' and ')
+        }` : '');
     return sql;
 };
 

package/lib/table-max-id-trg.sql

@@ -1,20 +1,21 @@
-create or replace function table_name_max_id_trg() returns trigger
+create or replace function "table_name_max_id_trg"() returns trigger
   language plpgsql
 as
 $body$
 declare
-  curr_max bigint;
+  v_curr_max bigint;
 begin
-  if new.id_name is null then
-    select max(id_name) into curr_max
-      from table_name;
-    new.id_name := coalesce(curr_max + 1, 1);
+  if new."id_name" is null then
+    select max("id_name") into v_curr_max
+      from "table_name" x
+      where true;
+    new."id_name" := coalesce(v_curr_max + 1, v_first_id);
   end if;
   return new;
 end;
 $body$;
 
-create trigger table_name_max_id_trg 
+create trigger "table_name_max_id_trg" 
   before insert
-  on table_name
-  for each row execute function table_name_max_id_trg();
+  on "table_name"
+  for each row execute function "table_name_max_id_trg"();

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "pg-triggers",
   "description": "Triggers for postgres",
-  "version": "0.4.2",
+  "version": "0.4.3",
   "author": "Emilio Platzer <emilioplatzer@gmail.com>",
   "repository": "emilioplatzer/pg-triggers",
   "license": "MIT",