envio 2.22.0 → 2.22.2

package/src/PgStorage.res.js

@@ -0,0 +1,125 @@
+// Generated by ReScript, PLEASE EDIT WITH CARE
+'use strict';
+
+var $$Array = require("rescript/lib/js/array.js");
+var Table = require("./db/Table.res.js");
+var Utils = require("./Utils.res.js");
+var Schema = require("./db/Schema.res.js");
+var Belt_Array = require("rescript/lib/js/belt_Array.js");
+
+function makeCreateIndexSqlUnsafe(tableName, indexFields, pgSchema) {
+  var indexName = tableName + "_" + indexFields.join("_");
+  var index = Belt_Array.map(indexFields, (function (idx) {
+            return "\"" + idx + "\"";
+          })).join(", ");
+  return "CREATE INDEX IF NOT EXISTS \"" + indexName + "\" ON \"" + pgSchema + "\".\"" + tableName + "\"(" + index + ");";
+}
+
+function makeCreateTableIndicesSqlUnsafe(table, pgSchema) {
+  var tableName = table.tableName;
+  var createIndex = function (indexField) {
+    return makeCreateIndexSqlUnsafe(tableName, [indexField], pgSchema);
+  };
+  var createCompositeIndex = function (indexFields) {
+    return makeCreateIndexSqlUnsafe(tableName, indexFields, pgSchema);
+  };
+  var singleIndices = Table.getSingleIndices(table);
+  var compositeIndices = Table.getCompositeIndices(table);
+  return Belt_Array.map(singleIndices, createIndex).join("\n") + Belt_Array.map(compositeIndices, createCompositeIndex).join("\n");
+}
+
+function makeCreateTableSqlUnsafe(table, pgSchema) {
+  var fieldsMapped = Belt_Array.map(Table.getFields(table), (function (field) {
+            var defaultValue = field.defaultValue;
+            var fieldType = field.fieldType;
+            var fieldName = Table.getDbFieldName(field);
+            var tmp;
+            tmp = fieldType === "TIMESTAMP" || fieldType === "TIMESTAMP WITH TIME ZONE" || fieldType === "JSONB" || fieldType === "SERIAL" || fieldType === "TEXT" || fieldType === "DOUBLE PRECISION" || fieldType === "NUMERIC" || fieldType === "BOOLEAN" || fieldType === "INTEGER" || fieldType === "TIMESTAMP WITH TIME ZONE NULL" || fieldType.startsWith("NUMERIC(") ? fieldType : "\"" + pgSchema + "\"." + fieldType;
+            return "\"" + fieldName + "\" " + tmp + (
+                    field.isArray ? "[]" : ""
+                  ) + (
+                    defaultValue !== undefined ? " DEFAULT " + defaultValue : (
+                        field.isNullable ? "" : " NOT NULL"
+                      )
+                  );
+          })).join(", ");
+  var primaryKeyFieldNames = Table.getPrimaryKeyFieldNames(table);
+  var primaryKey = Belt_Array.map(primaryKeyFieldNames, (function (field) {
+            return "\"" + field + "\"";
+          })).join(", ");
+  return "CREATE TABLE IF NOT EXISTS \"" + pgSchema + "\".\"" + table.tableName + "\"(" + fieldsMapped + (
+          primaryKeyFieldNames.length !== 0 ? ", PRIMARY KEY(" + primaryKey + ")" : ""
+        ) + ");";
+}
+
+function makeInitializeTransaction(pgSchema, pgUser, staticTables, entities, enums, cleanRun) {
+  var allTables = $$Array.copy(staticTables);
+  var allEntityTables = [];
+  entities.forEach(function (entity) {
+        allEntityTables.push(entity.table);
+        allTables.push(entity.table);
+        allTables.push(entity.entityHistory.table);
+      });
+  var derivedSchema = Schema.make(allEntityTables);
+  var query = {
+    contents: (
+      cleanRun ? "DROP SCHEMA IF EXISTS \"" + pgSchema + "\" CASCADE;\nCREATE SCHEMA \"" + pgSchema + "\";" : "CREATE SCHEMA IF NOT EXISTS \"" + pgSchema + "\";"
+    ) + ("GRANT ALL ON SCHEMA \"" + pgSchema + "\" TO " + pgUser + ";\nGRANT ALL ON SCHEMA \"" + pgSchema + "\" TO public;")
+  };
+  enums.forEach(function (enumConfig) {
+        var enumCreateQuery = "CREATE TYPE \"" + pgSchema + "\"." + enumConfig.name + " AS ENUM(" + enumConfig.variants.map(function (v) {
+                return "'" + v + "'";
+              }).join(", ") + ");";
+        query.contents = query.contents + "\n" + (
+          cleanRun ? enumCreateQuery : "IF NOT EXISTS (\n  SELECT 1 FROM pg_type \n  WHERE typname = '" + enumConfig.name.toLowerCase() + "' \n  AND typnamespace = (SELECT oid FROM pg_namespace WHERE nspname = '" + pgSchema + "')\n) THEN \n    " + enumCreateQuery + "\nEND IF;"
+        );
+      });
+  allTables.forEach(function (table) {
+        query.contents = query.contents + "\n" + makeCreateTableSqlUnsafe(table, pgSchema);
+      });
+  allTables.forEach(function (table) {
+        var indices = makeCreateTableIndicesSqlUnsafe(table, pgSchema);
+        if (indices !== "") {
+          query.contents = query.contents + "\n" + indices;
+          return ;
+        }
+        
+      });
+  var functionsQuery = {
+    contents: ""
+  };
+  entities.forEach(function (entity) {
+        functionsQuery.contents = functionsQuery.contents + "\n" + entity.entityHistory.createInsertFnQuery;
+        Table.getDerivedFromFields(entity.table).forEach(function (derivedFromField) {
+              var indexField = Utils.unwrapResultExn(Schema.getDerivedFromFieldName(derivedSchema, derivedFromField));
+              query.contents = query.contents + "\n" + makeCreateIndexSqlUnsafe(derivedFromField.derivedFromEntity, [indexField], pgSchema);
+            });
+      });
+  return [cleanRun ? query.contents : "DO $$ BEGIN " + query.contents + " END $$;"].concat(functionsQuery.contents !== "" ? [functionsQuery.contents] : []);
+}
+
+function make(sql, pgSchema, pgUser) {
+  var isInitialized = async function () {
+    var schemas = await sql.unsafe("SELECT schema_name FROM information_schema.schemata WHERE schema_name = '" + pgSchema + "';");
+    return Utils.$$Array.notEmpty(schemas);
+  };
+  var initialize = async function (entities, staticTables, enums, cleanRun) {
+    var queries = makeInitializeTransaction(pgSchema, pgUser, staticTables, entities, enums, cleanRun);
+    await sql.begin(function (sql) {
+          return queries.map(function (query) {
+                      return sql.unsafe(query);
+                    });
+        });
+  };
+  return {
+          isInitialized: isInitialized,
+          initialize: initialize
+        };
+}
+
+exports.makeCreateIndexSqlUnsafe = makeCreateIndexSqlUnsafe;
+exports.makeCreateTableIndicesSqlUnsafe = makeCreateTableIndicesSqlUnsafe;
+exports.makeCreateTableSqlUnsafe = makeCreateTableSqlUnsafe;
+exports.makeInitializeTransaction = makeInitializeTransaction;
+exports.make = make;
+/* Table Not a pure module */