package logic

// ============================================================================
// logic 共享能力层：跨模块复用的运行时横切能力（组合式，非继承）
//
// 【架构约定 · 企业级纪律】
//  1. 本文件按能力分区：文件字段 / 富文本 / 枚举文本 / 唯一校验 / 事务 / 分块查询 / 树
//  2. 依赖方向单向：base 只允许依赖 dao / utility / 内部 consts，禁止引用任何模块文件
//  3. 导出函数上限 15 个：超出即按能力拆为 internal/logic/base/ 子包（防"万能工具箱"腐化）
//  4. CheckUnique 仅为 UX 快速报错（先查后插存在 TOCTOU 竞态），并发唯一性以数据库
//     唯一索引（uk_*）为准，调用点必须保证对应 uk 索引存在，冲突时以错误1062翻译提示
//  5. 富文本占位符 [IMG_URL] 是持久化契约（见 architecture.md），还原只能走 ExpandContentFields
// ============================================================================

import (
	"context"
	"fmt"
	"os"
	"path/filepath"
	"reflect"
	"regexp"
	"sort"
	"strings"
	"sync"
	"time"

	"github.com/gogf/gf/v2/database/gdb"
	"github.com/gogf/gf/v2/errors/gerror"
	"github.com/gogf/gf/v2/frame/g"

	"easygoadmin/internal/consts"
	"easygoadmin/internal/dao"
	"easygoadmin/utility/upload"
)

// ===========================================================================
// 一、文件字段（保存前：临时文件转正式目录）
//    声明式用法（ModuleSpec 逃生舱手写 Add/Update 内联场景）：
//      FileFields: []string{"avatar"}, FileSaveDir: "user"   // 字段名单 + 模块统一目录
//      SaveFileFields(ctx, data, spec.FileFields, spec.FileSaveDir)  // 临时 → 正式，原地更新
// ===========================================================================

// SaveFileFields 单文件字段批量处理
// 值为临时文件URL时移动到正式目录，其余原样保留
func SaveFileFields(ctx context.Context, data map[string]any, fields []string, saveDir string) {
	for _, field := range fields {
		if v, ok := data[field]; ok {
			if s, ok2 := v.(string); ok2 && s != "" {
				data[field] = upload.SaveFromTemp(ctx, s, saveDir)
			}
		}
	}
}

// SaveMultiFileFields 多文件字段（逗号分隔）批量处理
// 各文件为临时文件URL时移动到正式目录，其余原样保留
func SaveMultiFileFields(ctx context.Context, data map[string]any, fields []string, saveDir string) {
	for _, field := range fields {
		if v, ok := data[field]; ok {
			if s, ok2 := v.(string); ok2 && s != "" {
				data[field] = upload.SaveFromTempMulti(ctx, s, saveDir)
			}
		}
	}
}

// SaveFileField 单值便捷版
// 供模块内联场景使用，临时文件URL转正式目录
func SaveFileField(ctx context.Context, fileUrl, saveDir string) string {
	return upload.SaveFromTemp(ctx, fileUrl, saveDir)
}

// SaveMultiFileField 逗号分隔多文件串便捷版
// 供模块内联场景使用，临时文件URL批量转正式目录
func SaveMultiFileField(ctx context.Context, fileUrls, saveDir string) string {
	return upload.SaveFromTempMulti(ctx, fileUrls, saveDir)
}

// ===========================================================================
// 二、富文本（存库占位符化 / 读取还原）
//    存库契约：正文内本站文件引用统一写为 [IMG_URL]/{相对uploadDir路径} 占位符
//      - temp 文件URL  → 移动转正后写占位符
//      - 本站正式文件   → 剥离域名/静态前缀写占位符
//      - 外站 URL      → 不处理（规避 SSRF）
//    读取契约：ExpandContentFields 是唯一还原出口（占位符 → 静态访问基址拼接）
// ===========================================================================

var (
	contentSrcRe  = regexp.MustCompile(`(?i)<(?:img|video|source|embed)[^>]+src=["']?([^"'>\s]+)`)
	contentHrefRe = regexp.MustCompile(`(?i)<a[^>]+href=["']?([^"'>\s]+\.\w{2,5})["']?`)
)

// LocalizeContentFields 富文本存库前本地化
// 临时文件转正，并将本站URL统一写为 [IMG_URL] 占位符
func LocalizeContentFields(ctx context.Context, content, saveDir string) string {
	if content == "" {
		return content
	}
	urlMap := map[string]string{}
	collect := func(re *regexp.Regexp) {
		for _, m := range re.FindAllStringSubmatch(content, -1) {
			if len(m) < 2 {
				continue
			}
			url := m[1]
			if _, seen := urlMap[url]; seen {
				continue
			}
			if newURL := localizeOne(ctx, url, saveDir); newURL != "" {
				urlMap[url] = newURL
			}
		}
	}
	collect(contentSrcRe)
	collect(contentHrefRe)
	for old, newV := range urlMap {
		content = strings.ReplaceAll(content, old, newV)
	}
	return content
}

