package tui

import (
    "ai-agent/internal/config"
)

type Mode int

const (
    ModeAsk   Mode = iota
    ModePlan
    ModeBuild
)

type ModeConfig struct {
    Label               string
    SystemPromptPrefix  string
    AllowTools          bool
    PreferredCapability config.ModelCapability
}

func DefaultModeConfigs() [3]ModeConfig {
    return [3]ModeConfig{
        {
            Label:               "ASK",
            SystemPromptPrefix:  "Provide direct, concise answers. Use tools when the user asks about files or the codebase.",
            AllowTools:          true,
            PreferredCapability: config.CapabilitySimple,
        },
        {
            Label:               "PLAN",
            SystemPromptPrefix:  "Help the user plan and design. Break down tasks into steps. Use tools to read and explore, but do not modify files.",
            AllowTools:          true,
            PreferredCapability: config.CapabilityComplex,
        },
        {
            Label:               "BUILD",
            SystemPromptPrefix:  "Execute tasks using all available tools.",
            AllowTools:          true,
            PreferredCapability: config.CapabilityAdvanced,
        },
    }
}