package core

import (
	"context"
	"errors"
	"fmt"
	"sync"

	"github.com/cdle/sillyplus/core/common"
	"github.com/cdle/sillyplus/utils"
	"google.golang.org/grpc/metadata"
)

var senderRegisters sync.Map

func getRegisterSenderByCtx(ctx context.Context, uuid string) (common.Sender, error) {
	// if uuid == "" { //临时---====+++测试
	// 	plt, _ := GetAdapter("terminal", "default")
	// 	return plt.Sender2(map[string]string{
	// 		CONETNT:    "hello",
	// 		USER_ID:    "user_id",
	// 		CHAT_ID:    "chat_id",
	// 		MESSAGE_ID: "MESSAGE_ID",
	// 	}), nil
	// }
	md, ok := metadata.FromIncomingContext(ctx)
	if !ok {
		return nil, fmt.Errorf("missing metadata")
	}
	runtimeID := md.Get("RUNTIME_ID")
	if runtimeID == nil {
		return nil, fmt.Errorf("missing runtime_id")
	}
	runtime_id := runtimeID[0]
	return getRegisterSender(runtime_id, uuid)
}

func getRegisterSender(runtime_id, uuid string) (common.Sender, error) {
	if uuid == "" {
		return &CustomSender{
			F: &Factory{
				botid:  "default",
				botplt: "*",
			},
		}, nil
	}
	var sm *sync.Map
	v, ok := senderRegisters.Load(runtime_id)
	if ok {
		sm = v.(*sync.Map)
		v, ok := sm.Load(uuid)
		if ok {
			return v.(common.Sender), nil
		}
	}
	return nil, errors.New("NOT FOUND SENDER")
}

func createSenderRegister(runtime_id string) func(common.Sender) string {
	var sm = new(sync.Map)
	senderRegisters.Store(runtime_id, sm)
	return func(s common.Sender) string {
		uuid := utils.GenUUID()
		sm.Store(uuid, s)
		return uuid
	}
}

func getSenderRegister(runtime_id string) (func(common.Sender) string, error) {
	var sm *sync.Map
	v, ok := senderRegisters.Load(runtime_id)
	if !ok {
		return nil, errors.New("INVALID RUNTIME")

	}
	sm = v.(*sync.Map)
	return func(s common.Sender) string {
		uuid := utils.GenUUID()
		sm.Store(uuid, s)
		return uuid
	}, nil
}

func deleteSenderRegister(runtime_id string) {
	v, ok := senderRegisters.Load(runtime_id)
	if ok {
		sm := v.(*sync.Map)
		sm.Range(func(key, value any) bool {
			// fmt.Println(key)
			sm.Delete(key)
			return true
		})
	}

	senderRegisters.Delete(runtime_id)
}

func getSenderRegisterByCtx(ctx context.Context) (string, func(common.Sender) string, error) {
	var senderRegister func(common.Sender) string
	md, ok := metadata.FromIncomingContext(ctx)
	if !ok {
		return "", senderRegister, fmt.Errorf("missing metadata")
	}
	runtimeID := md.Get("RUNTIME_ID")
	if runtimeID == nil {
		return "", senderRegister, fmt.Errorf("missing runtime_id")
	}
	runtime_id := runtimeID[0]
	f, err := getSenderRegister(runtime_id)
	return runtime_id, f, err
}