package system

import (
	"fmt"

	"github.com/yohamta/donburi/filter"
	"joylink.club/ecs"
	"joylink.club/rtsssimulation/components"
	"joylink.club/rtsssimulation/umi"
)

func FindEntityById(world ecs.World, id string) *ecs.Entry {
	query := ecs.NewQuery(filter.Contains(components.DeviceIdentityComponent))
	return QueryEntityById(world, query, id)
}
func QueryEntityById(world ecs.World, q *ecs.Query, id string) *ecs.Entry {
	var entry *ecs.Entry = nil
	func() {
		defer simpleRecover()
		q.Each(world, func(e *ecs.Entry) {
			if id == components.DeviceIdentityComponent.Get(e).Id {
				entry = e
				panic(fmt.Sprintf("找到实体[%s]，结束查找", id))
			}
		})
	}()
	//
	return entry
}

// 捕获panic并恢复执行
func simpleRecover() {
	recover()
}

// /////////////////////////////////////////////////////////////////////////
// 模型仓库查询
var worldModelStorageQuery *ecs.Query = ecs.NewQuery(filter.Contains(components.ModelStorageRefComponent))

func WorldModelStorage(world ecs.World) umi.IModelManager {
	e, ok := worldModelStorageQuery.First(world)
	if ok {
		return components.ModelStorageRefComponent.Get(e).ModelManager
	} else {
		return nil
	}
}

// //////////////////////////////////////////////////////////////////////////
// 继电器系统
type relaySystem struct {
	// key-设备id
	relayQuery map[string]*ecs.Query
}

// 获取设备的相关继电器的查询
func (me *relaySystem) getDeviceRelayQuery(deviceEntry *ecs.Entry) *ecs.Query {
	id := components.DeviceIdentityComponent.Get(deviceEntry).Id
	query, ok := me.relayQuery[id]
	if !ok {
		query = ecs.NewQuery(filter.Contains(components.RelayTagHandlerComponent.Get(deviceEntry).Tag))
		me.relayQuery[id] = query
	}
	return query
}