From d40b74875d02f11c1d1b7358021196f8a3b0bc9e Mon Sep 17 00:00:00 2001
From: zhanghan2021 <zhanghan@kylinos.cn>
Date: Tue, 25 Feb 2025 14:47:20 +0800
Subject: [PATCH] add permissions for prometheus plugin

---
 server/main.go                |  5 +++--
 server/service/permissions.go | 28 ++++++++++++++++++++++++++++
 2 files changed, 31 insertions(+), 2 deletions(-)
 create mode 100644 server/service/permissions.go

diff --git a/server/main.go b/server/main.go
index 46a7255..59a70ee 100644
--- a/server/main.go
+++ b/server/main.go
@@ -51,8 +51,9 @@ func main() {
 	plugin.Client = client.DefaultClient(plugin.Init(config.Config().PluginPrometheus, config.Config().PrometheusServer))
 	router.RegisterAPIs(server)
 	router.StaticRouter(server)
-	service.GetTags()       // pilotgo机器列表tag标签
-	service.AddExtentions() // 添加扩展点
+	service.GetTags()        // pilotgo机器列表tag标签
+	service.AddExtentions()  // 添加扩展点
+	service.AddPermissions() // 添加权限
 
 	if err := server.Run(config.Config().HttpServer.Addr); err != nil {
 		logger.Fatal("failed to run server: %v", err)
diff --git a/server/service/permissions.go b/server/service/permissions.go
new file mode 100644
index 0000000..bc39711
--- /dev/null
+++ b/server/service/permissions.go
@@ -0,0 +1,28 @@
+/*
+ * Copyright (c) KylinSoft  Co., Ltd. 2024.All rights reserved.
+ * PilotGo-plugin-prometheus licensed under the Mulan Permissive Software License, Version 2.
+ * See LICENSE file for more details.
+ * Author: zhanghan2021 <zhanghan@kylinos.cn>
+ * Date: Tue Feb 25 16:23:18 2025 +0800
+ */
+package service
+
+import (
+	"gitee.com/openeuler/PilotGo/sdk/common"
+	"openeuler.org/PilotGo/prometheus-plugin/server/plugin"
+)
+
+func AddPermissions() {
+	var pe []common.Permission
+	p1 := common.Permission{
+		Resource: "monitor_operate",
+		Operate:  "button",
+	}
+	p2 := common.Permission{
+		Resource: "monitor",
+		Operate:  "menu",
+	}
+
+	p := append(pe, p1, p2)
+	plugin.Client.RegisterPermission(p)
+}
-- 
Gitee