package response

import (
	"gitee.com/xlizy/common-go/utils/common"
	rpcApi "gitee.com/xlizy/rpc-interface/pbs"
	"reflect"
)

func RpcRsp(rsp, data any) *Response {
	rpcResValue := reflect.ValueOf(rsp).Elem().FieldByName("Result")
	rpcRes := rpcResValue.Interface().(*rpcApi.Result)
	res := &Response{
		Success:      rpcRes.Success,
		Code:         rpcRes.Code,
		Msg:          rpcRes.Msg,
		ResponseTime: rpcRes.ResponseTime,
		TraceId:      rpcRes.TraceId,
	}
	if data != nil {
		rpcDataValue := reflect.ValueOf(rsp).Elem().FieldByName("Data")
		if !rpcDataValue.IsNil() {
			common.DeepCopy(data, rpcDataValue.Interface())
		}
		res.Data = data
	}
	return res
}

func RpcRspByDataFieldName(rsp, data any, dataFieldName string) *Response {
	rpcResValue := reflect.ValueOf(rsp).Elem().FieldByName("Result")
	rpcRes := rpcResValue.Interface().(*rpcApi.Result)
	res := &Response{
		Success:      rpcRes.Success,
		Code:         rpcRes.Code,
		Msg:          rpcRes.Msg,
		ResponseTime: rpcRes.ResponseTime,
		TraceId:      rpcRes.TraceId,
	}
	rpcDataValue := reflect.ValueOf(rsp).Elem().FieldByName(dataFieldName)
	if !rpcDataValue.IsNil() {
		common.DeepCopy(data, rpcDataValue.Interface())
	}
	res.Data = data
	return res
}