package config

import "reasonix/internal/provider"

func withOpenCodeGoChatContextOverrides(base map[string]ProviderModelOverride) map[string]ProviderModelOverride {
	if base == nil {
		base = map[string]ProviderModelOverride{}
	}
	for id, lim := range provider.OpenCodeGoChatModels() {
		ov := base[id]
		ov.ContextWindow = lim.Context
		base[id] = ov
	}
	return base
}

func openCodeGoAnthropicContextOverrides() map[string]ProviderModelOverride {
	out := map[string]ProviderModelOverride{}
	for id, lim := range provider.OpenCodeGoAnthropicModels() {
		out[id] = ProviderModelOverride{ContextWindow: lim.Context}
	}
	return out
}

func openCodeGoResponsesContextOverrides() map[string]ProviderModelOverride {
	out := map[string]ProviderModelOverride{}
	for id, lim := range provider.OpenCodeGoResponsesModels() {
		out[id] = ProviderModelOverride{ContextWindow: lim.Context}
	}
	out["grok-4.5"] = ProviderModelOverride{ContextWindow: 500_000, SupportedEfforts: []string{"low", "medium", "high"}, DefaultEffort: "high"}
	out["gpt-5.6-luna"] = ProviderModelOverride{ContextWindow: 1_050_000, SupportedEfforts: []string{"none", "low", "medium", "high", "xhigh", "max"}, DefaultEffort: "high"}
	out["muse-spark-1.2-contributor"] = ProviderModelOverride{ContextWindow: 1_048_576, SupportedEfforts: []string{"minimal", "low", "medium", "high", "xhigh"}, DefaultEffort: "high"}
	return out
}

// openCodeGoRecommendedChatOverrides keeps the one-click setup conservative:
// most coding turns stay on a low reasoning effort and a bounded output cap.
// Users can still raise the effort or output budget on the editable provider
// entry when a task genuinely needs it.
func openCodeGoRecommendedChatOverrides() map[string]ProviderModelOverride {
	out := map[string]ProviderModelOverride{}
	for id, lim := range provider.OpenCodeGoChatModels() {
		out[id] = ProviderModelOverride{ContextWindow: lim.Context, MaxOutputTokens: minPositive(lim.MaxOutput, 32_768)}
	}
	for id, ov := range map[string]ProviderModelOverride{
		"glm-5.3": {
			ReasoningProtocol: ReasoningProtocolOpenAI,
			SupportedEfforts:  []string{"low", "high", "max"},
			DefaultEffort:     "low",
		},
		"glm-5.2": {
			ReasoningProtocol: ReasoningProtocolOpenAI,
			SupportedEfforts:  []string{"high", "max"},
			DefaultEffort:     "high",
		},
		"glm-5.1": {
			ReasoningProtocol: ReasoningProtocolOpenAI,
			SupportedEfforts:  []string{"low", "high"},
			DefaultEffort:     "low",
		},
		"deepseek-v4-flash": {
			ReasoningProtocol: ReasoningProtocolDeepSeek,
			SupportedEfforts:  []string{"disabled", "high", "max"},
			DefaultEffort:     "high",
		},
		"deepseek-v4-pro": {
			ReasoningProtocol: ReasoningProtocolDeepSeek,
			SupportedEfforts:  []string{"disabled", "high", "max"},
			DefaultEffort:     "high",
		},
		"kimi-k2.6": {
			ReasoningProtocol: ReasoningProtocolOpenAI,
			SupportedEfforts:  []string{"low", "medium", "high"},
			DefaultEffort:     "low",
		},
		"kimi-k2.7-code": {
			ReasoningProtocol: ReasoningProtocolOpenAI,
			SupportedEfforts:  []string{"low", "medium", "high"},
			DefaultEffort:     "low",
		},
		"kimi-k3": kimiK3DirectOverride(),
		"hy3": {
			ReasoningProtocol: ReasoningProtocolOpenAI,
			SupportedEfforts:  []string{"none", "low", "high"},
			DefaultEffort:     "low",
		},
	} {
		base := out[id]
		base.ReasoningProtocol = ov.ReasoningProtocol
		base.SupportedEfforts = append([]string(nil), ov.SupportedEfforts...)
		base.DefaultEffort = ov.DefaultEffort
		if ov.ContextWindow > 0 {
			base.ContextWindow = ov.ContextWindow
		}
		out[id] = base
	}
	return out
}

func openCodeGoRecommendedAnthropicOverrides() map[string]ProviderModelOverride {
	out := map[string]ProviderModelOverride{}
	for id, lim := range provider.OpenCodeGoAnthropicModels() {
		out[id] = ProviderModelOverride{ContextWindow: lim.Context, MaxOutputTokens: minPositive(lim.MaxOutput, 32_768)}
	}
	return out
}

