57 lines
1.6 KiB
TypeScript
57 lines
1.6 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 table 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() : addTableModulesToModule(options),
|
||
|
|
]);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Adds the required modules to the relative module.
|
||
|
|
*/
|
||
|
|
function addTableModulesToModule(options: Schema) {
|
||
|
|
return async (host: Tree) => {
|
||
|
|
const isStandalone = await isStandaloneSchematic(host, options);
|
||
|
|
|
||
|
|
if (!isStandalone) {
|
||
|
|
const modulePath = (await findModuleFromOptions(host, options))!;
|
||
|
|
addModuleImportToModule(host, modulePath, 'MatTableModule', '@angular/material/table');
|
||
|
|
addModuleImportToModule(
|
||
|
|
host,
|
||
|
|
modulePath,
|
||
|
|
'MatPaginatorModule',
|
||
|
|
'@angular/material/paginator',
|
||
|
|
);
|
||
|
|
addModuleImportToModule(host, modulePath, 'MatSortModule', '@angular/material/sort');
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|