package gdconf

import (
	"os"
	"sort"
	"strconv"
	"sync"
	"sync/atomic"
	"time"

	"hk4e/pkg/logger"

	lua "github.com/yuin/gopher-lua"
)

// 场景详情配置数据

const (
	SceneGroupLoaderLimit = 4 // 加载文件的并发数 此操作很耗内存 调大之前请确保你的机器内存足够
)

var OBJECT_ID_COUNTER uint64

type SceneLuaConfig struct {
	Id          int32
	SceneConfig *SceneConfig     // 地图配置
	BlockMap    map[int32]*Block // 所有的区块
}

type Vector struct {
	X float32 `json:"x"`
	Y float32 `json:"y"`
	Z float32 `json:"z"`
}

type SceneConfig struct {
	BeginPos     *Vector `json:"begin_pos"`
	Size         *Vector `json:"size"`
	BornPos      *Vector `json:"born_pos"`
	BornRot      *Vector `json:"born_rot"`
	DieY         float32 `json:"die_y"`
	VisionAnchor *Vector `json:"vision_anchor"`
}

type Block struct {
	Id               int32
	BlockRange       *BlockRange      // 区块范围坐标
	GroupMap         map[int32]*Group // 所有的group
	groupMapLoadLock sync.Mutex
}

type BlockRange struct {
	Min *Vector `json:"min"`
	Max *Vector `json:"max"`
}

type Group struct {
	Id            int32        `json:"id"`
	RefreshId     int32        `json:"refresh_id"`
	Area          int32        `json:"area"`
	Pos           *Vector      `json:"pos"`
	DynamicLoad   bool         `json:"dynamic_load"`
	IsReplaceable *Replaceable `json:"is_replaceable"`
	MonsterList   []*Monster   `json:"monsters"` // 怪物
	NpcList       []*Npc       `json:"npcs"`     // NPC
	GadgetList    []*Gadget    `json:"gadgets"`  // 物件
	RegionList    []*Region    `json:"regions"`
	TriggerList   []*Trigger   `json:"triggers"`
	LuaStr        string       `json:"-"`
	LuaState      *lua.LState  `json:"-"`
}

const (
	LuaStateLruKeepNum = 1000
)

type LuaStateLru struct {
	GroupId    int32
	AccessTime int64
}

type LuaStateLruList []*LuaStateLru

func (l LuaStateLruList) Len() int {
	return len(l)
}

func (l LuaStateLruList) Less(i, j int) bool {
	return l[i].AccessTime < l[j].AccessTime
}

func (l LuaStateLruList) Swap(i, j int) {
	l[i], l[j] = l[j], l[i]
}

func LuaStateLruRemove() {
	removeNum := len(CONF.LuaStateLruMap) - LuaStateLruKeepNum
	if removeNum <= 0 {
		return
	}
	luaStateLruList := make(LuaStateLruList, 0)
	for _, luaStateLru := range CONF.LuaStateLruMap {
		luaStateLruList = append(luaStateLruList, luaStateLru)
	}
	sort.Stable(luaStateLruList)
	for i := 0; i < removeNum; i++ {
		luaStateLru := luaStateLruList[i]
		group := GetSceneGroup(luaStateLru.GroupId)
		group.LuaState = nil
		delete(CONF.LuaStateLruMap, luaStateLru.GroupId)
	}
	logger.Info("lua state lru remove finish, remove num: %v", removeNum)
}

func (g *Group) GetLuaState() *lua.LState {
	CONF.LuaStateLruMap[g.Id] = &LuaStateLru{
		GroupId:    g.Id,
		AccessTime: time.Now().UnixMilli(),
	}
	if g.LuaState == nil {
		g.LuaState = newLuaState(g.LuaStr)
		scriptLib := g.LuaState.NewTable()
		g.LuaState.SetGlobal("ScriptLib", scriptLib)
		for _, scriptLibFunc := range SCRIPT_LIB_FUNC_LIST {
			g.LuaState.SetField(scriptLib, scriptLibFunc.fnName, g.LuaState.NewFunction(scriptLibFunc.fn))
		}
	}
	return g.LuaState
}

type Replaceable struct {
	Value      bool  `json:"value"`
	Version    int32 `json:"version"`
	NewBinOnly bool  `json:"new_bin_only"`
}

type Monster struct {
	ConfigId  int32   `json:"config_id"`
	MonsterId int32   `json:"monster_id"`
	Pos       *Vector `json:"pos"`
	Rot       *Vector `json:"rot"`
	Level     int32   `json:"level"`
	AreaId    int32   `json:"area_id"`
	ObjectId  uint64  `json:"-"`
}

