WAF规则中国家/地区、省份、城市、ISP增加候选项检索和选择

internal/web/actions/default/ui/cityOptions.go

@@ -0,0 +1,44 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package ui
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/maps"
+)
+
+type CityOptionsAction struct {
+	actionutils.ParentAction
+}
+
+func (this *CityOptionsAction) RunPost(params struct{}) {
+	citiesResp, err := this.RPC().RegionCityRPC().FindAllEnabledRegionCities(this.AdminContext(), &pb.FindAllEnabledRegionCitiesRequest{
+		IncludeRegionProvince: true,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var cityMaps = []maps.Map{}
+	for _, city := range citiesResp.RegionCities {
+		if city.Codes == nil {
+			city.Codes = []string{}
+		}
+
+		var fullname = city.Name
+		if city.RegionProvince != nil && len(city.RegionProvince.Name) > 0 && city.RegionProvince.Name != city.Name {
+			fullname = city.RegionProvince.Name + " " + fullname
+		}
+
+		cityMaps = append(cityMaps, maps.Map{
+			"id":       city.Id,
+			"name":     city.Name,
+			"fullname": fullname,
+			"codes":    city.Codes,
+		})
+	}
+	this.Data["cities"] = cityMaps
+
+	this.Success()
+}

internal/web/actions/default/ui/countryOptions.go

@@ -0,0 +1,43 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package ui
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/maps"
+	"strings"
+)
+
+type CountryOptionsAction struct {
+	actionutils.ParentAction
+}
+
+func (this *CountryOptionsAction) RunPost(params struct{}) {
+	countriesResp, err := this.RPC().RegionCountryRPC().FindAllEnabledRegionCountries(this.AdminContext(), &pb.FindAllEnabledRegionCountriesRequest{})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var countryMaps = []maps.Map{}
+	for _, country := range countriesResp.RegionCountries {
+		if country.Codes == nil {
+			country.Codes = []string{}
+		}
+
+		var letter = ""
+		if len(country.Pinyin) > 0 && len(country.Pinyin) > 0 && len(country.Pinyin[0]) > 0 {
+			letter = strings.ToUpper(country.Pinyin[0][:1])
+		}
+
+		countryMaps = append(countryMaps, maps.Map{
+			"id":       country.Id,
+			"name":     country.Name,
+			"fullname": letter + " " + country.Name,
+			"codes":    country.Codes,
+		})
+	}
+	this.Data["countries"] = countryMaps
+
+	this.Success()
+}

internal/web/actions/default/ui/download.go

@@ -4,7 +4,7 @@ import (
 	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
 )
 
-// 下载指定的文本内容
+// DownloadAction 下载指定的文本内容
 type DownloadAction struct {
 	actionutils.ParentAction
 }

internal/web/actions/default/ui/init.go

@@ -25,6 +25,10 @@ func init() {
 			Post("/hideTip", new(HideTipAction)).
 			Post("/theme", new(ThemeAction)).
 			Post("/validateIPs", new(ValidateIPsAction)).
+			Post("/providerOptions", new(ProviderOptionsAction)).
+			Post("/countryOptions", new(CountryOptionsAction)).
+			Post("/provinceOptions", new(ProvinceOptionsAction)).
+			Post("/cityOptions", new(CityOptionsAction)).
 			EndAll()
 
 		// 开发环境下总是动态加载，以便于调试

internal/web/actions/default/ui/providerOptions.go

@@ -0,0 +1,36 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package ui
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/maps"
+)
+
+type ProviderOptionsAction struct {
+	actionutils.ParentAction
+}
+
+func (this *ProviderOptionsAction) RunPost(params struct{}) {
+	providersResp, err := this.RPC().RegionProviderRPC().FindAllEnabledRegionProviders(this.AdminContext(), &pb.FindAllEnabledRegionProvidersRequest{})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	var providerMaps = []maps.Map{}
+	for _, provider := range providersResp.RegionProviders {
+		if provider.Codes == nil {
+			provider.Codes = []string{}
+		}
+		providerMaps = append(providerMaps, maps.Map{
+			"id":    provider.Id,
+			"name":  provider.Name,
+			"codes": provider.Codes,
+		})
+	}
+	this.Data["providers"] = providerMaps
+
+	this.Success()
+}

internal/web/actions/default/ui/provinceOptions.go

@@ -0,0 +1,35 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package ui
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/maps"
+)
+
+type ProvinceOptionsAction struct {
+	actionutils.ParentAction
+}
+
+func (this *ProvinceOptionsAction) RunPost(params struct{}) {
+	provincesResp, err := this.RPC().RegionProvinceRPC().FindAllEnabledRegionProvincesWithCountryId(this.AdminContext(), &pb.FindAllEnabledRegionProvincesWithCountryIdRequest{RegionCountryId: ChinaCountryId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var provinceMaps = []maps.Map{}
+	for _, province := range provincesResp.RegionProvinces {
+		if province.Codes == nil {
+			province.Codes = []string{}
+		}
+		provinceMaps = append(provinceMaps, maps.Map{
+			"id":    province.Id,
+			"name":  province.Name,
+			"codes": province.Codes,
+		})
+	}
+	this.Data["provinces"] = provinceMaps
+
+	this.Success()
+}