// 提示词业务模块，统一封装用户提示词规则、选择关系和持久化操作。
package prompt

import (
	"encoding/json"
	"errors"
	"strings"

	"github.com/google/uuid"
	"gorm.io/gorm"
)

// Service 提供提示词模块的公开业务接口，并隐藏底层数据库实现。
type Service struct {
	db *gorm.DB
}

// CustomPromptInput 表示用户自定义提示词的可编辑内容。
type CustomPromptInput struct {
	Name    string
	Type    string
	Content string
}

// CustomPrompt 表示返回给接口层的用户自定义提示词。
type CustomPrompt struct {
	ID       uuid.UUID `json:"id"`
	Name     string    `json:"name"`
	Type     string    `json:"type"`
	Content  string    `json:"content"`
	Scope    string    `json:"scope"`
	Editable bool      `json:"editable"`
}

// SystemPromptInput 表示管理端可维护的系统提示词内容。
type SystemPromptInput struct {
	Name    string `json:"name"`
	Type    string `json:"type"`
	Content string `json:"content"`
}

// SystemPrompt 表示管理端保存后返回的系统提示词。
type SystemPrompt struct {
	ID      string `json:"id"`
	Name    string `json:"name"`
	Type    string `json:"type"`
	Content string `json:"content"`
}

// VersionedPromptInput 表示旧版提示词工作流中的草稿和版本字段。
type VersionedPromptInput struct {
	Code        string
	Name        string
	Category    string
	Type        string
	Content     string
	Variables   []string
	VersionNote string
}

// UserPromptList 表示指定类型的可用提示词及用户当前选择。
type UserPromptList struct {
	Prompts    []map[string]any
	SelectedID *uuid.UUID
}

// PersistenceError 标识应由接口层转换为内部错误的数据库故障。
type PersistenceError struct {
	Err error
}

// Error 返回底层数据库错误文本，供日志与测试定位。
func (e PersistenceError) Error() string {
	return e.Err.Error()
}

// Unwrap 暴露底层错误，保留 errors.Is 和 errors.As 语义。
func (e PersistenceError) Unwrap() error {
	return e.Err
}

var fixedDramaParsePromptTypes = []string{"剧本解析", "角色、场景、道具解析"}

// NewService 创建提示词服务，数据库连接仅在模块内部使用。
func NewService(db *gorm.DB) *Service {
	return &Service{db: db}
}

// IsFixedDramaParseType 判断提示词类型是否由后端固定配置且禁止用户编辑。
func IsFixedDramaParseType(promptType string) bool {
	for _, fixedType := range fixedDramaParsePromptTypes {
		if strings.TrimSpace(promptType) == fixedType {
			return true
		}
	}
	return false
}

// IsPersistenceError 判断错误是否属于需要隐藏细节的数据库故障。
func IsPersistenceError(err error) bool {
	var target PersistenceError
	return errors.As(err, &target)
}

// ListSystemPrompts 查询管理端可维护的系统提示词，不返回后端固定配置。
func (s *Service) ListSystemPrompts(keyword, promptType string, page, size int) ([]map[string]any, int64, error) {
	query := s.db.Table("prompts p").Where("p.deleted_at IS NULL AND p.scope='system' AND p.type NOT IN ?", fixedDramaParsePromptTypes)
	if keyword = strings.TrimSpace(keyword); keyword != "" {
		query = query.Where("p.name::text ILIKE ? OR p.content ILIKE ?", "%"+keyword+"%", "%"+keyword+"%")
	}
	if promptType = strings.TrimSpace(promptType); promptType != "" {
		query = query.Where("p.type=?", promptType)
	}
	var total int64
	if err := query.Count(&total).Error; err != nil {
		return nil, 0, err
	}
	items := make([]map[string]any, 0)
	err := query.Select("p.id,p.name,p.type,p.content,p.scope,p.owner_user_id,p.created_at,p.updated_at").
		Order("p.updated_at DESC").Offset((page - 1) * size).Limit(size).Find(&items).Error
	return items, total, err
}

