feat: Allow overriding authentication scheme

broadinstitute · Jun 4, 2024 · d9d467e · d9d467e
1 parent 1a4e18e
commit d9d467e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/dockerHashing/src/main/scala/cromwell/docker/registryv2/DockerRegistryV2Abstract.scala b/dockerHashing/src/main/scala/cromwell/docker/registryv2/DockerRegistryV2Abstract.scala
@@ -157,6 +157,11 @@ abstract class DockerRegistryV2Abstract(override val config: DockerRegistryConfi
       .handleErrorWith(tryOCIManifest)
   }
 
+  /**
+   * Gets the authorization scheme to use for the token request.
+   */
+  protected def getAuthorizationScheme(dockerImageIdentifier: DockerImageIdentifier): AuthScheme = AuthScheme.Bearer
+
   /**
     * Returns true if this flow is able to process this docker image,
     * false otherwise
@@ -241,7 +246,7 @@ abstract class DockerRegistryV2Abstract(override val config: DockerRegistryConfi
                               manifestHeader: Accept
   ): IO[Request[IO]] = {
     val authorizationHeader: Option[Authorization] =
-      token.map(t => Authorization(Credentials.Token(AuthScheme.Bearer, t)))
+      token.map(t => Authorization(Credentials.Token(getAuthorizationScheme(imageId), t)))
     val request = Method.GET(
       buildManifestUri(imageId),
       List(