Support GitHub Enterprise.

BUG=477915865
GWSQ_IGNORE: vcarbonaro@google.com
PiperOrigin-RevId: 874716222
Change-Id: I5766381792022f125c580615fcc12dbd2983cc78

Author: vincentcarbonaro

java/com/google/copybara/git/GitHubOptions.java

@@ -22,7 +22,6 @@
 import com.google.api.client.http.javanet.NetHttpTransport;
 import com.google.common.annotations.VisibleForTesting;
 import com.google.common.base.Preconditions;
-import com.google.common.base.VerifyException;
 import com.google.common.collect.ImmutableList;
 import com.google.copybara.GeneralOptions;
 import com.google.copybara.LazyResourceLoader;
@@ -227,7 +226,8 @@ protected GitHubApiTransport newTransport(
     if (gitHubHostName.equals("github.com")) {
       return newTransport(repo, storePath, console);
     }
-    throw new VerifyException("Non-mygithub.libinneed.workers.dev Rest API is not yet supported.");
+    return new GitHubApiTransportImpl(
+        repo, newHttpTransport(), storePath, gitHubApiBearerAuth, console, gitHubHostName);
   }
 
   protected HttpTransport newHttpTransport() {

java/com/google/copybara/git/github/api/GitHubApiTransportImpl.java

@@ -27,6 +27,7 @@
 import com.google.api.client.json.JsonFactory;
 import com.google.api.client.json.JsonObjectParser;
 import com.google.api.client.json.gson.GsonFactory;
+import com.google.common.annotations.VisibleForTesting;
 import com.google.common.base.Preconditions;
 import com.google.common.collect.ImmutableListMultimap;
 import com.google.common.collect.Iterables;
@@ -55,9 +56,11 @@ public class GitHubApiTransportImpl implements GitHubApiTransport {
   private static final FluentLogger logger = FluentLogger.forEnclosingClass();
 
   private static final JsonFactory JSON_FACTORY = new GsonFactory();
-  private static final String API_URL = "https://api.github.com";
-  private static final String GITHUB_WEB_URL = "https://github.com";
+  private static final String GITHUB_DOT_COM_API_URL = "https://api.github.com";
+  private static final String GITHUB_DOT_COM_WEB_URL = "https://github.com";
 
+  private final String apiUrl;
+  private final String webUrl;
   private final GitRepository repo;
   private final HttpTransport httpTransport;
   private final String storePath;
@@ -71,6 +74,24 @@ public GitHubApiTransportImpl(GitRepository repo, HttpTransport httpTransport,
     this.storePath = storePath;
     this.console = Preconditions.checkNotNull(console);
     this.bearerAuth = bearerAuth;
+    this.apiUrl = GITHUB_DOT_COM_API_URL;
+    this.webUrl = GITHUB_DOT_COM_WEB_URL;
+  }
+
+  public GitHubApiTransportImpl(
+      GitRepository repo,
+      HttpTransport httpTransport,
+      String storePath,
+      boolean bearerAuth,
+      Console console,
+      String webUrl) {
+    this.repo = Preconditions.checkNotNull(repo);
+    this.httpTransport = Preconditions.checkNotNull(httpTransport);
+    this.storePath = storePath;
+    this.console = Preconditions.checkNotNull(console);
+    this.bearerAuth = bearerAuth;
+    this.webUrl = buildGhesWebUrl(Preconditions.checkNotNull(webUrl));
+    this.apiUrl = buildGhesApiUrl(this.webUrl);
   }
 
   @SuppressWarnings("unchecked")
@@ -87,7 +108,7 @@ public <T> T get(String path, Type responseType, ImmutableListMultimap<String, S
       Object responseObj = GsonParserUtil.parseHttpResponse(response, responseType, false);
       if (responseObj instanceof PaginatedPayload) {
         return (T)
-            ((PaginatedPayload) responseObj).annotatePayload(API_URL, maybeGetLinkHeader(response));
+            ((PaginatedPayload) responseObj).annotatePayload(apiUrl, maybeGetLinkHeader(response));
       }
       return (T) responseObj;
     } catch (HttpResponseException e) {
@@ -152,7 +173,7 @@ public <T> T post(String path, Object request, Type responseType, String request
           response, responseType, false);
       if (responseObj instanceof PaginatedPayload) {
         return (T)
-            ((PaginatedPayload) responseObj).annotatePayload(API_URL, maybeGetLinkHeader(response));
+            ((PaginatedPayload) responseObj).annotatePayload(apiUrl, maybeGetLinkHeader(response));
       }
       return (T) responseObj;
 
@@ -221,7 +242,7 @@ private HttpRequestFactory getHttpRequestFactory(
 
   private GenericUrl getFullEndpointURL(String path) {
     String maybePrefix = path.startsWith("/") ? "" : "/";
-    return new GenericUrl(URI.create(API_URL + maybePrefix + path));
+    return new GenericUrl(URI.create(apiUrl + maybePrefix + path));
   }
 
   /**
@@ -231,10 +252,10 @@ private GenericUrl getFullEndpointURL(String path) {
    */
   private UserPassword getCredentials() throws RepoException, ValidationException {
     try {
-      return repo.credentialFill(API_URL);
+      return repo.credentialFill(apiUrl);
     } catch (ValidationException e) {
       try {
-        return repo.credentialFill(GITHUB_WEB_URL);
+        return repo.credentialFill(webUrl);
       } catch (ValidationException e1) {
         // Ugly, but helpful...
         throw new ValidationException(String.format(
@@ -252,4 +273,22 @@ private UserPassword getCredentials() throws RepoException, ValidationException
       }
     }
   }
+
+  private static String buildGhesWebUrl(String hostName) {
+    return "https://" + hostName;
+  }
+
+  private static String buildGhesApiUrl(String hostName) {
+    return hostName + "/api/v3";
+  }
+
+  @VisibleForTesting
+  public String getApiUrl() {
+    return apiUrl;
+  }
+
+  @VisibleForTesting
+  public String getWebUrl() {
+    return webUrl;
+  }
 }

javatests/com/google/copybara/git/github/api/GitHubApiTransportImplTest.java

@@ -133,6 +133,27 @@ public LowLevelHttpResponse execute() throws IOException {
     assertThat(headers).containsEntry("authorization", ImmutableList.of("Bearer SECRET"));
   }
 
+  @Test
+  public void testDefaultUrlsPopulated() throws Exception {
+    httpTransport = createMockHttpTransport(new IOException());
+    GitHubApiTransportImpl transport =
+        new GitHubApiTransportImpl(repo, httpTransport, "store", false, new TestingConsole());
+
+    assertThat(transport.getApiUrl()).isEqualTo("https://api.github.com");
+    assertThat(transport.getWebUrl()).isEqualTo("https://github.com");
+  }
+
+  @Test
+  public void testGhesUrlsPopulated() throws Exception {
+    httpTransport = createMockHttpTransport(new IOException());
+    GitHubApiTransportImpl transport =
+        new GitHubApiTransportImpl(
+            repo, httpTransport, "store", false, new TestingConsole(), "github.enterprise.com");
+
+    assertThat(transport.getApiUrl()).isEqualTo("https://github.enterprise.com/api/v3");
+    assertThat(transport.getWebUrl()).isEqualTo("https://github.enterprise.com");
+  }
+
   private void runTestThrowsHttpResponseException(Callable<?> c) throws Exception {
     HttpResponseException ex =
         new HttpResponseException.Builder(STATUS_CODE, ERROR_MESSAGE, new HttpHeaders()).build();