prostgles-server 4.2.159 → 4.2.161

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (107) hide show
  1. package/dist/Auth/setEmailProvider.js +2 -2
  2. package/dist/Auth/setEmailProvider.js.map +1 -1
  3. package/lib/Auth/AuthHandler.ts +436 -0
  4. package/lib/Auth/AuthTypes.ts +280 -0
  5. package/lib/Auth/getSafeReturnURL.ts +35 -0
  6. package/lib/Auth/sendEmail.ts +83 -0
  7. package/lib/Auth/setAuthProviders.ts +128 -0
  8. package/lib/Auth/setEmailProvider.ts +85 -0
  9. package/lib/Auth/setupAuthRoutes.ts +161 -0
  10. package/lib/DBEventsManager.ts +178 -0
  11. package/lib/DBSchemaBuilder.ts +225 -0
  12. package/lib/DboBuilder/DboBuilder.ts +319 -0
  13. package/lib/DboBuilder/DboBuilderTypes.ts +361 -0
  14. package/lib/DboBuilder/QueryBuilder/Functions.ts +1153 -0
  15. package/lib/DboBuilder/QueryBuilder/QueryBuilder.ts +288 -0
  16. package/lib/DboBuilder/QueryBuilder/getJoinQuery.ts +263 -0
  17. package/lib/DboBuilder/QueryBuilder/getNewQuery.ts +271 -0
  18. package/lib/DboBuilder/QueryBuilder/getSelectQuery.ts +136 -0
  19. package/lib/DboBuilder/QueryBuilder/prepareHaving.ts +22 -0
  20. package/lib/DboBuilder/QueryStreamer.ts +250 -0
  21. package/lib/DboBuilder/TableHandler/DataValidator.ts +428 -0
  22. package/lib/DboBuilder/TableHandler/TableHandler.ts +205 -0
  23. package/lib/DboBuilder/TableHandler/delete.ts +115 -0
  24. package/lib/DboBuilder/TableHandler/insert.ts +183 -0
  25. package/lib/DboBuilder/TableHandler/insertTest.ts +78 -0
  26. package/lib/DboBuilder/TableHandler/onDeleteFromFileTable.ts +62 -0
  27. package/lib/DboBuilder/TableHandler/runInsertUpdateQuery.ts +134 -0
  28. package/lib/DboBuilder/TableHandler/update.ts +126 -0
  29. package/lib/DboBuilder/TableHandler/updateBatch.ts +49 -0
  30. package/lib/DboBuilder/TableHandler/updateFile.ts +48 -0
  31. package/lib/DboBuilder/TableHandler/upsert.ts +34 -0
  32. package/lib/DboBuilder/ViewHandler/ViewHandler.ts +393 -0
  33. package/lib/DboBuilder/ViewHandler/count.ts +38 -0
  34. package/lib/DboBuilder/ViewHandler/find.ts +153 -0
  35. package/lib/DboBuilder/ViewHandler/getExistsCondition.ts +73 -0
  36. package/lib/DboBuilder/ViewHandler/getExistsFilters.ts +74 -0
  37. package/lib/DboBuilder/ViewHandler/getInfo.ts +32 -0
  38. package/lib/DboBuilder/ViewHandler/getTableJoinQuery.ts +84 -0
  39. package/lib/DboBuilder/ViewHandler/parseComplexFilter.ts +96 -0
  40. package/lib/DboBuilder/ViewHandler/parseFieldFilter.ts +105 -0
  41. package/lib/DboBuilder/ViewHandler/parseJoinPath.ts +208 -0
  42. package/lib/DboBuilder/ViewHandler/prepareSortItems.ts +163 -0
  43. package/lib/DboBuilder/ViewHandler/prepareWhere.ts +90 -0
  44. package/lib/DboBuilder/ViewHandler/size.ts +37 -0
  45. package/lib/DboBuilder/ViewHandler/subscribe.ts +118 -0
  46. package/lib/DboBuilder/ViewHandler/validateViewRules.ts +70 -0
  47. package/lib/DboBuilder/dboBuilderUtils.ts +222 -0
  48. package/lib/DboBuilder/getColumns.ts +114 -0
  49. package/lib/DboBuilder/getCondition.ts +201 -0
  50. package/lib/DboBuilder/getSubscribeRelatedTables.ts +190 -0
  51. package/lib/DboBuilder/getTablesForSchemaPostgresSQL.ts +426 -0
  52. package/lib/DboBuilder/insertNestedRecords.ts +355 -0
  53. package/lib/DboBuilder/parseUpdateRules.ts +187 -0
  54. package/lib/DboBuilder/prepareShortestJoinPaths.ts +186 -0
  55. package/lib/DboBuilder/runSQL.ts +182 -0
  56. package/lib/DboBuilder/runTransaction.ts +50 -0
  57. package/lib/DboBuilder/sqlErrCodeToMsg.ts +254 -0
  58. package/lib/DboBuilder/uploadFile.ts +69 -0
  59. package/lib/Event_Trigger_Tags.ts +118 -0
  60. package/lib/FileManager/FileManager.ts +358 -0
  61. package/lib/FileManager/getValidatedFileType.ts +69 -0
  62. package/lib/FileManager/initFileManager.ts +187 -0
  63. package/lib/FileManager/upload.ts +62 -0
  64. package/lib/FileManager/uploadStream.ts +79 -0
  65. package/lib/Filtering.ts +463 -0
  66. package/lib/JSONBValidation/validate_jsonb_schema_sql.ts +502 -0
  67. package/lib/JSONBValidation/validation.ts +143 -0
  68. package/lib/Logging.ts +127 -0
  69. package/lib/PostgresNotifListenManager.ts +143 -0
  70. package/lib/Prostgles.ts +485 -0
  71. package/lib/ProstglesTypes.ts +196 -0
  72. package/lib/PubSubManager/PubSubManager.ts +609 -0
  73. package/lib/PubSubManager/addSub.ts +138 -0
  74. package/lib/PubSubManager/addSync.ts +141 -0
  75. package/lib/PubSubManager/getCreatePubSubManagerError.ts +72 -0
  76. package/lib/PubSubManager/getPubSubManagerInitQuery.ts +662 -0
  77. package/lib/PubSubManager/initPubSubManager.ts +79 -0
  78. package/lib/PubSubManager/notifListener.ts +173 -0
  79. package/lib/PubSubManager/orphanTriggerCheck.ts +70 -0
  80. package/lib/PubSubManager/pushSubData.ts +55 -0
  81. package/lib/PublishParser/PublishParser.ts +162 -0
  82. package/lib/PublishParser/getFileTableRules.ts +124 -0
  83. package/lib/PublishParser/getSchemaFromPublish.ts +141 -0
  84. package/lib/PublishParser/getTableRulesWithoutFileTable.ts +177 -0
  85. package/lib/PublishParser/publishTypesAndUtils.ts +399 -0
  86. package/lib/RestApi.ts +127 -0
  87. package/lib/SchemaWatch/SchemaWatch.ts +90 -0
  88. package/lib/SchemaWatch/createSchemaWatchEventTrigger.ts +3 -0
  89. package/lib/SchemaWatch/getValidatedWatchSchemaType.ts +45 -0
  90. package/lib/SchemaWatch/getWatchSchemaTagList.ts +27 -0
  91. package/lib/SyncReplication.ts +557 -0
  92. package/lib/TableConfig/TableConfig.ts +468 -0
  93. package/lib/TableConfig/getColumnDefinitionQuery.ts +111 -0
  94. package/lib/TableConfig/getConstraintDefinitionQueries.ts +95 -0
  95. package/lib/TableConfig/getFutureTableSchema.ts +64 -0
  96. package/lib/TableConfig/getPGIndexes.ts +53 -0
  97. package/lib/TableConfig/getTableColumnQueries.ts +129 -0
  98. package/lib/TableConfig/initTableConfig.ts +326 -0
  99. package/lib/index.ts +13 -0
  100. package/lib/initProstgles.ts +319 -0
  101. package/lib/onSocketConnected.ts +102 -0
  102. package/lib/runClientRequest.ts +129 -0
  103. package/lib/shortestPath.ts +122 -0
  104. package/lib/typeTests/DBoGenerated.d.ts +320 -0
  105. package/lib/typeTests/dboTypeCheck.ts +81 -0
  106. package/lib/utils.ts +15 -0
  107. package/package.json +1 -1
