multica/server/internal/handler/qoder_catalog.go

package handler

import (
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"io"
	"net/http"
	"net/url"
	"strings"
	"time"

	"github.com/go-chi/chi/v5"
	"github.com/jackc/pgx/v5"
	"github.com/multica-ai/multica/server/internal/qoderruntime"
	db "github.com/multica-ai/multica/server/pkg/db/generated"
)

const qoderAPIURL = "https://api.qoder.com/api/v1/cloud"

// Managed connections may only reach Qoder's public regional APIs, preventing SSRF.
var qoderAPIURLs = map[string]bool{
	qoderAPIURL:                             true,
	"https://api.qoder.com.cn/api/v1/cloud": true,
}

func qoderBaseURL(raw string) (string, bool) {
	base := strings.TrimRight(strings.TrimSpace(raw), "/")
	if base == "" {
		return qoderAPIURL, true
	}
	return base, qoderAPIURLs[base]
}

type qoderCatalogItem struct {
	ID   string `json:"id"`
	Name string `json:"name"`
}

// Only names and identifiers leave the server. Provider configurations may contain secrets.
func fetchQoderCatalog(ctx context.Context, baseURL, token, kind string) ([]qoderCatalogItem, error) {
	client := &http.Client{Timeout: 15 * time.Second, CheckRedirect: func(*http.Request, []*http.Request) error { return http.ErrUseLastResponse }}
	items := []qoderCatalogItem{}
	cursor := ""
	seen := map[string]bool{}
	ctx, cancel := context.WithTimeout(ctx, 30*time.Second)
	defer cancel()
	for page := 0; page < 10; page++ {
		query := url.Values{"limit": {"100"}}
		if cursor != "" {
			query.Set("after_id", cursor)
		}
		req, err := http.NewRequestWithContext(ctx, http.MethodGet, baseURL+"/"+kind+"?"+query.Encode(), nil)
		if err != nil {
			return nil, fmt.Errorf("invalid Qoder Cloud Agents catalog request")
		}
		req.Header.Set("Authorization", "Bearer "+token)
		res, err := client.Do(req)
		if err != nil {
			return nil, fmt.Errorf("Qoder Cloud Agents connection failed")
		}
		if res.StatusCode != http.StatusOK {
			res.Body.Close()
			return nil, fmt.Errorf("Qoder Cloud Agents catalog returned HTTP %d", res.StatusCode)
		}
		var result struct {
			Data []struct {
				ID         string  `json:"id"`
				Name       string  `json:"name"`
				ArchivedAt *string `json:"archived_at"`
			} `json:"data"`
			HasMore bool   `json:"has_more"`
			LastID  string `json:"last_id"`
		}
		raw, err := io.ReadAll(io.LimitReader(res.Body, (2<<20)+1))
		res.Body.Close()
		if err != nil || len(raw) > 2<<20 || json.Unmarshal(raw, &result) != nil || result.Data == nil {
			return nil, fmt.Errorf("invalid Qoder Cloud Agents catalog response")
		}
		for _, item := range result.Data {
			prefix := "env_"
			if kind == "agents" {
				prefix = "agent_"
			}
			if !strings.HasPrefix(item.ID, prefix) {
				return nil, fmt.Errorf("invalid Qoder Cloud Agents catalog identity")
			}
			if item.ArchivedAt != nil || seen[item.ID] {
				continue
			}
			seen[item.ID] = true
			name := strings.TrimSpace(item.Name)
			if name == "" {
				name = item.ID
			}
			items = append(items, qoderCatalogItem{ID: item.ID, Name: name})
		}
		if !result.HasMore {
			return items, nil
		}
		if result.LastID == "" || result.LastID == cursor || len(result.Data) == 0 {
			return nil, fmt.Errorf("invalid Qoder Cloud Agents pagination")
		}
		cursor = result.LastID
	}
	return nil, fmt.Errorf("Qoder Cloud Agents catalog exceeds pagination limit")
}

func (h *Handler) ListQoderAgents(w http.ResponseWriter, r *http.Request) {
	ws, ok := parseUUIDOrBadRequest(w, chi.URLParam(r, "id"), "workspace_id")
	if !ok {
		return
	}
	if h.QoderRateLimiter != nil && !h.QoderRateLimiter.Allow(r.Context(), "qoder-catalog:"+uuidToString(ws)) {
		w.Header().Set("Retry-After", "60")
		writeError(w, http.StatusTooManyRequests, "Qoder Cloud Agents catalog request limit reached")
		return
	}

	cfg, _, _, _, err := h.readQoderSettings(r.Context(), uuidToString(ws))
	if errors.Is(err, pgx.ErrNoRows) {
		writeError(w, 409, "configure a Qoder Cloud Agents connection first")
		return
	}
	if err != nil {
		writeError(w, 500, "cannot read Qoder Cloud Agents configuration")
		return
	}
	base, ok := qoderBaseURL(cfg.BaseURL)
	if !ok {
		writeError(w, 409, "unsupported Qoder Cloud Agents API URL")
		return
	}
	items, err := fetchQoderCatalog(r.Context(), base, cfg.QoderToken, "agents")
	if err != nil {
		writeError(w, 502, err.Error())
		return
	}
	writeJSON(w, 200, map[string]any{"items": items})
}

