probo/pkg/server/api/api.go

// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.

package api

import (
	"errors"
	"net/http"
	"time"

	"github.com/go-chi/chi/v5"
	"github.com/go-chi/cors"
	"github.com/vektah/gqlparser/v2/ast"
	"go.gearno.de/kit/httpserver"
	"go.gearno.de/kit/log"
	"go.probo.inc/probo/pkg/auth"
	"go.probo.inc/probo/pkg/authz"
	"go.probo.inc/probo/pkg/connector"
	"go.probo.inc/probo/pkg/probo"
	"go.probo.inc/probo/pkg/saferedirect"
	console_v1 "go.probo.inc/probo/pkg/server/api/console/v1"
	mcp_v1 "go.probo.inc/probo/pkg/server/api/mcp/v1"
	trust_v1 "go.probo.inc/probo/pkg/server/api/trust/v1"
	"go.probo.inc/probo/pkg/trust"
)

type (
	ConsoleAuthConfig struct {
		CookieName      string
		CookieDomain    string
		SessionDuration time.Duration
		CookieSecret    string
		CookieSecure    bool
	}

	TrustAuthConfig struct {
		CookieName        string
		CookieDomain      string
		CookieDuration    time.Duration
		TokenDuration     time.Duration
		ReportURLDuration time.Duration
		TokenSecret       string
		Scope             string
		TokenType         string
		CookieSecure      bool
	}

	Config struct {
		AllowedOrigins    []string
		Probo             *probo.Service
		Auth              *auth.Service
		Authz             *authz.Service
		Trust             *trust.Service
		SAML              *auth.SAMLService
		ConsoleAuth       ConsoleAuthConfig
		TrustAuth         TrustAuthConfig
		MCPConfig         MCPConfig
		ConnectorRegistry *connector.ConnectorRegistry
		SafeRedirect      *saferedirect.SafeRedirect
		CustomDomainCname string
		Logger            *log.Logger
	}

	MCPConfig struct {
		Version        string
		RequestTimeout time.Duration
		MaxRequestSize int64
	}

	Server struct {
		cfg               Config
		trustAPIHandler   http.Handler
		consoleAPIHandler http.Handler
		mcpAPIHandler     http.Handler
	}
)

var (
	ErrMissingProboService = errors.New("server configuration requires a valid probo.Service instance")
	ErrMissingAuthService  = errors.New("server configuration requires a valid auth.Service instance")
	ErrMissingAuthzService = errors.New("server configuration requires a valid authz.Service instance")
)

// GetConsoleSchema returns the GraphQL schema for the console API
func GetConsoleSchema() *ast.Schema {
	return console_v1.GetSchema()
}

func methodNotAllowed(w http.ResponseWriter, r *http.Request) {
	defer r.Body.Close()

	httpserver.RenderJSON(
		w,
		http.StatusMethodNotAllowed,
		map[string]string{
			"error": "method not allowed",
		},
	)
}

func notFound(w http.ResponseWriter, r *http.Request) {
	defer r.Body.Close()

	httpserver.RenderJSON(
		w,
		http.StatusNotFound,
		map[string]string{
			"error": "not found",
		},
	)
}

