From 819ffe8d990a282a986176fd54cac4315c9a0aa2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=99=88=E5=B0=9A=E6=9D=B0?= <305986045@qq.com>
Date: Wed, 14 May 2025 15:58:16 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E8=B0=83=E6=95=B4=E9=83=A8=E7=BD=B2?=
 =?UTF-8?q?=E3=80=91=E5=AE=9D=E5=A1=94WAF=E7=BD=91=E7=AB=99=E8=AF=81?=
 =?UTF-8?q?=E4=B9=A6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/internal/cert/deploy/btwaf.go      | 8 ++++----
 backend/internal/cert/deploy/btwaf_test.go | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/backend/internal/cert/deploy/btwaf.go b/backend/internal/cert/deploy/btwaf.go
index 01594bf..6fc56c2 100644
--- a/backend/internal/cert/deploy/btwaf.go
+++ b/backend/internal/cert/deploy/btwaf.go
@@ -87,13 +87,13 @@ func RequestBtWaf(data *map[string]any, method, providerID, requestUrl string) (
 	return res, nil
 }
 
-func GetBTWafSiteList(page int, pageSize int, siteName string) ([]any, error) {
+func GetBTWafSiteList(page int, pageSize int, siteName string, providerId string) ([]any, error) {
 	data := map[string]any{
 		"p":         page,
 		"p_size":    pageSize,
 		"site_name": siteName,
 	}
-	response, err := RequestBtWaf(&data, "POST", "1", "api/wafmastersite/get_site_list")
+	response, err := RequestBtWaf(&data, "POST", providerId, "api/wafmastersite/get_site_list")
 	res := response["res"].(map[string]any)
 	if err != nil {
 		return nil, err
@@ -102,7 +102,7 @@ func GetBTWafSiteList(page int, pageSize int, siteName string) ([]any, error) {
 	return res["list"].([]any), nil
 }
 
-//btwaf不支持通过API设置SSL
+// btwaf不支持通过API设置SSL
 func DeployBtWaf(cfg map[string]any) error {
 	cert, ok := cfg["certificate"].(map[string]any)
 	if !ok {
@@ -166,7 +166,7 @@ func DeployBtWafSite(cfg map[string]any) error {
 		return fmt.Errorf("参数错误：siteName")
 	}
 
-	sitelist, err := GetBTWafSiteList(1, 10, siteName)
+	sitelist, err := GetBTWafSiteList(1, 10, siteName, providerID)
 	if len(sitelist) == 0 || err != nil {
 		return fmt.Errorf("找不到网站：%s", siteName)
 	}
diff --git a/backend/internal/cert/deploy/btwaf_test.go b/backend/internal/cert/deploy/btwaf_test.go
index 7ac1b08..463f304 100644
--- a/backend/internal/cert/deploy/btwaf_test.go
+++ b/backend/internal/cert/deploy/btwaf_test.go
@@ -33,7 +33,7 @@ func TestBTWAFSite(t *testing.T) {
 //}
 
 func TestGetBTWAFSiteList(t *testing.T) {
-	res, err := GetBTWafSiteList(1, 10, "xxx.cn")
+	res, err := GetBTWafSiteList(1, 10, "xxx.cn", "1")
 	for _, site := range res {
 		sites := site.(map[string]any)
 		fmt.Printf("网站名: %s   ID：%s  \n", sites["site_name"], sites["site_id"])