Initial commit (code only without large binaries)

EdgeAdmin/internal/web/actions/default/clusters/clusterutils/cluster_helper.go

@@ -0,0 +1,211 @@
+package clusterutils
+
+import (
+	teaconst "github.com/TeaOSLab/EdgeAdmin/internal/const"
+	"github.com/TeaOSLab/EdgeAdmin/internal/utils/numberutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/helpers"
+	"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/dao"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/actions"
+	"github.com/iwind/TeaGo/logs"
+	"github.com/iwind/TeaGo/maps"
+	"net/http"
+	"strconv"
+)
+
+// ClusterHelper 单个集群的帮助
+type ClusterHelper struct {
+	helpers.LangHelper
+}
+
+func NewClusterHelper() *ClusterHelper {
+	return &ClusterHelper{}
+}
+
+func (this *ClusterHelper) BeforeAction(actionPtr actions.ActionWrapper) (goNext bool) {
+	var action = actionPtr.Object()
+	if action.Request.Method != http.MethodGet {
+		return true
+	}
+
+	action.Data["teaMenu"] = "clusters"
+
+	var selectedTabbar = action.Data.GetString("mainTab")
+	var clusterId = action.ParamInt64("clusterId")
+	var clusterIdString = strconv.FormatInt(clusterId, 10)
+	action.Data["clusterId"] = clusterId
+
+	if clusterId > 0 {
+		var ctx = actionPtr.(actionutils.ActionInterface).AdminContext()
+		cluster, err := dao.SharedNodeClusterDAO.FindEnabledNodeCluster(ctx, clusterId)
+		if err != nil {
+			logs.Error(err)
+			return
+		}
+		if cluster == nil {
+			action.WriteString("can not find cluster")
+			return
+		}
+		action.Data["currentClusterName"] = cluster.Name
+
+		clusterInfo, err := dao.SharedNodeClusterDAO.FindEnabledNodeClusterConfigInfo(ctx, clusterId)
+		if err != nil {
+			logs.Error(err)
+			return
+		}
+		if clusterInfo == nil {
+			action.WriteString("can not find cluster info")
+			return
+		}
+
+		var nodeId = action.ParamInt64("nodeId")
+		var isInCluster = nodeId <= 0
+
+		var tabbar = actionutils.NewTabbar()
+		{
+			var url = "/clusters"
+			if !isInCluster {
+				url = "/clusters/cluster/nodes?clusterId=" + clusterIdString
+			}
+			tabbar.Add("", "", url, "arrow left", false)
+		}
+		{
+			var url = "/clusters/cluster?clusterId=" + clusterIdString
+			if !isInCluster {
+				url = "/clusters/cluster/nodes?clusterId=" + clusterIdString
+			}
+
+			var item = tabbar.Add(cluster.Name, "", url, "angle right", true)
+			item.IsTitle = true
+		}
+		if teaconst.IsPlus {
+			{
+				var item = tabbar.Add(this.Lang(actionPtr, codes.NodeClusterMenu_TabClusterDashboard), "", "/clusters/cluster/boards?clusterId="+clusterIdString, "chart line area", selectedTabbar == "board")
+				item.IsDisabled = !isInCluster
+			}
+		}
+		{
+			var item = tabbar.Add(this.Lang(actionPtr, codes.NodeClusterMenu_TabClusterNodes), "", "/clusters/cluster/nodes?clusterId="+clusterIdString, "server", selectedTabbar == "node")
+			item.IsDisabled = !isInCluster
+		}
+
+		{
+			var item = tabbar.Add(this.Lang(actionPtr, codes.NodeClusterMenu_TabClusterSettings), "", "/clusters/cluster/settings?clusterId="+clusterIdString, "setting", selectedTabbar == "setting")
+			item.IsDisabled = !isInCluster
+		}
+		{
+			var item = tabbar.Add(this.Lang(actionPtr, codes.NodeClusterMenu_TabClusterDelete), "", "/clusters/cluster/delete?clusterId="+clusterIdString, "trash", selectedTabbar == "delete")
+			item.IsDisabled = !isInCluster
+		}
+		actionutils.SetTabbar(action, tabbar)
+
+		// 左侧菜单
+		var secondMenuItem = action.Data.GetString("secondMenuItem")
+		switch selectedTabbar {
+		case "setting":
+			var menuItems = this.createSettingMenu(cluster, clusterInfo, secondMenuItem, actionPtr)
+			action.Data["leftMenuItems"] = menuItems
+
+			// 当前菜单
+			action.Data["leftMenuActiveItem"] = nil
+			for _, item := range menuItems {
+				if item.GetBool("isActive") {
+					action.Data["leftMenuActiveItem"] = item
+					break
+				}
+			}
+		}
+	}
+
+	return true
+}
+
+// 设置菜单
+func (this *ClusterHelper) createSettingMenu(cluster *pb.NodeCluster, info *pb.FindEnabledNodeClusterConfigInfoResponse, selectedItem string, actionPtr actions.ActionWrapper) (items []maps.Map) {
+	clusterId := numberutils.FormatInt64(cluster.Id)
+	items = append(items, maps.Map{
+		"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingBasic),
+		"url":      "/clusters/cluster/settings?clusterId=" + clusterId,
+		"isActive": selectedItem == "basic",
+		"isOn":     true,
+	})
+
+	items = append(items, maps.Map{
+		"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingDNS),
+		"url":      "/clusters/cluster/settings/dns?clusterId=" + clusterId,
+		"isActive": selectedItem == "dns",
+		"isOn":     cluster.DnsDomainId > 0 || len(cluster.DnsName) > 0,
+	})
+	items = append(items, maps.Map{
+		"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingHealthCheck),
+		"url":      "/clusters/cluster/settings/health?clusterId=" + clusterId,
+		"isActive": selectedItem == "health",
+		"isOn":     info != nil && info.HealthCheckIsOn,
+	})
+
+	items = append(items, maps.Map{
+		"name": "-",
+	})
+
+	items = append(items, maps.Map{
+		"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingServiceGlobal),
+		"url":      "/clusters/cluster/settings/global-server-config?clusterId=" + clusterId,
+		"isActive": selectedItem == "globalServerConfig",
+		"isOn":     true,
+	})
+
+	items = append(items, maps.Map{
+		"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingCachePolicy),
+		"url":      "/clusters/cluster/settings/cache?clusterId=" + clusterId,
+		"isActive": selectedItem == "cache",
+		"isOn":     cluster.HttpCachePolicyId > 0,
+	})
+	items = append(items, maps.Map{
+		"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingWAFPolicy),
+		"url":      "/clusters/cluster/settings/waf?clusterId=" + clusterId,
+		"isActive": selectedItem == "waf",
+		"isOn":     cluster.HttpFirewallPolicyId > 0,
+	})
+
+	items = append(items, maps.Map{
+		"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingWAFActions),
+		"url":      "/clusters/cluster/settings/firewall-actions?clusterId=" + clusterId,
+		"isActive": selectedItem == "firewallAction",
+		"isOn":     info != nil && info.HasFirewallActions,
+	})
+
+	items = append(items, maps.Map{
+		"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingWebP),
+		"url":      "/clusters/cluster/settings/webp?clusterId=" + clusterId,
+		"isActive": selectedItem == "webp",
+		"isOn":     info != nil && info.WebPIsOn,
+	})
+
+	items = this.filterMenuItems1(items, info, clusterId, selectedItem, actionPtr)
+
+	items = append(items, maps.Map{
+		"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingMetrics),
+		"url":      "/clusters/cluster/settings/metrics?clusterId=" + clusterId,
+		"isActive": selectedItem == "metric",
+		"isOn":     info != nil && info.HasMetricItems,
+	})
+
+	items = append(items, maps.Map{
+		"name":     "-",
+		"url":      "",
+		"isActive": false,
+	})
+
+	items = append(items, maps.Map{
+		"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingDDoSProtection),
+		"url":      "/clusters/cluster/settings/ddos-protection?clusterId=" + clusterId,
+		"isActive": selectedItem == "ddosProtection",
+		"isOn":     info != nil && info.HasDDoSProtection,
+	})
+
+	items = this.filterMenuItems2(items, info, clusterId, selectedItem, actionPtr)
+
+	return
+}

