1 files changed, 14 insertions, 1 deletions

diff --git a/cmd/web/server.go b/cmd/web/server.go
index e930257..4867970 100644
--- a/cmd/web/server.go
+++ b/cmd/web/server.go
@@ -171,15 +171,25 @@ func webMain(cfg app.Config, embeddedTemplates, embeddedClients fs.FS, appVersio
                 Webauthn:     wauthn,
         }
 
+        // TODO: Clean up this hack and expose these to echo
+        hostname := fmt.Sprintf("https://%s", cfg.Hostnames[0])
+        if strings.HasPrefix(cfg.Hostnames[0], "dev.") {
+                hostname += ":8070"
+        }
+
         o2dc := &controllers.OAuth2DeviceController[*app.Session]{
                 Logger:            s.Logger,
                 AuthSessions:      authSessionStore,
                 OauthClients:      oauthClientStore,
-                Hostname:          fmt.Sprintf("https://%s", cfg.Hostnames[0]), // TODO
+                Hostname:          hostname,
                 PollSeconds:       cfg.OauthDevicePollSecs,
                 SessionExpiration: cfg.OauthSessionTimeout,
         }
 
+        od := controllers.Oauth2DiscoveryController{
+                Hostname: hostname,
+        }
+
         ph := &controllers.ProxyHandler{
                 Logger: s.Logger,
                 Users:  userStore,
@@ -260,5 +270,8 @@ func webMain(cfg app.Config, embeddedTemplates, embeddedClients fs.FS, appVersio
                 pg.GET("/:host/:port", ph.Handle)
         }
 
+        s.GET(models.Oauth2MetadataPath, od.Handle)
+        s.GET(models.Oauth2MetadataCompatPath, od.Handle)
+
         s.RunForever(!cfg.DisableBackgroundJobs)
 }