114 lines
3.5 KiB
Go
114 lines
3.5 KiB
Go
package v1
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gitea.badhouseplants.net/softplayer/softplayer-backend/internal/controllers"
|
|
"gitea.badhouseplants.net/softplayer/softplayer-backend/internal/tools/logger"
|
|
accounts "gitea.badhouseplants.net/softplayer/softplayer-go-proto/pkg/accounts/v1"
|
|
"github.com/golang/protobuf/ptypes/empty"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/metadata"
|
|
"google.golang.org/grpc/status"
|
|
"google.golang.org/protobuf/types/known/emptypb"
|
|
)
|
|
|
|
func NewAccountNoAuthRPCImpl(ctrl *controllers.AccountController) *AccountsNoAuthServer {
|
|
return &AccountsNoAuthServer{
|
|
ctrl: ctrl,
|
|
}
|
|
}
|
|
|
|
type AccountsNoAuthServer struct {
|
|
accounts.UnimplementedAccountsNoAuthServiceServer
|
|
ctrl *controllers.AccountController
|
|
}
|
|
|
|
func (a *AccountsNoAuthServer) SignIn(ctx context.Context, in *accounts.SignInRequest) (*empty.Empty, error) {
|
|
id, err := a.ctrl.Login(ctx, in.GetEmail(), in.GetPassword())
|
|
if err != nil {
|
|
return nil, status.Error(codes.Aborted, "Couldn't create a user")
|
|
}
|
|
accessToken, err := a.ctrl.GenerateAccessToken(id)
|
|
if err != nil {
|
|
return nil, status.Error(codes.Aborted, "Couldn't generate an access token")
|
|
}
|
|
|
|
refreshToken, err := a.ctrl.GenerateRefreshToken(ctx, id)
|
|
if err != nil {
|
|
return nil, status.Error(codes.Aborted, "Couldn't generate an access token")
|
|
}
|
|
|
|
header := metadata.New(map[string]string{
|
|
"X-Access-Token": accessToken,
|
|
"X-Refresh-Token": refreshToken,
|
|
})
|
|
if err := grpc.SetHeader(ctx, header); err != nil {
|
|
return nil, status.Error(codes.Aborted, "Couldn't set metadata")
|
|
}
|
|
return &emptypb.Empty{}, nil
|
|
}
|
|
|
|
// Create a new account in Softplayer
|
|
func (a *AccountsNoAuthServer) SignUp(ctx context.Context, in *accounts.SignUpRequest) (*empty.Empty, error) {
|
|
data := &controllers.AccountData{
|
|
Password: in.GetPassword(),
|
|
Email: in.GetEmail(),
|
|
}
|
|
id, err := a.ctrl.Create(ctx, data)
|
|
if err != nil {
|
|
return nil, status.Error(codes.Aborted, "Couldn't create a user")
|
|
}
|
|
|
|
accessToken, err := a.ctrl.GenerateAccessToken(id)
|
|
if err != nil {
|
|
return nil, status.Error(codes.Aborted, "Couldn't generate an access token")
|
|
}
|
|
|
|
refreshToken, err := a.ctrl.GenerateRefreshToken(ctx, id)
|
|
if err != nil {
|
|
return nil, status.Error(codes.Aborted, "Couldn't generate an access token")
|
|
}
|
|
|
|
header := metadata.New(map[string]string{
|
|
"X-Access-Token": accessToken,
|
|
"X-Refresh-Token": refreshToken,
|
|
})
|
|
if err := grpc.SetHeader(ctx, header); err != nil {
|
|
return nil, status.Error(codes.Aborted, "Couldn't set metadata")
|
|
}
|
|
return &emptypb.Empty{}, nil
|
|
}
|
|
|
|
func (a *AccountsAuthServer) RefreshToken(ctx context.Context, in *empty.Empty) (*empty.Empty, error) {
|
|
log := logger.FromContext(ctx)
|
|
uuid, err := a.ctrl.ValidateRefreshToken(ctx, tokenID, userID)
|
|
if err != nil {
|
|
return nil, status.Error(codes.Unauthenticated, "refresh token is invalid")
|
|
}
|
|
accessToken, err := a.ctrl.GenerateAccessToken(uuid)
|
|
if err != nil {
|
|
log.Error(err, "Couldn't generate an access token")
|
|
return nil, status.Error(codes.Aborted, "Couldn't generate Access Token")
|
|
}
|
|
|
|
refreshToken, err := a.ctrl.GenerateRefreshToken(ctx, uuid)
|
|
if err != nil {
|
|
log.Error(err, "Couldn't generate a refresh token")
|
|
return nil, status.Error(codes.Aborted, "Couldn't generate Access Token")
|
|
}
|
|
|
|
header := metadata.Pairs(
|
|
"access-token", accessToken,
|
|
"refreshToken", refreshToken,
|
|
)
|
|
|
|
if err := grpc.SetHeader(ctx, header); err != nil {
|
|
log.Error(err, "Couldn't set headers")
|
|
return nil, status.Error(codes.Unknown, "Couldn't set headers")
|
|
}
|
|
|
|
return &emptypb.Empty{}, nil
|
|
}
|