EdgeAdmin/internal/web/actions/default/clusters/clusterutils/cluster_helper_ext.go

@@ -0,0 +1,18 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build !plus
+
+package clusterutils
+
+import (
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/actions"
+	"github.com/iwind/TeaGo/maps"
+)
+
+func (this *ClusterHelper) filterMenuItems1(items []maps.Map, info *pb.FindEnabledNodeClusterConfigInfoResponse, clusterIdString string, selectedItem string, actionPtr actions.ActionWrapper) []maps.Map {
+	return items
+}
+
+func (this *ClusterHelper) filterMenuItems2(items []maps.Map, info *pb.FindEnabledNodeClusterConfigInfoResponse, clusterIdString string, selectedItem string, actionPtr actions.ActionWrapper) []maps.Map {
+	return items
+}

EdgeAdmin/internal/web/actions/default/clusters/clusterutils/cluster_helper_ext_plus.go

@@ -0,0 +1,102 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build plus
+
+package clusterutils
+
+import (
+	teaconst "github.com/TeaOSLab/EdgeAdmin/internal/const"
+	"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	plusutils "github.com/TeaOSLab/EdgePlus/pkg/utils"
+	"github.com/iwind/TeaGo/actions"
+	"github.com/iwind/TeaGo/maps"
+)
+
+func (this *ClusterHelper) filterMenuItems1(items []maps.Map, info *pb.FindEnabledNodeClusterConfigInfoResponse, clusterIdString string, selectedItem string, actionPtr actions.ActionWrapper) []maps.Map {
+	if teaconst.IsPlus {
+		items = append(items, maps.Map{
+			"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingUAM),
+			"url":      "/clusters/cluster/settings/uam?clusterId=" + clusterIdString,
+			"isActive": selectedItem == "uam",
+			"isOn":     info != nil && info.UamIsOn,
+		})
+		items = append(items, maps.Map{
+			"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingCC),
+			"url":      "/clusters/cluster/settings/cc?clusterId=" + clusterIdString,
+			"isActive": selectedItem == "cc",
+			"isOn":     info != nil && info.HttpCCIsOn,
+		})
+		items = append(items, maps.Map{
+			"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingPages),
+			"url":      "/clusters/cluster/settings/pages?clusterId=" + clusterIdString,
+			"isActive": selectedItem == "pages",
+			"isOn":     info != nil && info.HasHTTPPagesPolicy,
+		})
+		items = append(items, maps.Map{
+			"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingHTTP3),
+			"url":      "/clusters/cluster/settings/http3?clusterId=" + clusterIdString,
+			"isActive": selectedItem == "http3",
+			"isOn":     info != nil && info.Http3IsOn,
+		})
+	}
+	return items
+}
+
+func (this *ClusterHelper) filterMenuItems2(items []maps.Map, info *pb.FindEnabledNodeClusterConfigInfoResponse, clusterIdString string, selectedItem string, actionPtr actions.ActionWrapper) []maps.Map {
+	if teaconst.IsPlus {
+		items = append(items, maps.Map{
+			"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingSecurityPolicy),
+			"url":      "/clusters/cluster/settings/network-security?clusterId=" + clusterIdString,
+			"isActive": selectedItem == "networkSecurity",
+			"isOn":     info != nil && info.HasNetworkSecurityPolicy, // TODO 将来 加入 info.HasDDoSProtection
+		})
+
+		items = append(items, maps.Map{
+			"name": "-",
+		})
+		if plusutils.CheckComponent(plusutils.ComponentCodeScheduling) {
+			items = append(items, maps.Map{
+				"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingSchedule),
+				"url":      "/clusters/cluster/settings/schedule?clusterId=" + clusterIdString,
+				"isActive": selectedItem == "schedule",
+				"isOn":     false, // TODO
+			})
+		}
+		items = append(items, maps.Map{
+			"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingThresholds),
+			"url":      "/clusters/cluster/settings/thresholds?clusterId=" + clusterIdString,
+			"isActive": selectedItem == "threshold",
+			"isOn":     info != nil && info.HasThresholds,
+		})
+
+		items = append(items, maps.Map{
+			"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingNotification),
+			"url":      "/clusters/cluster/settings/message?clusterId=" + clusterIdString,
+			"isActive": selectedItem == "message",
+			"isOn":     info != nil && info.HasMessageReceivers,
+		})
+
+		items = append(items, maps.Map{
+			"name":     "-",
+			"url":      "",
+			"isActive": false,
+		})
+
+		items = append(items, maps.Map{
+			"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingSystemService),
+			"url":      "/clusters/cluster/settings/services?clusterId=" + clusterIdString,
+			"isActive": selectedItem == "service",
+			"isOn":     info != nil && info.HasSystemServices,
+		})
+
+		{
+			items = append(items, maps.Map{
+				"name":     this.Lang(actionPtr, codes.NodeClusterMenu_SettingTOA),
+				"url":      "/clusters/cluster/settings/toa?clusterId=" + clusterIdString,
+				"isActive": selectedItem == "toa",
+				"isOn":     info != nil && info.IsTOAEnabled,
+			})
+		}
+	}
+	return items
+}

