增加企业版认证相关API

internal/rpc/services/service_authority_key.go

@@ -0,0 +1,76 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+
+package services
+
+import (
+	"context"
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+// AuthorityKeyService 版本认证
+type AuthorityKeyService struct {
+	BaseService
+}
+
+// UpdateAuthorityKey 设置Key
+func (this *AuthorityKeyService) UpdateAuthorityKey(ctx context.Context, req *pb.UpdateAuthorityKeyRequest) (*pb.RPCSuccess, error) {
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAuthority)
+	if err != nil {
+		return nil, err
+	}
+	var tx = this.NullTx()
+	err = models.SharedAuthorityKeyDAO.UpdateKey(tx, req.Value, req.DayFrom, req.DayTo, req.Hostname, req.MacAddresses)
+	if err != nil {
+		return nil, err
+	}
+	return this.Success()
+}
+
+// ReadAuthorityKey 读取Key
+func (this *AuthorityKeyService) ReadAuthorityKey(ctx context.Context, req *pb.ReadAuthorityKeyRequest) (*pb.ReadAuthorityKeyResponse, error) {
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin, rpcutils.UserTypeMonitor, rpcutils.UserTypeProvider, rpcutils.UserTypeDNS)
+	if err != nil {
+		return nil, err
+	}
+	var tx = this.NullTx()
+	key, err := models.SharedAuthorityKeyDAO.ReadKey(tx)
+	if err != nil {
+		return nil, err
+	}
+	if key == nil {
+		return &pb.ReadAuthorityKeyResponse{AuthorityKey: nil}, nil
+	}
+
+	macAddresses := []string{}
+	if len(key.MacAddresses) > 0 {
+		err = json.Unmarshal([]byte(key.MacAddresses), &macAddresses)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	return &pb.ReadAuthorityKeyResponse{AuthorityKey: &pb.AuthorityKey{
+		Value:        key.Value,
+		DayFrom:      key.DayFrom,
+		DayTo:        key.DayTo,
+		Hostname:     key.Hostname,
+		MacAddresses: macAddresses,
+		UpdatedAt:    int64(key.UpdatedAt),
+	}}, nil
+}
+
+// ResetAuthorityKey 重置Key
+func (this *AuthorityKeyService) ResetAuthorityKey(ctx context.Context, req *pb.ResetAuthorityKeyRequest) (*pb.RPCSuccess, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+	err = models.SharedAuthorityKeyDAO.ResetKey(nil)
+	if err != nil {
+		return nil, err
+	}
+	return this.Success()
+}