From f5fea6c34d8f89c5736f944e8fa8c45ac75fd067 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E7=A5=A5=E8=B6=85?= <iwind.liu@gmail.com>
Date: Sat, 27 May 2023 15:33:08 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=9B=E5=BB=BA=E8=8A=82=E7=82=B9=E8=A1=A8?=
 =?UTF-8?q?=E5=8D=95=E6=98=BE=E7=A4=BA=E5=89=A9=E4=BD=99=E9=85=8D=E9=A2=9D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../actions/default/clusters/cluster/createBatch.go   | 11 +++++++++++
 .../default/clusters/cluster/createBatch_ext.go       |  8 ++++++++
 .../actions/default/clusters/cluster/createNode.go    | 11 +++++++++++
 .../default/clusters/cluster/createNode_ext.go        |  8 ++++++++
 web/views/@default/clusters/cluster/createBatch.html  |  5 ++++-
 web/views/@default/clusters/cluster/createNode.html   |  3 +++
 6 files changed, 45 insertions(+), 1 deletion(-)
 create mode 100644 internal/web/actions/default/clusters/cluster/createBatch_ext.go
 create mode 100644 internal/web/actions/default/clusters/cluster/createNode_ext.go

diff --git a/internal/web/actions/default/clusters/cluster/createBatch.go b/internal/web/actions/default/clusters/cluster/createBatch.go
index c982fd9e..40fbfc81 100644
--- a/internal/web/actions/default/clusters/cluster/createBatch.go
+++ b/internal/web/actions/default/clusters/cluster/createBatch.go
@@ -39,6 +39,17 @@ func (this *CreateBatchAction) RunGet(params struct {
 	}
 	this.Data["leftMenuItems"] = leftMenuItems
 
+	// 限额
+	maxNodes, leftNodes, err := this.findNodesQuota()
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	this.Data["quota"] = maps.Map{
+		"maxNodes":  maxNodes,
+		"leftNodes": leftNodes,
+	}
+
 	this.Show()
 }
 
diff --git a/internal/web/actions/default/clusters/cluster/createBatch_ext.go b/internal/web/actions/default/clusters/cluster/createBatch_ext.go
new file mode 100644
index 00000000..59a4bd79
--- /dev/null
+++ b/internal/web/actions/default/clusters/cluster/createBatch_ext.go
@@ -0,0 +1,8 @@
+// Copyright 2023 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build !plus
+
+package cluster
+
+func (this *CreateBatchAction) findNodesQuota() (maxNodes int32, leftNodes int32, err error) {
+	return
+}
diff --git a/internal/web/actions/default/clusters/cluster/createNode.go b/internal/web/actions/default/clusters/cluster/createNode.go
index dd300878..f8b04a8e 100644
--- a/internal/web/actions/default/clusters/cluster/createNode.go
+++ b/internal/web/actions/default/clusters/cluster/createNode.go
@@ -92,6 +92,17 @@ func (this *CreateNodeAction) RunGet(params struct {
 	// 安装文件下载
 	this.Data["installerFiles"] = clusterutils.ListInstallerFiles()
 
+	// 限额
+	maxNodes, leftNodes, err := this.findNodesQuota()
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	this.Data["quota"] = maps.Map{
+		"maxNodes":  maxNodes,
+		"leftNodes": leftNodes,
+	}
+
 	this.Show()
 }
 
diff --git a/internal/web/actions/default/clusters/cluster/createNode_ext.go b/internal/web/actions/default/clusters/cluster/createNode_ext.go
new file mode 100644
index 00000000..d0e08006
--- /dev/null
+++ b/internal/web/actions/default/clusters/cluster/createNode_ext.go
@@ -0,0 +1,8 @@
+// Copyright 2023 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build !plus
+
+package cluster
+
+func (this *CreateNodeAction) findNodesQuota() (maxNodes int32, leftNodes int32, err error) {
+	return
+}
diff --git a/web/views/@default/clusters/cluster/createBatch.html b/web/views/@default/clusters/cluster/createBatch.html
index ff5811f6..1eda3eca 100644
--- a/web/views/@default/clusters/cluster/createBatch.html
+++ b/web/views/@default/clusters/cluster/createBatch.html
@@ -6,7 +6,10 @@
 	<form method="post" class="ui form" data-tea-action="$" data-tea-success="success">
 		<csrf-token></csrf-token>
 		<input type="hidden" name="clusterId" :value="clusterId"/>
-		<table class="ui table definition selectable">
+
+        <div v-if="quota.maxNodes > 0 && quota.leftNodes >= 0"><span style="color: #959da6">当前授权最多支持节点数：{{quota.maxNodes}}个，剩余节点数：{{quota.leftNodes}}个。</span></div>
+
+        <table class="ui table definition selectable">
 			<tr>
 				<td class="title">节点IP列表</td>
 				<td>
diff --git a/web/views/@default/clusters/cluster/createNode.html b/web/views/@default/clusters/cluster/createNode.html
index d5d40ac6..c0ee6d28 100644
--- a/web/views/@default/clusters/cluster/createNode.html
+++ b/web/views/@default/clusters/cluster/createNode.html
@@ -19,6 +19,9 @@
     <!-- 填写信息 -->
 	<form method="post" class="ui form" data-tea-action="$" data-tea-success="success" v-show="step == 'info'">
 		<input type="hidden" name="clusterId" :value="clusterId"/>
+
+        <div v-if="quota.maxNodes > 0 && quota.leftNodes >= 0"><span style="color: #959da6">当前授权最多节点数：{{quota.maxNodes}}个，剩余节点数：{{quota.leftNodes}}个。</span></div>
+
 		<table class="ui table definition selectable">
 			<tr>
 				<td class="title">节点名称 *</td>