package controller

import (
	"context"

	"slogan-agent/common"
	"slogan-agent/styleagent/model/dto"
	"slogan-agent/styleagent/model/entity"
	"slogan-agent/styleagent/service"

	"github.com/gogf/gf/v2/frame/g"
)

type body_measurement struct{}

var BodyMeasurement = new(body_measurement)

func (c *body_measurement) Save(ctx context.Context, req *dto.BodyMeasurementSaveReq) (res *struct{}, err error) {
	if err := service.BodyMeasurementService.Save(ctx, common.GetUserId(g.RequestFromCtx(ctx)), &entity.BodyMeasurement{
		Height:    req.Height,
		Weight:    req.Weight,
		SkinTone:  req.SkinTone,
		Bust:      req.Bust,
		Waist:     req.Waist,
		Hip:       req.Hip,
		Shoulder:  req.Shoulder,
		FitParams: req.FitParams,
	}); err != nil {
		return nil, err
	}
	return &struct{}{}, nil
}

func (c *body_measurement) Get(ctx context.Context, req *dto.BodyMeasurementGetReq) (res *dto.BodyMeasurementGetRes, err error) {
	b, err := service.BodyMeasurementService.Get(ctx, common.GetUserId(g.RequestFromCtx(ctx)))
	if err != nil || b == nil {
		return &dto.BodyMeasurementGetRes{}, nil
	}
	return &dto.BodyMeasurementGetRes{
		Height:    b.Height,
		Weight:    b.Weight,
		SkinTone:  b.SkinTone,
		Bust:      b.Bust,
		Waist:     b.Waist,
		Hip:       b.Hip,
		Shoulder:  b.Shoulder,
		FitParams: b.FitParams,
	}, nil
}