Initial Commit

1 files changed, 94 insertions, 0 deletions

diff --git a/backend/internal/server2/idb/user/v1/user.go b/backend/internal/server2/idb/user/v1/user.go
new file mode 100644
index 0000000..1866944
--- /dev/null
+++ b/backend/internal/server2/idb/user/v1/user.go
@@ -0,0 +1,94 @@
+package user
+
+import (
+	"context"
+	"errors"
+
+	pb "ibd-trader/api/gen/idb/user/v1"
+	"ibd-trader/internal/database"
+
+	"github.com/mennanov/fmutils"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+	"google.golang.org/protobuf/proto"
+)
+
+type Server struct {
+	pb.UnimplementedUserServiceServer
+
+	db database.UserStore
+}
+
+func New(db database.UserStore) *Server {
+	return &Server{db: db}
+}
+
+func (u *Server) CreateUser(ctx context.Context, request *pb.CreateUserRequest) (*pb.CreateUserResponse, error) {
+	err := u.db.AddUser(ctx, request.Subject)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "unable to create user: %v", err)
+	}
+
+	user, err := u.db.GetUser(ctx, request.Subject)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "unable to get user: %v", err)
+	}
+
+	return &pb.CreateUserResponse{
+		User: &pb.User{
+			Subject:     user.Subject,
+			IbdUsername: user.IBDUsername,
+			IbdPassword: nil,
+		},
+	}, nil
+}
+
+func (u *Server) GetUser(ctx context.Context, request *pb.GetUserRequest) (*pb.GetUserResponse, error) {
+	user, err := u.db.GetUser(ctx, request.Subject)
+	if errors.Is(err, database.ErrUserNotFound) {
+		return nil, status.New(codes.NotFound, "user not found").Err()
+	}
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "unable to get user: %v", err)
+	}
+
+	return &pb.GetUserResponse{
+		User: &pb.User{
+			Subject:     user.Subject,
+			IbdUsername: user.IBDUsername,
+			IbdPassword: nil,
+		},
+	}, nil
+}
+
+func (u *Server) UpdateUser(ctx context.Context, request *pb.UpdateUserRequest) (*pb.UpdateUserResponse, error) {
+	request.UpdateMask.Normalize()
+	if !request.UpdateMask.IsValid(request.User) {
+		return nil, status.Errorf(codes.InvalidArgument, "invalid update mask")
+	}
+
+	existingUserRes, err := u.GetUser(ctx, &pb.GetUserRequest{Subject: request.User.Subject})
+	if err != nil {
+		return nil, err
+	}
+	existingUser := existingUserRes.User
+
+	newUser := proto.Clone(existingUser).(*pb.User)
+	fmutils.Overwrite(request.User, newUser, request.UpdateMask.Paths)
+
+	// if IDB creds are both set and are different, update them
+	if (newUser.IbdPassword != nil && newUser.IbdUsername != nil) &&
+		(newUser.IbdPassword != existingUser.IbdPassword ||
+			newUser.IbdUsername != existingUser.IbdUsername) {
+		// Update IBD creds
+		err = u.db.AddIBDCreds(ctx, newUser.Subject, *newUser.IbdUsername, *newUser.IbdPassword)
+		if err != nil {
+			return nil, status.Errorf(codes.Internal, "unable to update user: %v", err)
+		}
+	}
+
+	newUser.IbdPassword = nil
+	return &pb.UpdateUserResponse{
+		User: newUser,
+	}, nil
+}