73 lines
2.3 KiB
TypeScript
73 lines
2.3 KiB
TypeScript
import { BaseProcessor, ProcessorContext, ProcessorResult } from './base-processor';
|
|
|
|
export class DirectiveCollectorProcessor extends BaseProcessor {
|
|
get name(): string {
|
|
return 'directive-collector-processor';
|
|
}
|
|
|
|
async process(context: ProcessorContext): Promise<ProcessorResult> {
|
|
if (!context.source.includes('_quarcComponent')) {
|
|
return this.noChange(context.source);
|
|
}
|
|
|
|
if (!context.source.includes('imports:')) {
|
|
return this.noChange(context.source);
|
|
}
|
|
|
|
let source = context.source;
|
|
let modified = false;
|
|
|
|
const scopeIdPattern = /static\s+_scopeId\s*=\s*'[^']*';/g;
|
|
|
|
let match;
|
|
const replacements: { position: number; insert: string }[] = [];
|
|
|
|
while ((match = scopeIdPattern.exec(source)) !== null) {
|
|
const scopeIdEnd = match.index + match[0].length;
|
|
|
|
const beforeScopeId = source.substring(0, match.index);
|
|
const quarcComponentMatch = beforeScopeId.match(/static\s+_quarcComponent\s*=\s*\[([^\]]*(?:\[[^\]]*\][^\]]*)*)\];[^]*$/);
|
|
|
|
if (!quarcComponentMatch) {
|
|
continue;
|
|
}
|
|
|
|
const componentOptions = quarcComponentMatch[1];
|
|
|
|
const importsMatch = componentOptions.match(/imports\s*:\s*\[([^\]]*)\]/);
|
|
if (!importsMatch) {
|
|
continue;
|
|
}
|
|
|
|
const importsContent = importsMatch[1];
|
|
const importNames = this.parseImportNames(importsContent);
|
|
|
|
if (importNames.length === 0) {
|
|
continue;
|
|
}
|
|
|
|
const directivesProperty = `\n static _quarcDirectives = [${importNames.join(', ')}];`;
|
|
|
|
replacements.push({
|
|
position: scopeIdEnd,
|
|
insert: directivesProperty,
|
|
});
|
|
}
|
|
|
|
for (let i = replacements.length - 1; i >= 0; i--) {
|
|
const r = replacements[i];
|
|
source = source.slice(0, r.position) + r.insert + source.slice(r.position);
|
|
modified = true;
|
|
}
|
|
|
|
return modified ? this.changed(source) : this.noChange(source);
|
|
}
|
|
|
|
private parseImportNames(importsContent: string): string[] {
|
|
return importsContent
|
|
.split(',')
|
|
.map(s => s.trim())
|
|
.filter(s => s.length > 0 && /^[A-Z]/.test(s));
|
|
}
|
|
}
|