// SaveSystemPrompt 校验并新增或更新管理端系统提示词。
func (s *Service) SaveSystemPrompt(promptID string, input SystemPromptInput) (SystemPrompt, error) {
	input.Name = strings.TrimSpace(input.Name)
	input.Type = strings.TrimSpace(input.Type)
	input.Content = strings.TrimSpace(input.Content)
	if input.Name == "" || input.Type == "" || input.Content == "" {
		return SystemPrompt{}, errors.New("name、type、content不能为空")
	}
	if IsFixedDramaParseType(input.Type) {
		return SystemPrompt{}, errors.New("该提示词已固定在后端代码中，不支持管理")
	}
	if promptID == "" {
		promptID = uuid.NewString()
		if err := s.db.Exec("INSERT INTO prompts(id,name,type,content,scope,owner_user_id) VALUES(?,?,?,?,'system',NULL)", promptID, input.Name, input.Type, input.Content).Error; err != nil {
			return SystemPrompt{}, err
		}
	} else if err := s.db.Exec("UPDATE prompts SET name=?,type=?,content=?,updated_at=CURRENT_TIMESTAMP WHERE id=? AND scope='system' AND deleted_at IS NULL", input.Name, input.Type, input.Content, promptID).Error; err != nil {
		return SystemPrompt{}, err
	}
	return SystemPrompt{ID: promptID, Name: input.Name, Type: input.Type, Content: input.Content}, nil
}

// DeleteSystemPrompt 删除管理端指定的系统提示词。
func (s *Service) DeleteSystemPrompt(promptID string) error {
	return s.db.Exec("DELETE FROM prompts WHERE id=? AND scope='system'", promptID).Error
}

// SaveVersionedPrompt 保存旧版管理接口使用的提示词草稿及版本快照。
func (s *Service) SaveVersionedPrompt(adminID uuid.UUID, promptID string, input VersionedPromptInput) (string, error) {
	if strings.TrimSpace(input.Category) == "" {
		input.Category = input.Type
	}
	if strings.TrimSpace(input.Type) == "" {
		input.Type = input.Category
	}
	err := s.db.Transaction(func(tx *gorm.DB) error {
		version := 1
		if promptID == "" {
			promptID = uuid.NewString()
			if err := tx.Exec("INSERT INTO prompts(id,code,name,category,type,content,status) VALUES(?,?,?,?,?,?,'draft')", promptID, input.Code, input.Name, input.Category, input.Type, input.Content).Error; err != nil {
				return err
			}
		} else {
			if err := tx.Raw("SELECT coalesce(max(version),0)+1 FROM prompt_versions WHERE prompt_id=?", promptID).Scan(&version).Error; err != nil {
				return err
			}
			if err := tx.Exec("UPDATE prompts SET code=?,name=?,category=?,type=?,content=? WHERE id=? AND deleted_at IS NULL", input.Code, input.Name, input.Category, input.Type, input.Content, promptID).Error; err != nil {
				return err
			}
		}
		variables, _ := json.Marshal(input.Variables)
		versionID := uuid.NewString()
		if err := tx.Exec("INSERT INTO prompt_versions(id,prompt_id,version,content,variables,version_note,status,operator_id) VALUES(?,?,?,?,?::jsonb,?,'draft',?)", versionID, promptID, version, input.Content, string(variables), input.VersionNote, adminID).Error; err != nil {
			return err
		}
		return tx.Exec("UPDATE prompts SET current_version_id=?,status='draft' WHERE id=?", versionID, promptID).Error
	})
	return promptID, err
}

