orchid-orm 1.4.20 → 1.4.22

package/dist/index.js

@@ -1594,8 +1594,38 @@ const getImportPath = (from, to) => {
 
 const libraryName = "orchid-orm";
 const importKey = "orchidORM";
-const updateMainFile = async (path, tablePath, ast) => {
-  const content = await fs__default["default"].readFile(path, "utf-8");
+const newFile = (options) => `import { orchidORM } from 'orchid-orm';
+
+export const db = orchidORM(
+  {
+    ${optionsToString(options)}
+  },
+  {
+  }
+);
+`;
+const optionsToString = (options) => {
+  const lines = [];
+  for (const key in options) {
+    const value = options[key];
+    if (typeof value !== "object" && typeof value !== "function") {
+      lines.push(
+        `${key}: ${typeof value === "string" ? pqb.singleQuote(value) : value},`
+      );
+    }
+  }
+  return lines.join("\n    ");
+};
+const updateMainFile = async (filePath, tablePath, ast, options) => {
+  const result = await fs__default["default"].readFile(filePath, "utf-8").then(
+    (content2) => ({ error: void 0, content: content2 }),
+    (error) => {
+      return { error, content: void 0 };
+    }
+  );
+  if (result.error && result.error.code !== "ENOENT")
+    throw result.error;
+  const content = result.content || newFile(options);
   const statements = ts.getStatements(content);
   const importName = ts.import.getStatementsImportedName(
     statements,
@@ -1613,26 +1643,33 @@ const updateMainFile = async (path, tablePath, ast) => {
   }
   const spaces = ts.spaces.getAtLine(content, object.end);
   const context = {
-    path,
+    filePath,
     tablePath,
     statements,
     object,
     content,
     spaces
   };
+  let write;
   if (ast.type === "table") {
     if (ast.action === "create") {
-      return fs__default["default"].writeFile(path, createTable$1(context, ast));
+      write = createTable$1(context, ast);
     } else {
-      return fs__default["default"].writeFile(path, dropTable(context, ast));
+      write = dropTable(context, ast);
     }
   }
+  if (write) {
+    if (result.error) {
+      await fs__default["default"].mkdir(path__default["default"].dirname(filePath), { recursive: true });
+    }
+    await fs__default["default"].writeFile(filePath, write);
+  }
 };
-const createTable$1 = ({ path, tablePath, statements, object, content, spaces }, ast) => {
+const createTable$1 = ({ filePath, tablePath, statements, object, content, spaces }, ast) => {
   const key = toCamelCase(ast.name);
   const value = toPascalCase(ast.name);
   const changes = new FileChanges(content);
-  const importPath = getImportPath(path, tablePath(ast.name));
+  const importPath = getImportPath(filePath, tablePath(ast.name));
   const importPos = ts.import.getEndPos(statements);
   changes.add(
     importPos,
@@ -1650,9 +1687,9 @@ ${spaces}`;
   changes.add(object.properties.end, insert);
   return changes.apply();
 };
-const dropTable = ({ path, tablePath, statements, object, content }, ast) => {
+const dropTable = ({ filePath, tablePath, statements, object, content }, ast) => {
   const changes = new FileChanges(content);
-  const importPath = getImportPath(path, tablePath(ast.name));
+  const importPath = getImportPath(filePath, tablePath(ast.name));
   const tableClassName = toPascalCase(ast.name);
   const importNames = [];
   for (const node of ts.import.iterateWithSource(statements, importPath)) {
@@ -1717,7 +1754,11 @@ const createTable = async (_a) => {
   ]);
   const tablePath = params.tablePath(ast.name);
   const baseTablePath = getImportPath(tablePath, params.baseTablePath);
-  const props = [`table = ${pqb.singleQuote(ast.name)};`];
+  const props = [];
+  if (ast.schema) {
+    props.push(`schema = ${pqb.singleQuote(ast.schema)};`);
+  }
+  props.push(`table = ${pqb.singleQuote(ast.name)};`);
   if (ast.noPrimaryKey === "ignore") {
     props.push("noPrimaryKey = true;");
   }
@@ -1731,8 +1772,9 @@ const createTable = async (_a) => {
 `,
     `export class ${toPascalCase(ast.name)} extends ${params.baseTableName} {`,
     props,
-    "}"
+    "}\n"
   ];
+  await fs__default["default"].mkdir(path__default["default"].dirname(tablePath), { recursive: true });
   await fs__default["default"].writeFile(tablePath, pqb.codeToString(code, "", "  "));
 };
 
@@ -2159,17 +2201,38 @@ const renameTable = async (_a) => {
   const changes = new FileChanges(content);
   const statements = ts.getStatements(content);
   const className = toPascalCase(ast.from);
+  const changeSchema = ast.fromSchema !== ast.toSchema;
   for (const node of ts.class.iterate(statements)) {
     if (((_a2 = node.name) == null ? void 0 : _a2.escapedText) !== className)
       continue;
+    const addSchema = changeSchema && ast.toSchema && !node.members.some((member) => ts.prop.getName(member) === "schema");
+    if (addSchema && ast.toSchema) {
+      changes.add(
+        node.members.pos,
+        `
+  schema = ${pqb.singleQuote(ast.toSchema)};`
+      );
+    }
     for (const member of node.members) {
       const name = ts.prop.getName(member);
-      if (name !== "table")
+      if (name !== "table" && !(changeSchema && name === "schema"))
         continue;
       const { initializer: value } = member;
       if (!value)
         continue;
-      changes.replace(value.pos, value.end, ` ${pqb.singleQuote(ast.to)}`);
+      if (name === "schema") {
+        if (ast.toSchema) {
+          changes.replace(
+            value.pos,
+            value.end,
+            ` ${pqb.singleQuote(ast.toSchema)}`
+          );
+        } else {
+          changes.remove(member.pos, member.end);
+        }
+      } else {
+        changes.replace(value.pos, value.end, ` ${pqb.singleQuote(ast.to)}`);
+      }
     }
   }
   await fs__default["default"].writeFile(params.tablePath(ast.to), changes.apply());
@@ -2205,6 +2268,32 @@ const updateTableFile = async (params) => {
   }
 };
 
+const createBaseTableFile = async ({
+  baseTableName,
+  baseTablePath
+}) => {
+  await fs__default["default"].mkdir(path__default["default"].dirname(baseTablePath), { recursive: true });
+  await fs__default["default"].writeFile(
+    baseTablePath,
+    `import { createBaseTable } from 'orchid-orm';
+import { columnTypes } from 'pqb';
+
+export const ${baseTableName} = createBaseTable({
+  columnTypes: {
+    ...columnTypes,
+  },
+});
+`,
+    {
+      flag: "wx"
+    }
+  ).catch((err) => {
+    if (err.code === "EEXIST")
+      return;
+    throw err;
+  });
+};
+
 var __defProp = Object.defineProperty;
 var __defProps = Object.defineProperties;
 var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
@@ -2231,11 +2320,20 @@ const appCodeUpdater = (config) => {
     tablePath: (name) => path__namespace.resolve(config.tablePath(name)),
     mainFilePath: path__namespace.resolve(config.mainFilePath)
   });
-  return async (ast) => {
-    await Promise.all([
-      updateMainFile(params.mainFilePath, params.tablePath, ast),
+  return async ({ ast, options, cache: cacheObject }) => {
+    const promises = [
+      updateMainFile(params.mainFilePath, params.tablePath, ast, options),
       updateTableFile(__spreadProps(__spreadValues({}, params), { ast }))
-    ]);
+    ];
+    const cache = cacheObject;
+    if (!cache.createdBaseTable) {
+      promises.push(
+        createBaseTableFile(params).then(() => {
+          cache.createdBaseTable = true;
+        })
+      );
+    }
+    await Promise.all(promises);
   };
 };