package models

import (
	"maps"
	"time"
)

// OnEnterActionType represents the type of action to execute when entering a step.
type OnEnterActionType string

const (
	OnEnterEnablePlanMode    OnEnterActionType = "enable_plan_mode"
	OnEnterAutoStartAgent    OnEnterActionType = "auto_start_agent"
	OnEnterResetAgentContext OnEnterActionType = "reset_agent_context"
	// OnEnterSetSessionMode declares the agent's session permission mode (e.g.
	// "default", "acceptEdits") for a step on entry. The target mode is carried
	// in the action Config under the "mode" key. See issue #1183.
	OnEnterSetSessionMode OnEnterActionType = "set_session_mode"

	// Phase 2 (ADR-0004) — generic actions are also permitted on on_enter
	// so review/approval steps can clear decisions and fan out runs to
	// participants when the task arrives at the step.
	OnEnterClearDecisions             OnEnterActionType = "clear_decisions"
	OnEnterQueueRunForEachParticipant OnEnterActionType = "queue_run_for_each_participant"
	OnEnterQueueRun                   OnEnterActionType = "queue_run"
)

// OnTurnStartActionType represents the type of action to execute when a user sends a message.
type OnTurnStartActionType string

const (
	OnTurnStartMoveToNext     OnTurnStartActionType = "move_to_next"
	OnTurnStartMoveToPrevious OnTurnStartActionType = "move_to_previous"
	OnTurnStartMoveToStep     OnTurnStartActionType = "move_to_step"
)

// OnTurnStartAction represents an action to execute when a user sends a message.
type OnTurnStartAction struct {
	Type   OnTurnStartActionType  `json:"type" yaml:"type"`
	Config map[string]interface{} `json:"config,omitempty" yaml:"config,omitempty"`
}

// OnTurnCompleteActionType represents the type of action to execute when an agent turn completes.
type OnTurnCompleteActionType string

const (
	OnTurnCompleteMoveToNext      OnTurnCompleteActionType = "move_to_next"
	OnTurnCompleteMoveToPrevious  OnTurnCompleteActionType = "move_to_previous"
	OnTurnCompleteMoveToStep      OnTurnCompleteActionType = "move_to_step"
	OnTurnCompleteDisablePlanMode OnTurnCompleteActionType = "disable_plan_mode"
)

// OnEnterAction represents an action to execute when entering a step.
type OnEnterAction struct {
	Type   OnEnterActionType      `json:"type" yaml:"type"`
	Config map[string]interface{} `json:"config,omitempty" yaml:"config,omitempty"`
}

// OnTurnCompleteAction represents an action to execute when an agent turn completes.
type OnTurnCompleteAction struct {
	Type   OnTurnCompleteActionType `json:"type" yaml:"type"`
	Config map[string]interface{}   `json:"config,omitempty" yaml:"config,omitempty"`
}

// OnExitActionType represents the type of action to execute when leaving a step.
type OnExitActionType string

const (
	OnExitDisablePlanMode OnExitActionType = "disable_plan_mode"
)

// OnExitAction represents an action to execute when leaving a step.
type OnExitAction struct {
	Type   OnExitActionType       `json:"type" yaml:"type"`
	Config map[string]interface{} `json:"config,omitempty" yaml:"config,omitempty"`
}

// GenericActionType represents the type of a Phase 2 (ADR-0004) action that
// can appear under any of the new event-driven triggers (on_comment,
// on_blocker_resolved, on_children_completed, on_approval_resolved,
// on_heartbeat, on_budget_alert, on_agent_error). Actions are compiled into
// the engine's typed Action structs by engine.CompileStep.
type GenericActionType string

const (
	// GenericActionQueueRun queues a run on a target task/agent.
	GenericActionQueueRun GenericActionType = "queue_run"
	// GenericActionClearDecisions clears recorded decisions for the
	// (task, step) pair. Typically used by a Review step's on_enter to
	// start fresh after a rejection round.
	GenericActionClearDecisions GenericActionType = "clear_decisions"
	// GenericActionQueueRunForEachParticipant fans out queue_run over
	// every participant of the step matching a configured role.
	GenericActionQueueRunForEachParticipant GenericActionType = "queue_run_for_each_participant"
)

// GenericAction is the persisted shape of a Phase 2 action used in the
// new event-driven triggers. Config carries the action-specific parameters
// (target, task_id, reason, payload, role, …) interpreted by
// engine.CompileStep.
type GenericAction struct {
	Type   GenericActionType      `json:"type" yaml:"type"`
	Config map[string]interface{} `json:"config,omitempty" yaml:"config,omitempty"`
}

