feat(core): add a generator for dotnet restore

Add a generator to restore packages and tools for builders to use
nx-dotnet · May 18, 2021 · 96082a1 · 96082a1
1 parent 6ab5d4a
commit 96082a1
Show file tree

Hide file tree

Showing 5 changed files with 83 additions and 0 deletions.
diff --git a/packages/core/generators.json b/packages/core/generators.json
@@ -34,6 +34,11 @@
       "factory": "./src/generators/nuget-reference/generator",
       "schema": "./src/generators/nuget-reference/schema.json",
       "description": "nuget-reference generator"
+    },
+    "restore": {
+      "factory": "./src/generators/restore/generator",
+      "schema": "./src/generators/restore/schema.json",
+      "description": "Restores NuGet packages and .NET tools used by the workspace"
     }
   }
 }
diff --git a/packages/core/src/generators/restore/generator.spec.ts b/packages/core/src/generators/restore/generator.spec.ts
@@ -0,0 +1,38 @@
+import { Tree } from '@nrwl/devkit';
+import { createTreeWithEmptyWorkspace } from '@nrwl/devkit/testing';
+
+import { prompt } from 'inquirer';
+
+import { updateConfig, getNxDotnetProjects } from '@nx-dotnet/utils';
+
+import generator from './generator';
+
+import PromptUI = require('inquirer/lib/ui/prompt');
+
+jest.mock('../../../../dotnet/src/lib/core/dotnet.client');
+jest.mock('../../../../utils/src/lib/utility-functions/workspace');
+jest.mock('inquirer');
+
+describe('restore generator', () => {
+  let appTree: Tree;
+
+  beforeEach(() => {
+    appTree = createTreeWithEmptyWorkspace();
+    updateConfig(appTree, { nugetPackages: {} });
+
+    (prompt as jest.MockedFunction<typeof prompt>)
+      .mockReset()
+      .mockImplementation((async () => {
+        return {};
+      }) as () => Promise<unknown> & { ui: PromptUI });
+
+    (getNxDotnetProjects as jest.MockedFunction<typeof getNxDotnetProjects>)
+      .mockReset()
+      .mockImplementation(() => new Map());
+  });
+
+  it('should run successfully', async () => {
+    await generator(appTree);
+    expect(true).toBeTruthy();
+  });
+});
diff --git a/packages/core/src/generators/restore/generator.ts b/packages/core/src/generators/restore/generator.ts
@@ -0,0 +1,21 @@
+import { Tree } from '@nrwl/devkit';
+import { DotNetClient, dotnetFactory } from '@nx-dotnet/dotnet';
+import {
+  getNxDotnetProjects,
+  getProjectFilesForProject,
+} from '@nx-dotnet/utils';
+
+export default async function (
+  host: Tree,
+  dotnetClient = new DotNetClient(dotnetFactory()),
+) {
+  const projects = getNxDotnetProjects(host);
+  for (const project of projects.values()) {
+    const projectFiles = getProjectFilesForProject(host, project);
+    for (const file of projectFiles) {
+      dotnetClient.restorePackages(file);
+    }
+  }
+
+  dotnetClient.restoreTools();
+}
diff --git a/packages/core/src/generators/restore/schema.json b/packages/core/src/generators/restore/schema.json
@@ -0,0 +1,9 @@
+{
+  "$schema": "http://json-schema.org/schema",
+  "cli": "nx",
+  "id": "Restore",
+  "title": "",
+  "type": "object",
+  "description": "Restores NuGet packages and .NET tools used by the workspace.",
+  "properties": {}
+}
diff --git a/packages/dotnet/src/lib/core/dotnet.client.ts b/packages/dotnet/src/lib/core/dotnet.client.ts
@@ -120,6 +120,16 @@ export class DotNetClient {
     return this.logAndExecute(cmd);
   }
 
+  restorePackages(project: string): Buffer {
+    const cmd = `${this.cliCommand.command} restore ${project}`;
+    return this.logAndExecute(cmd);
+  }
+
+  restoreTools(): Buffer {
+    const cmd = `${this.cliCommand.command} tool restore`;
+    return this.logAndExecute(cmd);
+  }
+
   private logAndExecute(cmd: string): Buffer {
     console.log(`Executing Command: ${cmd}`);
     return execSync(cmd, { stdio: 'inherit' });