1 files changed, 127 insertions, 0 deletions
diff --git a/remote/test/puppeteer/packages/ng-schematics/src/schematics/ng-add/index.ts b/remote/test/puppeteer/packages/ng-schematics/src/schematics/ng-add/index.ts
new file mode 100644
index 0000000000..49c56bfc2f
--- /dev/null
+++ b/remote/test/puppeteer/packages/ng-schematics/src/schematics/ng-add/index.ts
@@ -0,0 +1,127 @@
+/**
+ * Copyright 2022 Google Inc. All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import {chain, Rule, SchematicContext, Tree} from '@angular-devkit/schematics';
+import {NodePackageInstallTask} from '@angular-devkit/schematics/tasks';
+import {of} from 'rxjs';
+import {concatMap, map, scan} from 'rxjs/operators';
+
+import {
+  addBaseFiles,
+  addFrameworkFiles,
+  getNgCommandName,
+} from '../utils/files.js';
+import {getAngularConfig} from '../utils/json.js';
+import {
+  addPackageJsonDependencies,
+  addPackageJsonScripts,
+  getDependenciesFromOptions,
+  getPackageLatestNpmVersion,
+  DependencyType,
+  type NodePackage,
+  updateAngularJsonScripts,
+} from '../utils/packages.js';
+import {type SchematicsOptions} from '../utils/types.js';
+
+// You don't have to export the function as default. You can also have more than one rule
+// factory per file.
+export function ngAdd(options: SchematicsOptions): Rule {
+  return (tree: Tree, context: SchematicContext) => {
+    return chain([
+      addDependencies(options),
+      addPuppeteerFiles(options),
+      addOtherFiles(options),
+      updateScripts(options),
+      updateAngularConfig(options),
+    ])(tree, context);
+  };
+}
+
+function addDependencies(options: SchematicsOptions): Rule {
+  return (tree: Tree, context: SchematicContext) => {
+    context.logger.debug('Adding dependencies to "package.json"');
+    const dependencies = getDependenciesFromOptions(options);
+
+    return of(...dependencies).pipe(
+      concatMap((packageName: string) => {
+        return getPackageLatestNpmVersion(packageName);
+      }),
+      scan((array, nodePackage) => {
+        array.push(nodePackage);
+        return array;
+      }, [] as NodePackage[]),
+      map(packages => {
+        context.logger.debug('Updating dependencies...');
+        addPackageJsonDependencies(tree, packages, DependencyType.Dev);
+        context.addTask(new NodePackageInstallTask());
+
+        return tree;
+      })
+    );
+  };
+}
+
+function updateScripts(options: SchematicsOptions): Rule {
+  return (tree: Tree, context: SchematicContext): Tree => {
+    context.logger.debug('Updating "package.json" scripts');
+    const angularJson = getAngularConfig(tree);
+    const projects = Object.keys(angularJson['projects']);
+
+    if (projects.length === 1) {
+      const name = getNgCommandName(options);
+      const prefix = options.isDefaultTester ? '' : `run ${projects[0]}:`;
+      return addPackageJsonScripts(tree, [
+        {
+          name,
+          script: `ng ${prefix}${name}`,
+        },
+      ]);
+    }
+    return tree;
+  };
+}
+
+function addPuppeteerFiles(options: SchematicsOptions): Rule {
+  return (tree: Tree, context: SchematicContext) => {
+    context.logger.debug('Adding Puppeteer base files.');
+    const {projects} = getAngularConfig(tree);
+
+    return addBaseFiles(tree, context, {
+      projects,
+      options,
+    });
+  };
+}
+
+function addOtherFiles(options: SchematicsOptions): Rule {
+  return (tree: Tree, context: SchematicContext) => {
+    context.logger.debug('Adding Puppeteer additional files.');
+    const {projects} = getAngularConfig(tree);
+
+    return addFrameworkFiles(tree, context, {
+      projects,
+      options,
+    });
+  };
+}
+
+function updateAngularConfig(options: SchematicsOptions): Rule {
+  return (tree: Tree, context: SchematicContext): Tree => {
+    context.logger.debug('Updating "angular.json".');
+
+    return updateAngularJsonScripts(tree, options);
+  };
+}