Files
evening_detective_server/internal/app/server.go
T
2026-07-06 15:56:10 +07:00

112 lines
2.8 KiB
Go

package app
import (
"context"
"evening_detective_server/internal/modules/processor_jwt"
"evening_detective_server/internal/services/users_service"
proto "evening_detective_server/proto"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
"google.golang.org/protobuf/types/known/timestamppb"
)
type server struct {
proto.UnsafeEveningDetectiveServerServer
usersService *users_service.UsersService
}
func NewServer(
usersService *users_service.UsersService,
) proto.EveningDetectiveServerServer {
return &server{
usersService: usersService,
}
}
func (s *server) Ping(_ context.Context, _ *proto.PingReq) (*proto.PingRsp, error) {
return &proto.PingRsp{}, nil
}
func (s *server) Echo(_ context.Context, req *proto.EchoReq) (*proto.EchoRsp, error) {
return &proto.EchoRsp{Text: req.Text}, nil
}
func (s *server) Signup(ctx context.Context, req *proto.SignupReq) (*proto.SignupRsp, error) {
err := s.usersService.AddUser(ctx, req.Username, req.Email)
if err != nil {
return &proto.SignupRsp{
Error: err.Error(),
}, nil
}
return &proto.SignupRsp{}, nil
}
func (s *server) RefreshPassword(ctx context.Context, req *proto.RefreshPasswordReq) (*proto.RefreshPasswordRsp, error) {
err := s.usersService.RefreshPassword(ctx, req.Email)
if err != nil {
return &proto.RefreshPasswordRsp{
Error: err.Error(),
}, nil
}
return &proto.RefreshPasswordRsp{}, nil
}
func (s *server) Login(ctx context.Context, req *proto.LoginReq) (*proto.LoginRsp, error) {
accessToken, refreshToken, err := s.usersService.Login(ctx, req.Email, req.Password)
if err != nil {
return &proto.LoginRsp{
Error: err.Error(),
}, nil
}
return &proto.LoginRsp{
AccessToken: accessToken,
RefreshToken: refreshToken,
}, nil
}
func (s *server) Refresh(ctx context.Context, req *proto.RefreshReq) (*proto.RefreshRsp, error) {
accessToken, refreshToken, err := s.usersService.Refresh(ctx, req.RefreshToken)
if err != nil {
return &proto.RefreshRsp{
Error: err.Error(),
}, nil
}
return &proto.RefreshRsp{
AccessToken: accessToken,
RefreshToken: refreshToken,
}, nil
}
func (s *server) GetUsers(ctx context.Context, req *proto.GetUsersReq) (*proto.GetUsersRsp, error) {
claims := ctx.Value("claims").(*processor_jwt.JWTClaims)
if !claims.HasRole("admin") {
return nil, status.Errorf(codes.PermissionDenied, "permission denied")
}
users, err := s.usersService.GetUsers(ctx)
if err != nil {
return &proto.GetUsersRsp{
Error: err.Error(),
}, nil
}
resUsers := make([]*proto.User, 0, len(users))
for _, user := range users {
resUsers = append(
resUsers,
&proto.User{
Id: int32(user.ID),
Username: user.Username,
Email: user.Email,
Roles: user.Roles,
IsActive: user.IsActive,
CreatedAt: timestamppb.New(user.CreatedAt),
},
)
}
return &proto.GetUsersRsp{
Users: resUsers,
}, nil
}