// ApplyPromptAction 执行旧版提示词接口的发布、停用或版本回滚事务。
func (s *Service) ApplyPromptAction(adminID uuid.UUID, promptID, action, versionID, versionNote string) error {
	return s.db.Transaction(func(tx *gorm.DB) error {
		switch action {
		case "publish":
			if err := tx.Exec("UPDATE prompt_versions SET status=CASE WHEN id=(SELECT current_version_id FROM prompts WHERE id=?) THEN 'published' ELSE 'retired' END WHERE prompt_id=?", promptID, promptID).Error; err != nil {
				return err
			}
			return tx.Exec("UPDATE prompts SET status='published' WHERE id=?", promptID).Error
		case "disable":
			return tx.Exec("UPDATE prompts SET status='disabled' WHERE id=?", promptID).Error
		case "rollback":
			var source struct {
				Content   string
				Variables string
			}
			if err := tx.Raw("SELECT content,variables::text FROM prompt_versions WHERE id=? AND prompt_id=?", versionID, promptID).Scan(&source).Error; err != nil {
				return err
			}
			var version int
			if err := tx.Raw("SELECT coalesce(max(version),0)+1 FROM prompt_versions WHERE prompt_id=?", promptID).Scan(&version).Error; err != nil {
				return err
			}
			newVersionID := uuid.NewString()
			if err := tx.Exec("INSERT INTO prompt_versions(id,prompt_id,version,content,variables,version_note,status,operator_id) VALUES(?,?,?,?,?::jsonb,?,'draft',?)", newVersionID, promptID, version, source.Content, source.Variables, versionNote, adminID).Error; err != nil {
				return err
			}
			return tx.Exec("UPDATE prompts SET current_version_id=?,status='draft' WHERE id=?", newVersionID, promptID).Error
		default:
			return errors.New("不支持的提示词操作")
		}
	})
}

// ListPromptHistory 查询旧版提示词接口保留的版本历史。
func (s *Service) ListPromptHistory(promptID string) ([]map[string]any, error) {
	items := make([]map[string]any, 0)
	err := s.db.Table("prompt_versions v").
		Select("v.id,v.version,v.content,v.variables,v.version_note,v.status,v.created_at,a.username AS operator").
		Joins("LEFT JOIN admin_users a ON a.id=v.operator_id").Where("v.prompt_id=?", promptID).
		Order("v.version DESC").Find(&items).Error
	return items, err
}

// ListUserPrompts 查询系统提示词、当前用户自定义提示词及用户选择。
func (s *Service) ListUserPrompts(userID uuid.UUID, promptType string) (UserPromptList, error) {
	if IsFixedDramaParseType(promptType) {
		return UserPromptList{Prompts: []map[string]any{}}, nil
	}
	var prompts []map[string]any
	err := s.db.Table("prompts").Select("id,name,type,content,scope,owner_user_id,created_at,updated_at,(scope='user') AS editable").
		Where("deleted_at IS NULL AND type=? AND (scope='system' OR (scope='user' AND owner_user_id=?))", promptType, userID).
		Order("CASE WHEN scope='system' THEN 0 ELSE 1 END, updated_at DESC").Find(&prompts).Error
	if err != nil {
		return UserPromptList{}, err
	}
	var selected struct{ PromptID *uuid.UUID }
	// 用户未设置偏好时返回空属于正常情况，用 Limit(1).Find 避免触发 GORM 的 record not found 日志
	s.db.Table("user_prompt_preferences").Select("prompt_id").Where("user_id=? AND prompt_type=?", userID, promptType).Limit(1).Find(&selected)
	return UserPromptList{Prompts: prompts, SelectedID: selected.PromptID}, nil
}

// SelectUserPrompt 更新用户对指定提示词类型的选择，空值表示关闭选择。
func (s *Service) SelectUserPrompt(userID uuid.UUID, promptType string, promptID *uuid.UUID) error {
	if IsFixedDramaParseType(promptType) {
		return errors.New("该提示词由平台固定配置，不支持选择")
	}
	if promptID == nil {
		if err := s.db.Exec("DELETE FROM user_prompt_preferences WHERE user_id=? AND prompt_type=?", userID, promptType).Error; err != nil {
			return PersistenceError{Err: err}
		}
		return nil
	}
	var count int64
	if err := s.db.Table("prompts").Where("id=? AND type=? AND deleted_at IS NULL AND (scope='system' OR (scope='user' AND owner_user_id=?))", *promptID, promptType, userID).Count(&count).Error; err != nil {
		return PersistenceError{Err: err}
	}
	if count == 0 {
		return gorm.ErrRecordNotFound
	}
	if err := s.db.Exec(`INSERT INTO user_prompt_preferences(id,user_id,prompt_id,prompt_type) VALUES(gen_random_uuid(),?,?,?) ON CONFLICT(user_id,prompt_type) DO UPDATE SET prompt_id=excluded.prompt_id,updated_at=CURRENT_TIMESTAMP`, userID, *promptID, promptType).Error; err != nil {
		return PersistenceError{Err: err}
	}
	return nil
}

