moteva/server/internal/logic/generate_project_logic.go

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

package logic

import (
	"context"

	"img_infinite_canvas/internal/svc"
	"img_infinite_canvas/internal/types"

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

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

func NewGenerateProjectLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GenerateProjectLogic {
	return &GenerateProjectLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *GenerateProjectLogic) GenerateProject(req *types.GenerateRequest) (resp *types.GenerateResponse, err error) {
	if err := validateRequiredPrompt(req.Prompt, false); err != nil {
		return nil, err
	}

	result, err := l.svcCtx.DesignService.Generate(l.ctx, req.Id, req.Prompt, req.Mode, req.ImageModel, req.ImageSize, req.ImageQuality, int(req.ImageCount))
	if err != nil {
		return nil, err
	}

	return &types.GenerateResponse{
		Project:        toAPIProject(result.Project),
		Message:        toAPIMessage(result.Message),
		GeneratedNodes: toAPINodes(result.GeneratedNodes),
	}, nil
}