func openCodeGoRecommendedResponsesOverrides() map[string]ProviderModelOverride {
	out := map[string]ProviderModelOverride{}
	for id, lim := range provider.OpenCodeGoResponsesModels() {
		out[id] = ProviderModelOverride{ContextWindow: lim.Context, MaxOutputTokens: minPositive(lim.MaxOutput, 32_768)}
	}
	for id, levels := range map[string][]string{
		"grok-4.5":                   {"low", "medium", "high"},
		"gpt-5.6-luna":               {"none", "low", "medium", "high", "xhigh", "max"},
		"muse-spark-1.2-contributor": {"minimal", "low", "medium", "high", "xhigh"},
	} {
		ov := out[id]
		ov.SupportedEfforts = append([]string(nil), levels...)
		ov.DefaultEffort = "low"
		out[id] = ov
	}
	return out
}

func minPositive(value, cap int) int {
	if value <= 0 {
		return cap
	}
	if value < cap {
		return value
	}
	return cap
}

// opencodeGoRecommendedPreset is the low-friction entry point for new users.
// It installs all three official wire routes under one shared credential while
// keeping each route's model catalog isolated. The entries remain ordinary
// editable providers after installation.
var opencodeGoRecommendedPreset = ProviderPreset{
	ID:             "opencode-go-recommended",
	Label:          "OpenCode Go (Recommended)",
	Description:    "One key, three safe routes: Chat, Anthropic Messages, and Responses. Defaults use a low-cost effort and a 32K output cap.",
	KeyEnv:         "OPENCODE_GO_API_KEY",
	Recommended:    true,
	BillingMode:    "subscription_equivalent",
	DisplayGroup:   "opencode",
	DisplaySection: "go",
	DisplayTier:    "primary",
	RouteKind:      "bundle",
	DisplayOrder:   0,
	Entries: []ProviderEntry{
		{
			Name:            "opencode-go",
			Kind:            "openai",
			BaseURL:         "https://opencode.ai/zen/go/v1",
			Models:          opencodeGoModels,
			VisionModels:    opencodeGoVisionModels,
			Default:         "glm-5.3",
			APIKeyEnv:       "OPENCODE_GO_API_KEY",
			ContextWindow:   128_000,
			MaxOutputTokens: 32_768,
			BillingMode:     "subscription_equivalent",
			ModelOverrides:  openCodeGoRecommendedChatOverrides(),
		},
		{
			Name:            "opencode-go-anthropic",
			Kind:            "anthropic",
			BaseURL:         "https://opencode.ai/zen/go",
			Models:          []string{"qwen3.8-max", "qwen3.7-plus", "qwen3.7-max", "qwen3.6-plus", "minimax-m3", "minimax-m2.7", "minimax-m2.5"},
			VisionModels:    []string{"qwen3.8-max", "qwen3.7-plus", "qwen3.6-plus"},
			Default:         "qwen3.7-plus",
			APIKeyEnv:       "OPENCODE_GO_API_KEY",
			Thinking:        "adaptive",
			ContextWindow:   262_144,
			MaxOutputTokens: 32_768,
			BillingMode:     "subscription_equivalent",
			ModelOverrides:  openCodeGoRecommendedAnthropicOverrides(),
		},
		{
			Name:             "opencode-go-responses",
			Kind:             "responses",
			BaseURL:          "https://opencode.ai/zen/go/v1",
			Models:           []string{"grok-4.5", "gpt-5.6-luna", "muse-spark-1.2-contributor"},
			VisionModels:     []string{"grok-4.5", "gpt-5.6-luna", "muse-spark-1.2-contributor"},
			Default:          "grok-4.5",
			APIKeyEnv:        "OPENCODE_GO_API_KEY",
			ResponsesMode:    "stateless",
			ContextWindow:    500_000,
			MaxOutputTokens:  32_768,
			SupportedEfforts: []string{"low", "medium", "high"},
			DefaultEffort:    "low",
			BillingMode:      "subscription_equivalent",
			ModelOverrides:   openCodeGoRecommendedResponsesOverrides(),
		},
	},
}

