package taskrpc

import (
	"encoding/json"
	"errors"
	"strconv"
	"time"

	"git.tetele.net/tgo/crypter"
	"github.com/golang/protobuf/proto"
)

func SetResData(data interface{}, res *Response) {
	res_data_json, err := json.Marshal(data)
	if err == nil {

		encryData := crypter.DesEn(string(res_data_json), DES_KEY)

		now_str := strconv.FormatInt(time.Now().Unix(), 10)

		res_sign := Sign(encryData, now_str)

		res.Data = proto.String(encryData)
		res.Time = proto.String(now_str)
		res.Sign = proto.String(res_sign)
	}

}

func SetReqData(arg interface{}) (*Request, error) {
	data_json, err := json.Marshal(arg)
	if err != nil {
		return nil, err
	}

	now_int64 := time.Now().Unix()

	encryData := crypter.DesEn(string(data_json), DES_KEY)

	now := strconv.FormatInt(now_int64, 10)

	sign := Sign(encryData, now)

	return &Request{proto.String(encryData), proto.String(now), proto.String(sign), nil}, nil

}

func GetReqData(req *Request) (string, error) {
	res_data := req.GetData()

	if res_data != "" {

		time_int64, err := strconv.ParseInt(req.GetTime(), 10, 64)
		if err != nil {
			return "", err
		}

		now_int64 := time.Now().Unix()

		if now_int64-time_int64 > 10 || time_int64-now_int64 > 10 {
			//时间误差前后10秒，返回
			return "", errors.New("返回时间错误")
		}

		check_sign := CheckSign(req.GetSign(), res_data, req.GetTime())
		if !check_sign {
			return "", errors.New("返回数据签名错误")
		}

		//解密
		return crypter.DesDe(res_data, DES_KEY), nil

	}

	return "", nil
}

func GetResData(res *Response) (string, error) {
	res_data := res.GetData()

	if res_data != "" {

		time_int64, err := strconv.ParseInt(res.GetTime(), 10, 64)
		if err != nil {
			return "", err
		}

		now_int64 := time.Now().Unix()

		if now_int64-time_int64 > 10 || time_int64-now_int64 > 10 {
			//时间误差前后10秒，返回
			return "", errors.New("返回时间错误")
		}

		check_sign := CheckSign(res.GetSign(), res_data, res.GetTime())
		if !check_sign {
			return "", errors.New("返回数据签名错误")
		}

		//解密
		return crypter.DesDe(res_data, DES_KEY), nil

	}

	return "", nil
}