Initial commit (code only without large binaries)

EdgeAdmin/internal/web/actions/default/messages/badge.go

@@ -0,0 +1,22 @@
+package messages
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+type BadgeAction struct {
+	actionutils.ParentAction
+}
+
+func (this *BadgeAction) RunPost(params struct{}) {
+	countResp, err := this.RPC().MessageRPC().CountUnreadMessages(this.AdminContext(), &pb.CountUnreadMessagesRequest{})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Data["count"] = countResp.Count
+
+	this.Success()
+}

EdgeAdmin/internal/web/actions/default/messages/helper.go

@@ -0,0 +1,11 @@
+package messages
+
+import "github.com/iwind/TeaGo/actions"
+
+type Helper struct {
+}
+
+func (this *Helper) BeforeAction(actionPtr actions.ActionWrapper) {
+	action := actionPtr.Object()
+	action.Data["teaMenu"] = "message"
+}

EdgeAdmin/internal/web/actions/default/messages/index.go

@@ -0,0 +1,72 @@
+package messages
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/maps"
+	timeutil "github.com/iwind/TeaGo/utils/time"
+)
+
+type IndexAction struct {
+	actionutils.ParentAction
+}
+
+func (this *IndexAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *IndexAction) RunGet(params struct{}) {
+	countResp, err := this.RPC().MessageRPC().CountUnreadMessages(this.AdminContext(), &pb.CountUnreadMessagesRequest{})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	count := countResp.Count
+
+	page := this.NewPage(count)
+	this.Data["page"] = page.AsHTML()
+
+	listResp, err := this.RPC().MessageRPC().ListUnreadMessages(this.AdminContext(), &pb.ListUnreadMessagesRequest{
+		Offset: page.Offset,
+		Size:   page.Size,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	messages := []maps.Map{}
+	for _, message := range listResp.Messages {
+		clusterMap := maps.Map{}
+		if message.NodeCluster != nil {
+			clusterMap = maps.Map{
+				"id":   message.NodeCluster.Id,
+				"name": message.NodeCluster.Name,
+			}
+		}
+
+		nodeMap := maps.Map{}
+		if message.Node != nil {
+			nodeMap = maps.Map{
+				"id":   message.Node.Id,
+				"name": message.Node.Name,
+			}
+		}
+
+		messages = append(messages, maps.Map{
+			"id":       message.Id,
+			"role":     message.Role,
+			"isRead":   message.IsRead,
+			"body":     message.Body,
+			"level":    message.Level,
+			"datetime": timeutil.FormatTime("Y-m-d H:i:s", message.CreatedAt),
+			"params":   string(message.ParamsJSON),
+			"type":     message.Type,
+			"cluster":  clusterMap,
+			"node":     nodeMap,
+		})
+	}
+	this.Data["messages"] = messages
+
+	this.Show()
+}

EdgeAdmin/internal/web/actions/default/messages/init.go

@@ -0,0 +1,21 @@
+package messages
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/configloaders"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/helpers"
+	"github.com/iwind/TeaGo"
+)
+
+func init() {
+	TeaGo.BeforeStart(func(server *TeaGo.Server) {
+		server.
+			Helper(helpers.NewUserMustAuth(configloaders.AdminModuleCodeCommon)).
+			Helper(new(Helper)).
+			Prefix("/messages").
+			GetPost("", new(IndexAction)).
+			Post("/badge", new(BadgeAction)).
+			Post("/readAll", new(ReadAllAction)).
+			Post("/readPage", new(ReadPageAction)).
+			EndAll()
+	})
+}

EdgeAdmin/internal/web/actions/default/messages/readAll.go

@@ -0,0 +1,24 @@
+package messages
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+type ReadAllAction struct {
+	actionutils.ParentAction
+}
+
+func (this *ReadAllAction) RunPost(params struct{}) {
+	// 创建日志
+	defer this.CreateLogInfo(codes.Message_LogReadAll)
+
+	_, err := this.RPC().MessageRPC().UpdateAllMessagesRead(this.AdminContext(), &pb.UpdateAllMessagesReadRequest{})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}

EdgeAdmin/internal/web/actions/default/messages/readPage.go

@@ -0,0 +1,29 @@
+package messages
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+type ReadPageAction struct {
+	actionutils.ParentAction
+}
+
+func (this *ReadPageAction) RunPost(params struct {
+	MessageIds []int64
+}) {
+	// 创建日志
+	defer this.CreateLogInfo(codes.Message_LogReadMessages)
+
+	_, err := this.RPC().MessageRPC().UpdateMessagesRead(this.AdminContext(), &pb.UpdateMessagesReadRequest{
+		MessageIds: params.MessageIds,
+		IsRead:     true,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}