From 951fd51fe6f0646283bccca94de02c9014f8bb93 Mon Sep 17 00:00:00 2001
From: samwaf <samwafgo@gmail.com>
Date: Mon, 25 Dec 2023 14:34:01 +0800
Subject: [PATCH] =?UTF-8?q?feat:=E9=A6=96=E9=A1=B5=E4=BF=A1=E6=81=AF?=
 =?UTF-8?q?=E6=8F=90=E7=A4=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/waf_stat.go                               |  5 +++
 localwaf/src/apis/stats.ts                    | 10 ++++-
 .../dashboard/base/components/MiddleChart.vue |  1 -
 localwaf/src/pages/dashboard/base/index.vue   | 40 +++++++++++++++++--
 model/response/waf_stat.go                    |  9 +++++
 router/waf_stat.go                            |  1 +
 service/waf_service/waf_account.go            | 17 ++++++++
 service/waf_service/waf_host.go               | 13 ++++++
 service/waf_service/waf_rule.go               |  2 +-
 service/waf_service/waf_stat.go               |  5 +++
 10 files changed, 95 insertions(+), 8 deletions(-)

diff --git a/api/waf_stat.go b/api/waf_stat.go
index 1259ff4..f41c7ea 100644
--- a/api/waf_stat.go
+++ b/api/waf_stat.go
@@ -9,6 +9,11 @@ import (
 type WafStatApi struct {
 }
 
+// 获取系统基本信息
+func (w *WafStatApi) StatSysinfoApi(c *gin.Context) {
+
+	response.OkWithDetailed(wafStatService.StatHomeSysinfo(), "获取成功", c)
+}
 func (w *WafStatApi) StatHomeSumDayApi(c *gin.Context) {
 
 	wafStat, _ := wafStatService.StatHomeSumDayApi()
diff --git a/localwaf/src/apis/stats.ts b/localwaf/src/apis/stats.ts
index ef9a5b7..4acd542 100644
--- a/localwaf/src/apis/stats.ts
+++ b/localwaf/src/apis/stats.ts
@@ -23,8 +23,14 @@ export function wafstatsumdaytopiprangeapi(params) {
     params: params
   })
 }
