From db7211c31407c4965e697501d7e548bd349b6c16 Mon Sep 17 00:00:00 2001
From: samwaf <samwafgo@gmail.com>
Date: Fri, 21 Nov 2025 08:53:34 +0800
Subject: [PATCH] feat: export bad ip

#470
---
 api/waf_common.go                   | 9 +++++++--
 service/waf_service/waf_block_ip.go | 7 +++++++
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/api/waf_common.go b/api/waf_common.go
index 5596f72..9904c7e 100644
--- a/api/waf_common.go
+++ b/api/waf_common.go
@@ -10,14 +10,15 @@ import (
 	"SamWaf/model/request"
 	"errors"
 	"fmt"
-	"github.com/gin-gonic/gin"
-	"github.com/xuri/excelize/v2"
 	"log"
 	"net/http"
 	"os"
 	"reflect"
 	"strconv"
 	"time"
+
+	"github.com/gin-gonic/gin"
+	"github.com/xuri/excelize/v2"
 )
 
 type WafCommonApi struct {
@@ -141,6 +142,10 @@ func getStructTypeValueByName(name string) (reflect.Type, reflect.Value) {
 		// 模拟获取数据
 		webHosts := wafHostService.GetAllHostApi()
 		return reflect.TypeOf(model.Hosts{}), reflect.ValueOf(webHosts)
+	case "ipblock":
+		// 获取所有IP黑名单数据
+		ipBlocks := wafIpBlockService.GetAllBlockIpApi()
+		return reflect.TypeOf(model.IPBlockList{}), reflect.ValueOf(ipBlocks)
 	default:
 		return nil, reflect.ValueOf(nil)
 	}
diff --git a/service/waf_service/waf_block_ip.go b/service/waf_service/waf_block_ip.go
index c5c8d54..fff62cf 100644
--- a/service/waf_service/waf_block_ip.go
+++ b/service/waf_service/waf_block_ip.go
@@ -172,3 +172,10 @@ func (receiver *WafBlockIpService) GetHostCodes() ([]string, error) {
 		Pluck("host_code", &hostCodes).Error
 	return hostCodes, err
 }
+
+// GetAllBlockIpApi 获取所有IP黑名单数据（用于导出）
+func (receiver *WafBlockIpService) GetAllBlockIpApi() []model.IPBlockList {
+	var list []model.IPBlockList
+	global.GWAF_LOCAL_DB.Where("user_code = ? AND tenant_id = ?", global.GWAF_USER_CODE, global.GWAF_TENANT_ID).Find(&list)
+	return list
+}