45 lines
1.1 KiB
Go
45 lines
1.1 KiB
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)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return &types.GenerateResponse{
|
||
|
|
Project: toAPIProject(result.Project),
|
||
|
|
Message: toAPIMessage(result.Message),
|
||
|
|
GeneratedNodes: toAPINodes(result.GeneratedNodes),
|
||
|
|
}, nil
|
||
|
|
}
|