52 lines
1.6 KiB
TypeScript
52 lines
1.6 KiB
TypeScript
|
import * as ts from 'typescript';
|
||
|
|
import { getMethodsForDecorator } from '../helpers';
|
||
|
|
|
||
|
|
function transformImports(file: ts.SourceFile, ctx: ts.TransformationContext, ngcBuild?: boolean) {
|
||
|
|
// remove angular imports
|
||
|
|
if (!ngcBuild) {
|
||
|
|
file.statements = (file.statements as any).filter((s: any) => !(s.kind === ts.SyntaxKind.ImportDeclaration && s.moduleSpecifier.text === '@angular/core'));
|
||
|
|
}
|
||
|
|
|
||
|
|
// find the @ionic-native/core import statement
|
||
|
|
const importStatement = (file.statements as any).find((s: any) => {
|
||
|
|
return s.kind === ts.SyntaxKind.ImportDeclaration && s.moduleSpecifier.text === '@ionic-native/core';
|
||
|
|
});
|
||
|
|
|
||
|
|
// we're only interested in files containing @ionic-native/core import statement
|
||
|
|
if (!importStatement) return file;
|
||
|
|
|
||
|
|
let decorators: string[] = [];
|
||
|
|
|
||
|
|
const decoratorRegex: RegExp = /@([a-zA-Z]+)\(/g;
|
||
|
|
|
||
|
|
let m;
|
||
|
|
|
||
|
|
while ((m = decoratorRegex.exec(file.text)) !== null) {
|
||
|
|
if (m.index === decoratorRegex.lastIndex) {
|
||
|
|
decoratorRegex.lastIndex++;
|
||
|
|
}
|
||
|
|
if (m && m[1] && decorators.indexOf(m[1]) === -1 && m[1] !== 'Plugin') decorators.push(m[1]);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (decorators.length) {
|
||
|
|
let methods = [];
|
||
|
|
|
||
|
|
decorators.forEach(d => methods = getMethodsForDecorator(d).concat(methods));
|
||
|
|
|
||
|
|
importStatement.importClause.namedBindings.elements = [
|
||
|
|
ts.createIdentifier('IonicNativePlugin'),
|
||
|
|
...methods.map(m => ts.createIdentifier(m))
|
||
|
|
];
|
||
|
|
}
|
||
|
|
|
||
|
|
return file;
|
||
|
|
}
|
||
|
|
|
||
|
|
export function importsTransformer(ngcBuild?: boolean) {
|
||
|
|
return (ctx: ts.TransformationContext) => {
|
||
|
|
return tsSourceFile => {
|
||
|
|
return transformImports(tsSourceFile, ctx, ngcBuild);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|