整理代码

addins/goscr/ec_comp.go

@@ -213,3 +213,27 @@ func ComponentScriptT[T any](script string) pt.ComponentAttribute {
 
 	return pt.Component(types.ZeroT[T]()).SetName(scriptIdent).SetExtra(map[string]any{"script_pkg": scriptPkg, "script_ident": scriptIdent})
 }
+
+// GetComponentScript 获取组件脚本
+func GetComponentScript(entity ec.Entity, name string) func() *ComponentBehavior {
+	return GetComponentScriptT[*ComponentBehavior](entity, name)
+}
+
+// GetComponentScriptT 获取组件脚本
+func GetComponentScriptT[T interface{ This() func() T }](entity ec.Entity, name string) func() T {
+	if entity == nil {
+		panic(fmt.Errorf("%s: entity is nil", exception.ErrArgs))
+	}
+
+	comp := entity.GetComponent(name)
+	if comp == nil {
+		return nil
+	}
+
+	behavior, ok := comp.(T)
+	if !ok {
+		return nil
+	}
+
+	return behavior.This()
+}

addins/goscr/ec_entity.go

@@ -196,3 +196,22 @@ func EntityScriptT[T any](prototype, script string) pt.EntityAttribute {
 
 	return pt.Entity(prototype).SetExtra(map[string]any{"script_pkg": scriptPkg, "script_ident": scriptIdent})
 }
+
+// GetEntityScript 获取实体脚本
+func GetEntityScript(entity ec.Entity) func() *EntityBehavior {
+	return GetEntityScriptT[*EntityBehavior](entity)
+}
+
+// GetEntityScriptT 获取实体脚本
+func GetEntityScriptT[T interface{ This() func() T }](entity ec.Entity) func() T {
+	if entity == nil {
+		panic(fmt.Errorf("%s: entity is nil", exception.ErrArgs))
+	}
+
+	behavior, ok := entity.(T)
+	if !ok {
+		return nil
+	}
+
+	return behavior.This()
+}