softplayer-backend/api/v1/accounts_no_auth.go

package v1

import (
	"context"

	"gitea.badhouseplants.net/softplayer/softplayer-backend/internal/controllers"
	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) {
	//uuid, err := a.ctrl.ValidateRefreshToken(ctx, , 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 nil, status.Error(codes.Unimplemented, "Endpoint is not Unimplemented yet")
}