refactor: re-impl sdk3rd

2025-06-17 10:39:39 +08:00
parent 9421da2cde
commit 299a722aa9
280 changed files with 6923 additions and 4658 deletions
--- a/internal/pkg/sdk3rd/dogecloud/api_bind_cdn_cert.go
+++ b/internal/pkg/sdk3rd/dogecloud/api_bind_cdn_cert.go
@@ -0,0 +1,36 @@
+package dogecloud
+
+import (
+	"context"
+	"net/http"
+)
+
+type BindCdnCertRequest struct {
+	CertId int64  `json:"id"`
+	Domain string `json:"domain"`
+}
+
+type BindCdnCertResponse struct {
+	apiResponseBase
+}
+
+func (c *Client) BindCdnCert(req *BindCdnCertRequest) (*BindCdnCertResponse, error) {
+	return c.BindCdnCertWithContext(context.Background(), req)
+}
+
+func (c *Client) BindCdnCertWithContext(ctx context.Context, req *BindCdnCertRequest) (*BindCdnCertResponse, error) {
+	httpreq, err := c.newRequest(http.MethodPost, "/cdn/cert/bind.json")
+	if err != nil {
+		return nil, err
+	} else {
+		httpreq.SetBody(req)
+		httpreq.SetContext(ctx)
+	}
+
+	result := &BindCdnCertResponse{}
+	if _, err := c.doRequestWithResult(httpreq, result); err != nil {
+		return result, err
+	}
+
+	return result, nil
+}
--- a/internal/pkg/sdk3rd/dogecloud/api_upload_cdn_cert.go
+++ b/internal/pkg/sdk3rd/dogecloud/api_upload_cdn_cert.go
@@ -0,0 +1,41 @@
+package dogecloud
+
+import (
+	"context"
+	"net/http"
+)
+
+type UploadCdnCertRequest struct {
+	Note        string `json:"note"`
+	Certificate string `json:"cert"`
+	PrivateKey  string `json:"private"`
+}
+
+type UploadCdnCertResponse struct {
+	apiResponseBase
+
+	Data *struct {
+		Id int64 `json:"id"`
+	} `json:"data,omitempty"`
+}
+
+func (c *Client) UploadCdnCert(req *UploadCdnCertRequest) (*UploadCdnCertResponse, error) {
+	return c.UploadCdnCertWithContext(context.Background(), req)
+}
+
+func (c *Client) UploadCdnCertWithContext(ctx context.Context, req *UploadCdnCertRequest) (*UploadCdnCertResponse, error) {
+	httpreq, err := c.newRequest(http.MethodPost, "/cdn/cert/upload.json")
+	if err != nil {
+		return nil, err
+	} else {
+		httpreq.SetBody(req)
+		httpreq.SetContext(ctx)
+	}
+
+	result := &UploadCdnCertResponse{}
+	if _, err := c.doRequestWithResult(httpreq, result); err != nil {
+		return result, err
+	}
+
+	return result, nil
+}
--- a/internal/pkg/sdk3rd/dogecloud/client.go
+++ b/internal/pkg/sdk3rd/dogecloud/client.go
@@ -8,176 +8,124 @@ import (
 	"fmt"
 	"io"
 	"net/http"
-	"net/url"
-	"strings"
+	"time"
+
+	"github.com/go-resty/resty/v2"
 )

-const dogeHost = "https://api.dogecloud.com"
-
 type Client struct {
-	accessKey string
-	secretKey string
+	client *resty.Client
 }

