refactor: change type signature of cli plugin

r2don · Oct 22, 2023 · f61875f · f61875f
1 parent b5951bf
commit f61875f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/lib/plugin/compiler-plugin.ts b/lib/plugin/compiler-plugin.ts
@@ -11,24 +11,24 @@ function isFilenameMatched(patterns: string[], filename: string): boolean {
 export const before: (
   options: Record<string, any> | undefined,
   program: ts.Program,
-) => (ctx: ts.TransformationContext) => ts.Transformer<any> = (
+) => ts.TransformerFactory<ts.SourceFile> | ts.CustomTransformerFactory = (
   options,
   program,
 ) => {
   const mergedOption = mergePluginOptions(options);
 
-  return (ctx: ts.TransformationContext): ts.Transformer<any> => {
-    return (sf: ts.SourceFile) => {
+  return (ctx) => {
+    return (sourceFile) => {
       if (
         isFilenameMatched(
           mergedOption.interfaceFilenameSuffix as string[],
-          sf.fileName,
+          sourceFile.fileName,
         )
       ) {
-        return httpInterfaceVisitor.visit(sf, ctx, program);
+        return httpInterfaceVisitor.visit(sourceFile, ctx, program);
       }
 
-      return sf;
+      return sourceFile;
     };
   };
 };
diff --git a/lib/plugin/visitors/http-interface.visitor.ts b/lib/plugin/visitors/http-interface.visitor.ts
@@ -40,7 +40,7 @@ export class HttpInterfaceVisitor {
     sourceFile: ts.SourceFile,
     ctx: ts.TransformationContext,
     program: ts.Program,
-  ): ts.Node {
+  ): ts.SourceFile {
     this.importSet.clear();
     const factory = ctx.factory;
     const typeChecker = program.getTypeChecker();
@@ -61,7 +61,7 @@ export class HttpInterfaceVisitor {
       return ts.visitEachChild(node, visitNode, ctx);
     };
 
-    return ts.visitNode(sourceFile, visitNode);
+    return ts.visitNode(sourceFile, visitNode) as ts.SourceFile;
   }
 
   private updateSourceFile(