// CreateCustomPrompt 校验并创建当前用户拥有的自定义提示词。
func (s *Service) CreateCustomPrompt(userID uuid.UUID, input CustomPromptInput) (CustomPrompt, error) {
	input = normalizeCustomPromptInput(input)
	if input.Name == "" || input.Type == "" || input.Content == "" {
		return CustomPrompt{}, errors.New("name、type、content不能为空")
	}
	if IsFixedDramaParseType(input.Type) {
		return CustomPrompt{}, errors.New("该提示词由平台固定配置，不支持自定义")
	}
	item := CustomPrompt{ID: uuid.New(), Name: input.Name, Type: input.Type, Content: input.Content, Scope: "user", Editable: true}
	if err := s.db.Exec("INSERT INTO prompts(id,name,type,content,scope,owner_user_id) VALUES(?,?,?,?,'user',?)", item.ID, item.Name, item.Type, item.Content, userID).Error; err != nil {
		return CustomPrompt{}, err
	}
	return item, nil
}

// UpdateCustomPrompt 校验并更新当前用户拥有的自定义提示词。
func (s *Service) UpdateCustomPrompt(userID, promptID uuid.UUID, input CustomPromptInput) (CustomPrompt, error) {
	input = normalizeCustomPromptInput(input)
	if input.Name == "" || input.Content == "" {
		return CustomPrompt{}, errors.New("name、content不能为空")
	}
	if IsFixedDramaParseType(input.Type) {
		return CustomPrompt{}, errors.New("该提示词由平台固定配置，不支持自定义")
	}
	result := s.db.Exec("UPDATE prompts SET name=?,type=COALESCE(NULLIF(?,''),type),content=?,updated_at=CURRENT_TIMESTAMP WHERE id=? AND scope='user' AND owner_user_id=? AND deleted_at IS NULL", input.Name, input.Type, input.Content, promptID, userID)
	if result.Error != nil {
		return CustomPrompt{}, result.Error
	}
	if result.RowsAffected == 0 {
		return CustomPrompt{}, gorm.ErrRecordNotFound
	}
	return CustomPrompt{ID: promptID, Name: input.Name, Type: input.Type, Content: input.Content, Scope: "user", Editable: true}, nil
}

// DeleteCustomPrompt 在同一事务中删除用户选择关系及其自定义提示词。
func (s *Service) DeleteCustomPrompt(userID, promptID uuid.UUID) error {
	return s.db.Transaction(func(tx *gorm.DB) error {
		if err := tx.Exec("DELETE FROM user_prompt_preferences WHERE prompt_id=? AND user_id=?", promptID, userID).Error; err != nil {
			return err
		}
		result := tx.Exec("DELETE FROM prompts WHERE id=? AND scope='user' AND owner_user_id=?", promptID, userID)
		if result.Error != nil {
			return result.Error
		}
		if result.RowsAffected == 0 {
			return gorm.ErrRecordNotFound
		}
		return nil
	})
}

// normalizeCustomPromptInput 清理用户输入两端空白，保证校验和写入使用一致值。
func normalizeCustomPromptInput(input CustomPromptInput) CustomPromptInput {
	input.Name = strings.TrimSpace(input.Name)
	input.Type = strings.TrimSpace(input.Type)
	input.Content = strings.TrimSpace(input.Content)
	return input
}
