52 lines
1.5 KiB
TypeScript
52 lines
1.5 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.dev/license
|
|
*/
|
|
|
|
import {chain, noop, Rule, Tree} from '@angular-devkit/schematics';
|
|
import {
|
|
addModuleImportToModule,
|
|
buildComponent,
|
|
findModuleFromOptions,
|
|
isStandaloneSchematic,
|
|
} from '@angular/cdk/schematics';
|
|
import {Schema} from './schema';
|
|
|
|
/**
|
|
* Scaffolds a new tree component.
|
|
* Internally it bootstraps the base component schematic
|
|
*/
|
|
export default function (options: Schema): Rule {
|
|
return chain([
|
|
buildComponent(
|
|
{...options},
|
|
{
|
|
template:
|
|
'./__path__/__name@dasherize@if-flat__/__name@dasherize__.component.html.template',
|
|
stylesheet:
|
|
'./__path__/__name@dasherize@if-flat__/__name@dasherize__.component.__style__.template',
|
|
},
|
|
),
|
|
options.skipImport ? noop() : addTreeModulesToModule(options),
|
|
]);
|
|
}
|
|
|
|
/**
|
|
* Adds the required modules to the relative module.
|
|
*/
|
|
function addTreeModulesToModule(options: Schema) {
|
|
return async (host: Tree) => {
|
|
const isStandalone = await isStandaloneSchematic(host, options);
|
|
|
|
if (!isStandalone) {
|
|
const modulePath = (await findModuleFromOptions(host, options))!;
|
|
addModuleImportToModule(host, modulePath, 'MatTreeModule', '@angular/material/tree');
|
|
addModuleImportToModule(host, modulePath, 'MatIconModule', '@angular/material/icon');
|
|
addModuleImportToModule(host, modulePath, 'MatButtonModule', '@angular/material/button');
|
|
}
|
|
};
|
|
}
|