feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

backend/domain/openauth/openapiauth/api_auth_impl.go

@@ -0,0 +1,141 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * 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 openapiauth
+
+import (
+	"context"
+	"time"
+
+	"gorm.io/gorm"
+
+	"github.com/coze-dev/coze-studio/backend/domain/openauth/openapiauth/entity"
+	"github.com/coze-dev/coze-studio/backend/domain/openauth/openapiauth/internal/dal"
+	"github.com/coze-dev/coze-studio/backend/domain/openauth/openapiauth/internal/dal/model"
+	"github.com/coze-dev/coze-studio/backend/infra/contract/idgen"
+	"github.com/coze-dev/coze-studio/backend/pkg/lang/slices"
+	"github.com/coze-dev/coze-studio/backend/pkg/logs"
+)
+
+type apiAuthImpl struct {
+	IDGen idgen.IDGenerator
+	DB    *gorm.DB
+	dao   *dal.ApiKeyDAO
+}
+
+type Components struct {
+	IDGen idgen.IDGenerator
+	DB    *gorm.DB
+}
+
+func NewService(c *Components) APIAuth {
+	return &apiAuthImpl{
+		IDGen: c.IDGen,
+		DB:    c.DB,
+		dao:   dal.NewApiKeyDAO(c.IDGen, c.DB),
+	}
+}
+
+func (a *apiAuthImpl) Create(ctx context.Context, req *entity.CreateApiKey) (*entity.ApiKey, error) {
+	apiKeyData, err := a.dao.Create(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+	return apiKeyData, nil
+}
+
+func (a *apiAuthImpl) Delete(ctx context.Context, req *entity.DeleteApiKey) error {
+
+	return a.dao.Delete(ctx, req.ID, req.UserID)
+
+}
+func (a *apiAuthImpl) Get(ctx context.Context, req *entity.GetApiKey) (*entity.ApiKey, error) {
+
+	apiKey, err := a.dao.Get(ctx, req.ID)
+	logs.CtxInfof(ctx, "apiKey=%v, err:%v", apiKey, err)
+	if err != nil {
+		return nil, err
+	}
+	if apiKey == nil {
+		return nil, nil
+	}
+	return a.buildPoData2ApiKey([]*model.APIKey{apiKey})[0], nil
+}
+
+func (a *apiAuthImpl) buildPoData2ApiKey(apiKey []*model.APIKey) []*entity.ApiKey {
+
+	apiKeyData := slices.Transform(apiKey, func(a *model.APIKey) *entity.ApiKey {
+		return &entity.ApiKey{
+			ID:         a.ID,
+			Name:       a.Name,
+			ApiKey:     a.APIKey,
+			UserID:     a.UserID,
+			ExpiredAt:  a.ExpiredAt,
+			CreatedAt:  a.CreatedAt,
+			LastUsedAt: a.LastUsedAt,
+		}
+	})
+
+	return apiKeyData
+}
+
+func (a *apiAuthImpl) List(ctx context.Context, req *entity.ListApiKey) (*entity.ListApiKeyResp, error) {
+	resp := &entity.ListApiKeyResp{
+		ApiKeys: make([]*entity.ApiKey, 0),
+		HasMore: false,
+	}
+	apiKey, hasMore, err := a.dao.List(ctx, req.UserID, int(req.Limit), int(req.Page))
+	if err != nil {
+		return nil, err
+	}
+	resp.ApiKeys = a.buildPoData2ApiKey(apiKey)
+	resp.HasMore = hasMore
+
+	return resp, nil
+}
+func (a *apiAuthImpl) CheckPermission(ctx context.Context, req *entity.CheckPermission) (*entity.ApiKey, error) {
+
+	apiKey, err := a.dao.FindByKey(ctx, req.ApiKey)
+	if err != nil {
+		return nil, err
+	}
+	if apiKey.APIKey != req.ApiKey {
+		return nil, nil
+	}
+	apiKeyDo := &entity.ApiKey{
+		ID:        apiKey.ID,
+		Name:      apiKey.Name,
+		UserID:    apiKey.UserID,
+		ExpiredAt: apiKey.ExpiredAt,
+		CreatedAt: apiKey.CreatedAt,
+	}
+	return apiKeyDo, nil
+}
+
+func (a *apiAuthImpl) Save(ctx context.Context, sm *entity.SaveMeta) error {
+
+	updateColumn := make(map[string]any)
+	if sm.Name != nil {
+		updateColumn["name"] = sm.Name
+	}
+	if sm.LastUsedAt != nil {
+		updateColumn["last_used_at"] = sm.LastUsedAt
+	}
+	updateColumn["updated_at"] = time.Now().Unix()
+	err := a.dao.Update(ctx, sm.ID, sm.UserID, updateColumn)
+
+	return err
+}