// localizeOne 单个URL本地化处理
// 临时文件转正为相对路径写占位符；本站正式文件剥离访问基址写占位符；其他返回空表示跳过
func localizeOne(ctx context.Context, url, saveDir string) string {
	if strings.Contains(url, "/temp/") {
		newURL := upload.SaveFromTemp(ctx, url, saveDir)
		if newURL == "" || strings.HasPrefix(newURL, "http://") || strings.HasPrefix(newURL, "https://") {
			return ""
		}
		return "[IMG_URL]/" + strings.TrimLeft(newURL, "/")
	}
	// 本站正式文件：剥离访问基址（domainUrl）后剩余的 "/" 开头路径即相对路径；
	// 未剥离成功的（仍含协议头或未匹配基址）视为外站URL，跳过
	rest := strings.TrimPrefix(url, upload.GetAccessBase(ctx))
	if strings.HasPrefix(rest, "/") && !strings.Contains(rest, "://") {
		return "[IMG_URL]" + rest
	}
	return ""
}

// ExpandContentFields 富文本读取侧还原
// 将 [IMG_URL]/{相对路径} 拼接访问基址，是唯一的还原出口
func ExpandContentFields(ctx context.Context, content string) string {
	if content == "" {
		return content
	}
	return strings.ReplaceAll(content, "[IMG_URL]", upload.GetAccessBase(ctx))
}

// ===========================================================================
// 三、枚举文本（字典值 → 显示名，详情/列表补 {field}Text）
// ===========================================================================

// dictTextCache 进程内字典文本缓存
// 读多写少，TTL 300 秒，修改字典后最迟 5 分钟生效
var dictTextCache sync.Map // key: {code}:{value} → dictTextEntry

// dictTextEntry 字典文本缓存条目
type dictTextEntry struct {
	text    string    // 显示名
	expires time.Time // 过期时间
}

// DictGetText 按字典编码与值获取显示名
// dict_item 查询为唯一真相，未命中时返回原值文本
func DictGetText(ctx context.Context, dictCode string, value any) string {
	val := strings.TrimSpace(fmt.Sprintf("%v", value))
	if dictCode == "" || val == "" {
		return val
	}
	key := dictCode + ":" + val
	if e, ok := dictTextCache.Load(key); ok {
		ent := e.(dictTextEntry)
		if time.Now().Before(ent.expires) {
			return ent.text
		}
	}
	dict, err := dao.Dict.Ctx(ctx).Where("code", dictCode).One()
	if err != nil || len(dict) == 0 {
		return val
	}
	item, err := dao.DictItem.Ctx(ctx).Where("dict_id", dict["id"].Int64()).Where("value", val).One()
	if err != nil || len(item) == 0 {
		return val
	}
	text := item["name"].String()
	dictTextCache.Store(key, dictTextEntry{text: text, expires: time.Now().Add(300 * time.Second)})
	return text
}

// SerializeMapOnRead 详情/列表读取后补 {field}Text 显示名
// 直接在传入的 map 上原地更新
func SerializeMapOnRead(ctx context.Context, data map[string]any, fieldDict map[string]string) {
	for field, dictCode := range fieldDict {
		if v, ok := data[field]; ok {
			data[field+"Text"] = DictGetText(ctx, dictCode, v)
		}
	}
}

// LogTypeText 操作日志类型转中文名
//
// 该字段存储的是注解常量值（1=新增…40=解绑），与字典编码体系不一致，
// 故使用静态映射表而非字典；未知值回退原数字。
func LogTypeText(v int) string {
	if n, ok := consts.LogTypeNames[v]; ok {
		return n
	}
	return fmt.Sprintf("%d", v)
}

// ===========================================================================
// 四、唯一校验（TOCTOU 说明见文件头纪律4）
//    m 传入 dao.X.Ctx(ctx)（自带 del_flag/租户过滤）
// ===========================================================================

// CheckUnique 唯一性快速校验
// label 为字段中文名，例如"角色编码"
func CheckUnique(ctx context.Context, m *gdb.Model, column string, value any, excludeId int64, label string) error {
	mm := m.Where(column, value)
	if excludeId > 0 {
		mm = mm.WhereNot("id", excludeId)
	}
	n, err := mm.Count()
	if err != nil {
		return err
	}
	if n > 0 {
		return gerror.NewCodef(consts.CodeRecordExists, "%s已存在", label)
	}
	return nil
}

