38 lines
876 B
Go
38 lines
876 B
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"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.ExpiredAt != nil {
|
|
u.ExpiredAt = user.ExpiredAt.Format("Y-m-d H:i:s")
|
|
}
|
|
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("暂未实现")
|
|
}
|