type Npc struct {
	ConfigId int32   `json:"config_id"`
	NpcId    int32   `json:"npc_id"`
	Pos      *Vector `json:"pos"`
	Rot      *Vector `json:"rot"`
	AreaId   int32   `json:"area_id"`
	ObjectId uint64  `json:"-"`
}

type Gadget struct {
	ConfigId  int32   `json:"config_id"`
	GadgetId  int32   `json:"gadget_id"`
	Pos       *Vector `json:"pos"`
	Rot       *Vector `json:"rot"`
	Level     int32   `json:"level"`
	AreaId    int32   `json:"area_id"`
	PointType int32   `json:"point_type"` // 关联GatherData表
	ObjectId  uint64  `json:"-"`
}

type Region struct {
	ConfigId   int32     `json:"config_id"`
	Shape      int32     `json:"shape"`
	Radius     float32   `json:"radius"`
	Size       *Vector   `json:"size"`
	Pos        *Vector   `json:"pos"`
	Height     float32   `json:"height"`
	PointArray []*Vector `json:"point_array"`
	AreaId     int32     `json:"area_id"`
}

type Trigger struct {
	ConfigId     int32  `json:"config_id"`
	Name         string `json:"name"`
	Event        int32  `json:"event"`
	Source       string `json:"source"`
	Condition    string `json:"condition"`
	Action       string `json:"action"`
	TriggerCount int32  `json:"trigger_count"`
}

func (g *GameDataConfig) loadGroup(group *Group, block *Block, sceneId int32, blockId int32) {
	sceneLuaPrefix := g.luaPrefix + "scene/"
	sceneIdStr := strconv.Itoa(int(sceneId))
	groupId := group.Id
	groupIdStr := strconv.Itoa(int(groupId))
	groupLuaData, err := os.ReadFile(sceneLuaPrefix + sceneIdStr + "/scene" + sceneIdStr + "_group" + groupIdStr + ".lua")
	if err != nil {
		logger.Error("open file error: %v, sceneId: %v, blockId: %v, groupId: %v", err, sceneId, blockId, groupId)
		return
	}
	group.LuaStr = string(groupLuaData)
	luaState := newLuaState(group.LuaStr)
	// monsters
	group.MonsterList = make([]*Monster, 0)
	ok := getSceneLuaConfigTable[*[]*Monster](luaState, "monsters", &group.MonsterList)
	if !ok {
		logger.Error("get monsters object error, sceneId: %v, blockId: %v, groupId: %v", sceneId, blockId, groupId)
		luaState.Close()
		return
	}
	for _, monster := range group.MonsterList {
		monster.ObjectId = atomic.AddUint64(&OBJECT_ID_COUNTER, 1)
	}
	// npcs
	group.NpcList = make([]*Npc, 0)
	ok = getSceneLuaConfigTable[*[]*Npc](luaState, "npcs", &group.NpcList)
	if !ok {
		logger.Error("get npcs object error, sceneId: %v, blockId: %v, groupId: %v", sceneId, blockId, groupId)
		luaState.Close()
		return
	}
	for _, npc := range group.NpcList {
		npc.ObjectId = atomic.AddUint64(&OBJECT_ID_COUNTER, 1)
	}
	// gadgets
	group.GadgetList = make([]*Gadget, 0)
	ok = getSceneLuaConfigTable[*[]*Gadget](luaState, "gadgets", &group.GadgetList)
	if !ok {
		logger.Error("get gadgets object error, sceneId: %v, blockId: %v, groupId: %v", sceneId, blockId, groupId)
		luaState.Close()
		return
	}
	for _, gadget := range group.GadgetList {
		gadget.ObjectId = atomic.AddUint64(&OBJECT_ID_COUNTER, 1)
	}
	// regions
	group.RegionList = make([]*Region, 0)
	ok = getSceneLuaConfigTable[*[]*Region](luaState, "regions", &group.RegionList)
	if !ok {
		logger.Error("get regions object error, sceneId: %v, blockId: %v, groupId: %v", sceneId, blockId, groupId)
		luaState.Close()
		return
	}
	// triggers
	group.TriggerList = make([]*Trigger, 0)
	ok = getSceneLuaConfigTable[*[]*Trigger](luaState, "triggers", &group.TriggerList)
	if !ok {
		logger.Error("get triggers object error, sceneId: %v, blockId: %v, groupId: %v", sceneId, blockId, groupId)
		luaState.Close()
		return
	}
	luaState.Close()
	block.groupMapLoadLock.Lock()
	block.GroupMap[group.Id] = group
	block.groupMapLoadLock.Unlock()
}

