Merge pull request #1 from Altinity/add_cors_to_sse

Add cors to sse

Author: Slach

.gitignore

@@ -365,3 +365,4 @@ $RECYCLE.BIN/
 
 # Local configs
 *.secret.yaml
+.aider*

README.md

@@ -1,6 +1,7 @@
 <div align="center">
 
-![Build Binaries](https://github.com/centralmind/gateway/actions/workflows/build-binaries.yml/badge.svg) &nbsp; <a href="https://discord.gg/XFhaUG4F5x"><img src="https://dcbadge.limes.pink/api/server/https://discord.gg/XFhaUG4F5x" height="20"></a> &nbsp;&nbsp;<a href="https://t.me/+TM3T1SikjzA4ZWVi"><img src="https://img.shields.io/badge/telegram-%E2%9D%A4%EF%B8%8F-252850?style=plastic&logo=telegram" height=20></a> &nbsp;&nbsp; <a href="https://docs.centralmind.ai"><img src="https://img.shields.io/badge/Full%20Documentation-blue?style=for-the-badge&logo=rocket&logoColor=white" height="20"></a>
+![Build Binaries](https://github.com/centralmind/gateway/actions/workflows/build-binaries.yml/badge.svg) &nbsp; <a href="https://discord.gg/XFhaUG4F5x"><img src="https://dcbadge.limes.pink/api/server/https://discord.gg/XFhaUG4F5x" height="20"></a> &nbsp;&nbsp;<a href="https://t.me/+TM3T1SikjzA4ZWVi"><img src="https://img.shields.io/badge/telegram-%E2%9D%A4%EF%B8%8F-252850?style=plastic&logo=telegram" height=20></a> &nbsp;&nbsp; <a href="https://docs.centralmind.ai"><img src="https://img.shields.io/badge/Full%20Documentation-blue?style=for-the-badge&logo=rocket&logoColor=white" height="20"></a>&nbsp;&nbsp; <a href="https://cursor.com/install-mcp?name=CentralMind%20Database%20Gateway&config=eyJjb21tYW5kIjoiZG9ja2VyIHJ1biAtaSAtLXBsYXRmb3JtIGxpbnV4L2FtZDY0IGdoY3IuaW8vY2VudHJhbG1pbmQvZ2F0ZXdheTp2MC4yLjE4IC0tY29ubmVjdGlvbi1zdHJpbmcgcG9zdGdyZXNxbDovL215X3VzZXI6bXlfcGFzc0Bsb2NhbGhvc3Q6NTQzMi9teWRiIHN0YXJ0IHN0ZGlvIn0%3D"><img height="21" src="https://cursor.com/deeplink/mcp-install-dark.svg"></a>
+
 
 </div>
 
@@ -15,7 +16,7 @@ Simple way to expose your database to AI-Agent via MCP or OpenAPI 3.1 protocols.
 
 ```bash
 docker run --platform linux/amd64 -p 9090:9090 \
-  ghcr.io/centralmind/gateway:v0.2.14 start \
+  ghcr.io/centralmind/gateway:v0.2.18 start \
   --connection-string "postgres://db-user:db-password@db-host/db-name?sslmode=require"
 ```
 

cors/cors.go

@@ -0,0 +1,21 @@
+package cors
+
+import "net/http"
+
+// ApplyCORSHeaders adds the standard CORS headers to a response
+// For handlers that are not wrapped in middleware
+func ApplyCORSHeaders(w http.ResponseWriter, allowedMethods string) {
+	w.Header().Set("Access-Control-Allow-Origin", "*")
+	w.Header().Set("Access-Control-Allow-Methods", allowedMethods+", OPTIONS")
+	w.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization, mcp-protocol-version")
+}
+
+// HandlePreflight checks if the request is a preflight OPTIONS request and handles it
+// Returns true if the request was handled (caller should return immediately)
+func HandlePreflight(w http.ResponseWriter, r *http.Request) bool {
+	if r.Method == http.MethodOptions {
+		w.WriteHeader(http.StatusOK)
+		return true
+	}
+	return false
+}

plugins/oauth/README.md

@@ -202,6 +202,8 @@ oauth:
   client_id: "xxx"
   client_secret: "xxx"
   user_info_url: "https://your-tenant.auth0.com/userinfo"
+  provider_auth_url: "https://your-tenant.auth0.com/authorize"
+  provider_token_url: "https://your-tenant.auth0.com/oauth/token"
   scopes:
     - "openid"
     - "profile"

plugins/oauth/auth_metadata.go

@@ -15,19 +15,26 @@ type Metadata struct {
 }
 
 func NewMetadata(issuer url.URL, authorizationEndpoint, tokenEndpoint string, registrationEndpoint string) Metadata {
+	buildURL := func(endpoint string) string {
+		if u, err := url.Parse(endpoint); err == nil && u.IsAbs() {
+			return endpoint
+		}
+		return (&url.URL{Scheme: issuer.Scheme, Host: issuer.Host, Path: endpoint}).String()
+	}
+
 	metadata := Metadata{
 		Issuer:                            issuer.String(),
-		AuthorizationEndpoint:             (&url.URL{Scheme: issuer.Scheme, Host: issuer.Host, Path: authorizationEndpoint}).String(),
+		AuthorizationEndpoint:             buildURL(authorizationEndpoint),
 		ResponseTypesSupported:            []string{"code"},
 		CodeChallengeMethodsSupported:     []string{"S256"},
-		TokenEndpoint:                     (&url.URL{Scheme: issuer.Scheme, Host: issuer.Host, Path: tokenEndpoint}).String(),
+		TokenEndpoint:                     buildURL(tokenEndpoint),
 		TokenEndpointAuthMethodsSupported: []string{"client_secret_post"},
 		GrantTypesSupported:               []string{"authorization_code", "refresh_token"},
 	}
 
 	// Add registration endpoint if provided
 	if registrationEndpoint != "" {
-		metadata.RegistrationEndpoint = (&url.URL{Scheme: issuer.Scheme, Host: issuer.Host, Path: registrationEndpoint}).String()
+		metadata.RegistrationEndpoint = buildURL(registrationEndpoint)
 	}
 
 	return metadata

plugins/oauth/config.go

@@ -147,16 +147,16 @@ func (c *Config) WithDefaults() {
 		c.TokenHeader = "Authorization"
 	}
 	if c.AuthURL == "" {
-		c.AuthURL = "/oauth/authorize"
+		c.AuthURL = "/oauth/authorize/"
 	}
 	if c.CallbackURL == "" {
-		c.CallbackURL = "/oauth/callback"
+		c.CallbackURL = "/oauth/callback/"
 	}
 	if c.TokenURL == "" {
-		c.TokenURL = "/oauth/token"
+		c.TokenURL = "/oauth/token/"
 	}
 	if c.RegisterURL == "" {
-		c.RegisterURL = "/oauth/register"
+		c.RegisterURL = "/oauth/register/"
 	}
 	if c.ClientRegistration.ClientSecretExpirySeconds == 0 {
 		c.ClientRegistration.ClientSecretExpirySeconds = 30 * 24 * 60 * 60 // 30 days

plugins/oauth/http_helpers.go

@@ -1,42 +1,18 @@
 package oauth
 
 import (
+	"github.com/centralmind/gateway/cors"
 	"net/http"
 )
 
 // CORSMiddleware applies standard CORS headers to the response
 func CORSMiddleware(handler http.Handler) http.Handler {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		// Add CORS headers
-		w.Header().Set("Access-Control-Allow-Origin", "*")
-		w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS")
-		w.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization")
-
-		// Handle preflight OPTIONS request
-		if r.Method == "OPTIONS" {
-			w.WriteHeader(http.StatusOK)
+		cors.ApplyCORSHeaders(w, "GET, POST")
+		if cors.HandlePreflight(w, r) {
 			return
 		}
-
 		// Call the original handler
 		handler.ServeHTTP(w, r)
 	})
 }
-
-// ApplyCORSHeaders adds the standard CORS headers to a response
-// For handlers that are not wrapped in middleware
-func ApplyCORSHeaders(w http.ResponseWriter, allowedMethods string) {
-	w.Header().Set("Access-Control-Allow-Origin", "*")
-	w.Header().Set("Access-Control-Allow-Methods", allowedMethods+", OPTIONS")
-	w.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization")
-}
-
-// HandlePreflight checks if the request is a preflight OPTIONS request and handles it
-// Returns true if the request was handled (caller should return immediately)
-func HandlePreflight(w http.ResponseWriter, r *http.Request) bool {
-	if r.Method == "OPTIONS" {
-		w.WriteHeader(http.StatusOK)
-		return true
-	}
-	return false
-}

plugins/oauth/plugin.go

@@ -148,9 +148,17 @@ func (p *Plugin) RegisterRoutes(mux *http.ServeMux) {
 	if err != nil {
 		return
 	}
+
+	getPath := func(rawURL string) string {
+		if u, urlParseErr := url.Parse(rawURL); urlParseErr == nil && u.Path != "" {
+			return u.Path
+		}
+		return rawURL
+	}
+
 	// Register HTTP handlers with CORS middleware
-	mux.Handle(p.config.AuthURL, CORSMiddleware(http.HandlerFunc(p.HandleAuthorize)))
-	mux.Handle(p.config.CallbackURL, CORSMiddleware(http.HandlerFunc(p.HandleCallback)))
+	mux.Handle(getPath(p.config.AuthURL), CORSMiddleware(http.HandlerFunc(p.HandleAuthorize)))
+	mux.Handle(getPath(p.config.CallbackURL), CORSMiddleware(http.HandlerFunc(p.HandleCallback)))
 
 	// Set up and register the token endpoint
 	tokenPath := p.config.TokenURL // Use the configured token URL
@@ -160,7 +168,7 @@ func (p *Plugin) RegisterRoutes(mux *http.ServeMux) {
 
 	// Register the token handler with CORS middleware
 	tokenHandler := http.HandlerFunc(p.HandleToken)
-	mux.Handle(tokenPath, CORSMiddleware(tokenHandler))
+	mux.Handle(getPath(tokenPath), CORSMiddleware(tokenHandler))
 
 	// Register dynamic client registration endpoint if enabled
 	if p.config.ClientRegistration.Enabled {
@@ -169,7 +177,7 @@ func (p *Plugin) RegisterRoutes(mux *http.ServeMux) {
 
 		// Register the handler with CORS middleware
 		registrationHandler := http.HandlerFunc(p.HandleRegister)
-		mux.Handle(p.config.RegisterURL, CORSMiddleware(registrationHandler))
+		mux.Handle(getPath(p.config.RegisterURL), CORSMiddleware(registrationHandler))
 	}
 
 	// Register the well-known endpoint with CORS middleware

server/sse.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"github.com/centralmind/gateway/cors"
 	"github.com/centralmind/gateway/xcontext"
 	"net/http"
 	"net/http/httptest"
@@ -87,10 +88,8 @@ func (s *SSEServer) handleSSE(w http.ResponseWriter, r *http.Request) {
 	w.Header().Set("Content-Type", "text/event-stream")
 	w.Header().Set("Cache-Control", "no-cache")
 	w.Header().Set("Connection", "keep-alive")
-	w.Header().Set("Access-Control-Allow-Origin", "*")
-
-	if r.Method == http.MethodOptions {
-		w.WriteHeader(http.StatusOK)
+	cors.ApplyCORSHeaders(w, "GET")
+	if cors.HandlePreflight(w, r) {
 		return
 	}