package builtin

import (
	"gitee.com/blizzard1413/dsl/function"
	"gitee.com/blizzard1413/dsl/types"
)

// RegisterObject 注册所有对象函数
func RegisterObject(registry function.Registry) error {
	objectFuncs := []struct {
		name string
		fn   func([]interface{}) (interface{}, error)
		sig  *function.Signature
	}{
		{"Object.keys", objectKeys, function.NewSignature([]types.Type{types.NewObjectType(nil)}, types.NewArrayType(types.NewStringType()))},
		{"Object.values", objectValues, function.NewSignature([]types.Type{types.NewObjectType(nil)}, types.NewArrayType(types.NewAnyType()))},
		{"Object.entries", objectEntries, function.NewSignature([]types.Type{types.NewObjectType(nil)}, types.NewArrayType(types.NewArrayType(types.NewAnyType())))},
	}

	for _, of := range objectFuncs {
		if err := registry.RegisterNative(of.name, of.sig, of.fn); err != nil {
			return err
		}
	}

	return nil
}

// objectKeys 返回对象的所有键
func objectKeys(args []interface{}) (interface{}, error) {
	obj := toObject(args[0])

	keys := make([]interface{}, 0, len(obj))
	for key := range obj {
		keys = append(keys, key)
	}

	return keys, nil
}

// objectValues 返回对象的所有值
func objectValues(args []interface{}) (interface{}, error) {
	obj := toObject(args[0])

	values := make([]interface{}, 0, len(obj))
	for _, value := range obj {
		values = append(values, value)
	}

	return values, nil
}

// objectEntries 返回对象的所有键值对
func objectEntries(args []interface{}) (interface{}, error) {
	obj := toObject(args[0])

	entries := make([]interface{}, 0, len(obj))
	for key, value := range obj {
		entry := []interface{}{key, value}
		entries = append(entries, entry)
	}

	return entries, nil
}