@@ -0,0 +1,662 @@
1
+
2
+ import { tryCatch } from "prostgles-types";
3
+ import { pgp } from "../DboBuilder/DboBuilderTypes";
4
+ import { asValue, NOTIF_CHANNEL, NOTIF_TYPE, PubSubManager } from "./PubSubManager";
5
+ const { version } = require("../../package.json");
6
+ import { getAppCheckQuery } from "./orphanTriggerCheck";
7
+ import { DboBuilder } from "../DboBuilder/DboBuilder";
8
+
9
+ export const DB_OBJ_NAMES = {
10
+ trigger_add_remove_func: "prostgles.trigger_add_remove_func",
11
+ data_watch_func: "prostgles.prostgles_trigger_function",
12
+ schema_watch_func: "prostgles.schema_watch_func",
13
+ schema_watch_trigger: "prostgles_schema_watch_trigger_new",
14
+ schema_watch_trigger_drop: "prostgles_schema_watch_trigger_new_drop"
15
+ } as const;
16
+
17
+ const PROSTGLES_SCHEMA_EXISTS_QUERY = `
18
+ SELECT 1
19
+ FROM information_schema.columns
20
+ WHERE table_schema = 'prostgles'
21
+ AND table_name = 'versions'
22
+ AND column_name = 'schema_md5'
23
+ `;
24
+ const PROSTGLES_SCHEMA_VERSION_OK_QUERY = `
25
+ SELECT 1
26
+ FROM prostgles.versions
27
+ WHERE (string_to_array(version, '.')::int[] > string_to_array(\${version}, '.')::int[])
28
+ OR (string_to_array(version, '.')::int[] = string_to_array(\${version}, '.')::int[])
29
+ AND schema_md5 = \${schema_md5}
30
+ `;
31
+
32
+ const getInitQuery = (debugMode: boolean | undefined, pgVersion: number) => {
33
+
34
+ const canReplaceTriggers = pgVersion >= 140006;
35
+ const createTriggerQuery = canReplaceTriggers ?
36
+ `CREATE OR REPLACE TRIGGER %1$I`:
37
+ `
38
+ DROP TRIGGER IF EXISTS %1$I ON %2$s;
39
+ CREATE TRIGGER %1$I
40
+ `;
41
+
42
+ return `
43
+ BEGIN; -- TRANSACTION ISOLATION LEVEL SERIALIZABLE;
44
+
45
+ --SET TRANSACTION ISOLATION LEVEL SERIALIZABLE;
46
+
47
+ /*
48
+ * ${PubSubManager.EXCLUDE_QUERY_FROM_SCHEMA_WATCH_ID}
49
+ */
50
+
51
+ DO
52
+ $do$
53
+ BEGIN
54
+
55
+ /* Drop older version. */
56
+ IF EXISTS (SELECT 1 FROM information_schema.schemata WHERE schema_name = 'prostgles') THEN
57
+
58
+ /* The seemingly useless IF nesting is done to prevent pg evaluating the entire condition and throw a 'schema_md5 column does not exist' */
59
+ IF
60
+ /* Backwards compatibility. Cannot check schema version */
61
+ NOT EXISTS(
62
+ ${PROSTGLES_SCHEMA_EXISTS_QUERY}
63
+ )
64
+ THEN
65
+ DROP SCHEMA IF EXISTS prostgles CASCADE;
66
+ ELSIF
67
+ /* There is no newer version or same same version but different schema */
68
+ NOT EXISTS (
69
+ ${PROSTGLES_SCHEMA_VERSION_OK_QUERY}
70
+ )
71
+ THEN
72
+ DROP SCHEMA IF EXISTS prostgles CASCADE;
73
+
74
+ END IF;
75
+
76
+ END IF;
77
+
78
+
79
+ IF NOT EXISTS (
80
+ SELECT 1
81
+ FROM information_schema.schemata
82
+ WHERE schema_name = 'prostgles'
83
+ )
84
+ THEN
85
+
86
+ CREATE SCHEMA IF NOT EXISTS prostgles;
87
+ COMMENT ON SCHEMA prostgles IS 'Used by prostgles-server to enable data/schema change tracking through subscribe/sync/watchSchema';
88
+
89
+ CREATE TABLE IF NOT EXISTS prostgles.versions(
90
+ version TEXT PRIMARY KEY,
91
+ schema_md5 TEXT NOT NULL,
92
+ added_at TIMESTAMP NOT NULL DEFAULT now()
93
+ );
94
+ COMMENT ON TABLE prostgles.versions IS 'Stores the prostgles schema creation query hash and package version number to identify when a newer schema needs to be re-created';
95
+
96
+ INSERT INTO prostgles.versions(version, schema_md5)
97
+ VALUES(${asValue(version)}, \${schema_md5})
98
+ ON CONFLICT DO NOTHING;
99
+
100
+
101
+ CREATE OR REPLACE FUNCTION prostgles.random_string(length INTEGER DEFAULT 33) RETURNS TEXT AS $$
102
+ DECLARE
103
+ chars TEXT[] := '{0,1,2,3,4,5,6,7,8,9,A,B,C,D,E,F,G,H,I,J,K,L,M,N,O,P,Q,R,S,T,U,V,W,X,Y,Z,a,b,c,d,e,f,g,h,i,j,k,l,m,n,o,p,q,r,s,t,u,v,w,x,y,z}';
104
+ result TEXT := '';
105
+ i INTEGER := 0;
106
+ BEGIN
107
+
108
+ IF length < 0 THEN
109
+ RAISE exception 'Given length cannot be less than 0';
110
+ END IF;
111
+
112
+ FOR i IN 1..length LOOP
113
+ result := result || chars[1+random()*(array_length(chars, 1)-1)];
114
+ END LOOP;
115
+
116
+ RETURN result;
117
+
118
+ END;
119
+ $$ language plpgsql;
120
+ COMMENT ON FUNCTION prostgles.random_string IS 'UUIDs without installing pgcrypto';
121
+
122
+
123
+ CREATE OR REPLACE FUNCTION prostgles.debug(VARIADIC args TEXT[]) RETURNS VOID AS $$
124
+ BEGIN
125
+
126
+ --PERFORM pg_notify('debug', concat_ws(' ', args));
127
+ IF
128
+ NOT EXISTS (
129
+ SELECT 1
130
+ FROM information_schema.tables
131
+ WHERE table_schema = 'prostgles'
132
+ AND table_name = 'debug'
133
+ )
134
+ THEN
135
+ CREATE TABLE IF NOT EXISTS prostgles.debug(m TEXT);
136
+ END IF;
137
+
138
+ INSERT INTO prostgles.debug(m) VALUES(concat_ws(' ', args));
139
+
140
+ END;
141
+ $$ LANGUAGE plpgsql;
142
+ COMMENT ON FUNCTION prostgles.debug IS 'Used for internal debugging';
143
+
144
+
145
+ CREATE OR REPLACE FUNCTION prostgles.user(key text default null)
146
+ RETURNS TEXT AS $$
147
+ DECLARE user_text text;
148
+ DECLARE user_jsonb JSONB = '{}'::JSONB;
149
+ BEGIN
150
+ user_text := current_setting('prostgles.user', true);
151
+ IF length(user_text) > 0 THEN
152
+ user_jsonb := user_text::JSONB;
153
+ END IF;
154
+
155
+ IF length(key) > 0 THEN
156
+ RETURN jsonb_extract_path(user_jsonb, key);
157
+ END IF;
158
+ RETURN user_jsonb;
159
+ END;
160
+ $$ LANGUAGE plpgsql;
161
+ COMMENT ON FUNCTION prostgles."user" IS 'Used for row level security';
162
+
163
+ CREATE OR REPLACE FUNCTION prostgles.user_id()
164
+ RETURNS UUID AS $$
165
+ BEGIN
166
+ RETURN prostgles.user('id')::UUID;
167
+ END;
168
+ $$ LANGUAGE plpgsql;
169
+ COMMENT ON FUNCTION prostgles.user_id IS 'Session user id';
170
+
171
+ CREATE OR REPLACE FUNCTION prostgles.user_type()
172
+ RETURNS TEXT AS $$
173
+ BEGIN
174
+ RETURN prostgles.user('type')::TEXT;
175
+ END;
176
+ $$ LANGUAGE plpgsql;
177
+ COMMENT ON FUNCTION prostgles.user_type IS 'Session user type';
178
+
179
+ CREATE TABLE IF NOT EXISTS prostgles.apps (
180
+ id TEXT PRIMARY KEY DEFAULT prostgles.random_string(),
181
+ added TIMESTAMP DEFAULT NOW(),
182
+ application_name TEXT,
183
+ watching_schema_tag_names _TEXT,
184
+ check_frequency_ms INTEGER NOT NULL
185
+ );
186
+ COMMENT ON TABLE prostgles.apps IS 'Keep track of prostgles server apps connected to db to combine common triggers. Heartbeat used due to no logout triggers in postgres';
187
+
188
+ CREATE TABLE IF NOT EXISTS prostgles.app_triggers (
189
+ app_id TEXT NOT NULL,
190
+ table_name TEXT NOT NULL,
191
+ condition TEXT NOT NULL,
192
+ condition_hash TEXT NOT NULL,
193
+
194
+ /* The view from the root subscription, found in the condition.
195
+ We need this because old_table/new_table data is not reflected in the view inside the AFTER trigger
196
+ */
197
+ related_view_name TEXT,
198
+ related_view_def TEXT, /* view definition */
199
+
200
+ inserted TIMESTAMP NOT NULL DEFAULT NOW(),
201
+ last_used TIMESTAMP NOT NULL DEFAULT NOW(),
202
+ PRIMARY KEY (app_id, table_name, condition_hash)
203
+ );
204
+ COMMENT ON TABLE prostgles.app_triggers IS 'Tables and conditions that are currently subscribed/synced';
205
+
206
+
207
+ CREATE OR REPLACE VIEW prostgles.v_triggers AS
208
+ SELECT *
209
+ , (ROW_NUMBER() OVER( ORDER BY table_name, condition ))::text AS id
210
+ , ROW_NUMBER() OVER(PARTITION BY app_id, table_name ORDER BY table_name, condition ) - 1 AS c_id
211
+ FROM prostgles.app_triggers;
212
+ COMMENT ON VIEW prostgles.v_triggers IS 'Augment trigger table with natural IDs and per app IDs';
213
+
214
+
215
+ CREATE OR REPLACE FUNCTION ${DB_OBJ_NAMES.data_watch_func}() RETURNS TRIGGER
216
+ AS $$
217
+
218
+ DECLARE t_ids TEXT[];
219
+ DECLARE c_ids INTEGER[];
220
+ DECLARE err_c_ids INTEGER[];
221
+ DECLARE unions TEXT := '';
222
+ DECLARE query TEXT := '';
223
+ DECLARE v_trigger RECORD;
224
+ DECLARE has_errors BOOLEAN := FALSE;
225
+
226
+ DECLARE err_text TEXT;
227
+ DECLARE err_detail TEXT;
228
+ DECLARE err_hint TEXT;
229
+
230
+ DECLARE view_def_query TEXT := '';
231
+
232
+ DECLARE escaped_table TEXT;
233
+
234
+ BEGIN
235
+
236
+ --PERFORM pg_notify('debug', concat_ws(' ', 'TABLE', TG_TABLE_NAME, TG_OP));
237
+
238
+ escaped_table := concat_ws('.', CASE WHEN TG_TABLE_SCHEMA <> CURRENT_SCHEMA THEN format('%I', TG_TABLE_SCHEMA) END, format('%I', TG_TABLE_NAME));
239
+
240
+ SELECT string_agg(
241
+ format(
242
+ $c$
243
+ SELECT CASE WHEN EXISTS(
244
+ SELECT 1
245
+ FROM %s
246
+ WHERE %s
247
+ ) THEN %s::text END AS t_ids
248
+ $c$,
249
+ table_name,
250
+ condition,
251
+ id
252
+ ),
253
+ E' UNION \n '
254
+ )
255
+ INTO unions
256
+ FROM prostgles.v_triggers
257
+ WHERE table_name = escaped_table;
258
+
259
+
260
+ /* unions = 'old_table union new_table' or any one of the tables */
261
+ IF unions IS NOT NULL THEN
262
+
263
+ SELECT
264
+ format(
265
+ E'WITH %I AS (\n %s \n) ',
266
+ TG_TABLE_NAME,
267
+ concat_ws(
268
+ E' UNION ALL \n ',
269
+ CASE WHEN (TG_OP = 'DELETE' OR TG_OP = 'UPDATE') THEN ' SELECT * FROM old_table ' END,
270
+ CASE WHEN (TG_OP = 'INSERT' OR TG_OP = 'UPDATE') THEN ' SELECT * FROM new_table ' END
271
+ )
272
+ )
273
+ ||
274
+ COALESCE((
275
+ SELECT ', ' || string_agg(format(E' %s AS ( \n %s \n ) ', related_view_name, related_view_def), ', ')
276
+ FROM (
277
+ SELECT DISTINCT related_view_name, related_view_def
278
+ FROM prostgles.v_triggers
279
+ WHERE table_name = escaped_table
280
+ AND related_view_name IS NOT NULL
281
+ AND related_view_def IS NOT NULL
282
+ ) t
283
+ ), '')
284
+ ||
285
+ format(
286
+ $c$
287
+ SELECT ARRAY_AGG(DISTINCT t.t_ids)
288
+ FROM (
289
+ %s
290
+ ) t
291
+ $c$,
292
+ unions
293
+ )
294
+ INTO query;
295
+
296
+ BEGIN
297
+ EXECUTE query INTO t_ids;
298
+
299
+ --RAISE NOTICE 'trigger fired ok';
300
+
301
+ EXCEPTION WHEN OTHERS THEN
302
+
303
+ has_errors := TRUE;
304
+
305
+ GET STACKED DIAGNOSTICS
306
+ err_text = MESSAGE_TEXT,
307
+ err_detail = PG_EXCEPTION_DETAIL,
308
+ err_hint = PG_EXCEPTION_HINT;
309
+
310
+ END;
311
+
312
+ --RAISE NOTICE 'has_errors: % ', has_errors;
313
+ --RAISE NOTICE 'unions: % , cids: %', unions, c_ids;
314
+
315
+ IF (t_ids IS NOT NULL OR has_errors) THEN
316
+
317
+ FOR v_trigger IN
318
+ SELECT app_id, string_agg(c_id::text, ',') as cids
319
+ FROM prostgles.v_triggers
320
+ WHERE id = ANY(t_ids)
321
+ OR has_errors
322
+ GROUP BY app_id
323
+ LOOP
324
+
325
+ PERFORM pg_notify(
326
+ ${asValue(NOTIF_CHANNEL.preffix)} || v_trigger.app_id ,
327
+ LEFT(concat_ws(
328
+ ${asValue(PubSubManager.DELIMITER)},
329
+
330
+ ${asValue(NOTIF_TYPE.data)},
331
+ COALESCE(escaped_table, 'MISSING'),
332
+ COALESCE(TG_OP, 'MISSING'),
333
+ CASE WHEN has_errors
334
+ THEN concat_ws('; ', 'error', err_text, err_detail, err_hint, 'query: ' || query )
335
+ ELSE COALESCE(v_trigger.cids, '')
336
+ END
337
+ ${debugMode? (", COALESCE(current_query(), 'current_query ??'), ' ', query") : ""}
338
+ ), 7999/4) -- Some chars are 2bytes -> 'Ω'
339
+ );
340
+ END LOOP;
341
+
342
+
343
+ IF has_errors THEN
344
+
345
+ DELETE FROM prostgles.app_triggers;
346
+ RAISE NOTICE 'trigger dropped due to exception: % % %', err_text, err_detail, err_hint;
347
+
348
+ END IF;
349
+
350
+ END IF;
351
+ END IF;
352
+
353
+ ${getAppCheckQuery()}
354
+
355
+ RETURN NULL;
356
+
357
+ END;
358
+
359
+ $$ LANGUAGE plpgsql;
360
+ COMMENT ON FUNCTION ${DB_OBJ_NAMES.data_watch_func} IS 'Prostgles internal function used to notify when data in the table changed';
361
+
362
+ CREATE OR REPLACE FUNCTION ${DB_OBJ_NAMES.trigger_add_remove_func}() RETURNS TRIGGER
363
+ AS $$
364
+
365
+ DECLARE operations TEXT[] := ARRAY['insert', 'update', 'delete'];
366
+ DECLARE op TEXT;
367
+ DECLARE query TEXT;
368
+ DECLARE trg_name TEXT;
369
+ DECLARE trw RECORD;
370
+ DECLARE app RECORD;
371
+ DECLARE start_time BIGINT;
372
+ DECLARE changed_triggers_count integer;
373
+
374
+ BEGIN
375
+
376
+ start_time := EXTRACT(EPOCH FROM now()) * 1000;
377
+
378
+ --RAISE NOTICE 'prostgles.app_triggers % ', TG_OP;
379
+
380
+ /* If no other listeners (app_triggers) left on table then DISABLE actual table data watch triggers */
381
+ IF TG_OP = 'DELETE' THEN
382
+
383
+ --RAISE NOTICE 'DELETE trigger_add_remove_func table: % ', ' ' || COALESCE((SELECT concat_ws(' ', string_agg(table_name, ' & '), count(*), min(inserted) ) FROM prostgles.app_triggers) , ' 0 ');
384
+ --RAISE NOTICE 'DELETE trigger_add_remove_func old_table: % ', '' || COALESCE((SELECT concat_ws(' ', string_agg(table_name, ' & '), count(*), min(inserted) ) FROM old_table), ' 0 ');
385
+
386
+ SELECT count(*)
387
+ FROM old_table
388
+ INTO changed_triggers_count;
389
+
390
+ /* Disable actual triggers if needed */
391
+ FOR trw IN
392
+ SELECT DISTINCT table_name
393
+ FROM old_table ot
394
+ WHERE NOT EXISTS (
395
+ SELECT 1
396
+ FROM prostgles.app_triggers t
397
+ WHERE t.table_name = ot.table_name
398
+ )
399
+ AND EXISTS (
400
+ SELECT trigger_name
401
+ FROM information_schema.triggers
402
+ WHERE trigger_name IN (
403
+ concat_ws('_', 'prostgles_triggers', table_name, 'insert'),
404
+ concat_ws('_', 'prostgles_triggers', table_name, 'update'),
405
+ concat_ws('_', 'prostgles_triggers', table_name, 'delete')
406
+ )
407
+ )
408
+ LOOP
409
+
410
+ FOREACH op IN ARRAY operations
411
+ LOOP
412
+ trg_name := concat_ws('_', 'prostgles_triggers', trw.table_name, op);
413
+
414
+ EXECUTE format(' ALTER TABLE %s DISABLE TRIGGER %I ;', trw.table_name, trg_name);
415
+ END LOOP;
416
+
417
+ END LOOP;
418
+
419
+ /* If newly added listeners on table then CREATE table data watch triggers */
420
+ ELSIF TG_OP = 'INSERT' THEN
421
+
422
+ SELECT count(*)
423
+ FROM new_table
424
+ INTO changed_triggers_count;
425
+
426
+ /* Loop through newly added tables to add data watch triggers */
427
+ FOR trw IN
428
+
429
+ SELECT DISTINCT table_name
430
+ FROM new_table nt
431
+
432
+ /* Table did not exist prior to this insert */
433
+ WHERE NOT EXISTS (
434
+ SELECT 1
435
+ FROM prostgles.app_triggers t
436
+ WHERE t.table_name = nt.table_name
437
+ AND t.inserted < nt.inserted -- exclude current record (this is an after trigger). Turn into before trigger?
438
+ )
439
+
440
+ /* Table is valid
441
+ AND EXISTS (
442
+ SELECT 1
443
+ FROM information_schema.tables
444
+ WHERE table_schema = 'public'
445
+ AND table_name = nt.table_name
446
+ )
447
+ */
448
+ LOOP
449
+
450
+ IF (
451
+ SELECT COUNT(*)
452
+ FROM information_schema.triggers
453
+ WHERE trigger_name IN (
454
+ 'prostgles_triggers_' || trw.table_name || '_insert',
455
+ 'prostgles_triggers_' || trw.table_name || '_update',
456
+ 'prostgles_triggers_' || trw.table_name || '_delete'
457
+ )
458
+ ) = 3
459
+ THEN
460
+ query := concat_ws(E'\n',
461
+ format(' ALTER TABLE %s ENABLE TRIGGER %I ;', trw.table_name, 'prostgles_triggers_' || trw.table_name || '_insert'),
462
+ format(' ALTER TABLE %s ENABLE TRIGGER %I ;', trw.table_name, 'prostgles_triggers_' || trw.table_name || '_update'),
463
+ format(' ALTER TABLE %s ENABLE TRIGGER %I ;', trw.table_name, 'prostgles_triggers_' || trw.table_name || '_delete')
464
+ );
465
+ ELSE
466
+
467
+ query := format(
468
+ $q$
469
+ ${createTriggerQuery}
470
+ AFTER INSERT ON %2$s
471
+ REFERENCING NEW TABLE AS new_table
472
+ FOR EACH STATEMENT EXECUTE PROCEDURE ${DB_OBJ_NAMES.data_watch_func}();
473
+ /* removed to allow less privileges for a user to create subscriptions
474
+ COMMENT ON TRIGGER %1$I ON %2$s IS 'Prostgles internal trigger used to notify when data in the table changed';
475
+ */
476
+ $q$,
477
+ 'prostgles_triggers_' || trw.table_name || '_insert', trw.table_name
478
+ ) || format(
479
+ $q$
480
+ ${createTriggerQuery}
481
+ AFTER UPDATE ON %2$s
482
+ REFERENCING OLD TABLE AS old_table NEW TABLE AS new_table
483
+ FOR EACH STATEMENT EXECUTE PROCEDURE ${DB_OBJ_NAMES.data_watch_func}();
484
+ --COMMENT ON TRIGGER %1$I ON %2$s IS 'Prostgles internal trigger used to notify when data in the table changed';
485
+ $q$,
486
+ 'prostgles_triggers_' || trw.table_name || '_update', trw.table_name
487
+ ) || format(
488
+ $q$
489
+ ${createTriggerQuery}
490
+ AFTER DELETE ON %2$s
491
+ REFERENCING OLD TABLE AS old_table
492
+ FOR EACH STATEMENT EXECUTE PROCEDURE ${DB_OBJ_NAMES.data_watch_func}();
493
+ --COMMENT ON TRIGGER %1$I ON %2$s IS 'Prostgles internal trigger used to notify when data in the table changed';
494
+ $q$,
495
+ 'prostgles_triggers_' || trw.table_name || '_delete', trw.table_name
496
+ );
497
+ END IF;
498
+
499
+
500
+ --RAISE NOTICE ' % ', query;
501
+
502
+
503
+ query := format(
504
+ $q$
505
+ DO $e$
506
+ BEGIN
507
+ /* ${ PubSubManager.EXCLUDE_QUERY_FROM_SCHEMA_WATCH_ID} */
508
+ %s
509
+
510
+ END $e$;
511
+ $q$,
512
+ query
513
+ ) ;
514
+
515
+
516
+ EXECUTE query;
517
+
518
+ END LOOP;
519
+
520
+ END IF;
521
+
522
+ /** Notify all apps about trigger table change */
523
+ IF changed_triggers_count > 0 THEN
524
+ FOR app IN
525
+ SELECT * FROM prostgles.apps
526
+ LOOP
527
+ PERFORM pg_notify(
528
+ ${asValue(NOTIF_CHANNEL.preffix)} || app.id,
529
+ LEFT(concat_ws(
530
+ ${asValue(PubSubManager.DELIMITER)},
531
+ ${asValue(NOTIF_TYPE.data_trigger_change)},
532
+ json_build_object(
533
+ 'TG_OP', TG_OP,
534
+ 'duration', (EXTRACT(EPOCH FROM now()) * 1000) - start_time,
535
+ 'query', ${debugMode? 'LEFT(current_query(), 400)' : "'Only shown in debug mode'"}
536
+ )
537
+ )::TEXT, 7999/4)
538
+ );
539
+ END LOOP;
540
+ END IF;
541
+
542
+ RETURN NULL;
543
+ END;
544
+
545
+ $$ LANGUAGE plpgsql;
546
+ COMMENT ON FUNCTION ${DB_OBJ_NAMES.trigger_add_remove_func} IS 'Used to add/remove table watch triggers concurrently ';
547
+
548
+ DROP TRIGGER IF EXISTS prostgles_triggers_insert ON prostgles.app_triggers;
549
+ CREATE TRIGGER prostgles_triggers_insert
550
+ AFTER INSERT ON prostgles.app_triggers
551
+ REFERENCING NEW TABLE AS new_table
552
+ FOR EACH STATEMENT EXECUTE PROCEDURE ${DB_OBJ_NAMES.trigger_add_remove_func}();
553
+
554
+ DROP TRIGGER IF EXISTS prostgles_triggers_delete ON prostgles.app_triggers;
555
+ CREATE TRIGGER prostgles_triggers_delete
556
+ AFTER DELETE ON prostgles.app_triggers
557
+ REFERENCING OLD TABLE AS old_table
558
+ FOR EACH STATEMENT EXECUTE PROCEDURE ${DB_OBJ_NAMES.trigger_add_remove_func}();
559
+
560
+
561
+ CREATE OR REPLACE FUNCTION ${DB_OBJ_NAMES.schema_watch_func}() RETURNS event_trigger AS $$
562
+
563
+ DECLARE curr_query TEXT := '';
564
+ DECLARE app RECORD;
565
+ DECLARE objects_changed BOOLEAN := false;
566
+
567
+ BEGIN
568
+
569
+ IF TG_event = 'ddl_command_end' THEN
570
+ objects_changed := EXISTS (
571
+ SELECT *
572
+ FROM pg_event_trigger_ddl_commands()
573
+ );
574
+ END IF;
575
+ IF TG_event = 'sql_drop' THEN
576
+ objects_changed := EXISTS (
577
+ SELECT *
578
+ FROM pg_event_trigger_dropped_objects()
579
+ );
580
+ END IF;
581
+
582
+ /*
583
+ This event trigger will outlive a prostgles app instance.
584
+ Must ensure it only fires if an app instance is running
585
+ */
586
+ IF
587
+ objects_changed
588
+ AND EXISTS (
589
+ SELECT 1
590
+ FROM information_schema.tables
591
+ WHERE table_schema = 'prostgles'
592
+ AND table_name = 'apps'
593
+ )
594
+ THEN
595
+
596
+ SELECT LEFT(COALESCE(current_query(), ''), 5000)
597
+ INTO curr_query;
598
+
599
+ FOR app IN
600
+ SELECT *
601
+ FROM prostgles.apps
602
+ WHERE tg_tag = ANY(watching_schema_tag_names)
603
+ AND curr_query NOT ILIKE '%${PubSubManager.EXCLUDE_QUERY_FROM_SCHEMA_WATCH_ID}%'
604
+ LOOP
605
+ PERFORM pg_notify(
606
+ ${asValue(NOTIF_CHANNEL.preffix)} || app.id,
607
+ LEFT(concat_ws(
608
+ ${asValue(PubSubManager.DELIMITER)},
609
+ ${asValue(NOTIF_TYPE.schema)},
610
+ tg_tag ,
611
+ TG_event,
612
+ ${debugMode? 'curr_query' : "'Only shown in debug mode'"}
613
+ ), 7999/4)
614
+ );
615
+ END LOOP;
616
+
617
+ ${getAppCheckQuery()}
618
+
619
+ END IF;
620
+
621
+ END;
622
+ $$ LANGUAGE plpgsql;
623
+ COMMENT ON FUNCTION ${DB_OBJ_NAMES.schema_watch_func} IS 'Prostgles internal function used to notify when schema has changed';
624
+
625
+ END IF;
626
+
627
+ END
628
+ $do$;
629
+
630
+ COMMIT;
631
+ `;
632
+
633
+ }
634
+
635
+ /**
636
+ * Initialize the prostgles schema and functions needed for realtime data and schema changes
637
+ * undefined returned if the database contains the apropriate prostgles schema
638
+ */
639
+ export const getPubSubManagerInitQuery = async function(this: DboBuilder): Promise<string | undefined> {
640
+
641
+ const versionNum = await this.db.one("SELECT current_setting('server_version_num')::int as val");
642
+ const initQuery = getInitQuery(this.prostgles.opts.DEBUG_MODE, versionNum.val);
643
+ const { schema_md5 = "none" } = await this.db.oneOrNone("SELECT md5($1) as schema_md5", [initQuery.trim()]);
644
+ const query = pgp.as.format(initQuery, { schema_md5, version });
645
+ const existingSchema = await this.db.any(PROSTGLES_SCHEMA_EXISTS_QUERY);
646
+ if(!existingSchema.length){
647
+ console.log("getPubSubManagerInitQuery: No prostgles.versions table found. Creating...");
648
+ return query;
649
+ }
650
+ const { existingSchemaVersions } = await tryCatch(async () => {
651
+ const existingSchemaVersions = await this.db.any(PROSTGLES_SCHEMA_VERSION_OK_QUERY, { schema_md5, version });
652
+ return {
653
+ existingSchemaVersions
654
+ }
655
+ });
656
+ if(!existingSchemaVersions?.length){
657
+ console.log("getPubSubManagerInitQuery: Outdated prostgles schema. Re-creating...");
658
+ return query;
659
+ }
660
+
661
+ return undefined;
662
+ }