Display environments and environment secrets

cschleiden · cschleiden · commit 58b52d243d3c · 2021-07-07T16:34:56.000+02:00
diff --git a/src/git/repository.ts b/src/git/repository.ts
@@ -1,17 +1,14 @@
-import * as vscode from "vscode";
-
-import { API, GitExtension, RefType } from "../typings/git";
-
 import { Octokit } from "@octokit/rest";
-import { Protocol } from "../external/protocol";
-import { flatten } from "../utils/array";
+import * as vscode from "vscode";
 import { getClient } from "../api/api";
 import { getSession } from "../auth/auth";
+import { Protocol } from "../external/protocol";
+import { API, GitExtension, RefType } from "../typings/git";
+import { flatten } from "../utils/array";
 
 async function getGitExtension(): Promise<API | undefined> {
-  const gitExtension = vscode.extensions.getExtension<GitExtension>(
-    "vscode.git"
-  );
+  const gitExtension =
+    vscode.extensions.getExtension<GitExtension>("vscode.git");
   if (gitExtension) {
     if (!gitExtension.isActive) {
       await gitExtension.activate();
@@ -84,6 +81,7 @@ export async function getGitHubProtocol(): Promise<Protocol | null> {
 export interface GitHubContext {
   client: Octokit;
 
+  id: number;
   owner: string;
   name: string;
 
@@ -116,6 +114,7 @@ export async function getGitHubContext(): Promise<GitHubContext | undefined> {
           client,
           name: protocolInfo.repositoryName,
           owner: protocolInfo.owner,
+          id: repoInfo.data.id,
           defaultBranch: `refs/heads/${repoInfo.data.default_branch}`,
           ownerIsOrg: repoInfo.data.owner?.type === "Organization",
           orgFeaturesEnabled:
diff --git a/src/model.ts b/src/model.ts
@@ -20,6 +20,17 @@ type OctokitData<
   >["data"][ResultProperty]
 >;
 
+type OctokitRepoData<
+  Operation extends keyof RestEndpointMethods["repos"],
+  ResultProperty extends keyof Await<
+    ReturnType<RestEndpointMethods["repos"][Operation]>
+  >["data"]
+> = GetElementType<
+  Await<
+    ReturnType<RestEndpointMethods["repos"][Operation]>
+  >["data"][ResultProperty]
+>;
+
 //
 // Domain types
 //
@@ -37,6 +48,13 @@ export type RepoSecret = OctokitData<"listRepoSecrets", "secrets">;
 
 export type OrgSecret = OctokitData<"listOrgSecrets", "secrets">;
 
+export type Environment = OctokitRepoData<"getAllEnvironments", "environments">;
+
+export type EnvironmentSecret = OctokitData<
+  "listEnvironmentSecrets",
+  "secrets"
+>;
+
 export type SelfHostedRunner = OctokitData<
   "listSelfHostedRunnersForRepo",
   "runners"
diff --git a/src/treeViews/settings.ts b/src/treeViews/settings.ts
@@ -1,8 +1,12 @@
 import * as vscode from "vscode";
-
-import { GitHubContext, getGitHubContext } from "../git/repository";
-import { OrgSecret, RepoSecret, SelfHostedRunner } from "../model";
-
+import { getGitHubContext, GitHubContext } from "../git/repository";
+import {
+  Environment,
+  EnvironmentSecret,
+  OrgSecret,
+  RepoSecret,
+  SelfHostedRunner,
+} from "../model";
 import { getAbsoluteIconPath } from "./icons";
 
 class OrgFeaturesNode extends vscode.TreeItem {
@@ -90,16 +94,58 @@ class OrgSecretNode extends vscode.TreeItem {
   }
 }
 
+class EnvironmentsNode extends vscode.TreeItem {
+  constructor(public readonly gitHubContext: GitHubContext) {
+    super("Environments", vscode.TreeItemCollapsibleState.Collapsed);
+
+    this.iconPath = new vscode.ThemeIcon("server-environment");
+  }
+}
+
+class EnvironmentNode extends vscode.TreeItem {
+  constructor(
+    public readonly gitHubContext: GitHubContext,
+    public readonly environment: Environment
+  ) {
+    super(environment.name, vscode.TreeItemCollapsibleState.Collapsed);
+
+    this.contextValue = "environment";
+  }
+}
+
+class EnvironmentSecretNode extends vscode.TreeItem {
+  constructor(
+    public readonly gitHubContext: GitHubContext,
+    public readonly secret: EnvironmentSecret
+  ) {
+    super(secret.name);
+
+    this.contextValue = "env-secret";
+  }
+}
+
+class EmptyEnvironmentSecretsNode extends vscode.TreeItem {
+  constructor() {
+    super("No environment secrets defined");
+  }
+}
+
 type SettingsExplorerNode =
   | OrgFeaturesNode
   | SelfHostedRunnersNode
   | SecretsNode
   | RepoSecretNode
-  | OrgSecretNode;
+  | OrgSecretNode
+  | EnvironmentsNode
+  | EnvironmentNode
+  | EnvironmentSecretNode
+  | EmptyEnvironmentSecretsNode;
 
 export class SettingsTreeProvider
-  implements vscode.TreeDataProvider<SettingsExplorerNode> {
-  private _onDidChangeTreeData = new vscode.EventEmitter<SettingsExplorerNode | null>();
+  implements vscode.TreeDataProvider<SettingsExplorerNode>
+{
+  private _onDidChangeTreeData =
+    new vscode.EventEmitter<SettingsExplorerNode | null>();
   readonly onDidChangeTreeData = this._onDidChangeTreeData.event;
 
   refresh(): void {
@@ -125,9 +171,14 @@ export class SettingsTreeProvider
       return [
         new SelfHostedRunnersNode(gitHubContext),
         new SecretsNode(gitHubContext),
+        new EnvironmentsNode(gitHubContext),
       ];
     }
 
+    //
+    // Secrets
+    //
+
     if (element instanceof SecretsNode) {
       const nodes = [new RepoSecretsNode(gitHubContext)];
 
@@ -164,17 +215,44 @@ export class SettingsTreeProvider
     }
 
     if (element instanceof SelfHostedRunnersNode) {
-      const result = await gitHubContext.client.actions.listSelfHostedRunnersForRepo(
-        {
+      const result =
+        await gitHubContext.client.actions.listSelfHostedRunnersForRepo({
           owner: gitHubContext.owner,
           repo: gitHubContext.name,
-        }
-      );
+        });
 
       const data = result.data.runners || [];
       return data.map((r) => new SelfHostedRunnerNode(gitHubContext, r));
     }
 
+    //
+    // Environments
+    //
+
+    if (element instanceof EnvironmentsNode) {
+      const result = await gitHubContext.client.repos.getAllEnvironments({
+        owner: gitHubContext.owner,
+        repo: gitHubContext.name,
+      });
+
+      const data = result.data.environments || [];
+      return data.map((e) => new EnvironmentNode(gitHubContext, e));
+    }
+
+    if (element instanceof EnvironmentNode) {
+      const result = await gitHubContext.client.actions.listEnvironmentSecrets({
+        repository_id: gitHubContext.id,
+        environment_name: element.environment.name,
+      });
+
+      const data = result.data.secrets;
+      if (!data) {
+        return [new EmptyEnvironmentSecretsNode()];
+      }
+
+      return data.map((s) => new EnvironmentSecretNode(gitHubContext, s));
+    }
+
     return [];
   }
 }
