models

type BaseContext struct {
	Type       EventType          `json:"type"`
	Repository *github.Repository `json:"repository"`
	Sender     *github.User       `json:"sender"`
	RawEvent   interface{}        `json:"raw_event"`
	Action     string             `json:"action"`
	DeliveryID string             `json:"delivery_id"`
	Timestamp  time.Time          `json:"timestamp"`
}

type CommandInfo struct {
	Command string `json:"command"`  // /code, /continue, /fix
	AIModel string `json:"ai_model"` // claude, gemini
	Args    string `json:"args"`     // 命令参数
	RawText string `json:"raw_text"` // 原始文本
}

type CommentContext struct {
	Repository     *github.Repository `json:"repository"`
	IssueNumber    int                `json:"issue_number"`
	CommentID      *int64             `json:"comment_id,omitempty"`
	InitialContent string             `json:"initial_content"`
	LastContent    string             `json:"last_content"`
	UpdateCount    int                `json:"update_count"`
	CreatedAt      time.Time          `json:"created_at"`
	LastUpdatedAt  *time.Time         `json:"last_updated_at,omitempty"`
}

type EventType string

type ExecutionMetrics struct {
	ToolCalls     int           `json:"tool_calls"`
	Duration      time.Duration `json:"duration"`
	Success       int           `json:"success"`
	Errors        int           `json:"errors"`
	LastExecution time.Time     `json:"last_execution"`
}

type ExecutionResult struct {
	Success      bool          `json:"success"`
	Output       string        `json:"output"`
	Error        string        `json:"error,omitempty"`
	FilesChanged []string      `json:"files_changed"`
	Duration     time.Duration `json:"duration"`
}

type GitHubContext interface {
	GetEventType() EventType
	GetRepository() *github.Repository
	GetSender() *github.User
	GetRawEvent() interface{}
	GetEventAction() string
	GetDeliveryID() string
	GetTimestamp() time.Time
}

type IssueCommentContext struct {
	BaseContext
	Issue   *github.Issue        `json:"issue"`
	Comment *github.IssueComment `json:"comment"`
	// 是否是PR评论（通过Issue.PullRequestLinks判断）
	IsPRComment bool `json:"is_pr_comment"`
}

type IssuesContext struct {
	BaseContext
	Issue *github.Issue `json:"issue"`
}

type JSONSchema struct {
	Type                 string                 `json:"type"`
	Properties           map[string]*JSONSchema `json:"properties,omitempty"`
	Required             []string               `json:"required,omitempty"`
	Items                *JSONSchema            `json:"items,omitempty"`
	Description          string                 `json:"description,omitempty"`
	Enum                 []interface{}          `json:"enum,omitempty"`
	AdditionalProperties bool                   `json:"additionalProperties,omitempty"`
}

type MCPContext struct {
	// 基础上下文
	Repository  GitHubContext `json:"repository"`
	Issue       interface{}   `json:"issue,omitempty"`
	PullRequest interface{}   `json:"pull_request,omitempty"`
	User        interface{}   `json:"user,omitempty"`

	// 工作环境
	WorkspacePath string            `json:"workspace_path,omitempty"`
	BranchName    string            `json:"branch_name,omitempty"`
	Metadata      map[string]string `json:"metadata,omitempty"`

	// 权限控制
	Permissions []string `json:"permissions,omitempty"`
	Constraints []string `json:"constraints,omitempty"`
}

type MCPError struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
	Data    string `json:"data,omitempty"`
}

type MCPRequest struct {
	ID     string                 `json:"id"`
	Method string                 `json:"method"`
	Params map[string]interface{} `json:"params"`
}

type MCPResponse struct {
	ID     string      `json:"id"`
	Result interface{} `json:"result,omitempty"`
	Error  *MCPError   `json:"error,omitempty"`
}

type MCPServerCapabilities struct {
	Tools     []Tool `json:"tools"`
	Resources []any  `json:"resources,omitempty"`
	Prompts   []any  `json:"prompts,omitempty"`
}

type MCPServerInfo struct {
	Name         string                `json:"name"`
	Version      string                `json:"version"`
	Description  string                `json:"description"`
	Capabilities MCPServerCapabilities `json:"capabilities"`
	CreatedAt    time.Time             `json:"created_at"`
}

