package key

import (
	"context"
	"encoding/json"
	"fmt"
	"gitee.com/lansexiongdi/ymt/api"
	"gitee.com/lansexiongdi/ymt/core"
	"io"
	"net/http"
)

const (
	orderMethod   = "/openapi/v1/key/order"
	queryMethod   = "/openapi/v1/key/query"
	discardMethod = "/openapi/v1/key/discard"
)

type Key api.Service

func (k *Key) Order(ctx context.Context, request *OrderRequest) (http.Header, *http.Response, *core.Response, error) {

	h, httpResponse, bodyBytes, err := k.Post(ctx, orderMethod, request)
	if err != nil {
		return nil, nil, nil, err
	}

	res, err := core.BuildResponse(bodyBytes)
	if err != nil {
		return h, httpResponse, nil, err
	}

	return h, httpResponse, res, nil
}

func (k *Key) Query(ctx context.Context, request *QueryRequest) (http.Header, *http.Response, *core.Response, error) {

	h, httpResponse, bodyBytes, err := k.Post(ctx, queryMethod, request)
	if err != nil {
		return nil, nil, nil, err
	}

	res, err := core.BuildResponse(bodyBytes)
	if err != nil {
		return h, httpResponse, nil, err
	}

	return h, httpResponse, res, nil
}

func (k *Key) Discard(ctx context.Context, request *DiscardRequest) (http.Header, *http.Response, *core.Response, error) {

	h, httpResponse, bodyBytes, err := k.Post(ctx, discardMethod, request)
	if err != nil {
		return nil, nil, nil, err
	}

	res, err := core.BuildResponse(bodyBytes)
	if err != nil {
		return h, httpResponse, nil, err
	}

	return h, httpResponse, res, nil
}

func (k *Key) Notify(_ context.Context, n *Notify) (*NotifyData, error) {

	if !k.CryptographySuite.Verifier.Verify(n.SignString(), n.Sign) {
		return nil, fmt.Errorf("verify sign fail")
	}

	return &n.Data, nil
}

func (k *Key) CallBack(ctx context.Context, req *http.Request) (*NotifyData, error) {

	body, err := io.ReadAll(req.Body)
	if err != nil {
		return nil, err
	}

	var n *Notify
	if err = json.Unmarshal(body, &n); err != nil {
		return nil, err
	}

	sign := req.Header.Get("Sign")

	if sign == "" {
		return k.Notify(ctx, n)
	}

	timestamp := req.Header.Get("Timestamp")
	if timestamp == "" {
		return nil, fmt.Errorf("timestamp is empty")
	}

	appid := req.Header.Get("Appid")
	if appid == "" {
		return nil, fmt.Errorf("appid is empty")
	}

	signType := req.Header.Get("Sign-Type")
	if signType == "" {
		return nil, fmt.Errorf("sign-type is empty")
	}

	if appid != k.Config.AppID {
		return nil, fmt.Errorf("appid is invalid")
	}
	if signType != string(k.Config.SignType) {
		return nil, fmt.Errorf("sign-type is invalid")
	}

	ciphertext, err := k.GetCiphertext(&n.Data)
	if err != nil {
		return nil, err
	}

	if !k.Verify(timestamp, ciphertext, sign) {
		return nil, fmt.Errorf("call back verify sign fail")
	}

	return &n.Data, nil
}