Enable GitHub Enterprise Server

Author: flmeyer

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 {isUseEnterprise} 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 = isUseEnterprise() ? 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,16 @@
 import * as vscode from "vscode";
+import {deactivateLanguageServer} from "../workflow/languageServer";
 
 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 => {
       if (e.affectsConfiguration(getSettingsKey("workflows.pinned"))) {
         pinnedWorkflowsChangeHandlers.forEach(h => h());
+      } else if (e.affectsConfiguration(getSettingsKey("use-enterprise"))) {
+        updateLanguageServerApiUrl();
       }
     })
   );
@@ -52,3 +56,19 @@ export function pinnedWorkflowsRefreshInterval(): number {
 export function getRemoteName(): string {
   return getConfiguration().get<string>(getSettingsKey("remote-name"), "origin");
 }
+
+export function isUseEnterprise(): boolean {
+  return getConfiguration().get<boolean>(getSettingsKey("use-enterprise"), false);
+}
+
+export function getGitHubApiUri(): string {
+  if (!isUseEnterprise()) 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`;
+}
+
+function updateLanguageServerApiUrl() {
+  // deactivateLanguageServer();
+  // TODO: restart language server with new URL configured
+  throw new Error("Can't yet change the api url parameter without restart.");
+}

src/git/repository.ts

@@ -4,7 +4,7 @@ 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, isUseEnterprise} from "../configuration/configuration";
 import {Protocol} from "../external/protocol";
 import {logDebug, logError} from "../log";
 import {API, GitExtension, RefType, RepositoryState} from "../typings/git";
@@ -74,7 +74,7 @@ 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 || isUseEnterprise())) {
           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, isUseEnterprise} 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: isUseEnterprise() ? getGitHubApiUri() : undefined,
     repos: ghContext?.repos.map(repo => ({
       id: repo.id,
       owner: repo.owner,