brizzle 0.2.7 → 0.2.8

package/dist/index.js

@@ -506,6 +506,24 @@ function getTimestampColumns(dialect, noTimestamps = false) {
     .$defaultFn(() => new Date())`;
   }
 }
+function extractImportsFromSchema(content) {
+  const importMatch = content.match(/import\s*\{([^}]+)\}\s*from\s*["']drizzle-orm\/[^"']+["']/);
+  if (!importMatch) {
+    return [];
+  }
+  return importMatch[1].split(",").map((s) => s.trim()).filter((s) => s.length > 0);
+}
+function updateSchemaImports(content, newImports, dialect) {
+  const existingImports = extractImportsFromSchema(content);
+  const mergedImports = Array.from(/* @__PURE__ */ new Set([...existingImports, ...newImports]));
+  const drizzleImport = getDrizzleImport(dialect);
+  const newImportLine = `import { ${mergedImports.join(", ")} } from "${drizzleImport}";`;
+  const importRegex = /import\s*\{[^}]+\}\s*from\s*["']drizzle-orm\/[^"']+["'];?/;
+  if (importRegex.test(content)) {
+    return content.replace(importRegex, newImportLine);
+  }
+  return newImportLine + "\n" + content;
+}
 function getRequiredImports(fields, dialect, options = {}) {
   const types = /* @__PURE__ */ new Set();
   types.add(getTableFunction(dialect));
@@ -782,17 +800,21 @@ function generateEnumField(field, columnName, dialect) {
 }
 function appendToSchema(schemaPath, modelName, tableName, fields, dialect, options) {
   const existingContent = readFile(schemaPath);
+  const newImports = getRequiredImports(fields, dialect, options);
+  const updatedContent = updateSchemaImports(existingContent, newImports, dialect);
   const enumDefinitions = generateEnumDefinitions(fields, dialect);
   const tableDefinition = generateTableDefinition(modelName, tableName, fields, dialect, options);
-  const newContent = existingContent + enumDefinitions + "\n" + tableDefinition + "\n";
+  const newContent = updatedContent + enumDefinitions + "\n" + tableDefinition + "\n";
   writeFile(schemaPath, newContent, { force: true, dryRun: options.dryRun });
 }
 function replaceInSchema(schemaPath, modelName, tableName, fields, dialect, options) {
   const existingContent = readFile(schemaPath);
   const cleanedContent = removeModelFromSchemaContent(existingContent, tableName);
+  const newImports = getRequiredImports(fields, dialect, options);
+  const updatedContent = updateSchemaImports(cleanedContent, newImports, dialect);
   const enumDefinitions = generateEnumDefinitions(fields, dialect);
   const tableDefinition = generateTableDefinition(modelName, tableName, fields, dialect, options);
-  const newContent = cleanedContent.trimEnd() + "\n" + enumDefinitions + "\n" + tableDefinition + "\n";
+  const newContent = updatedContent.trimEnd() + "\n" + enumDefinitions + "\n" + tableDefinition + "\n";
   writeFile(schemaPath, newContent, { force: true, dryRun: options.dryRun });
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "brizzle",
-  "version": "0.2.7",
+  "version": "0.2.8",
   "description": "Rails-like generators for Next.js + Drizzle ORM projects",
   "type": "module",
   "bin": {