EdgeAdmin/internal/web/actions/default/clusters/clusterutils/clusters_helper.go

@@ -0,0 +1,22 @@
+package clusterutils
+
+import (
+	"github.com/iwind/TeaGo/actions"
+	"net/http"
+)
+
+type ClustersHelper struct {
+}
+
+func NewClustersHelper() *ClustersHelper {
+	return &ClustersHelper{}
+}
+
+func (this *ClustersHelper) BeforeAction(actionPtr actions.ActionWrapper) {
+	var action = actionPtr.Object()
+	if action.Request.Method != http.MethodGet {
+		return
+	}
+
+	action.Data["teaMenu"] = "clusters"
+}

EdgeAdmin/internal/web/actions/default/clusters/clusterutils/installer.go

@@ -0,0 +1,66 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package clusterutils
+
+import (
+	teaconst "github.com/TeaOSLab/EdgeAdmin/internal/const"
+	"github.com/iwind/TeaGo/Tea"
+	"path/filepath"
+	"regexp"
+	"sort"
+	"strings"
+)
+
+type installerFile struct {
+	Name    string `json:"name"`
+	OS      string `json:"os"`
+	Arch    string `json:"arch"`
+	Version string `json:"version"`
+}
+
+func ListInstallerFiles() []*installerFile {
+	var dir = Tea.Root + "/edge-api/deploy"
+	matches, err := filepath.Glob(dir + "/edge-node-*.zip")
+	if err != nil {
+		return nil
+	}
+
+	var result = []*installerFile{}
+	var reg = regexp.MustCompile(`^edge-node-(\w+)-(\w+)-v([\w.]+)\.zip$`)
+	for _, match := range matches {
+		var baseName = filepath.Base(match)
+		var subMatches = reg.FindStringSubmatch(baseName)
+		if len(subMatches) >= 4 {
+			var osName = subMatches[1]
+			if len(osName) > 0 {
+				osName = strings.ToUpper(osName[:1]) + osName[1:]
+			}
+
+			var arch = subMatches[2]
+			if arch == "amd64" {
+				arch = "x86_64"
+			}
+
+			var version = subMatches[3]
+			if version != teaconst.Version { // 只能下载当前版本
+				continue
+			}
+
+			result = append(result, &installerFile{
+				Name:    subMatches[0],
+				OS:      osName,
+				Arch:    arch,
+				Version: version,
+			})
+		}
+	}
+
+	// 排序，将x86_64排在最上面
+	if len(result) > 0 {
+		sort.Slice(result, func(i, j int) bool {
+			return result[i].Arch == "x86_64"
+		})
+	}
+
+	return result
+}