+
+
+`);
+ return host;
+ };
+
+}
+
+/** 安装依赖 */
+function installNodeDeps(): (host: Tree, context: SchematicContext) => void {
+ return (host: Tree, context: SchematicContext) => {
+ context.addTask(new NodePackageInstallTask());
+ };
+}
+
+/** 将编译的 css 添加到 angular.json */
+function insertCompiledTheme(project: Project, host: Tree, workspace: Workspace): void {
+ const themePath = `node_modules/ng-zorro-antd/src/ng-zorro-antd.min.css`;
+
+ if (project.architect) {
+ addStyleToTarget(project.architect.build, host, themePath, workspace);
+ addStyleToTarget(project.architect.test, host, themePath, workspace);
+ } else {
+ throw new SchematicsException(`${project.name} does not have an architect configuration`);
+ }
+}
+
+/** Adds a style entry to the given target. */
+function addStyleToTarget(target: any, host: Tree, asset: string, workspace: Workspace): void {
+ const styleEntry = asset;
+
+ // We can't assume that any of these properties are defined, so safely add them as we go
+ // if necessary.
+ if (!target.options) {
+ target.options = { styles: [ styleEntry ] };
+ } else if (!target.options.styles) {
+ target.options.styles = [ styleEntry ];
+ } else {
+ const existingStyles = target.options.styles.map(s => typeof s === 'string' ? s : s.input);
+ const hasGivenTheme = existingStyles.find(s => s.includes(asset));
+
+ if (!hasGivenTheme) {
+ target.options.styles.splice(0, 0, styleEntry);
+ }
+ }
+
+ host.overwrite('angular.json', JSON.stringify(workspace, null, 2));
+}
diff --git a/schematics/ng-add/schema.json b/schematics/ng-add/schema.json
new file mode 100644
index 0000000000..41747d523a
--- /dev/null
+++ b/schematics/ng-add/schema.json
@@ -0,0 +1,25 @@
+{
+ "$schema": "http://json-schema.org/schema",
+ "id": "ngAdd",
+ "title": "add NG-ZORRO",
+ "type": "object",
+ "properties": {
+ "skipPackageJson": {
+ "type": "boolean",
+ "default": false,
+ "description": "Do not add ng-zorro dependencies to package.json (e.g., --skipPackageJson)"
+ },
+ "locale": {
+ "type": "string",
+ "default": "zh_CN",
+ "enum": ["ar_EG","bg_BG","ca_ES","cs_CZ","de_DE","el_GR","en_GB","en_US","es_ES","et_EE","fa_IR","fi_FI","fr_BE","fr_FR","is_IS","it_IT","ja_JP","ko_KR","nb_NO","nl_BE","nl_NL","pl_PL","pt_BR","pt_PT","sk_SK","sr_RS","sv_SE","th_TH","tr_TR","ru_RU","uk_UA","vi_VN","zh_CN","zh_TW"],
+ "description": "add locale code to module (e.g., --locale=en_US)"
+ },
+ "theme": {
+ "type": "boolean",
+ "default": false,
+ "description": "add theme.less"
+ }
+ },
+ "required": []
+}
\ No newline at end of file
diff --git a/schematics/ng-add/schema.ts b/schematics/ng-add/schema.ts
new file mode 100644
index 0000000000..ee150769bf
--- /dev/null
+++ b/schematics/ng-add/schema.ts
@@ -0,0 +1,12 @@
+
+export interface Schema {
+ /** Whether to skip package.json install. */
+ skipPackageJson: boolean;
+
+ theme: boolean;
+
+ /** Name of the project to target. */
+ project?: string;
+
+ locale: 'ar_EG' | 'bg_BG' | 'ca_ES' | 'cs_CZ' | 'de_DE' | 'el_GR' | 'en_GB' | 'en_US' | 'es_ES' | 'et_EE' | 'fa_IR' | 'fi_FI' | 'fr_BE' | 'fr_FR' | 'is_IS' | 'it_IT' | 'ja_JP' | 'ko_KR' | 'nb_NO' | 'nl_BE' | 'nl_NL' | 'pl_PL' | 'pt_BR' | 'pt_PT' | 'sk_SK' | 'sr_RS' | 'sv_SE' | 'th_TH' | 'tr_TR' | 'ru_RU' | 'uk_UA' | 'vi_VN' | 'zh_CN' | 'zh_TW';
+}
diff --git a/schematics/tsconfig.json b/schematics/tsconfig.json
new file mode 100644
index 0000000000..9c0ac514c7
--- /dev/null
+++ b/schematics/tsconfig.json
@@ -0,0 +1,25 @@
+{
+ "compilerOptions": {
+ "lib": ["es2017", "dom"],
+ "module": "commonjs",
+ "moduleResolution": "node",
+ "outDir": "../publish/schematics",
+ "noEmitOnError": false,
+ "skipDefaultLibCheck": true,
+ "skipLibCheck": true,
+ "sourceMap": true,
+ "target": "es6",
+ "types": [
+ "jasmine",
+ "node"
+ ]
+ },
+ "include": [
+ "**/*"
+ ],
+ "exclude": [
+ "**/*.component.ts",
+ "**/*spec*",
+ "template/**/*"
+ ]
+}
\ No newline at end of file
diff --git a/schematics/utils/ast.ts b/schematics/utils/ast.ts
new file mode 100755
index 0000000000..3eb9e8cf1c
--- /dev/null
+++ b/schematics/utils/ast.ts
@@ -0,0 +1,103 @@
+import {normalize} from '@angular-devkit/core';
+import {SchematicsException, Tree} from '@angular-devkit/schematics';
+import * as ts from 'typescript';
+import {addImportToModule} from './devkit-utils/ast-utils';
+import {InsertChange} from './devkit-utils/change';
+import {Project, getWorkspace} from './devkit-utils/config';
+import {findBootstrapModulePath, getAppModulePath} from './devkit-utils/ng-ast-utils';
+import {ModuleOptions, findModuleFromOptions as internalFindModule} from './devkit-utils/find-module';
+
+
+/** Reads file given path and returns TypeScript source file. */
+export function getSourceFile(host: Tree, path: string): ts.SourceFile {
+ const buffer = host.read(path);
+ if (!buffer) {
+ throw new SchematicsException(`Could not find file for path: ${path}`);
+ }
+ const content = buffer.toString();
+ return ts.createSourceFile(path, content, ts.ScriptTarget.Latest, true);
+}
+
+/** Import and add module to root app module. */
+export function addModuleImportToRootModule(host: Tree, moduleName: string, src: string, project: Project) {
+ const modulePath = getAppModulePath(host, project.architect.build.options.main);
+ addModuleImportToModule(host, modulePath, moduleName, src);
+}
+
+/**
+ * Import and add module to specific module path.
+ * @param host the tree we are updating
+ * @param modulePath src location of the module to import
+ * @param moduleName name of module to import
+ * @param src src location to import
+ */
+export function addModuleImportToModule(
+ host: Tree, modulePath: string, moduleName: string, src: string) {
+ const moduleSource = getSourceFile(host, modulePath);
+
+ if (!moduleSource) {
+ throw new SchematicsException(`Module not found: ${modulePath}`);
+ }
+
+ const changes = addImportToModule(moduleSource, modulePath, moduleName, src);
+ const recorder = host.beginUpdate(modulePath);
+
+ changes.forEach((change) => {
+ if (change instanceof InsertChange) {
+ recorder.insertLeft(change.pos, change.toAdd);
+ }
+ });
+
+ host.commitUpdate(recorder);
+}
+
+/** Gets the app index.html file */
+export function getIndexHtmlPath(host: Tree, project: Project): string {
+ const buildTarget = project.architect.build.options;
+
+ if (buildTarget.index && buildTarget.index.endsWith('index.html')) {
+ return buildTarget.index;
+ }
+
+ throw new SchematicsException('No index.html file was found.');
+}
+
+/** Get the root stylesheet file. */
+export function getStylesPath(host: Tree, project: Project): string {
+ const buildTarget = project.architect['build'];
+
+ if (buildTarget.options && buildTarget.options.styles && buildTarget.options.styles.length) {
+ const styles = buildTarget.options.styles.map(s => typeof s === 'string' ? s : s.input);
+
+ // First, see if any of the assets is called "styles.(le|sc|c)ss", which is the default
+ // "main" style sheet.
+ const defaultMainStylePath = styles.find(a => /styles\.(c|le|sc)ss/.test(a));
+ if (defaultMainStylePath) {
+ return normalize(defaultMainStylePath);
+ }
+
+ // If there was no obvious default file, use the first style asset.
+ const fallbackStylePath = styles.find(a => /\.(c|le|sc)ss/.test(a));
+ if (fallbackStylePath) {
+ return normalize(fallbackStylePath);
+ }
+ }
+
+ throw new SchematicsException('No style files could be found into which a theme could be added');
+}
+
+/** Wraps the internal find module from options with undefined path handling */
+export function findModuleFromOptions(host: Tree, options: any) {
+ const workspace = getWorkspace(host);
+ if (!options.project) {
+ options.project = Object.keys(workspace.projects)[0];
+ }
+
+ const project = workspace.projects[options.project];
+
+ if (options.path === undefined) {
+ options.path = `/${project.root}/src/app`;
+ }
+
+ return internalFindModule(host, options);
+}
diff --git a/schematics/utils/custom-theme.ts b/schematics/utils/custom-theme.ts
new file mode 100644
index 0000000000..b36a256ef5
--- /dev/null
+++ b/schematics/utils/custom-theme.ts
@@ -0,0 +1,4 @@
+export function createCustomTheme() {
+ return `@import "~ng-zorro-antd/src/ng-zorro-antd.less";
+{{content}}`;
+}
diff --git a/schematics/utils/devkit-utils/README.md b/schematics/utils/devkit-utils/README.md
new file mode 100755
index 0000000000..7c6ca2b68b
--- /dev/null
+++ b/schematics/utils/devkit-utils/README.md
@@ -0,0 +1,2 @@
+# NOTE
+This code is directly taken from [angular schematics package](https://github.com/angular/devkit/tree/master/packages/schematics/angular/utility).
\ No newline at end of file
diff --git a/schematics/utils/devkit-utils/ast-utils.ts b/schematics/utils/devkit-utils/ast-utils.ts
new file mode 100755
index 0000000000..253c9692dc
--- /dev/null
+++ b/schematics/utils/devkit-utils/ast-utils.ts
@@ -0,0 +1,480 @@
+/**
+ * @license
+ * Copyright Google Inc. 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
+ */
+import * as ts from 'typescript';
+import { Change, InsertChange } from './change';
+import { insertImport } from './route-utils';
+
+
+/**
+ * Find all nodes from the AST in the subtree of node of SyntaxKind kind.
+ * @param node
+ * @param kind
+ * @param max The maximum number of items to return.
+ * @return all nodes of kind, or [] if none is found
+ */
+export function findNodes(node: ts.Node, kind: ts.SyntaxKind, max = Infinity): ts.Node[] {
+ if (!node || max == 0) {
+ return [];
+ }
+
+ const arr: ts.Node[] = [];
+ if (node.kind === kind) {
+ arr.push(node);
+ max--;
+ }
+ if (max > 0) {
+ for (const child of node.getChildren()) {
+ findNodes(child, kind, max).forEach(node => {
+ if (max > 0) {
+ arr.push(node);
+ }
+ max--;
+ });
+
+ if (max <= 0) {
+ break;
+ }
+ }
+ }
+
+ return arr;
+}
+
+
+/**
+ * Get all the nodes from a source.
+ * @param sourceFile The source file object.
+ * @returns {Observable