rake-db 2.3.1 → 2.3.2

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # rake-db
 
+## 2.3.2
+
+### Patch Changes
+
+- Improve pulling db structure
+- Updated dependencies
+  - pqb@0.9.2
+
 ## 2.3.1
 
 ### Patch Changes

package/dist/index.esm.js

@@ -1,4 +1,4 @@
-import { singleQuote, quote, isRaw, getRaw, toArray, columnTypes, raw, getColumnTypes, getTableData, resetTableData, ColumnType, emptyObject, Operators, TransactionAdapter, logParamToLogObject, Adapter, columnsByType, instantiateColumn, codeToString, addCode, quoteObjectKey, primaryKeyToCode, indexToCode, foreignKeyToCode } from 'pqb';
+import { singleQuote, quote, isRaw, getRaw, toArray, columnTypes, raw, getColumnTypes, getTableData, resetTableData, ColumnType, emptyObject, Operators, TransactionAdapter, logParamToLogObject, Adapter, columnsByType, instantiateColumn, codeToString, addCode, quoteObjectKey, primaryKeyToCode, indexToCode, foreignKeyToCode, TimestampColumn } from 'pqb';
 import Enquirer from 'enquirer';
 import path from 'path';
 import { readdir, mkdir, writeFile } from 'fs/promises';
@@ -1771,8 +1771,8 @@ const matchMap = {
   p: "PARTIAL"
 };
 const fkeyActionMap = {
-  a: "NO ACTION",
-  r: void 0,
+  a: void 0,
+  r: "RESTRICT",
   c: "CASCADE",
   n: "SET NULL",
   d: "SET DEFAULT"
@@ -1807,6 +1807,8 @@ const structureToAst = async (db) => {
   }
   for (const table of tables) {
     const { schemaName, name } = table;
+    if (name === "schemaMigrations")
+      continue;
     const belongsToTable = makeBelongsToTable(schemaName, name);
     const columns = allColumns.filter(belongsToTable);
     const primaryKey = allPrimaryKeys.find(belongsToTable);
@@ -1815,13 +1817,13 @@ const structureToAst = async (db) => {
     const shape = {};
     for (let item of columns) {
       const isSerial = getIsSerial(item);
+      if (isSerial) {
+        item = __spreadProps(__spreadValues({}, item), { default: void 0 });
+      }
       const klass = columnsByType[getColumnType(item, isSerial)];
       if (!klass) {
         throw new Error(`Column type \`${item.type}\` is not supported`);
       }
-      if (isSerial) {
-        item = __spreadProps(__spreadValues({}, item), { default: void 0 });
-      }
       let column = instantiateColumn(klass, item);
       if ((primaryKey == null ? void 0 : primaryKey.columnNames.length) === 1 && (primaryKey == null ? void 0 : primaryKey.columnNames[0]) === item.name) {
         column = column.primaryKey();
@@ -1959,12 +1961,20 @@ const createSchema = (ast) => {
 const createTable = (ast) => {
   const code = [];
   addCode(code, `await db.createTable(${quoteSchemaTable(ast)}, (t) => ({`);
+  const hasTimestamps = isTimestamp(ast.shape.createdAt) && isTimestamp(ast.shape.updatedAt);
   for (const key in ast.shape) {
+    if (hasTimestamps && (key === "createdAt" || key === "updatedAt"))
+      continue;
     const line = [`${quoteObjectKey(key)}: `];
-    addCode(line, ast.shape[key].toCode("t"));
+    for (const part of ast.shape[key].toCode("t")) {
+      addCode(line, part);
+    }
     addCode(line, ",");
     code.push(line);
   }
+  if (hasTimestamps) {
+    code.push(["...t.timestamps(),"]);
+  }
   if (ast.primaryKey) {
     code.push([primaryKeyToCode(ast.primaryKey, "t")]);
   }
@@ -1977,10 +1987,18 @@ const createTable = (ast) => {
   addCode(code, "}));");
   return code;
 };
+const isTimestamp = (column) => {
+  if (!column)
+    return false;
+  const { default: def } = column.data;
+  return column instanceof TimestampColumn && !column.data.isNullable && def && typeof def === "object" && isRaw(def) && def.__raw === "now()";
+};
 
 const pullDbStructure = async (options, config) => {
-  const db = new DbStructure(new Adapter(options));
+  const adapter = new Adapter(options);
+  const db = new DbStructure(adapter);
   const ast = await structureToAst(db);
+  await adapter.close();
   const result = astToMigration(ast);
   if (!result)
     return;