From 4a4db417a0a25cd43e7a0f7c2e3ca21dd8d279e4 Mon Sep 17 00:00:00 2001
From: Will Morrison <will.morrison@starburstdata.com>
Date: Wed, 30 Oct 2024 09:04:50 -0400
Subject: [PATCH] Add bindings

---
 gateway-ha/src/main/java/io/trino/gateway/baseapp/BaseApp.java  | 2 ++
 .../io/trino/gateway/ha/router/OAuth2GatewayCookieProvider.java | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/gateway-ha/src/main/java/io/trino/gateway/baseapp/BaseApp.java b/gateway-ha/src/main/java/io/trino/gateway/baseapp/BaseApp.java
index aebdf4a67..b359ba461 100644
--- a/gateway-ha/src/main/java/io/trino/gateway/baseapp/BaseApp.java
+++ b/gateway-ha/src/main/java/io/trino/gateway/baseapp/BaseApp.java
@@ -31,6 +31,7 @@
 import io.trino.gateway.ha.resource.LoginResource;
 import io.trino.gateway.ha.resource.PublicResource;
 import io.trino.gateway.ha.resource.TrinoResource;
+import io.trino.gateway.ha.router.OAuth2GatewayCookieProvider;
 import io.trino.gateway.ha.security.AuthorizedExceptionMapper;
 import io.trino.gateway.proxyserver.ForProxy;
 import io.trino.gateway.proxyserver.ProxyRequestHandler;
@@ -121,6 +122,7 @@ public void configure(Binder binder)
         registerResources(binder);
         registerProxyResources(binder);
         jaxrsBinder(binder).bind(RoutingTargetHandler.class);
+        jaxrsBinder(binder).bind(OAuth2GatewayCookieProvider.class);
         addManagedApps(configuration, binder);
         jaxrsBinder(binder).bind(AuthorizedExceptionMapper.class);
         binder.bind(ProxyHandlerStats.class).in(Scopes.SINGLETON);
diff --git a/gateway-ha/src/main/java/io/trino/gateway/ha/router/OAuth2GatewayCookieProvider.java b/gateway-ha/src/main/java/io/trino/gateway/ha/router/OAuth2GatewayCookieProvider.java
index 3904b57cb..1a6d7d686 100644
--- a/gateway-ha/src/main/java/io/trino/gateway/ha/router/OAuth2GatewayCookieProvider.java
+++ b/gateway-ha/src/main/java/io/trino/gateway/ha/router/OAuth2GatewayCookieProvider.java
@@ -14,6 +14,7 @@
 package io.trino.gateway.ha.router;
 
 import com.google.inject.AbstractModule;
+import com.google.inject.Inject;
 import com.google.inject.Provides;
 import io.airlift.units.Duration;
 import io.trino.gateway.ha.config.HaGatewayConfiguration;
@@ -26,6 +27,7 @@ public class OAuth2GatewayCookieProvider
     private final List<String> deletePaths;
     private final Duration ttl;
 
+    @Inject
     public OAuth2GatewayCookieProvider(HaGatewayConfiguration configuration)
     {
         this.deletePaths = configuration.getOauth2GatewayCookieConfiguration().getDeletePaths();