package api

import (
	"gitee.com/ling_boogie/mahjong-helper/platform/majsoul/proto/lq"
	"github.com/golang/protobuf/proto"
)

func WrapMessage(name string, message proto.Message) (data []byte, err error) {
	data, err = proto.Marshal(message)
	if err != nil {
		return
	}
	wrap := lq.Wrapper{
		Name: name,
		Data: data,
	}
	return proto.Marshal(&wrap)
}

func UnwrapData(rawData []byte) (methodName string, data []byte, err error) {
	wrapper := lq.Wrapper{}
	if err = proto.Unmarshal(rawData, &wrapper); err != nil {
		return
	}
	return wrapper.GetName(), wrapper.GetData(), nil
}

// TODO: auto UnwrapMessage by methodName

func UnwrapMessage(rawData []byte, message proto.Message) error {
	methodName, data, err := UnwrapData(rawData)
	if err != nil {
		return err
	}
	// TODO: assert methodName when its not empty
	_ = methodName
	return proto.Unmarshal(data, message)
}