Add header support for HttpStreamFactory

This is a roll forward of 8ffded8

BUG=481544451
GWSQ_IGNORE: chriscampos@google.com
PiperOrigin-RevId: 882689078
Change-Id: I62e6d011bfcee3a7d70040a764c106103eff5e32

Author: chris-campos

java/com/google/copybara/remotefile/GclientHttpStreamFactory.java

@@ -22,6 +22,7 @@
 import com.google.api.client.http.HttpRequest;
 import com.google.api.client.http.HttpTransport;
 import com.google.api.client.http.javanet.NetHttpTransport;
+import com.google.common.collect.ImmutableMultimap;
 import com.google.copybara.credentials.CredentialIssuingException;
 import com.google.copybara.credentials.CredentialRetrievalException;
 import com.google.copybara.http.auth.AuthInterceptor;
@@ -50,7 +51,8 @@ public GclientHttpStreamFactory(HttpTransport javaNet, Duration timeout) {
   }
 
   @Override
-  public InputStream open(URL url, @Nullable AuthInterceptor auth)
+  public InputStream open(
+      URL url, @Nullable AuthInterceptor auth, ImmutableMultimap<String, String> headers)
       throws IOException, CredentialRetrievalException, CredentialIssuingException {
     HttpRequest req =
         javaNet
@@ -59,6 +61,7 @@ public InputStream open(URL url, @Nullable AuthInterceptor auth)
             .setReadTimeout((int) timeout.toMillis())
             .setConnectTimeout((int) timeout.toMillis())
             .setUseRawRedirectUrls(true);
+    headers.forEach((k, v) -> req.getHeaders().set(k, v));
     if (auth != null) {
       req.setInterceptor(auth.interceptor());
     }

java/com/google/copybara/remotefile/HttpStreamFactory.java

@@ -16,6 +16,7 @@
 
 package com.google.copybara.remotefile;
 
+import com.google.common.collect.ImmutableMultimap;
 import com.google.copybara.credentials.CredentialIssuingException;
 import com.google.copybara.credentials.CredentialRetrievalException;
 import com.google.copybara.http.auth.AuthInterceptor;
@@ -35,6 +36,19 @@ public interface HttpStreamFactory {
    * @param url The URL to open.
    * @param auth The interceptor to use for authentication. If null, no authentication will be used.
    */
-  InputStream open(URL url, @Nullable AuthInterceptor auth)
+  default InputStream open(URL url, @Nullable AuthInterceptor auth)
+      throws IOException, CredentialRetrievalException, CredentialIssuingException {
+    return open(url, auth, ImmutableMultimap.of());
+  }
+
+  /**
+   * Open the referenced URL and return the stream to the contents.
+   *
+   * @param url The URL to open.
+   * @param auth The interceptor to use for authentication. If null, no authentication will be used.
+   * @param headers The headers to send in the HTTP request.
+   */
+  InputStream open(
+      URL url, @Nullable AuthInterceptor auth, ImmutableMultimap<String, String> headers)
       throws IOException, CredentialRetrievalException, CredentialIssuingException;
 }