From d381e6d72db8d72fcac3dfdfb193e06ecccdb0d5 Mon Sep 17 00:00:00 2001
From: GoEdgeLab <goedgecloud@gmail.com>
Date: Sat, 27 May 2023 19:55:06 +0800
Subject: [PATCH] =?UTF-8?q?=E9=9B=86=E7=BE=A4=E5=88=97=E8=A1=A8=E9=A1=B5?=
 =?UTF-8?q?=E4=B9=9F=E5=A2=9E=E5=8A=A0=E2=80=9C=E5=88=9B=E5=BB=BA=E8=8A=82?=
 =?UTF-8?q?=E7=82=B9=E2=80=9D=E9=93=BE=E6=8E=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../default/clusters/cluster/createNode.go    |  5 ++
 .../actions/default/clusters/createNode.go    | 52 +++++++++++++++++++
 internal/web/actions/default/clusters/init.go |  1 +
 web/views/@default/clusters/@menu.html        |  3 +-
 web/views/@default/clusters/createNode.html   | 14 +++++
 web/views/@default/clusters/createNode.js     | 13 +++++
 6 files changed, 87 insertions(+), 1 deletion(-)
 create mode 100644 internal/web/actions/default/clusters/createNode.go
 create mode 100644 web/views/@default/clusters/createNode.html
 create mode 100644 web/views/@default/clusters/createNode.js

diff --git a/internal/web/actions/default/clusters/cluster/createNode.go b/internal/web/actions/default/clusters/cluster/createNode.go
index f8b04a8e..84b26db4 100644
--- a/internal/web/actions/default/clusters/cluster/createNode.go
+++ b/internal/web/actions/default/clusters/cluster/createNode.go
@@ -30,6 +30,11 @@ func (this *CreateNodeAction) Init() {
 func (this *CreateNodeAction) RunGet(params struct {
 	ClusterId int64
 }) {
+	if params.ClusterId <= 0 {
+		this.RedirectURL("/clusters")
+		return
+	}
+
 	var leftMenuItems = []maps.Map{
 		{
 			"name":     "单个创建",
diff --git a/internal/web/actions/default/clusters/createNode.go b/internal/web/actions/default/clusters/createNode.go
new file mode 100644
index 00000000..a791803d
--- /dev/null
+++ b/internal/web/actions/default/clusters/createNode.go
@@ -0,0 +1,52 @@
+// Copyright 2023 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package clusters
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/types"
+)
+
+type CreateNodeAction struct {
+	actionutils.ParentAction
+}
+
+func (this *CreateNodeAction) Init() {
+	this.Nav("", "cluster", "createNode")
+}
+
+func (this *CreateNodeAction) RunGet(params struct{}) {
+	// 集群总数
+	totalClustersResp, err := this.RPC().NodeClusterRPC().CountAllEnabledNodeClusters(this.AdminContext(), &pb.CountAllEnabledNodeClustersRequest{})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	this.Data["totalNodeClusters"] = totalClustersResp.Count
+
+	// 节点总数
+	totalNodesResp, err := this.RPC().NodeRPC().CountAllEnabledNodes(this.AdminContext(), &pb.CountAllEnabledNodesRequest{})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	this.Data["totalNodes"] = totalNodesResp.Count
+
+	// 如果只有一个默认集群，那么直接跳转到集群
+	clustersResp, err := this.RPC().NodeClusterRPC().ListEnabledNodeClusters(this.AdminContext(), &pb.ListEnabledNodeClustersRequest{
+		Offset:  0,
+		Size:    2,
+		Keyword: "",
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	if len(clustersResp.NodeClusters) == 1 {
+		this.RedirectURL("/clusters/cluster/createNode?clusterId=" + types.String(clustersResp.NodeClusters[0].Id))
+		return
+	}
+
+	this.Show()
+}
diff --git a/internal/web/actions/default/clusters/init.go b/internal/web/actions/default/clusters/init.go
index 81b86011..078900ec 100644
--- a/internal/web/actions/default/clusters/init.go
+++ b/internal/web/actions/default/clusters/init.go
@@ -16,6 +16,7 @@ func init() {
 			Prefix("/clusters").
 			Get("", new(IndexAction)).
 			GetPost("/create", new(CreateAction)).
+			GetPost("/createNode", new(CreateNodeAction)).
 			Post("/pin", new(PinAction)).
 			Get("/nodes", new(NodesAction)).
 
diff --git a/web/views/@default/clusters/@menu.html b/web/views/@default/clusters/@menu.html
index d70f75e1..166665d0 100644
--- a/web/views/@default/clusters/@menu.html
+++ b/web/views/@default/clusters/@menu.html
@@ -2,5 +2,6 @@
     <menu-item href="/clusters" code="index">集群&nbsp;<span class="small">({{totalNodeClusters}})</span></menu-item>
     <menu-item href="/clusters/nodes" code="node">节点&nbsp;<span class="small">({{totalNodes}})</span></menu-item>
     <span class="disabled item">|</span>
-	<menu-item href="/clusters/create" code="create">[创建集群]</menu-item>
+    <menu-item href="/clusters/create" code="create">[创建集群]</menu-item>
+    <menu-item href="/clusters/createNode" code="createNode">[创建节点]</menu-item>
 </first-menu>
diff --git a/web/views/@default/clusters/createNode.html b/web/views/@default/clusters/createNode.html
new file mode 100644
index 00000000..48e08267
--- /dev/null
+++ b/web/views/@default/clusters/createNode.html
@@ -0,0 +1,14 @@
+{$layout}
+{$template "menu"}
+
+<div class="margin"></div>
+<form class="ui form">
+    <table class="ui table definition selectable">
+        <tr>
+            <td class="title">选择所属集群 *</td>
+            <td><node-cluster-combo-box @change="changeCluster"></node-cluster-combo-box></td>
+        </tr>
+    </table>
+
+    <button type="button" class="ui button primary" :class="{disabled: clusterId <= 0}" @click.prevent="goNext">下一步 <i class="icon right arrow"></i></button>
+</form>
\ No newline at end of file
diff --git a/web/views/@default/clusters/createNode.js b/web/views/@default/clusters/createNode.js
new file mode 100644
index 00000000..1b4e7d7f
--- /dev/null
+++ b/web/views/@default/clusters/createNode.js
@@ -0,0 +1,13 @@
+Tea.context(function () {
+	this.clusterId = 0
+
+	this.changeCluster = function(clusterId) {
+		this.clusterId = clusterId
+	}
+
+	this.goNext = function () {
+		if (this.clusterId > 0) {
+			window.location = "/clusters/cluster/createNode?clusterId=" + this.clusterId
+		}
+	}
+})
\ No newline at end of file