-func NewClient(accessKey, secretKey string) *Client {
-	return &Client{accessKey: accessKey, secretKey: secretKey}
+func NewClient(accessKey, secretKey string) (*Client, error) {
+	if accessKey == "" {
+		return nil, fmt.Errorf("sdkerr: unset accessKey")
+	}
+	if secretKey == "" {
+		return nil, fmt.Errorf("sdkerr: unset secretKey")
+	}
+
+	client := resty.New().
+		SetBaseURL("https://api.dogecloud.com").
+		SetHeader("Accept", "application/json").
+		SetHeader("Content-Type", "application/json").
+		SetHeader("User-Agent", "certimate").
+		SetPreRequestHook(func(ctx *resty.Client, req *http.Request) error {
+			requestUrl := req.URL.Path
+			requestQuery := req.URL.Query().Encode()
+			if requestQuery != "" {
+				requestUrl += "?" + requestQuery
+			}
+
+			payload := ""
+			if req.Body != nil {
+				reader, err := req.GetBody()
+				if err != nil {
+					return err
+				}
+
+				defer reader.Close()
+
+				payloadb, err := io.ReadAll(reader)
+				if err != nil {
+					return err
+				}
+
+				payload = string(payloadb)
+			}
+
+			stringToSign := fmt.Sprintf("%s\n%s", requestUrl, payload)
+			mac := hmac.New(sha1.New, []byte(secretKey))
+			mac.Write([]byte(stringToSign))
+			sign := hex.EncodeToString(mac.Sum(nil))
+
+			req.Header.Set("Authorization", fmt.Sprintf("TOKEN %s:%s", accessKey, sign))
+
+			return nil
+		})
+
+	return &Client{client}, nil
 }

-func (c *Client) UploadCdnCert(note, cert, private string) (*UploadCdnCertResponse, error) {
-	req := &UploadCdnCertRequest{
-		Note:        note,
-		Certificate: cert,
-		PrivateKey:  private,
+func (c *Client) SetTimeout(timeout time.Duration) *Client {
+	c.client.SetTimeout(timeout)
+	return c
+}
+
+func (c *Client) newRequest(method string, path string) (*resty.Request, error) {
+	if method == "" {
+		return nil, fmt.Errorf("sdkerr: unset method")
+	}
+	if path == "" {
+		return nil, fmt.Errorf("sdkerr: unset path")
 	}

-	reqBts, err := json.Marshal(req)
-	if err != nil {
-		return nil, err
+	req := c.client.R()
+	req.Method = method
+	req.URL = path
+	return req, nil
+}
+
+func (c *Client) doRequest(req *resty.Request) (*resty.Response, error) {
+	if req == nil {
+		return nil, fmt.Errorf("sdkerr: nil request")
 	}

-	reqMap := make(map[string]interface{})
-	err = json.Unmarshal(reqBts, &reqMap)
-	if err != nil {
-		return nil, err
-	}
+	// WARN:
+	//   PLEASE DO NOT USE `req.SetResult` or `req.SetError` HERE! USE `doRequestWithResult` INSTEAD.

-	respBts, err := c.sendReq(http.MethodPost, "cdn/cert/upload.json", reqMap, true)
+	resp, err := req.Send()
 	if err != nil {
-		return nil, err
-	}
-
-	resp := &UploadCdnCertResponse{}
-	err = json.Unmarshal(respBts, resp)
-	if err != nil {
-		return nil, err
-	}
-	if resp.Code != nil && *resp.Code != 0 && *resp.Code != 200 {
-		return nil, fmt.Errorf("dogecloud api error, code: %d, msg: %s", *resp.Code, *resp.Message)
+		return resp, fmt.Errorf("sdkerr: failed to send request: %w", err)
+	} else if resp.IsError() {
+		return resp, fmt.Errorf("sdkerr: unexpected status code: %d, resp: %s", resp.StatusCode(), resp.String())
 	}

 	return resp, nil
 }

-func (c *Client) BindCdnCertWithDomain(certId int64, domain string) (*BindCdnCertResponse, error) {
-	req := &BindCdnCertRequest{
-		CertId: certId,
-		Domain: &domain,
+func (c *Client) doRequestWithResult(req *resty.Request, res apiResponse) (*resty.Response, error) {
+	if req == nil {
+		return nil, fmt.Errorf("sdkerr: nil request")
 	}

-	reqBts, err := json.Marshal(req)
+	resp, err := c.doRequest(req)
 	if err != nil {
-		return nil, err
-	}
-
-	reqMap := make(map[string]interface{})
-	err = json.Unmarshal(reqBts, &reqMap)
-	if err != nil {
-		return nil, err
-	}
-
-	respBts, err := c.sendReq(http.MethodPost, "cdn/cert/bind.json", reqMap, true)
-	if err != nil {
-		return nil, err
-	}
-
-	resp := &BindCdnCertResponse{}
-	err = json.Unmarshal(respBts, resp)
-	if err != nil {
-		return nil, err
-	}
-	if resp.Code != nil && *resp.Code != 0 && *resp.Code != 200 {
-		return nil, fmt.Errorf("dogecloud api error, code: %d, msg: %s", *resp.Code, *resp.Message)
-	}
-
-	return resp, nil
-}
-
-func (c *Client) BindCdnCertWithDomainId(certId int64, domainId int64) (*BindCdnCertResponse, error) {
-	req := &BindCdnCertRequest{
-		CertId:   certId,
-		DomainId: &domainId,
-	}
-
-	reqBts, err := json.Marshal(req)
-	if err != nil {
-		return nil, err
-	}
-
-	reqMap := make(map[string]interface{})
-	err = json.Unmarshal(reqBts, &reqMap)
-	if err != nil {
-		return nil, err
-	}
-
-	respBts, err := c.sendReq(http.MethodPost, "cdn/cert/bind.json", reqMap, true)
-	if err != nil {
-		return nil, err
-	}
-
-	resp := &BindCdnCertResponse{}
-	err = json.Unmarshal(respBts, resp)
-	if err != nil {
-		return nil, err
-	}
-	if resp.Code != nil && *resp.Code != 0 && *resp.Code != 200 {
-		return nil, fmt.Errorf("dogecloud api error, code: %d, msg: %s", *resp.Code, *resp.Message)
-	}
-
-	return resp, nil
-}
-
-// 调用多吉云的 API。
-// https://docs.dogecloud.com/cdn/api-access-token?id=go
-//
-// 入参：
-//   - method：GET 或 POST
-//   - path：是调用的 API 接口地址，包含 URL 请求参数 QueryString，例如：/console/vfetch/add.json?url=xxx&a=1&b=2
-//   - data：POST 的数据，对象，例如 {a: 1, b: 2}，传递此参数表示不是 GET 请求而是 POST 请求
-//   - jsonMode：数据 data 是否以 JSON 格式请求，默认为 false 则使用表单形式（a=1&b=2）
-func (c *Client) sendReq(method string, path string, data map[string]interface{}, jsonMode bool) ([]byte, error) {
-	body := ""
-	mime := ""
-	if jsonMode {
-		_body, err := json.Marshal(data)
-		if err != nil {
-			return nil, err
+		if resp != nil {
+			json.Unmarshal(resp.Body(), &res)
 		}
-		body = string(_body)
-		mime = "application/json"
-	} else {
-		values := url.Values{}
-		for k, v := range data {
-			values.Set(k, v.(string))
+		return resp, err
+	}
+
+	if len(resp.Body()) != 0 {
+		if err := json.Unmarshal(resp.Body(), &res); err != nil {
+			return resp, fmt.Errorf("sdkerr: failed to unmarshal response: %w", err)
+		} else {
+			if tcode := res.GetCode(); tcode != 0 && tcode != 200 {
+				return resp, fmt.Errorf("sdkerr: code='%d', msg='%s'", tcode, res.GetMessage())
+			}
 		}
-		body = values.Encode()
-		mime = "application/x-www-form-urlencoded"
 	}

-	path = strings.TrimPrefix(path, "/")
-	signStr := "/" + path + "\n" + body
-	hmacObj := hmac.New(sha1.New, []byte(c.secretKey))
-	hmacObj.Write([]byte(signStr))
-	sign := hex.EncodeToString(hmacObj.Sum(nil))
-	auth := fmt.Sprintf("TOKEN %s:%s", c.accessKey, sign)
-
-	req, err := http.NewRequest(method, fmt.Sprintf("%s/%s", dogeHost, path), strings.NewReader(body))
-	if err != nil {
-		return nil, err
-	}
-	req.Header.Set("Content-Type", mime)
-	req.Header.Set("Authorization", auth)
-
-	client := http.Client{}
-	resp, err := client.Do(req)
-	if err != nil {
-		return nil, err
-	}
-	defer resp.Body.Close()
-
-	bytes, err := io.ReadAll(resp.Body)
-	if err != nil {
-		return nil, err
-	}
-
-	return bytes, nil
+	return resp, nil
 }
--- a/internal/pkg/sdk3rd/dogecloud/models.go
+++ b/internal/pkg/sdk3rd/dogecloud/models.go
@@ -1,31 +0,0 @@
-package dogecloud
-
-type BaseResponse struct {
-	Code    *int    `json:"code,omitempty"`
-	Message *string `json:"msg,omitempty"`
-}
-
-type UploadCdnCertRequest struct {
-	Note        string `json:"note"`
-	Certificate string `json:"cert"`
-	PrivateKey  string `json:"private"`
-}
-
-type UploadCdnCertResponseData struct {
-	Id int64 `json:"id"`
-}
-
-type UploadCdnCertResponse struct {
-	BaseResponse
-	Data *UploadCdnCertResponseData `json:"data,omitempty"`
-}
-
-type BindCdnCertRequest struct {
-	CertId   int64   `json:"id"`
-	DomainId *int64  `json:"did,omitempty"`
-	Domain   *string `json:"domain,omitempty"`
-}
-
-type BindCdnCertResponse struct {
-	BaseResponse
-}
--- a/internal/pkg/sdk3rd/dogecloud/types.go
+++ b/internal/pkg/sdk3rd/dogecloud/types.go
@@ -0,0 +1,29 @@
+package dogecloud
+
+type apiResponse interface {
+	GetCode() int
+	GetMessage() string
+}
+
+type apiResponseBase struct {
+	Code    *int    `json:"code,omitempty"`
+	Message *string `json:"msg,omitempty"`
+}
+
+func (r *apiResponseBase) GetCode() int {
+	if r.Code == nil {
+		return 0
+	}
+
+	return *r.Code
+}
+
+func (r *apiResponseBase) GetMessage() string {
+	if r.Message == nil {
+		return ""
+	}
+
+	return *r.Message
+}
+
+var _ apiResponse = (*apiResponseBase)(nil)