// Code scaffolded by goctl. Safe to edit.
// goctl 1.9.2

package ai

import (
	"context"
	"fmt"

	"github.com/youruser/base/internal/svc"
	"github.com/youruser/base/internal/types"
	"github.com/youruser/base/model"

	"github.com/zeromicro/go-zero/core/logx"
)

type AiProviderUpdateLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

// 更新AI平台
func NewAiProviderUpdateLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AiProviderUpdateLogic {
	return &AiProviderUpdateLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *AiProviderUpdateLogic) AiProviderUpdate(req *types.AIProviderUpdateRequest) (resp *types.AIProviderInfo, err error) {
	provider, err := model.AIProviderFindOne(l.ctx, l.svcCtx.DB, req.Id)
	if err != nil {
		return nil, fmt.Errorf("AI平台不存在: %v", err)
	}

	provider.DisplayName = req.DisplayName
	provider.BaseUrl = req.BaseUrl
	provider.SdkType = req.SdkType
	provider.Protocol = req.Protocol
	provider.IsActive = req.IsActive
	provider.SortOrder = req.SortOrder

	if err := model.AIProviderUpdate(l.ctx, l.svcCtx.DB, provider); err != nil {
		return nil, fmt.Errorf("更新AI平台失败: %v", err)
	}

	return &types.AIProviderInfo{
		Id:          provider.Id,
		Name:        provider.Name,
		DisplayName: provider.DisplayName,
		BaseUrl:     provider.BaseUrl,
		SdkType:     provider.SdkType,
		Protocol:    provider.Protocol,
		IsActive:    provider.IsActive,
		SortOrder:   provider.SortOrder,
		CreatedAt:   provider.CreatedAt.Format("2006-01-02 15:04:05"),
		UpdatedAt:   provider.UpdatedAt.Format("2006-01-02 15:04:05"),
	}, nil
}