123 lines
4.2 KiB
TypeScript
123 lines
4.2 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright Google LLC All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by an MIT-style license that can be
|
|
* found in the LICENSE file at https://angular.io/license
|
|
*
|
|
* @fileoverview Schematics for ng-new project that builds with Bazel.
|
|
*/
|
|
|
|
import {virtualFs, workspaces} from '@angular-devkit/core';
|
|
import {chain, Rule, SchematicsException, Tree} from '@angular-devkit/schematics';
|
|
import {getWorkspace} from '@schematics/angular/utility/workspace';
|
|
import {Builders} from '@schematics/angular/utility/workspace-models';
|
|
|
|
import {Schema} from './schema';
|
|
|
|
|
|
export const localizePolyfill = `import '@angular/localize/init';`;
|
|
|
|
function getRelevantTargetDefinitions(
|
|
project: workspaces.ProjectDefinition, builderName: Builders): workspaces.TargetDefinition[] {
|
|
const definitions: workspaces.TargetDefinition[] = [];
|
|
project.targets.forEach((target: workspaces.TargetDefinition): void => {
|
|
if (target.builder === builderName) {
|
|
definitions.push(target);
|
|
}
|
|
});
|
|
return definitions;
|
|
}
|
|
|
|
function getOptionValuesForTargetDefinition(
|
|
definition: workspaces.TargetDefinition, optionName: string): string[] {
|
|
const optionValues: string[] = [];
|
|
if (definition.options && optionName in definition.options) {
|
|
let optionValue: unknown = definition.options[optionName];
|
|
if (typeof optionValue === 'string') {
|
|
optionValues.push(optionValue);
|
|
}
|
|
}
|
|
if (!definition.configurations) {
|
|
return optionValues;
|
|
}
|
|
Object.values(definition.configurations)
|
|
.forEach((configuration: Record<string, unknown>|undefined): void => {
|
|
if (configuration && optionName in configuration) {
|
|
const optionValue: unknown = configuration[optionName];
|
|
if (typeof optionValue === 'string') {
|
|
optionValues.push(optionValue);
|
|
}
|
|
}
|
|
});
|
|
return optionValues;
|
|
}
|
|
|
|
function getFileListForRelevantTargetDefinitions(
|
|
project: workspaces.ProjectDefinition, builderName: Builders, optionName: string): string[] {
|
|
const fileList: string[] = [];
|
|
const definitions = getRelevantTargetDefinitions(project, builderName);
|
|
definitions.forEach((definition: workspaces.TargetDefinition): void => {
|
|
const optionValues = getOptionValuesForTargetDefinition(definition, optionName);
|
|
optionValues.forEach((filePath: string): void => {
|
|
if (fileList.indexOf(filePath) === -1) {
|
|
fileList.push(filePath);
|
|
}
|
|
});
|
|
});
|
|
return fileList;
|
|
}
|
|
|
|
function prependToTargetFiles(
|
|
project: workspaces.ProjectDefinition, builderName: Builders, optionName: string, str: string) {
|
|
return (host: Tree) => {
|
|
const fileList = getFileListForRelevantTargetDefinitions(project, builderName, optionName);
|
|
|
|
fileList.forEach((path: string): void => {
|
|
const data = host.read(path);
|
|
if (!data) {
|
|
// If the file doesn't exist, just ignore it.
|
|
return;
|
|
}
|
|
|
|
const content = virtualFs.fileBufferToString(data);
|
|
if (content.includes(localizePolyfill) ||
|
|
content.includes(localizePolyfill.replace(/'/g, '"'))) {
|
|
// If the file already contains the polyfill (or variations), ignore it too.
|
|
return;
|
|
}
|
|
|
|
// Add string at the start of the file.
|
|
const recorder = host.beginUpdate(path);
|
|
recorder.insertLeft(0, str);
|
|
host.commitUpdate(recorder);
|
|
});
|
|
};
|
|
}
|
|
|
|
export default function(options: Schema): Rule {
|
|
return async (host: Tree) => {
|
|
if (!options.name) {
|
|
throw new SchematicsException('Option "name" is required.');
|
|
}
|
|
|
|
const workspace = await getWorkspace(host);
|
|
const project: workspaces.ProjectDefinition|undefined = workspace.projects.get(options.name);
|
|
if (!project) {
|
|
throw new SchematicsException(`Invalid project name (${options.name})`);
|
|
}
|
|
|
|
const localizeStr =
|
|
`/***************************************************************************************************
|
|
* Load \`$localize\` onto the global scope - used if i18n tags appear in Angular templates.
|
|
*/
|
|
${localizePolyfill}
|
|
`;
|
|
|
|
return chain([
|
|
prependToTargetFiles(project, Builders.Browser, 'polyfills', localizeStr),
|
|
prependToTargetFiles(project, Builders.Server, 'main', localizeStr),
|
|
]);
|
|
};
|
|
}
|