func NewServer(cfg Config) (*Server, error) {
	if cfg.Probo == nil {
		return nil, ErrMissingProboService
	}

	if cfg.Auth == nil {
		return nil, ErrMissingAuthService
	}

	if cfg.Authz == nil {
		return nil, ErrMissingAuthzService
	}

	trustAPIHandler := trust_v1.NewMux(
		cfg.Logger.Named("trust.v1"),
		cfg.Auth,
		cfg.Authz,
		cfg.Trust,
		console_v1.AuthConfig{
			CookieName:      cfg.ConsoleAuth.CookieName,
			CookieDomain:    cfg.ConsoleAuth.CookieDomain,
			SessionDuration: cfg.ConsoleAuth.SessionDuration,
			CookieSecret:    cfg.ConsoleAuth.CookieSecret,
			CookieSecure:    cfg.ConsoleAuth.CookieSecure,
		},
		trust_v1.TrustAuthConfig{
			CookieName:        cfg.TrustAuth.CookieName,
			CookieDomain:      cfg.TrustAuth.CookieDomain,
			CookieDuration:    cfg.TrustAuth.CookieDuration,
			TokenDuration:     cfg.TrustAuth.TokenDuration,
			ReportURLDuration: cfg.TrustAuth.ReportURLDuration,
			TokenSecret:       cfg.TrustAuth.TokenSecret,
			Scope:             cfg.TrustAuth.Scope,
			TokenType:         cfg.TrustAuth.TokenType,
			CookieSecure:      cfg.TrustAuth.CookieSecure,
		},
	)

	consoleAPIHandler := console_v1.NewMux(
		cfg.Logger.Named("console.v1"),
		cfg.Probo,
		cfg.Auth,
		cfg.Authz,
		console_v1.AuthConfig{
			CookieName:      cfg.ConsoleAuth.CookieName,
			CookieDomain:    cfg.ConsoleAuth.CookieDomain,
			SessionDuration: cfg.ConsoleAuth.SessionDuration,
			CookieSecret:    cfg.ConsoleAuth.CookieSecret,
			CookieSecure:    cfg.ConsoleAuth.CookieSecure,
		},
		cfg.ConnectorRegistry,
		cfg.SafeRedirect,
		cfg.CustomDomainCname,
		cfg.SAML,
	)

	mcpAPIHandler := mcp_v1.NewMux(
		cfg.Logger.Named("mcp.v1"),
		cfg.Probo,
		cfg.Auth,
		cfg.Authz,
		mcp_v1.Config{
			Version:        cfg.MCPConfig.Version,
			RequestTimeout: cfg.MCPConfig.RequestTimeout,
			MaxRequestSize: cfg.MCPConfig.MaxRequestSize,
		},
	)

	return &Server{
		cfg:               cfg,
		trustAPIHandler:   trustAPIHandler,
		consoleAPIHandler: consoleAPIHandler,
		mcpAPIHandler:     mcpAPIHandler,
	}, nil
}

func (s *Server) TrustAPIHandler() http.Handler {
	return s.trustAPIHandler
}

func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	corsOpts := cors.Options{
		AllowedOrigins:     s.cfg.AllowedOrigins,
		AllowedMethods:     []string{"GET", "POST", "PUT", "DELETE", "HEAD"},
		AllowedHeaders:     []string{"content-type", "traceparent", "authorization"},
		ExposedHeaders:     []string{"x-Request-id"},
		AllowCredentials:   true,
		MaxAge:             600, // 10 minutes (chrome >= 76 maximum value c.f. https://source.chromium.org/chromium/chromium/src/+/main:services/network/public/cpp/cors/preflight_result.cc;drc=52002151773d8cd9ffc5f557cd7cc880fddcae3e;l=36)
		OptionsPassthrough: false,
		Debug:              false,
	}

	w.Header().Set("X-Frame-Options", "DENY")
	w.Header().Set("X-XSS-Protection", "0")
	w.Header().Set("X-Content-Type-Options", "nosniff")
	w.Header().Set("Referrer-Policy", "no-referrer")
	w.Header().Set("Content-Security-Policy", "default-src 'self'")
	w.Header().Set("Permissions-Policy", "microphone=(), camera=(), geolocation=()")

	router := chi.NewRouter()
	router.MethodNotAllowed(methodNotAllowed)
	router.NotFound(notFound)

	router.Use(cors.Handler(corsOpts))

	router.Mount("/console/v1", s.consoleAPIHandler)
	router.Mount("/trust/v1", s.trustAPIHandler)
	router.Mount("/mcp/v1", s.mcpAPIHandler)

	router.ServeHTTP(w, r)
}