var opencodeGoAnthropicPreset = ProviderPreset{
	ID:             "opencode-go-anthropic",
	Label:          "OpenCode Go Anthropic",
	Description:    "OpenCode Go subscription Anthropic-compatible route for Qwen and MiniMax models.",
	KeyEnv:         "OPENCODE_GO_API_KEY",
	BillingMode:    "subscription_equivalent",
	DisplayGroup:   "opencode",
	DisplaySection: "go",
	DisplayTier:    "advanced",
	RouteKind:      "anthropic",
	DisplayOrder:   20,
	Entries: []ProviderEntry{{
		Name:            "opencode-go-anthropic",
		Kind:            "anthropic",
		BaseURL:         "https://opencode.ai/zen/go",
		Models:          []string{"qwen3.8-max", "qwen3.7-plus", "qwen3.7-max", "qwen3.6-plus", "minimax-m3", "minimax-m2.7", "minimax-m2.5"},
		VisionModels:    []string{"qwen3.8-max", "qwen3.7-plus", "qwen3.6-plus"},
		Default:         "qwen3.7-plus",
		APIKeyEnv:       "OPENCODE_GO_API_KEY",
		Thinking:        "adaptive",
		ContextWindow:   262144,
		MaxOutputTokens: 32_768,
		BillingMode:     "subscription_equivalent",
		ModelOverrides:  openCodeGoAnthropicContextOverrides(),
	}},
}

var opencodeGoResponsesPreset = ProviderPreset{
	ID:             "opencode-go-responses",
	Label:          "OpenCode Go Responses",
	Description:    "OpenCode Go Responses API route for Grok 4.5, GPT 5.6 Luna, and Muse Spark.",
	KeyEnv:         "OPENCODE_GO_API_KEY",
	BillingMode:    "subscription_equivalent",
	DisplayGroup:   "opencode",
	DisplaySection: "go",
	DisplayTier:    "advanced",
	RouteKind:      "responses",
	DisplayOrder:   30,
	Entries: []ProviderEntry{{
		Name:             "opencode-go-responses",
		Kind:             "responses",
		BaseURL:          "https://opencode.ai/zen/go/v1",
		Models:           []string{"grok-4.5", "gpt-5.6-luna", "muse-spark-1.2-contributor"},
		VisionModels:     []string{"grok-4.5", "gpt-5.6-luna", "muse-spark-1.2-contributor"},
		Default:          "grok-4.5",
		APIKeyEnv:        "OPENCODE_GO_API_KEY",
		ResponsesMode:    "stateless",
		ContextWindow:    500_000,
		MaxOutputTokens:  32_768,
		SupportedEfforts: []string{"low", "medium", "high"},
		DefaultEffort:    "high",
		BillingMode:      "subscription_equivalent",
		ModelOverrides:   openCodeGoResponsesContextOverrides(),
	}},
}

var opencodeGoDeepSeekAnthropicPreset = ProviderPreset{
	ID:             "opencode-go-deepseek-anthropic",
	Label:          "OpenCode Go DeepSeek Anthropic",
	Description:    "OpenCode Go Anthropic Messages route for DeepSeek Flash with server-side web search.",
	KeyEnv:         "OPENCODE_GO_API_KEY",
	BillingMode:    "subscription_equivalent",
	DisplayGroup:   "opencode",
	DisplaySection: "go",
	DisplayTier:    "advanced",
	RouteKind:      "search-anthropic",
	Optional:       true,
	DisplayOrder:   40,
	Entries: []ProviderEntry{{
		Name:             "opencode-go-deepseek-anthropic",
		Kind:             "anthropic",
		BaseURL:          "https://opencode.ai/zen/go",
		Models:           []string{"deepseek-v4-flash"},
		Default:          "deepseek-v4-flash",
		APIKeyEnv:        "OPENCODE_GO_API_KEY",
		Thinking:         "adaptive",
		WebSearch:        boolPointer(true),
		ContextWindow:    1_000_000,
		MaxOutputTokens:  32_768,
		SupportedEfforts: []string{"disabled", "high", "max"},
		DefaultEffort:    "high",
		BillingMode:      "subscription_equivalent",
	}},
}

var opencodeGoDeepSeekResponsesPreset = ProviderPreset{
	ID:             "opencode-go-deepseek-responses",
	Label:          "OpenCode Go DeepSeek Responses",
	Description:    "OpenCode Go stateless Responses API route for DeepSeek Flash with server-side web search.",
	KeyEnv:         "OPENCODE_GO_API_KEY",
	BillingMode:    "subscription_equivalent",
	DisplayGroup:   "opencode",
	DisplaySection: "go",
	DisplayTier:    "advanced",
	RouteKind:      "search-responses",
	Optional:       true,
	DisplayOrder:   50,
	Entries: []ProviderEntry{{
		Name:             "opencode-go-deepseek-responses",
		Kind:             "responses",
		BaseURL:          "https://opencode.ai/zen/go/v1",
		Models:           []string{"deepseek-v4-flash"},
		Default:          "deepseek-v4-flash",
		APIKeyEnv:        "OPENCODE_GO_API_KEY",
		ResponsesMode:    "stateless",
		WebSearch:        boolPointer(true),
		ContextWindow:    1_000_000,
		MaxOutputTokens:  32_768,
		SupportedEfforts: []string{"disabled", "high", "max"},
		DefaultEffort:    "high",
		BillingMode:      "subscription_equivalent",
	}},
}