Merge pull request #183 from Olfi01/main

Enable GitHub Enterprise Server

Author: felipesu19

package.json

@@ -86,6 +86,12 @@
           "description": "The name of the repository's git remote that points to GitHub",
           "default": "origin",
           "scope": "window"
+        },
+        "github-actions.use-enterprise": {
+          "type": "boolean",
+          "markdownDescription": "If this is set to true, use the auth provider for the GitHub Enterprise URL configured in `github-enterprise.uri`",
+          "default": false,
+          "scope": "window"
         }
       }
     },

src/api/api.ts

@@ -1,11 +1,13 @@
 import {Octokit} from "@octokit/rest";
 import {version} from "../../package.json";
+import {getGitHubApiUri} from "../configuration/configuration";
 
 export const userAgent = `VS Code GitHub Actions (${version})`;
 
 export function getClient(token: string): Octokit {
   return new Octokit({
     auth: token,
-    userAgent: userAgent
+    userAgent: userAgent,
+    baseUrl: getGitHubApiUri()
   });
 }

src/auth/auth.ts

@@ -1,6 +1,8 @@
 import * as vscode from "vscode";
+import {useEnterprise} from "../configuration/configuration";
 
 const AUTH_PROVIDER_ID = "github";
+const AUTH_PROVIDER_ID_ENTERPRISE = "github-enterprise";
 const DEFAULT_SCOPES = ["repo", "workflow"];
 
 let signInPrompted = false;
@@ -67,7 +69,8 @@ async function getSessionInternal(
     typeof createOrForceMessage === "string"
       ? {forceNewSession: {detail: createOrForceMessage}}
       : {createIfNone: createOrForceMessage};
-  return await vscode.authentication.getSession(AUTH_PROVIDER_ID, getScopes(), options);
+  const authProviderId = useEnterprise() ? AUTH_PROVIDER_ID_ENTERPRISE : AUTH_PROVIDER_ID;
+  return await vscode.authentication.getSession(authProviderId, getScopes(), options);
 }
 
 function getScopes(): string[] {

src/configuration/configuration.ts

@@ -1,12 +1,23 @@
 import * as vscode from "vscode";
+import {deactivateLanguageServer, initLanguageServer} from "../workflow/languageServer";
+import {resetGitHubContext} from "../git/repository";
 
 const settingsKey = "github-actions";
+const DEFAULT_GITHUB_API = "https://api.github.com";
 
 export function initConfiguration(context: vscode.ExtensionContext) {
   context.subscriptions.push(
-    vscode.workspace.onDidChangeConfiguration(e => {
+    vscode.workspace.onDidChangeConfiguration(async e => {
       if (e.affectsConfiguration(getSettingsKey("workflows.pinned"))) {
         pinnedWorkflowsChangeHandlers.forEach(h => h());
+      } else if (
+        e.affectsConfiguration(getSettingsKey("use-enterprise")) ||
+        (useEnterprise() &&
+          (e.affectsConfiguration("github-enterprise.uri") || e.affectsConfiguration(getSettingsKey("remote-name"))))
+      ) {
+        await updateLanguageServerApiUrl(context);
+        resetGitHubContext();
+        await vscode.commands.executeCommand("github-actions.explorer.refresh");
       }
     })
   );
@@ -52,3 +63,19 @@ export function pinnedWorkflowsRefreshInterval(): number {
 export function getRemoteName(): string {
   return getConfiguration().get<string>(getSettingsKey("remote-name"), "origin");
 }
+
+export function useEnterprise(): boolean {
+  return getConfiguration().get<boolean>(getSettingsKey("use-enterprise"), false);
+}
+
+export function getGitHubApiUri(): string {
+  if (!useEnterprise()) return DEFAULT_GITHUB_API;
+  const base = getConfiguration().get<string>("github-enterprise.uri", DEFAULT_GITHUB_API).replace(/\/$/, "");
+  return base === DEFAULT_GITHUB_API ? base : `${base}/api/v3`;
+}
+
+async function updateLanguageServerApiUrl(context: vscode.ExtensionContext) {
+  await deactivateLanguageServer();
+
+  await initLanguageServer(context);
+}

src/git/repository.ts

@@ -4,11 +4,12 @@ import {Octokit} from "@octokit/rest";
 import {canReachGitHubAPI} from "../api/canReachGitHubAPI";
 import {handleSamlError} from "../api/handleSamlError";
 import {getSession} from "../auth/auth";
-import {getRemoteName} from "../configuration/configuration";
+import {getRemoteName, useEnterprise} from "../configuration/configuration";
 import {Protocol} from "../external/protocol";
 import {logDebug, logError} from "../log";
 import {API, GitExtension, RefType, RepositoryState} from "../typings/git";
 import {RepositoryPermission, getRepositoryPermission} from "./repository-permissions";
+import {getGitHubApiUri} from "../configuration/configuration";
 
 interface GitHubUrls {
   workspaceUri: vscode.Uri;
@@ -74,7 +75,11 @@ export async function getGitHubUrls(): Promise<GitHubUrls[] | null> {
           remote = [r.state.remotes[0]];
         }
 
-        if (remote.length > 0 && remote[0].pushUrl?.indexOf("github.com") !== -1) {
+        if (
+          remote.length > 0 &&
+          (remote[0].pushUrl?.indexOf("github.com") !== -1 ||
+            (useEnterprise() && remote[0].pushUrl?.indexOf(new URL(getGitHubApiUri()).host) !== -1))
+        ) {
           const url = remote[0].pushUrl;
 
           return {

src/workflow/languageServer.ts

@@ -11,6 +11,7 @@ import {userAgent} from "../api/api";
 import {getSession} from "../auth/auth";
 import {getGitHubContext} from "../git/repository";
 import {WorkflowSelector} from "./documentSelector";
+import {getGitHubApiUri, useEnterprise} from "../configuration/configuration";
 
 let client: BaseLanguageClient;
 
@@ -26,6 +27,7 @@ export async function initLanguageServer(context: vscode.ExtensionContext) {
   const initializationOptions: InitializationOptions = {
     sessionToken: session?.accessToken,
     userAgent: userAgent,
+    gitHubApiUrl: useEnterprise() ? getGitHubApiUri() : undefined,
     repos: ghContext?.repos.map(repo => ({
       id: repo.id,
       owner: repo.owner,