1

2026-08-03 10:29:31 +08:00
parent 878468cd6d
commit a338865025
20 changed files with 613 additions and 398 deletions
@@ -0,0 +1,39 @@
+package controller
+
+import (
+	"context"
+
+	"slogan-agent/common"
+	"slogan-agent/styleagent/model/dto"
+	"slogan-agent/styleagent/service"
+
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+type avatar struct{}
+
+var Avatar = new(avatar)
+
+func (c *avatar) Build(ctx context.Context, req *dto.AvatarBuildReq) (res *dto.AvatarBuildRes, err error) {
+	a, err := service.AvatarService.Build(ctx, common.GetUserId(g.RequestFromCtx(ctx)))
+	if err != nil {
+		return nil, err
+	}
+	return &dto.AvatarBuildRes{AvatarId: a.Id, Status: a.BuildStatus}, nil
+}
+
+func (c *avatar) Get(ctx context.Context, req *dto.AvatarGetReq) (res *dto.AvatarGetRes, err error) {
+	a, err := service.AvatarService.Get(ctx, common.GetUserId(g.RequestFromCtx(ctx)))
+	if err != nil || a == nil {
+		return &dto.AvatarGetRes{}, nil
+	}
+	return &dto.AvatarGetRes{
+		FaceTemplateId: a.FaceTemplateId,
+		BodyTemplateId: a.BodyTemplateId,
+		SkinToneIndex:  a.SkinToneIndex,
+		GlbUrl:         a.GlbUrl,
+		FramesUrl:      a.FramesUrl,
+		BuildStatus:    a.BuildStatus,
+		Error:          a.Error,
+	}, nil
+}