func (h *Handler) ListQoderEnvironments(w http.ResponseWriter, r *http.Request) {
	ws, ok := parseUUIDOrBadRequest(w, chi.URLParam(r, "id"), "workspace_id")
	if !ok {
		return
	}
	if h.QoderRateLimiter != nil && !h.QoderRateLimiter.Allow(r.Context(), "qoder-catalog:"+uuidToString(ws)) {
		w.Header().Set("Retry-After", "60")
		writeError(w, http.StatusTooManyRequests, "Qoder Cloud Agents catalog request limit reached")
		return
	}

	var input struct {
		Token   string `json:"qoder_token"`
		BaseURL string `json:"base_url"`
	}
	dec := json.NewDecoder(http.MaxBytesReader(w, r.Body, 16<<10))
	dec.DisallowUnknownFields()
	if dec.Decode(&input) != nil {
		writeError(w, 400, "invalid Qoder PAT request")
		return
	}
	var extra any
	if dec.Decode(&extra) != io.EOF {
		writeError(w, 400, "request must contain one JSON object")
		return
	}
	cfg, _, _, _, err := h.readQoderSettings(r.Context(), uuidToString(ws))
	if err != nil && !errors.Is(err, pgx.ErrNoRows) {
		writeError(w, 500, "cannot read Qoder Cloud Agents configuration")
		return
	}
	requested := input.BaseURL
	if strings.TrimSpace(requested) == "" {
		requested = cfg.BaseURL
	}
	base, ok := qoderBaseURL(requested)
	if !ok {
		writeError(w, 400, "unsupported Qoder Cloud Agents API URL")
		return
	}
	token := strings.TrimSpace(input.Token)
	if token == "" {
		token = cfg.QoderToken
	}
	if token == "" || strings.ContainsAny(token, "\r\n\t ") {
		writeError(w, 400, "Qoder PAT is required")
		return
	}
	items, err := fetchQoderCatalog(r.Context(), base, token, "environments")
	if err != nil {
		writeError(w, 502, err.Error())
		return
	}
	writeJSON(w, 200, map[string]any{"items": items})
}

// qoderOverrides lists per-agent execution settings the Qoder bridge refuses
// at run time (see qoderruntime.prepare); saving them would create an agent
// that fails every task.
type qoderOverrides struct {
	skills, customEnv, customArgs     int
	mcpConfig                         json.RawMessage
	model, thinkingLevel, serviceTier string
}

func (o qoderOverrides) validate() error {
	mcp := strings.TrimSpace(string(o.mcpConfig))
	if o.skills > 0 || o.customEnv > 0 || o.customArgs > 0 || (mcp != "" && mcp != "null" && mcp != "{}") || o.model != "" || o.thinkingLevel != "" || o.serviceTier != "" {
		return fmt.Errorf("Qoder agents take skills, tools, credentials and model from the Qoder Agent; remove them from this agent")
	}
	return nil
}

// storedQoderOverrides reads an agent's saved execution settings. Skills live
// in their own table and are counted by the caller when they matter.
func storedQoderOverrides(agent db.Agent) qoderOverrides {
	var args []string
	_ = json.Unmarshal(agent.CustomArgs, &args)
	return qoderOverrides{
		customEnv:     len(unmarshalCustomEnv(agent)),
		customArgs:    len(args),
		mcpConfig:     agent.McpConfig,
		model:         agent.Model.String,
		thinkingLevel: agent.ThinkingLevel.String,
		serviceTier:   agent.ServiceTier.String,
	}
}

// rejectQoderOverrides writes a 400 and reports true when the agent runs on
// Qoder and o would leave it with settings the bridge refuses at run time.
func (h *Handler) rejectQoderOverrides(w http.ResponseWriter, r *http.Request, agent db.Agent, o qoderOverrides) bool {
	provider, ok := h.resolveAgentProvider(r, agent.WorkspaceID, agent.RuntimeID)
	if !ok || provider != "qoder_cloud" {
		return false
	}
	if err := o.validate(); err != nil {
		writeError(w, http.StatusBadRequest, err.Error())
		return true
	}
	return false
}

func validateQoderAgentConfig(raw []byte) error {
	if _, err := qoderruntime.ParseAgentConfig(raw); err != nil {
		if strings.HasPrefix(err.Error(), "default repository") {
			return err
		}
		return fmt.Errorf("select an agent from Qoder Cloud Agents in the agent runtime configuration")
	}
	return nil
}