drizzle-kit 0.24.0 → 0.24.1-3d751e8

package/api.mjs

@@ -7469,11 +7469,30 @@ var init_mysqlSchema = __esm({
   }
 });
 
+// src/extensions/vector.ts
+var vectorOps;
+var init_vector = __esm({
+  "src/extensions/vector.ts"() {
+    "use strict";
+    vectorOps = [
+      "vector_l2_ops",
+      "vector_ip_ops",
+      "vector_cosine_ops",
+      "vector_l1_ops",
+      "bit_hamming_ops",
+      "bit_jaccard_ops",
+      "halfvec_l2_ops",
+      "sparsevec_l2_ops"
+    ];
+  }
+});
+
 // src/serializer/pgSchema.ts
 var indexV2, columnV2, tableV2, enumSchemaV1, enumSchema, pgSchemaV2, references, columnV1, tableV1, pgSchemaV1, indexColumn, index2, indexV4, indexV5, indexV6, fk2, sequenceSchema, sequenceSquashed, columnV7, column2, columnSquashed, tableV32, compositePK2, uniqueConstraint2, tableV42, tableV5, tableV6, tableV7, table2, schemaHash2, kitInternals2, pgSchemaInternalV3, pgSchemaInternalV4, pgSchemaInternalV5, pgSchemaInternalV6, pgSchemaExternal, pgSchemaInternalV7, pgSchemaInternal, tableSquashed2, tableSquashedV42, pgSchemaSquashedV4, pgSchemaSquashedV6, pgSchemaSquashed, pgSchemaV3, pgSchemaV4, pgSchemaV5, pgSchemaV6, pgSchemaV7, pgSchema, backwardCompatiblePgSchema, PgSquasher, squashPgScheme, dryPg;
 var init_pgSchema = __esm({
   "src/serializer/pgSchema.ts"() {
     "use strict";
+    init_vector();
     init_global();
     init_lib();
     indexV2 = objectType({
@@ -7720,7 +7739,8 @@ var init_pgSchema = __esm({
             objectType({
               isArray: booleanType().optional(),
               dimensions: numberType().optional(),
-              rawType: stringType().optional()
+              rawType: stringType().optional(),
+              isDefaultAnExpression: booleanType().optional()
             }).optional()
           )
         }).optional()
@@ -7858,7 +7878,7 @@ var init_pgSchema = __esm({
       squashIdx: (idx) => {
         index2.parse(idx);
         return `${idx.name};${idx.columns.map(
-          (c) => `${c.expression}--${c.isExpression}--${c.asc}--${c.nulls}--${c.opclass}`
+          (c) => `${c.expression}--${c.isExpression}--${c.asc}--${c.nulls}--${c.opclass && vectorOps.includes(c.opclass) ? c.opclass : ""}`
         ).join(",,")};${idx.isUnique};${idx.concurrently};${idx.method};${idx.where};${JSON.stringify(idx.with)}`;
       },
       unsquashIdx: (input) => {
@@ -11507,7 +11527,7 @@ var init_sqlgenerator = __esm({
     isPgNativeType = (it) => {
       if (pgNativeTypes.has(it)) return true;
       const toCheck = it.replace(/ /g, "");
-      return toCheck.startsWith("varchar(") || toCheck.startsWith("char(") || toCheck.startsWith("numeric(") || toCheck.startsWith("timestamp(") || toCheck.startsWith("intervalyear(") || toCheck.startsWith("intervalmonth(") || toCheck.startsWith("intervalday(") || toCheck.startsWith("intervalhour(") || toCheck.startsWith("intervalminute(") || toCheck.startsWith("intervalsecond(") || toCheck.startsWith("intervalyeartomonth(") || toCheck.startsWith("intervaldaytohour(") || toCheck.startsWith("intervaldaytominute(") || toCheck.startsWith("intervaldaytosecond(") || toCheck.startsWith("intervalhourtominute(") || toCheck.startsWith("intervalhourtosecond(") || toCheck.startsWith("intervalminutetosecond(") || toCheck.startsWith("vector(") || toCheck.startsWith("geometry(") || /^(\w+)(\[\d*])+$/.test(it);
+      return toCheck.startsWith("varchar(") || toCheck.startsWith("char(") || toCheck.startsWith("numeric(") || toCheck.startsWith("timestamp(") || toCheck.startsWith("doubleprecision[") || toCheck.startsWith("intervalyear(") || toCheck.startsWith("intervalmonth(") || toCheck.startsWith("intervalday(") || toCheck.startsWith("intervalhour(") || toCheck.startsWith("intervalminute(") || toCheck.startsWith("intervalsecond(") || toCheck.startsWith("intervalyeartomonth(") || toCheck.startsWith("intervaldaytohour(") || toCheck.startsWith("intervaldaytominute(") || toCheck.startsWith("intervaldaytosecond(") || toCheck.startsWith("intervalhourtominute(") || toCheck.startsWith("intervalhourtosecond(") || toCheck.startsWith("intervalminutetosecond(") || toCheck.startsWith("vector(") || toCheck.startsWith("geometry(") || /^(\w+)(\[\d*])+$/.test(it);
     };
     Convertor = class {
     };
@@ -15850,7 +15870,7 @@ var init_schemaValidator = __esm({
 });
 
 // src/cli/validations/common.ts
-var sqliteDriversLiterals, prefixes, prefix, sqliteDriver, postgresDriver, driver, configMigrations, configCommonSchema, casing, introspectParams, configIntrospectCliSchema, configGenerateSchema, configPushSchema;
+var sqliteDriversLiterals, postgresqlDriversLiterals, prefixes, prefix, sqliteDriver, postgresDriver, driver, configMigrations, configCommonSchema, casing, introspectParams, configIntrospectCliSchema, configGenerateSchema, configPushSchema;
 var init_common = __esm({
   "src/cli/validations/common.ts"() {
     "use strict";
@@ -15862,6 +15882,10 @@ var init_common = __esm({
       literalType("d1-http"),
       literalType("expo")
     ];
+    postgresqlDriversLiterals = [
+      literalType("aws-data-api"),
+      literalType("pglite")
+    ];
     prefixes = [
       "index",
       "timestamp",
@@ -15874,7 +15898,7 @@ var init_common = __esm({
       const _2 = "";
     }
     sqliteDriver = unionType(sqliteDriversLiterals);
-    postgresDriver = literalType("aws-data-api");
+    postgresDriver = unionType(postgresqlDriversLiterals);
     driver = unionType([sqliteDriver, postgresDriver]);
     configMigrations = objectType({
       table: stringType().optional(),
@@ -19378,7 +19402,7 @@ function jaccardDistance(column4, value) {
   }
   return sql`${column4} <%> ${value}`;
 }
-var init_vector = __esm({
+var init_vector2 = __esm({
   "../drizzle-orm/dist/sql/functions/vector.js"() {
     "use strict";
     init_sql();
@@ -19390,7 +19414,7 @@ var init_functions = __esm({
   "../drizzle-orm/dist/sql/functions/index.js"() {
     "use strict";
     init_aggregate();
-    init_vector();
+    init_vector2();
   }
 });
 
@@ -21054,7 +21078,7 @@ var init_sparsevec = __esm({
 
 // ../drizzle-orm/dist/pg-core/columns/vector_extension/vector.js
 var _a122, _b95, PgVectorBuilder, _a123, _b96, PgVector;
-var init_vector2 = __esm({
+var init_vector3 = __esm({
   "../drizzle-orm/dist/pg-core/columns/vector_extension/vector.js"() {
     "use strict";
     init_entity();
@@ -21126,7 +21150,7 @@ var init_columns = __esm({
     init_bit();
     init_halfvec();
     init_sparsevec();
-    init_vector2();
+    init_vector3();
   }
 });
 
@@ -24624,24 +24648,6 @@ var init_pg_core = __esm({
   }
 });
 
-// src/extensions/vector.ts
-var vectorOps;
-var init_vector3 = __esm({
-  "src/extensions/vector.ts"() {
-    "use strict";
-    vectorOps = [
-      "vector_l2_ops",
-      "vector_ip_ops",
-      "vector_cosine_ops",
-      "vector_l1_ops",
-      "bit_hamming_ops",
-      "bit_jaccard_ops",
-      "halfvec_l2_ops",
-      "sparsevec_l2_ops"
-    ];
-  }
-});
-
 // src/serializer/pgSerializer.ts
 function stringFromIdentityProperty(field) {
   return typeof field === "string" ? field : typeof field === "undefined" ? void 0 : String(field);
@@ -24679,7 +24685,7 @@ function buildArrayString(array, sqlType) {
   }).join(",");
   return `{${values}}`;
 }
-var dialect4, indexName, generatePgSnapshot, trimChar, fromDatabase, columnToDefault, defaultForColumn;
+var dialect4, indexName, generatePgSnapshot, trimChar, fromDatabase, defaultForColumn;
 var init_pgSerializer = __esm({
   "src/serializer/pgSerializer.ts"() {
     "use strict";
@@ -24687,7 +24693,7 @@ var init_pgSerializer = __esm({
     init_dist();
     init_pg_core();
     init_pg_core();
-    init_vector3();
+    init_vector();
     init_outputs();
     init_utils();
     init_serializer();
@@ -24800,7 +24806,7 @@ ${withStyle.errorWarning(`We've found duplicated unique constraint names in ${so
                   columnToSet.default = `'${buildArrayString(
                     column4.default,
                     sqlTypeLowered
-                  )}'::${sqlTypeLowered}`;
+                  )}'`;
                 } else {
                   columnToSet.default = column4.default;
                 }
@@ -25147,7 +25153,7 @@ ${withStyle.errorWarning(
                    WHEN 'int2'::regtype THEN 'smallserial'
                 END
            ELSE format_type(a.atttypid, a.atttypmod)
-           END AS data_type, INFORMATION_SCHEMA.COLUMNS.table_name, 
+           END AS data_type, INFORMATION_SCHEMA.COLUMNS.table_name, ns.nspname as type_schema,
            pg_get_serial_sequence('"${tableSchema}"."${tableName}"', a.attname)::regclass as seq_name, INFORMATION_SCHEMA.COLUMNS.column_name, 
            INFORMATION_SCHEMA.COLUMNS.column_default, INFORMATION_SCHEMA.COLUMNS.data_type as additional_dt, 
            INFORMATION_SCHEMA.COLUMNS.udt_name as enum_name,
@@ -25158,6 +25164,7 @@ ${withStyle.errorWarning(
            INFORMATION_SCHEMA.COLUMNS.identity_cycle
    FROM  pg_attribute  a
    JOIN INFORMATION_SCHEMA.COLUMNS ON INFORMATION_SCHEMA.COLUMNS.column_name = a.attname
+   JOIN pg_type t ON t.oid = a.atttypid LEFT JOIN pg_namespace ns ON ns.oid = t.typnamespace
    WHERE  a.attrelid = '"${tableSchema}"."${tableName}"'::regclass and INFORMATION_SCHEMA.COLUMNS.table_name = '${tableName}' and INFORMATION_SCHEMA.COLUMNS.table_schema = '${tableSchema}'
    AND    a.attnum > 0
    AND    NOT a.attisdropped
@@ -25177,30 +25184,42 @@ ${withStyle.errorWarning(
             }
             const tableForeignKeys = await db.query(
               `SELECT
-          tc.table_schema,
-          tc.constraint_name,
-          tc.table_name,
-          kcu.column_name,
-          (
-              SELECT ccu.table_schema
-              FROM information_schema.constraint_column_usage ccu
-              WHERE ccu.constraint_name = tc.constraint_name
-              LIMIT 1
-          ) AS foreign_table_schema,
-          ccu.table_name AS foreign_table_name,
-          ccu.column_name AS foreign_column_name,
-          rc.delete_rule, 
-          rc.update_rule
-      FROM
-          information_schema.table_constraints AS tc
-          JOIN information_schema.key_column_usage AS kcu
-            ON tc.constraint_name = kcu.constraint_name
-            AND tc.table_schema = kcu.table_schema
-          JOIN information_schema.constraint_column_usage AS ccu
-            ON ccu.constraint_name = tc.constraint_name
-          JOIN information_schema.referential_constraints AS rc
-            ON ccu.constraint_name = rc.constraint_name
-    WHERE tc.constraint_type = 'FOREIGN KEY' AND tc.table_name='${tableName}' and tc.table_schema='${tableSchema}';`
+            con.contype AS constraint_type,
+            nsp.nspname AS constraint_schema,
+            con.conname AS constraint_name,
+            rel.relname AS table_name,
+            att.attname AS column_name,
+            fnsp.nspname AS foreign_table_schema,
+            frel.relname AS foreign_table_name,
+            fatt.attname AS foreign_column_name,
+            CASE con.confupdtype
+              WHEN 'a' THEN 'NO ACTION'
+              WHEN 'r' THEN 'RESTRICT'
+              WHEN 'n' THEN 'SET NULL'
+              WHEN 'c' THEN 'CASCADE'
+              WHEN 'd' THEN 'SET DEFAULT'
+            END AS update_rule,
+            CASE con.confdeltype
+              WHEN 'a' THEN 'NO ACTION'
+              WHEN 'r' THEN 'RESTRICT'
+              WHEN 'n' THEN 'SET NULL'
+              WHEN 'c' THEN 'CASCADE'
+              WHEN 'd' THEN 'SET DEFAULT'
+            END AS delete_rule
+          FROM
+            pg_catalog.pg_constraint con
+            JOIN pg_catalog.pg_class rel ON rel.oid = con.conrelid
+            JOIN pg_catalog.pg_namespace nsp ON nsp.oid = con.connamespace
+            LEFT JOIN pg_catalog.pg_attribute att ON att.attnum = ANY (con.conkey)
+              AND att.attrelid = con.conrelid
+            LEFT JOIN pg_catalog.pg_class frel ON frel.oid = con.confrelid
+            LEFT JOIN pg_catalog.pg_namespace fnsp ON fnsp.oid = frel.relnamespace
+            LEFT JOIN pg_catalog.pg_attribute fatt ON fatt.attnum = ANY (con.confkey)
+              AND fatt.attrelid = con.confrelid
+          WHERE
+            nsp.nspname = '${tableSchema}'
+            AND rel.relname = '${tableName}'
+            AND con.contype IN ('f');`
             );
             foreignKeysCount += tableForeignKeys.length;
             if (progressCallback) {
@@ -25212,8 +25231,8 @@ ${withStyle.errorWarning(
               const columnTo = fk4.foreign_column_name;
               const schemaTo = fk4.foreign_table_schema;
               const foreignKeyName = fk4.constraint_name;
-              const onUpdate = fk4.update_rule.toLowerCase();
-              const onDelete = fk4.delete_rule.toLowerCase();
+              const onUpdate = fk4.update_rule?.toLowerCase();
+              const onDelete = fk4.delete_rule?.toLowerCase();
               if (typeof foreignKeysToReturn[foreignKeyName] !== "undefined") {
                 foreignKeysToReturn[foreignKeyName].columnsFrom.push(columnFrom);
                 foreignKeysToReturn[foreignKeyName].columnsTo.push(columnTo);
@@ -25258,6 +25277,8 @@ ${withStyle.errorWarning(
               const columnDimensions = columnResponse.array_dimensions;
               const enumType2 = columnResponse.enum_name;
               let columnType = columnResponse.data_type;
+              const typeSchema = columnResponse.type_schema;
+              const defaultValueRes = columnResponse.column_default;
               const isGenerated = columnResponse.is_generated === "ALWAYS";
               const generationExpression = columnResponse.generation_expression;
               const isIdentity = columnResponse.is_identity === "YES";
@@ -25288,12 +25309,7 @@ ${withStyle.errorWarning(
                   columns: cprimaryKey.map((c) => c.column_name)
                 };
               }
-              const defaultValue = defaultForColumn(columnResponse);
-              const isSerial = columnType === "serial";
               let columnTypeMapped = columnType;
-              if (columnTypeMapped.startsWith("numeric(")) {
-                columnTypeMapped = columnTypeMapped.replace(",", ", ");
-              }
               if (columnAdditionalDT === "ARRAY") {
                 if (typeof internals.tables[tableName] === "undefined") {
                   internals.tables[tableName] = {
@@ -25321,6 +25337,34 @@ ${withStyle.errorWarning(
                   }
                 }
               }
+              const defaultValue = defaultForColumn(
+                columnResponse,
+                internals,
+                tableName
+              );
+              if (defaultValue === "NULL" || defaultValueRes && defaultValueRes.startsWith("(") && defaultValueRes.endsWith(")")) {
+                if (typeof internals.tables[tableName] === "undefined") {
+                  internals.tables[tableName] = {
+                    columns: {
+                      [columnName]: {
+                        isDefaultAnExpression: true
+                      }
+                    }
+                  };
+                } else {
+                  if (typeof internals.tables[tableName].columns[columnName] === "undefined") {
+                    internals.tables[tableName].columns[columnName] = {
+                      isDefaultAnExpression: true
+                    };
+                  } else {
+                    internals.tables[tableName].columns[columnName].isDefaultAnExpression = true;
+                  }
+                }
+              }
+              const isSerial = columnType === "serial";
+              if (columnTypeMapped.startsWith("numeric(")) {
+                columnTypeMapped = columnTypeMapped.replace(",", ", ");
+              }
               if (columnAdditionalDT === "ARRAY") {
                 for (let i = 1; i < Number(columnDimensions); i++) {
                   columnTypeMapped += "[]";
@@ -25334,7 +25378,7 @@ ${withStyle.errorWarning(
                   // filter vectors, but in future we should filter any extension that was installed by user
                   columnAdditionalDT === "USER-DEFINED" && !["vector", "geometry"].includes(enumType2) ? enumType2 : columnTypeMapped
                 ),
-                typeSchema: enumsToReturn[`${tableSchema}.${enumType2}`] !== void 0 ? enumsToReturn[`${tableSchema}.${enumType2}`].schema : void 0,
+                typeSchema: enumsToReturn[`${typeSchema}.${enumType2}`] !== void 0 ? enumsToReturn[`${typeSchema}.${enumType2}`].schema : void 0,
                 primaryKey: primaryKey.length === 1 && cprimaryKey.length < 2,
                 // default: isSerial ? undefined : defaultValue,
                 notNull: columnResponse.is_nullable === "NO",
@@ -25352,7 +25396,7 @@ ${withStyle.errorWarning(
                 } : void 0
               };
               if (identityName) {
-                delete sequencesToReturn[`${tableSchema}.${identityName}`];
+                delete sequencesToReturn[`${tableSchema}.${identityName.startsWith('"') && identityName.endsWith('"') ? identityName.slice(1, -1) : identityName}`];
                 delete sequencesToReturn[identityName];
               }
               if (!isSerial && typeof defaultValue !== "undefined") {
@@ -25505,59 +25549,71 @@ ${withStyle.errorWarning(
         internal: internals
       };
     };
-    columnToDefault = {
-      "numeric(": "::numeric",
-      // text: "::text",
-      // "character varying": "::character varying",
-      // "double precision": "::double precision",
-      // "time with time zone": "::time with time zone",
-      "time without time zone": "::time without time zone",
-      // "timestamp with time zone": "::timestamp with time zone",
-      "timestamp without time zone": "::timestamp without time zone",
-      "timestamp(": "::timestamp without time zone",
-      // date: "::date",
-      // interval: "::interval",
-      // character: "::bpchar",
-      // macaddr8: "::macaddr8",
-      // macaddr: "::macaddr",
-      // inet: "::inet",
-      // cidr: "::cidr",
-      // jsonb: "::jsonb",
-      // json: "::json",
-      "character(": "::bpchar"
-    };
-    defaultForColumn = (column4) => {
+    defaultForColumn = (column4, internals, tableName) => {
+      const columnName = column4.attname;
+      const isArray = internals?.tables[tableName]?.columns[columnName]?.isArray ?? false;
       if (column4.column_default === null) {
         return void 0;
       }
       if (column4.data_type === "serial" || column4.data_type === "smallserial" || column4.data_type === "bigserial") {
         return void 0;
       }
-      const hasDifferentDefaultCast = Object.keys(columnToDefault).find((it) => column4.data_type.startsWith(it));
+      if (column4.column_default.endsWith("[]")) {
+        column4.column_default = column4.column_default.slice(0, -2);
+      }
+      column4.column_default = column4.column_default.replace(/::(.*?)(?<![^\w"])(?=$)/, "");
       const columnDefaultAsString = column4.column_default.toString();
-      if (columnDefaultAsString.endsWith(
-        hasDifferentDefaultCast ? columnToDefault[hasDifferentDefaultCast] : column4.data_type
-      )) {
-        const nonPrefixPart = column4.column_default.length - (hasDifferentDefaultCast ? columnToDefault[hasDifferentDefaultCast] : `::${column4.data_type}`).length - 1;
-        const rt = column4.column_default.toString().substring(1, nonPrefixPart);
-        if (/^-?[\d.]+(?:e-?\d+)?$/.test(rt) && !column4.data_type.startsWith("numeric")) {
-          return Number(rt);
-        } else if (column4.data_type === "json" || column4.data_type === "jsonb") {
-          const jsonWithoutSpaces = JSON.stringify(JSON.parse(rt));
-          return `'${jsonWithoutSpaces}'${hasDifferentDefaultCast ? columnToDefault[hasDifferentDefaultCast] : `::${column4.data_type}`}`;
-        } else if (column4.data_type === "boolean") {
-          return column4.column_default === "true";
-        } else {
-          return `'${rt}'`;
-        }
-      } else {
-        if (/^-?[\d.]+(?:e-?\d+)?$/.test(columnDefaultAsString) && !column4.data_type.startsWith("numeric")) {
+      if (isArray) {
+        return `'{${columnDefaultAsString.slice(2, -2).split(/\s*,\s*/g).map((value) => {
+          if (["integer", "smallint", "bigint", "double precision", "real"].includes(column4.data_type.slice(0, -2))) {
+            return value;
+          } else if (column4.data_type.startsWith("timestamp")) {
+            return `${value}`;
+          } else if (column4.data_type.slice(0, -2) === "interval") {
+            return value.replaceAll('"', `"`);
+          } else if (column4.data_type.slice(0, -2) === "boolean") {
+            return value === "t" ? "true" : "false";
+          } else if (["json", "jsonb"].includes(column4.data_type.slice(0, -2))) {
+            return JSON.stringify(JSON.stringify(JSON.parse(JSON.parse(value)), null, 0));
+          } else {
+            return `"${value}"`;
+          }
+        }).join(",")}}'`;
+      }
+      if (["integer", "smallint", "bigint", "double precision", "real"].includes(column4.data_type)) {
+        if (/^-?[\d.]+(?:e-?\d+)?$/.test(columnDefaultAsString)) {
           return Number(columnDefaultAsString);
-        } else if (column4.data_type === "boolean") {
-          return column4.column_default === "true";
         } else {
-          return `${columnDefaultAsString}`;
+          if (typeof internals.tables[tableName] === "undefined") {
+            internals.tables[tableName] = {
+              columns: {
+                [columnName]: {
+                  isDefaultAnExpression: true
+                }
+              }
+            };
+          } else {
+            if (typeof internals.tables[tableName].columns[columnName] === "undefined") {
+              internals.tables[tableName].columns[columnName] = {
+                isDefaultAnExpression: true
+              };
+            } else {
+              internals.tables[tableName].columns[columnName].isDefaultAnExpression = true;
+            }
+          }
+          return columnDefaultAsString;
         }
+      } else if (column4.data_type === "json" || column4.data_type === "jsonb") {
+        const jsonWithoutSpaces = JSON.stringify(JSON.parse(columnDefaultAsString.slice(1, -1)));
+        return `'${jsonWithoutSpaces}'::${column4.data_type}`;
+      } else if (column4.data_type === "boolean") {
+        return column4.column_default === "true";
+      } else if (columnDefaultAsString === "NULL") {
+        return `NULL`;
+      } else if (columnDefaultAsString.startsWith("'") && columnDefaultAsString.endsWith("'")) {
+        return columnDefaultAsString;
+      } else {
+        return `${columnDefaultAsString.replace(/\\/g, "`\\")}`;
       }
     };
   }
@@ -33885,9 +33941,9 @@ var init_cli = __esm({
       tablesFilter: unionType([stringType(), stringType().array()]).optional(),
       schemaFilter: unionType([stringType(), stringType().array()]).optional().default(["public"]),
       extensionsFilters: literalType("postgis").array().optional(),
-      introspectCasing: casing,
+      casing,
       breakpoints: booleanType().optional().default(true),
-      database: objectType({
+      migrations: objectType({
         prefix: prefix.optional().default("index")
       }).optional()
     }).passthrough();
@@ -33981,6 +34037,10 @@ var init_postgres = __esm({
         database: stringType().min(1),
         secretArn: stringType().min(1),
         resourceArn: stringType().min(1)
+      }),
+      objectType({
+        driver: literalType("pglite"),
+        url: stringType().min(1)
       })
     ]);
   }