package restyclient

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

	resty "github.com/go-resty/resty/v2"
)

// CodeNotZeroErr ...
var CodeNotZeroErr = errors.New("resp code != 0")

// BKResult 蓝鲸返回规范的结构体
type BKResult[T any] struct {
	Result  bool   `json:"result"` // 部分蓝鲸接口有, 按需校验
	Code    any    `json:"code"`
	Message string `json:"message"`
	Data    *T     `json:"data"`
}

// NewBKResult create NewBKResult by resp
func NewBKResult[T any](resp *resty.Response) (*BKResult[T], error) {
	if !resp.IsSuccess() {
		return nil, fmt.Errorf("request failed, status: %s, message: %s", resp.Status(), resp.Body())
	}

	bkResult := new(BKResult[T])
	if err := json.Unmarshal(resp.Body(), bkResult); err != nil {
		return nil, err
	}

	if err := bkResult.ValidateCode(); err != nil {
		return nil, err
	}

	return bkResult, nil
}

// NewBKData only create data by resp
func NewBKData[T any](resp *resty.Response) (*T, error) {
	bkResult, err := NewBKResult[T](resp)
	if err != nil {
		return nil, err
	}

	return bkResult.Data, nil
}

// ValidateCode 返回结果是否OK
func (r *BKResult[T]) ValidateCode() error {
	var resultCode int

	switch code := r.Code.(type) {
	case int:
		resultCode = code
	case float64:
		resultCode = int(code)
	case string:
		c, err := strconv.Atoi(code)
		if err != nil {
			return err
		}
		resultCode = c
	default:
		return fmt.Errorf("conversion to int from %T not supported", code)
	}

	if resultCode != 0 {
		return fmt.Errorf("%w, code=%d, message=%s", CodeNotZeroErr, resultCode, r.Message)
	}
	return nil
}