backendtypes

type APIError struct {
	Code    string `json:"code"`
	Message string `json:"message"`
	Details string `json:"details,omitempty"`
}

type APIResponse struct {
	Success   bool        `json:"success"`
	Data      interface{} `json:"data,omitempty"`
	Error     *APIError   `json:"error,omitempty"`
	RequestID string      `json:"request_id,omitempty"`
	Timestamp time.Time   `json:"timestamp"`
}

type AuthConfig struct {
	Enabled     bool     `yaml:"enabled"`
	APIPassword string   `yaml:"api_password"`
	APIKeyEnv   string   `yaml:"api_key_env"`
	PublicPaths []string `yaml:"public_paths"`
}

type BackendConfig struct {
	Server     ServerConfig                     `yaml:"server"`
	Auth       AuthConfig                       `yaml:"auth"`
	Logging    LoggingConfig                    `yaml:"logging"`
	CORS       CORSConfig                       `yaml:"cors"`
	Providers  map[string]*types.ProviderConfig `yaml:"providers"`
	Extensions map[string]ExtensionConfig       `yaml:"extensions"`
}

type CORSConfig struct {
	Enabled        bool     `yaml:"enabled"`
	AllowedOrigins []string `yaml:"allowed_origins"`
	AllowedMethods []string `yaml:"allowed_methods"`
	AllowedHeaders []string `yaml:"allowed_headers"`
}

type ExtensionConfig struct {
	Enabled bool                   `yaml:"enabled"`
	Config  map[string]interface{} `yaml:"config"`
}

type GenerateRequest struct {
	Provider    string                 `json:"provider,omitempty"`
	Model       string                 `json:"model,omitempty"`
	Prompt      string                 `json:"prompt"`
	Messages    []types.ChatMessage    `json:"messages,omitempty"`
	MaxTokens   int                    `json:"max_tokens,omitempty"`
	Temperature float64                `json:"temperature,omitempty"`
	Stream      bool                   `json:"stream,omitempty"`
	Tools       []types.Tool           `json:"tools,omitempty"`
	Metadata    map[string]interface{} `json:"metadata,omitempty"`
}

type GenerateResponse struct {
	Content  string                 `json:"content"`
	Model    string                 `json:"model"`
	Provider string                 `json:"provider"`
	Usage    *UsageInfo             `json:"usage,omitempty"`
	Metadata map[string]interface{} `json:"metadata,omitempty"`
}

type HealthResponse struct {
	Status    string                    `json:"status"`
	Version   string                    `json:"version"`
	Uptime    string                    `json:"uptime"`
	Providers map[string]ProviderHealth `json:"providers,omitempty"`
}

type LoggingConfig struct {
	Level  string `yaml:"level"`
	Format string `yaml:"format"` // "json" or "text"
}

type ProviderConfigRequest struct {
	Type         string   `json:"type"`
	APIKey       string   `json:"api_key,omitempty"`
	APIKeys      []string `json:"api_keys,omitempty"`
	DefaultModel string   `json:"default_model,omitempty"`
	BaseURL      string   `json:"base_url,omitempty"`
	Enabled      bool     `json:"enabled"`
}

type ProviderHealth struct {
	Status  string `json:"status"`
	Latency int64  `json:"latency_ms"`
	Message string `json:"message,omitempty"`
}

type ProviderInfo struct {
	Name        string   `json:"name"`
	Type        string   `json:"type"`
	Enabled     bool     `json:"enabled"`
	Healthy     bool     `json:"healthy"`
	Models      []string `json:"models,omitempty"`
	Description string   `json:"description,omitempty"`
}

type QuotaAllResponse struct {
	// Quotas maps provider:model to quota information
	Quotas map[string]*QuotaResponse `json:"quotas"`

	// Summary provides overall summary
	Summary *QuotaSummaryResponse `json:"summary"`
}

type QuotaConfigResponse struct {
	Type       string                 `json:"type"`
	Period     string                 `json:"period"`
	Limit      int                    `json:"limit"`
	ResetAt    int64                  `json:"reset_at"` // Unix timestamp
	CustomData map[string]interface{} `json:"custom_data,omitempty"`
}

type QuotaHealthResponse struct {
	// Healthy indicates if quota tracking is operational
	Healthy bool `json:"healthy"`

	// Message provides additional health information
	Message string `json:"message,omitempty"`

	// ProvidersTracked is the number of providers being tracked
	ProvidersTracked int `json:"providers_tracked"`

	// LastUpdate is when quota information was last updated
	LastUpdate int64 `json:"last_update,omitempty"` // Unix timestamp
}