type PRAllComments struct {
	PRBody         string                       `json:"pr_body"`
	IssueComments  []*github.IssueComment       `json:"issue_comments"`
	ReviewComments []*github.PullRequestComment `json:"review_comments"`
	Reviews        []*github.PullRequestReview  `json:"reviews"`
}

type ProgressExecutionResult struct {
	Success        bool                   `json:"success"`
	Output         string                 `json:"output"`
	Error          string                 `json:"error,omitempty"`
	FilesChanged   []string               `json:"files_changed"`
	Duration       time.Duration          `json:"duration"`
	Summary        string                 `json:"summary"`
	CommitSHA      string                 `json:"commit_sha,omitempty"`
	BranchName     string                 `json:"branch_name,omitempty"`
	PullRequestURL string                 `json:"pull_request_url,omitempty"`
	TaskResults    []*Task                `json:"task_results"`
	Metadata       map[string]interface{} `json:"metadata,omitempty"`
}

type ProgressTracker struct {
	Tasks           []*Task       `json:"tasks"`
	CurrentTaskName string        `json:"current_task_name"`
	StartTime       time.Time     `json:"start_time"`
	EndTime         *time.Time    `json:"end_time,omitempty"`
	Status          TaskStatus    `json:"status"`
	Spinner         *SpinnerState `json:"spinner"`
	TotalDuration   time.Duration `json:"total_duration"`
}

type PullRequestContext struct {
	BaseContext
	PullRequest *github.PullRequest `json:"pull_request"`
}

type PullRequestReviewCommentContext struct {
	BaseContext
	PullRequest *github.PullRequest        `json:"pull_request"`
	Comment     *github.PullRequestComment `json:"comment"`
}

type PullRequestReviewContext struct {
	BaseContext
	PullRequest *github.PullRequest       `json:"pull_request"`
	Review      *github.PullRequestReview `json:"review"`
}

type PushContext struct {
	BaseContext
	Ref     string               `json:"ref"`
	Commits []*github.HeadCommit `json:"commits"`
	Before  string               `json:"before"`
	After   string               `json:"after"`
}

type ScheduleContext struct {
	BaseContext
	Cron string `json:"cron"`
}

type SpinnerState struct {
	Active     bool      `json:"active"`
	Message    string    `json:"message"`
	StartTime  time.Time `json:"start_time"`
	FrameIndex int       `json:"frame_index"`
}

type Task struct {
	Name        string            `json:"name"`
	Description string            `json:"description"`
	Status      TaskStatus        `json:"status"`
	StartTime   *time.Time        `json:"start_time,omitempty"`
	EndTime     *time.Time        `json:"end_time,omitempty"`
	Duration    time.Duration     `json:"duration"`
	Error       string            `json:"error,omitempty"`
	SubTasks    []*Task           `json:"sub_tasks,omitempty"`
	Progress    float64           `json:"progress"` // 0.0 - 1.0
	Metadata    map[string]string `json:"metadata,omitempty"`
}

type TaskStatus string

type Tool struct {
	Name        string      `json:"name"`
	Description string      `json:"description"`
	InputSchema *JSONSchema `json:"input_schema"`
}

type ToolCall struct {
	ID       string                 `json:"id"`
	Function ToolFunction           `json:"function"`
	Context  map[string]interface{} `json:"context,omitempty"`
}

type ToolFunction struct {
	Name      string                 `json:"name"`
	Arguments map[string]interface{} `json:"arguments"`
}

type ToolResult struct {
	ID      string      `json:"id"`
	Success bool        `json:"success"`
	Content interface{} `json:"content,omitempty"`
	Error   string      `json:"error,omitempty"`
	Type    string      `json:"type,omitempty"` // text, json, image, etc.
}

type WorkflowDispatchContext struct {
	BaseContext
	Inputs map[string]interface{} `json:"inputs"`
}

type Workspace struct {
	// github org name
	Org string `json:"org"`
	// github repo name
	Repo string `json:"repo"`
	// PR number
	PRNumber int `json:"pr_number"`
	// AI model name (claude or gemini)
	AIModel string `json:"ai_model"`
	// workspace path in local file system
	Path string `json:"path"`
	// session path in local file system
	SessionPath string `json:"session_path"`
	// github repo url
	Repository string `json:"repository"`
	// github branch name
	Branch      string              `json:"branch"`
	Issue       *github.Issue       `json:"issue"`
	PullRequest *github.PullRequest `json:"pull_request"`
	CreatedAt   time.Time           `json:"created_at"`
}