package socialwork_sdk

import (
	"encoding/base64"
	"encoding/json"
	"gitlab.workai.com.cn/chenang/socialwork-sdk/model"
	"log"
)

type OrderClient struct {
	Key                string `json:"key"`
	IV                 string `json:"iv"`
	PublicKey          string `json:"public_key"`
	CustomerPrivateKey string `json:"customer_private_key"`
	PtCode             string `json:"pt_code"`
	IP                 string `json:"ip"`
}

func NewOrderClient(key, iv, customerPrivateKey, publicKey, ip string) *OrderClient {
	return &OrderClient{
		Key:                key,
		IV:                 iv,
		PublicKey:          publicKey,
		CustomerPrivateKey: customerPrivateKey,
		IP:                 ip,
	}
}

func (o *OrderClient) EncryptAndSign(bs []byte) (string, string, error) {
	cipherText, err := aes_CBC_Encrypt(bs, []byte(o.Key), []byte(o.IV))
	if err != nil {
		log.Printf("Aes_CBC_Encrypt.failed.err=%v\n", err)
		return "", "", err
	}

	sign, err := rsaSignWithMd5Hex(cipherText, o.CustomerPrivateKey)
	if err != nil {
		log.Printf("RsaSignWithMd5Hex.failed.err=%v\n", err)
		return "", "", err
	}
	return cipherText, sign, nil
}

func (o *OrderClient) DecryptAndVerySign(cipherText, sign string) ([]byte, error) {
	_cipherText, err := base64.StdEncoding.DecodeString(cipherText)
	if err != nil {
		return nil, err
	}
	err = rsaVerifySignWithMd5Base64(cipherText, sign, o.PublicKey)
	if err != nil {
		return nil, err
	}
	content, err := aes_CBC_Decrypt(_cipherText, []byte(o.Key), []byte(o.IV))
	if err != nil {
		return nil, err
	}
	return content, nil
}

func (o *OrderClient) SendRequest(businessCode, reqBody string) (string, error) {
	//校验参数 reqbody
	checkErr := o.checkParams(businessCode, reqBody)
	if checkErr != nil {
		return "", checkErr
	}
	cipherText, sign, err := o.EncryptAndSign([]byte(reqBody))
	if err != nil {
		return "", err
	}

	orderModel := model.OrderModel{
		Data:         cipherText,
		Sign:         sign,
		PtCode:       o.PtCode,
		BusinessCode: businessCode,
	}

	_orderModel, err := json.Marshal(orderModel)
	if err != nil {
		return "", err
	}
	respBody, _, err := makeHttpRequestV4(post, o.IP+orderUrl, nil, map[string]interface{}{"jsonbody": string(_orderModel)}, nil)
	if err != nil {
		return "", err
	}

	return respBody, nil
}

func (o *OrderClient) checkParams(businessCode, reqBody string) error {
	switch businessCode {
	case "AUTHORIZE":
		var authorize model.Authorize
		if err := json.Unmarshal([]byte(reqBody),&authorize); err != nil  {
			return err
		}
		return nil
	case "CREATEORDER":
		var createOrder model.CreateOrder
		if err := json.Unmarshal([]byte(reqBody),&createOrder); err != nil  {
			return err
		}
		return nil
	case "ADDFREEDOMEEMP":
		var freedomEmps model.FreedomEmps
		if err := json.Unmarshal([]byte(reqBody),&freedomEmps); err != nil  {
			return err
		}
		return nil
	case "UPLOADATTACHMENT":
		var uploadFreedomEmps model.UploadFreedomEmps
		if err := json.Unmarshal([]byte(reqBody),&uploadFreedomEmps); err != nil  {
			return err
		}
		return nil
	case "UPLOADATTACHMENTBYURL":
		var uploadFreedomEmpsByUrl model.UploadFreedomEmpsByUrl
		if err := json.Unmarshal([]byte(reqBody),&uploadFreedomEmpsByUrl); err != nil  {
			return err
		}
		return nil
	case "ORDERINFO":
		var simpleOrderInfo model.SimpleOrderInfo
		if err := json.Unmarshal([]byte(reqBody),&simpleOrderInfo); err != nil  {
			return err
		}
		return nil
	case "FREEDOMEMOSTATUS":
		var freedomEmpStatus model.FreedomEmpStatus
		if err := json.Unmarshal([]byte(reqBody),&freedomEmpStatus); err != nil  {
			return err
		}
		return nil
	case "SUBACCOUNTBALANCE":
		var subAccountBalance model.SubAccountBalance
		if err := json.Unmarshal([]byte(reqBody),&subAccountBalance); err != nil  {
			return err
		}
		return nil
	case "UPDATEORDERDETAIL":
		var updateOrderDetail model.UpdateOrderDetail
		if err := json.Unmarshal([]byte(reqBody),&updateOrderDetail); err != nil  {
			return err
		}
		return nil
	case "EXISTRECEIPT":
		var receipt model.Receipt
		if err := json.Unmarshal([]byte(reqBody),&receipt); err != nil  {
			return err
		}
		return nil
	case "DOWNLOADRECEIPT":
		var downloadReceipt model.DownloadReceipt
		if err := json.Unmarshal([]byte(reqBody),&downloadReceipt); err != nil  {
			return err
		}
		return nil
	case "PAYORDER":
		var payOrder model.PayOrder
		if err := json.Unmarshal([]byte(reqBody),&payOrder); err != nil  {
			return err
		}
		return nil
	case "USERINFO":
		var getUserInfo model.GetUserInfo
		if err := json.Unmarshal([]byte(reqBody),&getUserInfo); err != nil  {
			return err
		}
		return nil
	case "PAYORDERV1":
		var mergeCreateOrder model.MergeCreateOrder
		if err := json.Unmarshal([]byte(reqBody),&mergeCreateOrder); err != nil  {
			return err
		}
		return nil
	case "UPDATEMOBILE":
		var updateMobile model.UpdateMobile
		if err := json.Unmarshal([]byte(reqBody),&updateMobile); err != nil  {
			return err
		}
		return nil

	case "CREATEEMPLOYMENTPOOL":
		var createEmploymentPool model.CreateEmploymentPool
		if err := json.Unmarshal([]byte(reqBody),&createEmploymentPool); err != nil  {
			return err
		}
		return nil
	}
	return nil
}