fix(@angular/cli): fully use add command registry option when installing packages

Fixes: #16766
(cherry picked from commit f4127b5)

Author: clydin

packages/angular/cli/commands/add-impl.ts

@@ -29,6 +29,14 @@ const npa = require('npm-package-arg');
 export class AddCommand extends SchematicCommand<AddCommandSchema> {
   readonly allowPrivateSchematics = true;
 
+  async initialize(options: AddCommandSchema & Arguments) {
+    if (options.registry) {
+      return super.initialize({ ...options, packageRegistry: options.registry });
+    } else {
+      return super.initialize(options);
+    }
+  }
+
   async run(options: AddCommandSchema & Arguments) {
     if (!options.collection) {
       this.logger.fatal(
@@ -156,7 +164,12 @@ export class AddCommand extends SchematicCommand<AddCommandSchema> {
     if (savePackage === false) {
       // Temporary packages are located in a different directory
       // Hence we need to resolve them using the temp path
-      const tempPath = installTempPackage(packageIdentifier.raw, this.logger, packageManager);
+      const tempPath = installTempPackage(
+        packageIdentifier.raw,
+        this.logger,
+        packageManager,
+        options.registry ? [`--registry="${options.registry}"`] : undefined,
+      );
       const resolvedCollectionPath = require.resolve(
         join(collectionName, 'package.json'),
         {
@@ -166,7 +179,13 @@ export class AddCommand extends SchematicCommand<AddCommandSchema> {
 
       collectionName = dirname(resolvedCollectionPath);
     } else {
-      installPackage(packageIdentifier.raw, this.logger, packageManager, savePackage);
+      installPackage(
+        packageIdentifier.raw,
+        this.logger,
+        packageManager,
+        savePackage,
+        options.registry ? [`--registry="${options.registry}"`] : undefined,
+      );
     }
 
     return this.executeSchematic(collectionName, options['--']);

packages/angular/cli/models/schematic-command.ts

@@ -53,6 +53,7 @@ export interface BaseSchematicSchema {
   force?: boolean;
   interactive?: boolean;
   defaults?: boolean;
+  packageRegistry?: string;
 }
 
 export interface RunSchematicOptions extends BaseSchematicSchema {
@@ -250,6 +251,7 @@ export abstract class SchematicCommand<
       force,
       dryRun,
       packageManager: await getPackageManager(this.workspace.root),
+      packageRegistry: options.packageRegistry,
       root: normalize(this.workspace.root),
       registry: new schema.CoreSchemaRegistry(formats.standardFormats),
       resolvePaths: !!this.workspace.configFile

packages/angular/cli/tasks/install-package.ts

@@ -68,6 +68,7 @@ export function installTempPackage(
   packageName: string,
   logger: logging.Logger,
   packageManager: PackageManager = PackageManager.Npm,
+  extraArgs?: string[],
 ): string {
   const tempPath = mkdtempSync(join(realpathSync(tmpdir()), 'angular-cli-packages-'));
 
@@ -100,6 +101,7 @@ export function installTempPackage(
   // Yarn will not append 'node_modules' to the path
   const prefixPath = packageManager === PackageManager.Yarn ? tempNodeModules : tempPath;
   const installArgs: string[] = [
+    ...(extraArgs || []),
     `${packageManagerArgs.prefix}="${prefixPath}"`,
     packageManagerArgs.noLockfile,
   ];

tests/legacy-cli/e2e/tests/commands/add/registry-option.ts

@@ -0,0 +1,18 @@
+import { expectFileToExist, rimraf, writeMultipleFiles } from '../../../utils/fs';
+import { ng } from '../../../utils/process';
+import { expectToFail } from '../../../utils/utils';
+
+export default async function () {
+  // forcibly remove in case another test doesn't clean itself up
+  await rimraf('node_modules/@angular/material');
+
+  // Setup an invalid registry
+  await writeMultipleFiles({
+    '.npmrc': 'registry=http://127.0.0.1:9999',
+  });
+
+  await expectToFail(() => ng('add', '@angular/pwa'));
+
+  await ng('add', '--registry=http://localhost:4873', '@angular/pwa');
+  await expectFileToExist('src/manifest.webmanifest');
+}