// StepEvents contains event-driven actions for a workflow step.
//
// The four kanban-era triggers (OnEnter, OnTurnStart, OnTurnComplete, OnExit)
// keep their typed action slices for backwards compatibility. The Phase 2
// (ADR-0004) triggers all use GenericAction so new action kinds can be added
// without further model changes — engine.CompileStep is the single point of
// translation into typed engine.Action structs.
type StepEvents struct {
	OnEnter        []OnEnterAction        `json:"on_enter,omitempty" yaml:"on_enter,omitempty"`
	OnTurnStart    []OnTurnStartAction    `json:"on_turn_start,omitempty" yaml:"on_turn_start,omitempty"`
	OnTurnComplete []OnTurnCompleteAction `json:"on_turn_complete,omitempty" yaml:"on_turn_complete,omitempty"`
	OnExit         []OnExitAction         `json:"on_exit,omitempty" yaml:"on_exit,omitempty"`

	// Phase 2 (ADR-0004) — new event-driven triggers. Empty slices keep
	// today's kanban behaviour; the engine simply finds no actions for the
	// trigger and exits.
	OnComment           []GenericAction `json:"on_comment,omitempty" yaml:"on_comment,omitempty"`
	OnBlockerResolved   []GenericAction `json:"on_blocker_resolved,omitempty" yaml:"on_blocker_resolved,omitempty"`
	OnChildrenCompleted []GenericAction `json:"on_children_completed,omitempty" yaml:"on_children_completed,omitempty"`
	OnApprovalResolved  []GenericAction `json:"on_approval_resolved,omitempty" yaml:"on_approval_resolved,omitempty"`
	OnHeartbeat         []GenericAction `json:"on_heartbeat,omitempty" yaml:"on_heartbeat,omitempty"`
	OnBudgetAlert       []GenericAction `json:"on_budget_alert,omitempty" yaml:"on_budget_alert,omitempty"`
	OnAgentError        []GenericAction `json:"on_agent_error,omitempty" yaml:"on_agent_error,omitempty"`
}

// ReviewStatus represents the review state of a session
type ReviewStatus string

const (
	ReviewStatusPending          ReviewStatus = "pending"
	ReviewStatusChangesRequested ReviewStatus = "changes_requested"
	ReviewStatusApproved         ReviewStatus = "approved"
)

// WorkflowTemplate represents a pre-defined workflow type that workflows can adopt
type WorkflowTemplate struct {
	ID          string `json:"id"`
	Name        string `json:"name"`
	Description string `json:"description"`
	IsSystem    bool   `json:"is_system"`
	// Hidden marks templates that produce hidden workflow instances
	// (excluded from management UI and pickers). Used by system-only flows.
	// Not persisted to DB; sourced from the embedded YAML at load time.
	Hidden    bool             `json:"hidden,omitempty"`
	Steps     []StepDefinition `json:"steps"` // JSON stored
	CreatedAt time.Time        `json:"created_at"`
	UpdatedAt time.Time        `json:"updated_at"`
}

// StepDefinition represents a step in a workflow template (stored as JSON in WorkflowTemplate)
type StepDefinition struct {
	ID                    string     `json:"id"`
	Name                  string     `json:"name"`
	Position              int        `json:"position"`
	Color                 string     `json:"color"`
	Prompt                string     `json:"prompt,omitempty"`
	Events                StepEvents `json:"events"`
	AllowManualMove       bool       `json:"allow_manual_move"`
	IsStartStep           bool       `json:"is_start_step"`
	ShowInCommandPanel    bool       `json:"show_in_command_panel"`
	AutoArchiveAfterHours int        `json:"auto_archive_after_hours,omitempty"`
	AgentProfileID        string     `json:"agent_profile_id,omitempty"`
	// StageType mirrors WorkflowStep.StageType for templates so the office
	// default + coordination workflows can declare their UX role
	// ("work", "review", "approval", "custom") in YAML.
	StageType StageType `json:"stage_type,omitempty"`
}

