package controller

import (
	"context"

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

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

type user struct{}

var User = new(user)

func (c *user) Register(ctx context.Context, req *dto.RegisterReq) (res *struct{}, err error) {
	_, err = service.UserService.Register(ctx, req.Account, req.Password, req.Name)
	if err != nil {
		return nil, err
	}
	return &struct{}{}, nil
}

func (c *user) Login(ctx context.Context, req *dto.LoginReq) (res *dto.LoginRes, err error) {
	user, token, err := service.UserService.Login(ctx, req.Account, req.Password)
	if err != nil {
		return nil, err
	}
	return &dto.LoginRes{
		Token: token,
		User:  &dto.LoginUser{Id: user.Id, Role: user.Role, Name: user.Name},
	}, nil
}

func (c *user) ChangePassword(ctx context.Context, req *dto.ChangePasswordReq) (res *struct{}, err error) {
	return nil, service.UserService.ChangePassword(ctx, common.GetUserId(g.RequestFromCtx(ctx)), req.OldPassword, req.NewPassword)
}

func (c *user) Profile(ctx context.Context, req *dto.ProfileReq) (res *dto.ProfileRes, err error) {
	user, err := dao.User.GetOne(ctx, common.GetUserId(g.RequestFromCtx(ctx)))
	if err != nil || user == nil {
		return nil, err
	}
	return &dto.ProfileRes{Id: user.Id, Role: user.Role, Name: user.Name, Username: user.Username, Phone: user.Phone}, nil
}