package controller

import (
	"context"
	"errors"

	"video-factory/shortdrama/dao"
	"video-factory/shortdrama/model/dto"
	"video-factory/shortdrama/service"
)

type auth struct{}

var Auth = new(auth)

func (c *auth) Login(ctx context.Context, req *dto.LoginReq) (res *dto.LoginRes, err error) {
	user, token, err := service.AuthService.Login(ctx, req.Account, req.Password)
	if err != nil {
		return nil, err
	}
	u := &dto.LoginUser{
		Id:   user.Id,
		Role: user.Role,
		Name: user.Name,
	}
	if user.Role == "customer" {
		u.Balance = service.AuthService.GetBalance(ctx, user.Id)
	}
	if user.Role == "agent" {
		if ap, _ := dao.AgentProfile.Get(ctx, user.Id); ap != nil {
			if ap.ExpiredAt != nil {
				u.ExpiredAt = ap.ExpiredAt.Format("Y-m-d H:i:s")
			}
			u.TierId = ap.TierId
			if tier, _ := dao.AgentTier.GetOne(ctx, ap.TierId); tier != nil {
				u.TierName = tier.Name
			}
		}
	}
	return &dto.LoginRes{Token: token, User: u}, nil
}

func (c *auth) ChangePassword(ctx context.Context, req *dto.ChangePasswordReq) (res *struct{}, err error) {
	// TODO: implement password change
	return nil, errors.New("暂未实现")
}