// WorkflowStep represents a step in a workflow
type WorkflowStep struct {
	ID                    string     `json:"id"`
	WorkflowID            string     `json:"workflow_id"`
	Name                  string     `json:"name"`
	Position              int        `json:"position"`
	Color                 string     `json:"color"`
	Prompt                string     `json:"prompt,omitempty"`
	Events                StepEvents `json:"events"`
	AllowManualMove       bool       `json:"allow_manual_move"`
	IsStartStep           bool       `json:"is_start_step"`
	ShowInCommandPanel    bool       `json:"show_in_command_panel"`
	AutoArchiveAfterHours int        `json:"auto_archive_after_hours,omitempty"`
	AgentProfileID        string     `json:"agent_profile_id,omitempty"`
	// StageType is a Phase 2 (ADR-0004) semantic hint for the frontend
	// ("work", "review", "approval", "custom"). The engine does not branch
	// on it. Stored as TEXT in workflow_steps.stage_type, defaulting to
	// "custom" so existing rows remain unchanged.
	StageType StageType `json:"stage_type,omitempty"`
	CreatedAt time.Time `json:"created_at"`
	UpdatedAt time.Time `json:"updated_at"`
}

// HasOnEnterAction checks if the step has a specific on_enter action type.
func (s *WorkflowStep) HasOnEnterAction(actionType OnEnterActionType) bool {
	for _, action := range s.Events.OnEnter {
		if action.Type == actionType {
			return true
		}
	}
	return false
}

// HasOnTurnStartAction checks if the step has any on_turn_start actions.
func (s *WorkflowStep) HasOnTurnStartAction() bool {
	return len(s.Events.OnTurnStart) > 0
}

// HasOnTurnCompleteAction checks if the step has a specific on_turn_complete action type.
func (s *WorkflowStep) HasOnTurnCompleteAction(actionType OnTurnCompleteActionType) bool {
	for _, action := range s.Events.OnTurnComplete {
		if action.Type == actionType {
			return true
		}
	}
	return false
}

// StepTransitionTrigger represents how a session moved between steps
type StepTransitionTrigger string

const (
	StepTransitionTriggerManual       StepTransitionTrigger = "manual"
	StepTransitionTriggerAutoComplete StepTransitionTrigger = "auto_complete"
	StepTransitionTriggerApproval     StepTransitionTrigger = "approval"
)

// SessionStepHistory represents an audit trail entry for session step transitions
type SessionStepHistory struct {
	ID         int64                  `json:"id"`
	SessionID  string                 `json:"session_id"`
	FromStepID *string                `json:"from_step_id,omitempty"`
	ToStepID   string                 `json:"to_step_id"`
	Trigger    StepTransitionTrigger  `json:"trigger"`
	ActorID    *string                `json:"actor_id,omitempty"`
	Metadata   map[string]interface{} `json:"metadata,omitempty"`
	CreatedAt  time.Time              `json:"created_at"`
}

// RemapStepEvents returns a copy of events with all step_id references
// in move_to_step actions replaced using the provided ID mapping.
func RemapStepEvents(events StepEvents, idMap map[string]string) StepEvents {
	result := StepEvents{}
	result.OnEnter = append(result.OnEnter, events.OnEnter...)
	for _, a := range events.OnTurnStart {
		if a.Type == OnTurnStartMoveToStep && a.Config != nil {
			if stepID, ok := a.Config["step_id"].(string); ok {
				if newID, found := idMap[stepID]; found {
					cfg := make(map[string]any, len(a.Config))
					maps.Copy(cfg, a.Config)
					cfg["step_id"] = newID
					a.Config = cfg
				}
			}
		}
		result.OnTurnStart = append(result.OnTurnStart, a)
	}
	for _, a := range events.OnTurnComplete {
		if a.Type == OnTurnCompleteMoveToStep && a.Config != nil {
			if stepID, ok := a.Config["step_id"].(string); ok {
				if newID, found := idMap[stepID]; found {
					cfg := make(map[string]any, len(a.Config))
					maps.Copy(cfg, a.Config)
					cfg["step_id"] = newID
					a.Config = cfg
				}
			}
		}
		result.OnTurnComplete = append(result.OnTurnComplete, a)
	}
	result.OnExit = append(result.OnExit, events.OnExit...)
	// Phase 2 (ADR-0004) — copy generic-action lists through. None of these
	// actions carry step_id references today, so a shallow copy suffices.
	result.OnComment = append(result.OnComment, events.OnComment...)
	result.OnBlockerResolved = append(result.OnBlockerResolved, events.OnBlockerResolved...)
	result.OnChildrenCompleted = append(result.OnChildrenCompleted, events.OnChildrenCompleted...)
	result.OnApprovalResolved = append(result.OnApprovalResolved, events.OnApprovalResolved...)
	result.OnHeartbeat = append(result.OnHeartbeat, events.OnHeartbeat...)
	result.OnBudgetAlert = append(result.OnBudgetAlert, events.OnBudgetAlert...)
	result.OnAgentError = append(result.OnAgentError, events.OnAgentError...)
	return result
}