Merge pull request #256 from Quramy/chore/break_for_each_statements

chore: Reactor insertLeadingBreakMarker fn
Quramy · Jan 15, 2024 · fcd0470 · fcd0470
2 parents 5801afc + b6f6c9a
commit fcd0470
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 7 deletions.
diff --git a/examples/example-prj/src/__generated__/fabbrica/index.d.ts b/examples/example-prj/src/__generated__/fabbrica/index.d.ts
diff --git a/packages/prisma-fabbrica/src/templates/ast-tools/comment.ts b/packages/prisma-fabbrica/src/templates/ast-tools/comment.ts
@@ -17,8 +17,19 @@ export function wrapWithTSDoc<T extends ts.Node>(text: string, node: T): T {
   );
 }
 
-export function insertLeadingBreakMarker<T extends ts.Node>(node: T): T {
-  const comments = ts.getSyntheticLeadingComments(node);
-  if (comments?.length) return node;
-  return ts.addSyntheticLeadingComment(node, ts.SyntaxKind.SingleLineCommentTrivia, "%BR%", true);
+function isNodeArray(x: any): x is ts.NodeArray<any> {
+  return Array.isArray(x);
+}
+
+export function insertLeadingBreakMarker<T extends ts.Node | ts.NodeArray<ts.Node>>(nodeOrNodeList: T): T {
+  const processNode = <S extends ts.Node>(node: S) => {
+    const comments = ts.getSyntheticLeadingComments(node);
+    if (comments?.length) return node;
+    return ts.addSyntheticLeadingComment(node, ts.SyntaxKind.SingleLineCommentTrivia, "%BR%", true) as S;
+  };
+  if (!isNodeArray(nodeOrNodeList)) {
+    return processNode(nodeOrNodeList) as T;
+  } else {
+    return nodeOrNodeList.map(processNode) as unknown as T;
+  }
 }
diff --git a/packages/prisma-fabbrica/src/templates/index.ts b/packages/prisma-fabbrica/src/templates/index.ts
@@ -106,7 +106,7 @@ export const importStatement = (specifier: string, prismaClientModuleSpecifier:
   `();
 
 export const symbols = () =>
-  template.statement`
+  template.sourceFile`
     const factoryFor = Symbol("factoryFor");
   `();
 
@@ -513,7 +513,7 @@ export function getSourceFile({
     ...modelEnums.map(enumName => importStatement(enumName, prismaClientModuleSpecifier)),
     ...header(prismaClientModuleSpecifier).statements,
     insertLeadingBreakMarker(buildDataOptions()),
-    insertLeadingBreakMarker(symbols()),
+    ...insertLeadingBreakMarker(symbols().statements),
     insertLeadingBreakMarker(modelFieldDefinitions(document.datamodel.models)),
     ...document.datamodel.models
       .reduce((acc, model) => {