chore: move '/internal/pkg' to '/pkg'

pkg/sdk3rd/btwaf/api_config_set_cert.go

@@ -0,0 +1,36 @@
+package btwaf
+
+import (
+	"context"
+	"net/http"
+)
+
+type ConfigSetCertRequest struct {
+	CertContent *string `json:"certContent,omitempty"`
+	KeyContent  *string `json:"keyContent,omitempty"`
+}
+
+type ConfigSetCertResponse struct {
+	apiResponseBase
+}
+
+func (c *Client) ConfigSetCert(req *ConfigSetCertRequest) (*ConfigSetCertResponse, error) {
+	return c.ConfigSetCertWithContext(context.Background(), req)
+}
+
+func (c *Client) ConfigSetCertWithContext(ctx context.Context, req *ConfigSetCertRequest) (*ConfigSetCertResponse, error) {
+	httpreq, err := c.newRequest(http.MethodPost, "/config/set_cert")
+	if err != nil {
+		return nil, err
+	} else {
+		httpreq.SetBody(req)
+		httpreq.SetContext(ctx)
+	}
+
+	result := &ConfigSetCertResponse{}
+	if _, err := c.doRequestWithResult(httpreq, result); err != nil {
+		return result, err
+	}
+
+	return result, nil
+}

pkg/sdk3rd/btwaf/api_get_site_list.go

@@ -0,0 +1,42 @@
+package btwaf
+
+import (
+	"context"
+	"net/http"
+)
+
+type GetSiteListRequest struct {
+	Page     *int32  `json:"p,omitempty"`
+	PageSize *int32  `json:"p_size,omitempty"`
+	SiteName *string `json:"site_name,omitempty"`
+}
+
+type GetSiteListResponse struct {
+	apiResponseBase
+
+	Result *struct {
+		List  []*SiteRecord `json:"list"`
+		Total int32         `json:"total"`
+	} `json:"res,omitempty"`
+}
+
+func (c *Client) GetSiteList(req *GetSiteListRequest) (*GetSiteListResponse, error) {
+	return c.GetSiteListWithContext(context.Background(), req)
+}
+
+func (c *Client) GetSiteListWithContext(ctx context.Context, req *GetSiteListRequest) (*GetSiteListResponse, error) {
+	httpreq, err := c.newRequest(http.MethodPost, "/wafmastersite/get_site_list")
+	if err != nil {
+		return nil, err
+	} else {
+		httpreq.SetBody(req)
+		httpreq.SetContext(ctx)
+	}
+
+	result := &GetSiteListResponse{}
+	if _, err := c.doRequestWithResult(httpreq, result); err != nil {
+		return result, err
+	}
+
+	return result, nil
+}

pkg/sdk3rd/btwaf/api_modify_site.go

@@ -0,0 +1,37 @@
+package btwaf
+
+import (
+	"context"
+	"net/http"
+)
+
+type ModifySiteRequest struct {
+	SiteId *string         `json:"site_id,omitempty"`
+	Type   *string         `json:"types,omitempty"`
+	Server *SiteServerInfo `json:"server,omitempty"`
+}
+
+type ModifySiteResponse struct {
+	apiResponseBase
+}
+
+func (c *Client) ModifySite(req *ModifySiteRequest) (*ModifySiteResponse, error) {
+	return c.ModifySiteWithContext(context.Background(), req)
+}
+
+func (c *Client) ModifySiteWithContext(ctx context.Context, req *ModifySiteRequest) (*ModifySiteResponse, error) {
+	httpreq, err := c.newRequest(http.MethodPost, "/wafmastersite/modify_site")
+	if err != nil {
+		return nil, err
+	} else {
+		httpreq.SetBody(req)
+		httpreq.SetContext(ctx)
+	}
+
+	result := &ModifySiteResponse{}
+	if _, err := c.doRequestWithResult(httpreq, result); err != nil {
+		return result, err
+	}
+
+	return result, nil
+}

pkg/sdk3rd/btwaf/client.go

