chore: Modify source format

Quramy · Jan 16, 2024 · dddae0a · dddae0a
1 parent 35d02d2
commit dddae0a
Showing 1 changed file with 7 additions and 15 deletions.
diff --git a/packages/prisma-fabbrica/src/templates/index.ts b/packages/prisma-fabbrica/src/templates/index.ts
@@ -94,7 +94,12 @@ export const header = (prismaClientModuleSpecifier: string) =>
     export { resetSequence, registerScalarFieldValueGenerator, resetScalarFieldValueGenerator } from "@quramy/prisma-fabbrica/lib/internal";
   `();
 
-export const genericTypeDeclarations = () =>
+export const importStatement = (specifier: string, prismaClientModuleSpecifier: string) =>
+  template.statement<ts.ImportDeclaration>`
+    import type { ${() => ast.identifier(specifier)} } from ${() => ast.stringLiteral(prismaClientModuleSpecifier)};
+  `();
+
+export const genericDeclarations = () =>
   template.sourceFile`
     type BuildDataOptions = {
       readonly seq: number;
@@ -105,22 +110,11 @@ export const genericTypeDeclarations = () =>
       onBeforeCreate?: (createInput: TCreateInput) => void | PromiseLike<void>;
       onAfterCreate?: (created: TCreated) => void | PromiseLike<void>;
     };
-  `();
-
-export const importStatement = (specifier: string, prismaClientModuleSpecifier: string) =>
-  template.statement<ts.ImportDeclaration>`
-    import type { ${() => ast.identifier(specifier)} } from ${() => ast.stringLiteral(prismaClientModuleSpecifier)};
-  `();
 
-export const initializer = () =>
-  template.sourceFile`
     const initializer = createInitializer();
     const { getClient } = initializer;
     export const { initialize } = initializer;
-  `();
 
-export const symbols = () =>
-  template.sourceFile`
     const factoryFor = Symbol("factoryFor");
   `();
 
@@ -560,9 +554,7 @@ export function getSourceFile({
     ...modelNames.map(modelName => importStatement(modelName, prismaClientModuleSpecifier)),
     ...modelEnums.map(enumName => importStatement(enumName, prismaClientModuleSpecifier)),
     ...header(prismaClientModuleSpecifier).statements,
-    ...insertLeadingBreakMarker(genericTypeDeclarations().statements),
-    ...insertLeadingBreakMarker(initializer().statements),
-    ...insertLeadingBreakMarker(symbols().statements),
+    ...insertLeadingBreakMarker(genericDeclarations().statements),
     insertLeadingBreakMarker(modelFieldDefinitions(document.datamodel.models)),
     ...document.datamodel.models
       .reduce((acc, model) => {