-
-
+//首页获取基本信息
+export function wafStatSysinfoapi(params) {
+  return request({
+    url: 'statsysinfo',
+    method: 'get',
+    params: params
+  })
+}
 //【数据分析】查询周期区间国家级别攻击和正常信息
 export function wafanalysisdaycountryrange(params) {
   return request({
diff --git a/localwaf/src/pages/dashboard/base/components/MiddleChart.vue b/localwaf/src/pages/dashboard/base/components/MiddleChart.vue
index e2880de..faa951f 100644
--- a/localwaf/src/pages/dashboard/base/components/MiddleChart.vue
+++ b/localwaf/src/pages/dashboard/base/components/MiddleChart.vue
@@ -88,7 +88,6 @@ export default {
 
     window.addEventListener('resize', this.updateContainer, false);
 
-
   },
 
   methods: {
diff --git a/localwaf/src/pages/dashboard/base/index.vue b/localwaf/src/pages/dashboard/base/index.vue
index eedf90f..db05df0 100644
--- a/localwaf/src/pages/dashboard/base/index.vue
+++ b/localwaf/src/pages/dashboard/base/index.vue
@@ -1,6 +1,11 @@
 <template>
   <div>
-    <t-alert theme="info" :message="infoMessage" >
+    <t-alert theme="error" :message="defaultAccountMessage" v-if="defaultAccountVisable" >
+      <template #operation>
+        <span @click="handleModifyDefaultPassWebOperation">马上修改</span>
+      </template>
+    </t-alert>
+    <t-alert theme="success" :message="emptyHostMessage"  v-if="emptyHostVisable">
       <template #operation>
         <span @click="handleCreateWebOperation">马上创建</span>
       </template>
@@ -13,12 +18,14 @@
    <!-- <rank-list class="row-container" /> -->
   </div>
 </template>
-<script>
+<script lang="ts">
 import TopPanel from './components/TopPanel.vue';
 import MiddleChart from './components/MiddleChart.vue';
 import RankList from './components/RankList.vue';
 import OutputOverview from './components/OutputOverview.vue';
-
+import {
+  wafStatSysinfoapi
+} from '@/apis/stats';
 export default {
   name: 'DashboardBase',
   components: {
@@ -31,9 +38,15 @@ export default {
     return {
       center: {lng: 0, lat: 0},
       zoom: 3,
-      infoMessage:"您当前未创建需要防护的网站，点击进行创建"
+      emptyHostMessage:"您当前未创建需要防护的网站，点击进行创建",
+      emptyHostVisable: false,
+      defaultAccountMessage:"您当前还是使用默认帐号和密码请尽快修改",
+      defaultAccountVisable:false,
     }
   },
+  mounted() {
+    this.loadSysInfo()
+  },
   methods: {
      handler ({BMap, map}) {
           console.log(BMap, map)
@@ -51,6 +64,25 @@ export default {
           },
         },
       );
+    },//引导修改默认密码
+    handleModifyDefaultPassWebOperation(){
+      this.$router.push(
+        {
+          path: '/account/Account',
+          query: {
+            sourcePage: "HomeFrist",
+          },
+        },
+      );
+    },
+    loadSysInfo(){
+      wafStatSysinfoapi({}).then((res)=>{
+        console.log("home res",res.data)
+        this.defaultAccountVisable = res.data.is_default_account
+        this.emptyHostVisable =  res.data.is_empty_host
+      } ).catch((e: Error) => {
+        console.log(e);
+      }).finally(() => {})
     },
     //end method
   },
diff --git a/model/response/waf_stat.go b/model/response/waf_stat.go
index 138639e..0046dd8 100644
--- a/model/response/waf_stat.go
+++ b/model/response/waf_stat.go
@@ -39,3 +39,12 @@ type WafAnalysisDayStats struct {
 	Name  string `json:"name"  form:"name"`
 	Value int64  `json:"value"  form:"value"`
 }
+
+/*
+*
+首页 获取系统基本信息
+*/
+type WafHomeSysinfoStat struct {
+	IsDefaultAccount bool `json:"is_default_account"  form:"is_default_account"`
+	IsEmptyHost      bool `json:"is_empty_host"  form:"is_empty_host"`
+}
diff --git a/router/waf_stat.go b/router/waf_stat.go
index b4e3c4a..79f24c9 100644
--- a/router/waf_stat.go
+++ b/router/waf_stat.go
@@ -15,6 +15,7 @@ func (receiver *StatRouter) InitStatRouter(group *gin.RouterGroup) {
 	router.GET("/samwaf/wafstatsumday", statApi.StatHomeSumDayApi)
 	router.GET("/samwaf/wafstatsumdayrange", statApi.StatHomeSumDayRangeApi)
 	router.GET("/samwaf/wafstatsumdaytopiprange", statApi.StatHomeSumDayTopIPRangeApi)
+	router.GET("/samwaf/statsysinfo", statApi.StatSysinfoApi)
 
 	//数据分析
 	router.GET("/samwaf/wafanalysisdaycountryrange", statApi.StatAnalysisDayCountryRangeApi)
diff --git a/service/waf_service/waf_account.go b/service/waf_service/waf_account.go
index 22f143b..15212c6 100644
--- a/service/waf_service/waf_account.go
+++ b/service/waf_service/waf_account.go
@@ -73,6 +73,23 @@ func (receiver *WafAccountService) GetInfoByLoginAccount(loginAccount string) mo
 	global.GWAF_LOCAL_DB.Where("login_account=? ", loginAccount).Find(&bean)
 	return bean
 }
+
+/*
+*
+ */
+func (receiver *WafAccountService) IsExistDefaultAccount() bool {
+	var total int64 = 0
+	err := global.GWAF_LOCAL_DB.Model(&model.Account{}).Where("login_account=? and login_password =?", global.GWAF_DEFAULT_ACCOUNT, global.GWAF_DEFAULT_ACCOUNT_PWD).Count(&total).Error
+	if err == nil {
+		if total > 0 {
+			return true
+		} else {
+			return false
+		}
+	} else {
+		return false
+	}
+}
 func (receiver *WafAccountService) GetDetailByIdApi(id string) model.Account {
 	var bean model.Account
 	global.GWAF_LOCAL_DB.Where("id=?", id).Find(&bean)
diff --git a/service/waf_service/waf_host.go b/service/waf_service/waf_host.go
index c2d2e1f..f292aa0 100644
--- a/service/waf_service/waf_host.go
+++ b/service/waf_service/waf_host.go
@@ -142,3 +142,16 @@ func (receiver *WafHostService) CheckPortExistApi(port int) int64 {
 	global.GWAF_LOCAL_DB.Model(&model.Hosts{}).Where("port=?", port).Count(&total)
 	return total
 }
+func (receiver *WafHostService) IsEmptyHost() bool {
+	var total int64 = 0
+	err := global.GWAF_LOCAL_DB.Model(&model.Hosts{}).Where("global_host=?", 0).Count(&total).Error
+	if err == nil {
+		if total == 0 {
+			return true
+		} else {
+			return false
+		}
+	} else {
+		return false
+	}
+}
diff --git a/service/waf_service/waf_rule.go b/service/waf_service/waf_rule.go
index e6c04dc..61bb78b 100644
--- a/service/waf_service/waf_rule.go
+++ b/service/waf_service/waf_rule.go
@@ -37,7 +37,7 @@ func (receiver *WafRuleService) AddApi(wafRuleAddReq request.WafRuleAddReq, rule
 		IsManualRule:    wafRuleAddReq.IsManualRule,
 		RuleStatus:      1,
 	}
-	global.GWAF_LOCAL_DB.Debug().Create(wafRule)
+	global.GWAF_LOCAL_DB.Create(wafRule)
 	return nil
 }
 
diff --git a/service/waf_service/waf_stat.go b/service/waf_service/waf_stat.go
index e35667c..fcb1144 100644
--- a/service/waf_service/waf_stat.go
+++ b/service/waf_service/waf_stat.go
@@ -120,3 +120,8 @@ func (receiver *WafStatService) StatAnalysisDayCountryRangeApi(req request.WafSt
 		req.StartDay, req.EndDay, req.AttackType).Select(" country as Name ,sum(count) as Value").Group("country").Order("sum(count) desc").Scan(&CountOfRange)
 	return CountOfRange
 }
+
+// 获取系统基本信息
+func (receiver *WafStatService) StatHomeSysinfo() response2.WafHomeSysinfoStat {
+	return response2.WafHomeSysinfoStat{IsDefaultAccount: WafAccountServiceApp.IsExistDefaultAccount(), IsEmptyHost: WafHostServiceApp.IsEmptyHost()}
+}