@@ -0,0 +1,118 @@
+package btwaf
+
+import (
+	"crypto/md5"
+	"crypto/tls"
+	"encoding/hex"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+
+	"github.com/go-resty/resty/v2"
+)
+
+type Client struct {
+	client *resty.Client
+}
+
+func NewClient(serverUrl, apiKey string) (*Client, error) {
+	if serverUrl == "" {
+		return nil, fmt.Errorf("sdkerr: unset serverUrl")
+	}
+	if _, err := url.Parse(serverUrl); err != nil {
+		return nil, fmt.Errorf("sdkerr: invalid serverUrl: %w", err)
+	}
+	if apiKey == "" {
+		return nil, fmt.Errorf("sdkerr: unset apiKey")
+	}
+
+	client := resty.New().
+		SetBaseURL(strings.TrimRight(serverUrl, "/")+"/api").
+		SetHeader("Accept", "application/json").
+		SetHeader("Content-Type", "application/json").
+		SetHeader("User-Agent", "certimate").
+		SetPreRequestHook(func(c *resty.Client, req *http.Request) error {
+			timestamp := fmt.Sprintf("%d", time.Now().Unix())
+			keyMd5 := md5.Sum([]byte(apiKey))
+			keyMd5Hex := strings.ToLower(hex.EncodeToString(keyMd5[:]))
+			signMd5 := md5.Sum([]byte(timestamp + keyMd5Hex))
+			signMd5Hex := strings.ToLower(hex.EncodeToString(signMd5[:]))
+			req.Header.Set("waf_request_time", timestamp)
+			req.Header.Set("waf_request_token", signMd5Hex)
+
+			return nil
+		})
+
+	return &Client{client}, nil
+}
+
+func (c *Client) SetTimeout(timeout time.Duration) *Client {
+	c.client.SetTimeout(timeout)
+	return c
+}
+
+func (c *Client) SetTLSConfig(config *tls.Config) *Client {
+	c.client.SetTLSClientConfig(config)
+	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")
+	}
+
+	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")
+	}
+
+	// WARN:
+	//   PLEASE DO NOT USE `req.SetResult` or `req.SetError` HERE! USE `doRequestWithResult` INSTEAD.
+
+	resp, err := req.Send()
+	if err != nil {
+		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) doRequestWithResult(req *resty.Request, res apiResponse) (*resty.Response, error) {
+	if req == nil {
+		return nil, fmt.Errorf("sdkerr: nil request")
+	}
+
+	resp, err := c.doRequest(req)
+	if err != nil {
+		if resp != nil {
+			json.Unmarshal(resp.Body(), &res)
+		}
+		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 code := res.GetCode(); code != 0 {
+				return resp, fmt.Errorf("sdkerr: api error: code='%d'", code)
+			}
+		}
+	}
+
+	return resp, nil
+}

pkg/sdk3rd/btwaf/types.go

@@ -0,0 +1,39 @@
+package btwaf
+
+type apiResponse interface {
+	GetCode() int32
+}
+
+type apiResponseBase struct {
+	Code *int32 `json:"code,omitempty"`
+}
+
+func (r *apiResponseBase) GetCode() int32 {
+	if r.Code == nil {
+		return 0
+	}
+
+	return *r.Code
+}
+
+var _ apiResponse = (*apiResponseBase)(nil)
+
+type SiteRecord struct {
+	SiteId     string `json:"site_id"`
+	SiteName   string `json:"site_name"`
+	Type       string `json:"types"`
+	Status     int32  `json:"status"`
+	CreateTime int64  `json:"create_time"`
+	UpdateTime int64  `json:"update_time"`
+}
+
+type SiteServerInfo struct {
+	ListenSSLPorts *[]int32           `json:"listen_ssl_port,omitempty"`
+	SSL            *SiteServerSSLInfo `json:"ssl,omitempty"`
+}
+
+type SiteServerSSLInfo struct {
+	IsSSL      *int32  `json:"is_ssl,omitempty"`
+	FullChain  *string `json:"full_chain,omitempty"`
+	PrivateKey *string `json:"private_key,omitempty"`
+}