// ===========================================================================
// 五、事务（gdb.WithTX 注入上下文，dao.Ctx(ctx) 自动走事务连接）
//    纪律：事务内不做 HTTP/文件IO（保持短事务）
// ===========================================================================

// WithTrans 事务包裹
// fn 返回错误时自动回滚
func WithTrans(ctx context.Context, fn func(ctx context.Context) error) error {
	tx, err := g.DB().Begin(ctx)
	if err != nil {
		return err
	}
	if err = fn(gdb.WithTX(ctx, tx)); err != nil {
		_ = tx.Rollback()
		return err
	}
	return tx.Commit()
}

// ===========================================================================
// 六、分块查询（导出防OOM）
//    build 每页重建查询（gdb Model 不可复用执行），快照语义：并发写入可能翻页漂移
// ===========================================================================

// ChunkQuery 分块遍历
// build(page, size) 返回当页模型，fn 处理每页结果
func ChunkQuery(ctx context.Context, build func(page, size int) *gdb.Model, size int, fn func(gdb.Result) error) error {
	if size <= 0 {
		size = 500
	}
	for page := 1; ; page++ {
		rows, err := build(page, size).All()
		if err != nil {
			return err
		}
		if len(rows) == 0 {
			return nil
		}
		if err = fn(rows); err != nil {
			return err
		}
		if len(rows) < size {
			return nil
		}
	}
}

// fillUserCount 按 user 表外键列统计引用数
// 批量填充 DTO 的 UserCount 字段，供 level、position 等带 userCount 展示的模块复用；
// list 元素需包含 Id 与 UserCount 字段
func fillUserCount(ctx context.Context, column string, list any) {
	rv := reflect.ValueOf(list)
	if rv.Kind() != reflect.Ptr || rv.Elem().Kind() != reflect.Slice {
		return
	}
	slice := rv.Elem()
	counts := make(map[int64]int)
	rows, err := dao.User.Ctx(ctx).Fields("id, " + column).All()
	if err == nil {
		for _, r := range rows {
			if v := r[column].Int64(); v > 0 {
				counts[v]++
			}
		}
	}
	for i := 0; i < slice.Len(); i++ {
		item := slice.Index(i)
		for item.Kind() == reflect.Ptr {
			item = item.Elem()
		}
		if item.Kind() != reflect.Struct {
			continue
		}
		idField := item.FieldByName("Id")
		cntField := item.FieldByName("UserCount")
		if idField.IsValid() && cntField.IsValid() && cntField.CanSet() && cntField.Kind() == reflect.Int {
			cntField.SetInt(int64(counts[idField.Int()]))
		}
	}
}

// ===========================================================================
// 七、树构建（泛型，供 menu/dept/category 等复用）
// ===========================================================================

// BuildTree 平铺列表转树
// 父节点不在集合内则挂根；childrenOf 返回子级切片指针；less 排序，为 nil 时保持原序
func BuildTree[T any](rows []T, idOf func(T) int64, pidOf func(T) int,
	childrenOf func(*T) *[]*T, less func(a, b T) bool) []*T {

	nodes := make(map[int64]*T, len(rows))
	ids := make([]int64, 0, len(rows))
	for _, r := range rows {
		item := r
		nodes[idOf(item)] = &item
		ids = append(ids, idOf(item))
	}
	roots := make([]*T, 0, len(ids))
	for _, idv := range ids {
		node := nodes[idv]
		if pid := pidOf(*node); pid != 0 {
			if parent, ok := nodes[int64(pid)]; ok {
				pc := childrenOf(parent)
				*pc = append(*pc, node)
				continue
			}
		}
		roots = append(roots, node)
	}
	if less != nil {
		sort.SliceStable(roots, func(i, j int) bool { return less(*roots[i], *roots[j]) })
	}
	return roots
}

// ===========================================================================
// 临时目录清理（企业级运营闭环：避免 temp 孤儿文件堆积）
// ===========================================================================

// CleanTemp 清理上传临时目录
// 删除 upload/temp 下超过 expireDays 天的日期子目录或散文件，返回删除数量
func CleanTemp(ctx context.Context, expireDays int) int {
	if expireDays <= 0 {
		expireDays = 7
	}
	base := upload.TempBaseDir(ctx)
	entries, err := os.ReadDir(base)
	if err != nil {
		return 0
	}
	deadline := time.Now().AddDate(0, 0, -expireDays)
	removed := 0
	for _, e := range entries {
		if !e.IsDir() {
			continue
		}
		// 目录名即日期（Yyyymmdd）
		if day, e2 := time.Parse("20060102", e.Name()); e2 == nil && day.Before(deadline) {
			if os.RemoveAll(filepath.Join(base, e.Name())) == nil {
				removed++
			}
		}
	}
	return removed
}