package auth

import (
	"bytes"
	"fmt"
	"golib.gaore.com/GaoreGo/gaore-common-sdk-go/sdk/requests"
	"golib.gaore.com/GaoreGo/gaore-common-sdk-go/sdk/utils"
	"net/url"
	"strings"
	"time"
)

func signRpcRequest(request requests.AcsRequest, signer Signer) (err error) {
	err = completeRpcSignParams(request, signer)
	if err != nil {
		return
	}
	if _, isContainsSign := request.GetQueryParams()["sign"]; isContainsSign {
		delete(request.GetQueryParams(), "sign")
	}

	stringToSign := buildRpcStringToSign(request)
	request.SetStringToSign(stringToSign)
	signature := signer.Sign(stringToSign, "&")
	request.GetQueryParams()["sign"] = signature
	return
}

func completeRpcSignParams(request requests.AcsRequest, signer Signer) (err error) {

	var accessKeyFrom string
	if accessKeyFrom, err = signer.GetAccessKeyFrom(); err != nil {
		return
	}

	queryParams := request.GetQueryParams()
	queryParams["access_time"] = fmt.Sprintf("%d", time.Now().Unix())
	queryParams["access_key"], err = signer.GetAccessKeyId()
	queryParams["access_from"] = accessKeyFrom

	if err != nil {
		return
	}

	request.GetHeaders()["Content-type"] = requests.Form
	request.GetHeaders()["Gr-Sdk-From"] = accessKeyFrom
	formString := utils.GetUrlFormedMap(request.GetFormParams())
	request.SetContent(bytes.NewBufferString(formString).Bytes())
	return
}

func buildRpcStringToSign(request requests.AcsRequest) (stringToSign string) {
	signParams := make(map[string]string)
	for key, value := range request.GetQueryParams() {
		signParams[key] = value
	}
	for key, value := range request.GetFormParams() {
		signParams[key] = value
	}

	stringToSign = utils.GetUrlFormedMap(signParams)
	stringToSign = strings.Replace(stringToSign, "+", "%20", -1)
	stringToSign = strings.Replace(stringToSign, "*", "%2A", -1)
	stringToSign = strings.Replace(stringToSign, "%7E", "~", -1)
	stringToSign = url.QueryEscape(stringToSign)
	stringToSign = request.GetMethod() + "&%2F&" + stringToSign
	return
}