func (g *GameDataConfig) loadSceneLuaConfig() {
	OBJECT_ID_COUNTER = 0
	g.SceneLuaConfigMap = make(map[int32]*SceneLuaConfig)
	g.GroupMap = make(map[int32]*Group)
	g.LuaStateLruMap = make(map[int32]*LuaStateLru)
	sceneLuaPrefix := g.luaPrefix + "scene/"
	for _, sceneData := range g.SceneDataMap {
		sceneId := sceneData.SceneId
		sceneIdStr := strconv.Itoa(int(sceneId))
		mainLuaData, err := os.ReadFile(sceneLuaPrefix + sceneIdStr + "/scene" + sceneIdStr + ".lua")
		if err != nil {
			logger.Info("open file error: %v, sceneId: %v", err, sceneId)
			continue
		}
		luaState := newLuaState(string(mainLuaData))
		sceneLuaConfig := new(SceneLuaConfig)
		sceneLuaConfig.Id = sceneId
		// scene_config
		sceneLuaConfig.SceneConfig = new(SceneConfig)
		ok := getSceneLuaConfigTable[*SceneConfig](luaState, "scene_config", sceneLuaConfig.SceneConfig)
		if !ok {
			logger.Error("get scene_config object error, sceneId: %v", sceneId)
			luaState.Close()
			continue
		}
		sceneLuaConfig.BlockMap = make(map[int32]*Block)
		// blocks
		blockIdList := make([]int32, 0)
		ok = getSceneLuaConfigTable[*[]int32](luaState, "blocks", &blockIdList)
		if !ok {
			logger.Error("get blocks object error, sceneId: %v", sceneId)
			luaState.Close()
			continue
		}
		// block_rects
		blockRectList := make([]*BlockRange, 0)
		ok = getSceneLuaConfigTable[*[]*BlockRange](luaState, "block_rects", &blockRectList)
		luaState.Close()
		if !ok {
			logger.Error("get block_rects object error, sceneId: %v", sceneId)
			continue
		}
		for index, blockId := range blockIdList {
			block := new(Block)
			block.Id = blockId
			if index >= len(blockRectList) {
				continue
			}
			block.BlockRange = blockRectList[index]
			blockIdStr := strconv.Itoa(int(block.Id))
			blockLuaData, err := os.ReadFile(sceneLuaPrefix + sceneIdStr + "/scene" + sceneIdStr + "_block" + blockIdStr + ".lua")
			if err != nil {
				logger.Error("open file error: %v, sceneId: %v, blockId: %v", err, sceneId, blockId)
				continue
			}
			luaState = newLuaState(string(blockLuaData))
			// groups
			block.GroupMap = make(map[int32]*Group)
			groupList := make([]*Group, 0)
			ok = getSceneLuaConfigTable[*[]*Group](luaState, "groups", &groupList)
			luaState.Close()
			if !ok {
				logger.Error("get groups object error, sceneId: %v, blockId: %v", sceneId, blockId)
				continue
			}
			// 因为group文件实在是太多了 有好几万个 所以这里并发同时加载
			wc := make(chan bool, SceneGroupLoaderLimit)
			wg := sync.WaitGroup{}
			for i := 0; i < len(groupList); i++ {
				group := groupList[i]
				wc <- true
				wg.Add(1)
				go func() {
					g.loadGroup(group, block, sceneId, blockId)
					<-wc
					wg.Done()
				}()
				g.GroupMap[group.Id] = group
			}
			wg.Wait()
			sceneLuaConfig.BlockMap[block.Id] = block
		}
		g.SceneLuaConfigMap[sceneId] = sceneLuaConfig
	}
	sceneCount := 0
	blockCount := 0
	groupCount := 0
	monsterCount := 0
	npcCount := 0
	gadgetCount := 0
	for _, scene := range g.SceneLuaConfigMap {
		for _, block := range scene.BlockMap {
			for _, group := range block.GroupMap {
				monsterCount += len(group.MonsterList)
				npcCount += len(group.NpcList)
				gadgetCount += len(group.GadgetList)
				groupCount++
			}
			blockCount++
		}
		sceneCount++
	}
	logger.Info("Scene count: %v, Block count: %v, Group count: %v, Monster count: %v, Npc count: %v, Gadget count: %v",
		sceneCount, blockCount, groupCount, monsterCount, npcCount, gadgetCount)
}

func GetSceneLuaConfigById(sceneId int32) *SceneLuaConfig {
	return CONF.SceneLuaConfigMap[sceneId]
}

func GetSceneLuaConfigMap() map[int32]*SceneLuaConfig {
	return CONF.SceneLuaConfigMap
}

func GetSceneGroup(groupId int32) *Group {
	groupConfig, exist := CONF.GroupMap[groupId]
	if !exist {
		return nil
	}
	return groupConfig
}