Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions example.config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@ server:
http:
enabled: true
port: 3476
expose_openapi: false
tls:
enabled: false
cert: /etc/letsencrypt/live/yourdomain.com/fullchain.pem
Expand Down
2 changes: 2 additions & 0 deletions internal/config/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ type (
HTTP struct {
Enabled bool `mapstructure:"enabled"` // Whether the HTTP server is enabled
Port string `mapstructure:"port"` // Port for the HTTP server
ExposeOpenAPI bool `mapstructure:"expose_openapi"` // expose OpenAPI configuration
TLSConfig TLSConfig `mapstructure:"tls"` // TLS configuration for the HTTP server
CORSAllowedOrigins []string `mapstructure:"cors_allowed_origins"` // List of allowed origins for CORS
CORSAllowedHeaders []string `mapstructure:"cors_allowed_headers"` // List of allowed headers for CORS
Expand Down Expand Up @@ -276,6 +277,7 @@ func DefaultConfig() *Config {
HTTP: HTTP{
Enabled: true,
Port: "3476",
ExposeOpenAPI: false,
TLSConfig: TLSConfig{
Enabled: false,
},
Expand Down
52 changes: 38 additions & 14 deletions internal/servers/server.go
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@ import (
"net"
"net/http"
"net/http/pprof"
"os"
"time"

"github.com/grpc-ecosystem/go-grpc-middleware/v2/interceptors/logging"
Expand Down Expand Up @@ -305,24 +306,47 @@ func (s *Container) Run(
}),
}

mux := runtime.NewServeMux(muxOpts...)
// Create the gRPC gateway mux
grpcMux := runtime.NewServeMux(muxOpts...)

if err = grpcV1.RegisterPermissionHandler(ctx, mux, conn); err != nil {
return err
}
if err = grpcV1.RegisterSchemaHandler(ctx, mux, conn); err != nil {
return err
handlers := []func(context.Context, *runtime.ServeMux, *grpc.ClientConn) error{
grpcV1.RegisterPermissionHandler,
grpcV1.RegisterSchemaHandler,
grpcV1.RegisterDataHandler,
grpcV1.RegisterBundleHandler,
grpcV1.RegisterTenancyHandler,
}
if err = grpcV1.RegisterDataHandler(ctx, mux, conn); err != nil {
return err
}
if err = grpcV1.RegisterBundleHandler(ctx, mux, conn); err != nil {
return err

for _, handler := range handlers {
if err = handler(ctx, grpcMux, conn); err != nil {
return fmt.Errorf("failed to register handler: %w", err)
}
}
if err = grpcV1.RegisterTenancyHandler(ctx, mux, conn); err != nil {
return err

// Create a new http.ServeMux for serving your OpenAPI file and gRPC gateway
httpMux := http.NewServeMux()
const openAPIPath = "./docs/api-reference/openapi.json"

if srv.HTTP.ExposeOpenAPI {
httpMux.HandleFunc("/openapi.json", func(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodGet {
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
return
}
w.Header().Set("Content-Type", "application/json")
w.Header().Set("Cache-Control", "public, max-age=3600")
if _, err := os.Stat(openAPIPath); os.IsNotExist(err) {
http.Error(w, "OpenAPI specification not found", http.StatusNotFound)
return
}

http.ServeFile(w, r, openAPIPath)
})
}

// Handle all gRPC gateway routes
httpMux.Handle("/", grpcMux)

httpServer = &http.Server{
Addr: ":" + srv.HTTP.Port,
Handler: cors.New(cors.Options{
Expand All @@ -333,7 +357,7 @@ func (s *Container) Run(
http.MethodGet, http.MethodPost,
http.MethodHead, http.MethodPatch, http.MethodDelete, http.MethodPut,
},
}).Handler(mux),
}).Handler(httpMux),
ReadHeaderTimeout: 5 * time.Second,
}

Expand Down