复制项目

internal/rpc/admin/applet.go

@@ -0,0 +1,159 @@
+// Copyright © 2023 OpenIM open source community. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package admin
+
+import (
+	"context"
+	"strings"
+	"time"
+
+	"github.com/openimsdk/tools/utils/datautil"
+
+	"github.com/google/uuid"
+	"github.com/openimsdk/tools/errs"
+
+	"git.imall.cloud/openim/chat/pkg/common/constant"
+	admindb "git.imall.cloud/openim/chat/pkg/common/db/table/admin"
+	"git.imall.cloud/openim/chat/pkg/common/mctx"
+	"git.imall.cloud/openim/chat/pkg/protocol/admin"
+	"git.imall.cloud/openim/chat/pkg/protocol/common"
+)
+
+func (o *adminServer) AddApplet(ctx context.Context, req *admin.AddAppletReq) (*admin.AddAppletResp, error) {
+	if _, err := mctx.CheckAdmin(ctx); err != nil {
+		return nil, err
+	}
+	if req.Name == "" {
+		return nil, errs.ErrArgs.WrapMsg("name empty")
+	}
+	if req.AppID == "" {
+		return nil, errs.ErrArgs.WrapMsg("appid empty")
+	}
+	if !(req.Status == constant.StatusOnShelf || req.Status == constant.StatusUnShelf) {
+		return nil, errs.ErrArgs.WrapMsg("invalid status")
+	}
+	m := admindb.Applet{
+		ID:         req.Id,
+		Name:       req.Name,
+		AppID:      req.AppID,
+		Icon:       req.Icon,
+		URL:        req.Url,
+		MD5:        req.Md5,
+		Size:       req.Size,
+		Version:    req.Version,
+		Priority:   req.Priority,
+		Status:     uint8(req.Status),
+		CreateTime: time.Now(),
+	}
+	if m.ID == "" {
+		m.ID = uuid.New().String()
+	}
+	if err := o.Database.CreateApplet(ctx, []*admindb.Applet{&m}); err != nil {
+		return nil, err
+	}
+	return &admin.AddAppletResp{}, nil
+}
+
+func (o *adminServer) DelApplet(ctx context.Context, req *admin.DelAppletReq) (*admin.DelAppletResp, error) {
+	if _, err := mctx.CheckAdmin(ctx); err != nil {
+		return nil, err
+	}
+	if len(req.AppletIds) == 0 {
+		return nil, errs.ErrArgs.WrapMsg("AppletIds empty")
+	}
+	applets, err := o.Database.FindApplet(ctx, req.AppletIds)
+	if err != nil {
+		return nil, err
+	}
+	if ids := datautil.Single(req.AppletIds, datautil.Slice(applets, func(e *admindb.Applet) string { return e.ID })); len(ids) > 0 {
+		return nil, errs.ErrArgs.WrapMsg("ids not found: " + strings.Join(ids, ", "))
+	}
+	if err := o.Database.DelApplet(ctx, req.AppletIds); err != nil {
+		return nil, err
+	}
+	return &admin.DelAppletResp{}, nil
+}
+
+func (o *adminServer) UpdateApplet(ctx context.Context, req *admin.UpdateAppletReq) (*admin.UpdateAppletResp, error) {
+	if _, err := mctx.CheckAdmin(ctx); err != nil {
+		return nil, err
+	}
+	_, err := o.Database.GetApplet(ctx, req.Id)
+	if err != nil {
+		return nil, err
+	}
+	update, err := ToDBAppletUpdate(req)
+	if err != nil {
+		return nil, err
+	}
+	if err := o.Database.UpdateApplet(ctx, req.Id, update); err != nil {
+		return nil, err
+	}
+	return &admin.UpdateAppletResp{}, nil
+}
+
+func (o *adminServer) FindApplet(ctx context.Context, req *admin.FindAppletReq) (*admin.FindAppletResp, error) {
+	if _, _, err := mctx.Check(ctx); err != nil {
+		return nil, err
+	}
+	applets, err := o.Database.FindOnShelf(ctx)
+	if err != nil {
+		return nil, err
+	}
+	resp := &admin.FindAppletResp{Applets: make([]*common.AppletInfo, 0, len(applets))}
+	for _, applet := range applets {
+		resp.Applets = append(resp.Applets, &common.AppletInfo{
+			Id:         applet.ID,
+			Name:       applet.Name,
+			AppID:      applet.AppID,
+			Icon:       applet.Icon,
+			Url:        applet.URL,
+			Md5:        applet.MD5,
+			Size:       applet.Size,
+			Version:    applet.Version,
+			Priority:   applet.Priority,
+			Status:     uint32(applet.Status),
+			CreateTime: applet.CreateTime.UnixMilli(),
+		})
+	}
+	return resp, nil
+}
+
+func (o *adminServer) SearchApplet(ctx context.Context, req *admin.SearchAppletReq) (*admin.SearchAppletResp, error) {
+	if _, err := mctx.CheckAdmin(ctx); err != nil {
+		return nil, err
+	}
+	total, applets, err := o.Database.SearchApplet(ctx, req.Keyword, req.Pagination)
+	if err != nil {
+		return nil, err
+	}
+	resp := &admin.SearchAppletResp{Total: uint32(total), Applets: make([]*common.AppletInfo, 0, len(applets))}
+	for _, applet := range applets {
+		resp.Applets = append(resp.Applets, &common.AppletInfo{
+			Id:         applet.ID,
+			Name:       applet.Name,
+			AppID:      applet.AppID,
+			Icon:       applet.Icon,
+			Url:        applet.URL,
+			Md5:        applet.MD5,
+			Size:       applet.Size,
+			Version:    applet.Version,
+			Priority:   applet.Priority,
+			Status:     uint32(applet.Status),
+			CreateTime: applet.CreateTime.UnixMilli(),
+		})
+	}
+	return resp, nil
+}