type QuotaHistoryRequest struct {
	// Provider is the name of the AI provider
	Provider string `json:"provider"`

	// Model is the model identifier (optional - if empty, returns all models)
	Model string `json:"model,omitempty"`

	// StartTime is the start of the time range (Unix timestamp)
	StartTime int64 `json:"start_time,omitempty"`

	// EndTime is the end of the time range (Unix timestamp)
	EndTime int64 `json:"end_time,omitempty"`

	// Limit is the maximum number of records to return
	Limit int `json:"limit,omitempty"`
}

type QuotaHistoryResponse struct {
	Provider    string                 `json:"provider"`
	Model       string                 `json:"model,omitempty"`
	Records     []*QuotaRecordResponse `json:"records"`
	TotalUsage  map[string]int         `json:"total_usage"`
	StartTime   int64                  `json:"start_time"` // Unix timestamp
	EndTime     int64                  `json:"end_time"`   // Unix timestamp
	RecordCount int                    `json:"record_count"`
}

type QuotaRecordResponse struct {
	ID        string         `json:"id"`
	Provider  string         `json:"provider"`
	Model     string         `json:"model"`
	Timestamp int64          `json:"timestamp"` // Unix timestamp
	Operation string         `json:"operation"`
	Usage     map[string]int `json:"usage"`
}

type QuotaRequest struct {
	// Provider is the name of the AI provider
	Provider string `json:"provider"`

	// Model is the model identifier (optional - if empty, returns provider-wide quota)
	Model string `json:"model,omitempty"`

	// FetchFromProvider if true, fetches real-time quota from the provider API
	// if the provider supports it. If false, returns cached quota information.
	FetchFromProvider bool `json:"fetch_from_provider,omitempty"`
}

type QuotaResponse struct {
	Provider             string                          `json:"provider"`
	ProviderType         string                          `json:"provider_type"`
	Model                string                          `json:"model"`
	Timestamp            int64                           `json:"timestamp"` // Unix timestamp
	Quotas               map[string]*QuotaUsageResponse  `json:"quotas"`
	ProviderQuotaConfigs map[string]*QuotaConfigResponse `json:"provider_quota_configs,omitempty"`
	CustomUsage          map[string]interface{}          `json:"custom_usage,omitempty"`
	Metadata             map[string]interface{}          `json:"metadata,omitempty"`

	// Computed fields
	AnyQuotaExceeded bool `json:"any_quota_exceeded"`
	Healthy          bool `json:"healthy"`
}

type QuotaSummaryRequest struct {
	// Provider filters to a specific provider (optional)
	Provider string `json:"provider,omitempty"`
}

type QuotaSummaryResponse struct {
	// TotalProviders is the number of providers with quota data
	TotalProviders int `json:"total_providers"`

	// ProviderModels maps provider names to their models
	ProviderModels map[string][]string `json:"provider_models"`

	// HistoryRecords is the total number of history records
	HistoryRecords int `json:"history_records"`

	// QuotaTypes counts quota types across all providers
	QuotaTypes map[string]int `json:"quota_types"`
}

type QuotaUsageResponse struct {
	Type             string  `json:"type"`   // "requests", "tokens", "input_tokens", "output_tokens", "daily", "custom"
	Period           string  `json:"period"` // "minute", "hour", "day", "week", "month", "custom"
	Used             int     `json:"used"`
	Limit            int     `json:"limit"`
	Remaining        int     `json:"remaining"`
	RemainingPercent float64 `json:"remaining_percent"`
	ResetAt          int64   `json:"reset_at"`          // Unix timestamp
	PeriodStartedAt  int64   `json:"period_started_at"` // Unix timestamp
}

type RegisterQuotaProviderRequest struct {
	// Name is the unique name for the quota provider
	Name string `json:"name"`

	// ProviderType is the type of the AI provider
	ProviderType types.ProviderType `json:"provider_type"`
}

type ServerConfig struct {
	Host            string        `yaml:"host"`
	Port            int           `yaml:"port"`
	Version         string        `yaml:"version"`
	ReadTimeout     time.Duration `yaml:"read_timeout"`
	WriteTimeout    time.Duration `yaml:"write_timeout"`
	ShutdownTimeout time.Duration `yaml:"shutdown_timeout"`
}

type SetQuotaRequest struct {
	// Provider is the name of the AI provider
	Provider string `json:"provider"`

	// Model is the model identifier (optional - if empty, applies to all models)
	Model string `json:"model,omitempty"`

	// Quotas is a map of quota type to limit value
	Quotas map[string]int `json:"quotas"`

	// ResetTime is the reset time for the quota (Unix timestamp, optional)
	ResetTime int64 `json:"reset_time,omitempty"`
}

type UsageInfo = types.Usage