From 020bba5c3861484a7e9108bec5ced992b6f6932b Mon Sep 17 00:00:00 2001
From: yanchao <ychao_1983@sina.com>
Date: Mon, 26 Dec 2022 13:55:23 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 routers/api/v1/api.go       |  2 ++
 routers/api/v1/user/repo.go | 14 ++++++++++++++
 2 files changed, 16 insertions(+)

diff --git a/routers/api/v1/api.go b/routers/api/v1/api.go
index b5ba895be..ba7346481 100755
--- a/routers/api/v1/api.go
+++ b/routers/api/v1/api.go
@@ -549,6 +549,8 @@ func RegisterRoutes(m *macaron.Macaron) {
 
 		}, reqToken())
 
+		m.Get("/compute-nodes", reqToken(), user.GetComputeNodes)
+
 		// Notifications
 		m.Group("/notifications", func() {
 			m.Combo("").
diff --git a/routers/api/v1/user/repo.go b/routers/api/v1/user/repo.go
index 4024bf96c..01f643992 100644
--- a/routers/api/v1/user/repo.go
+++ b/routers/api/v1/user/repo.go
@@ -5,6 +5,7 @@
 package user
 
 import (
+	"code.gitea.io/gitea/modules/modelarts"
 	"net/http"
 
 	"code.gitea.io/gitea/models"
@@ -146,3 +147,16 @@ func ListOrgRepos(ctx *context.APIContext) {
 
 	listUserRepos(ctx, ctx.Org.Organization, ctx.IsSigned)
 }
+
+func GetComputeNodes(ctx *context.APIContext) {
+	modelarts.InitMultiNode()
+	if modelarts.MultiNodeConfig != nil {
+		for _, info := range modelarts.MultiNodeConfig.Info {
+			if isInOrg, _ := models.IsOrganizationMemberByOrgName(info.Org, ctx.User.ID); isInOrg {
+				ctx.JSON(http.StatusOK, info.Node)
+				return
+			}
+		}
+	}
+	ctx.JSON(http.StatusOK, []int{1})
+}