- forwarding transformers afterDeclarations

#173
ezolenko · Sep 11, 2019 · 7765355 · 7765355
1 parent 1b9fa28
commit 7765355
Show file tree

Hide file tree

Showing 9 changed files with 15 additions and 4 deletions.
diff --git a/dist/host.d.ts.map b/dist/host.d.ts.map
diff --git a/dist/ioptions.d.ts b/dist/ioptions.d.ts
@@ -3,6 +3,7 @@ import * as tsTypes from "typescript";
 export interface ICustomTransformer {
     before?: tsTypes.TransformerFactory<tsTypes.SourceFile>;
     after?: tsTypes.TransformerFactory<tsTypes.SourceFile>;
+    afterDeclarations?: tsTypes.TransformerFactory<tsTypes.Bundle | tsTypes.SourceFile>;
 }
 export declare type TransformerFactoryCreator = (ls: tsTypes.LanguageService) => tsTypes.CustomTransformers | ICustomTransformer;
 export interface IOptions {

diff --git a/dist/ioptions.d.ts.map b/dist/ioptions.d.ts.map
diff --git a/dist/rollup-plugin-typescript2.cjs.js b/dist/rollup-plugin-typescript2.cjs.js
@@ -17300,13 +17300,16 @@ class LanguageServiceHost {
         const transformer = {
             before: [],
             after: [],
+            afterDeclarations: [],
         };
         for (const creator of this.transformers) {
             const factory = creator(this.service);
             if (factory.before)
                 transformer.before = lodash_11(transformer.before, factory.before);
             if (factory.after)
                 transformer.after = lodash_11(transformer.after, factory.after);
+            if (factory.afterDeclarations)
+                transformer.afterDeclarations = lodash_11(transformer.afterDeclarations, factory.afterDeclarations);
         }
         return transformer;
     }

diff --git a/dist/rollup-plugin-typescript2.cjs.js.map b/dist/rollup-plugin-typescript2.cjs.js.map
diff --git a/dist/rollup-plugin-typescript2.es.js b/dist/rollup-plugin-typescript2.es.js
@@ -17294,13 +17294,16 @@ class LanguageServiceHost {
         const transformer = {
             before: [],
             after: [],
+            afterDeclarations: [],
         };
         for (const creator of this.transformers) {
             const factory = creator(this.service);
             if (factory.before)
                 transformer.before = lodash_11(transformer.before, factory.before);
             if (factory.after)
                 transformer.after = lodash_11(transformer.after, factory.after);
+            if (factory.afterDeclarations)
+                transformer.afterDeclarations = lodash_11(transformer.afterDeclarations, factory.afterDeclarations);
         }
         return transformer;
     }

diff --git a/dist/rollup-plugin-typescript2.es.js.map b/dist/rollup-plugin-typescript2.es.js.map
diff --git a/src/host.ts b/src/host.ts
@@ -128,6 +128,7 @@ export class LanguageServiceHost implements tsTypes.LanguageServiceHost
 		{
 			before: [],
 			after: [],
+			afterDeclarations: [],
 		};
 
 		for (const creator of this.transformers)
@@ -137,6 +138,8 @@ export class LanguageServiceHost implements tsTypes.LanguageServiceHost
 				transformer.before = _.concat(transformer.before!, factory.before);
 			if (factory.after)
 				transformer.after = _.concat(transformer.after!, factory.after);
+			if (factory.afterDeclarations)
+				transformer.afterDeclarations = _.concat(transformer.afterDeclarations!, factory.afterDeclarations);
 		}
 
 		return transformer;

diff --git a/src/ioptions.ts b/src/ioptions.ts
@@ -5,6 +5,7 @@ export interface ICustomTransformer
 {
 	before?: tsTypes.TransformerFactory<tsTypes.SourceFile>;
 	after?: tsTypes.TransformerFactory<tsTypes.SourceFile>;
+	afterDeclarations?: tsTypes.TransformerFactory<tsTypes.Bundle | tsTypes.SourceFile>;
 }
 
 export type TransformerFactoryCreator = (ls: tsTypes.LanguageService) => tsTypes.CustomTransformers | ICustomTransformer;