add initial space logic/behaviour

internal/service/auth.go

@@ -36,6 +36,7 @@ type AuthService struct {
 	userRepository       repository.UserRepository
 	profileRepository    repository.ProfileRepository
 	tokenRepository      repository.TokenRepository
+	spaceService         *SpaceService
 	jwtSecret            string
 	jwtExpiry            time.Duration
 	tokenMagicLinkExpiry time.Duration
@@ -47,6 +48,7 @@ func NewAuthService(
 	userRepository repository.UserRepository,
 	profileRepository repository.ProfileRepository,
 	tokenRepository repository.TokenRepository,
+	spaceService *SpaceService,
 	jwtSecret string,
 	jwtExpiry time.Duration,
 	tokenMagicLinkExpiry time.Duration,
@@ -57,6 +59,7 @@ func NewAuthService(
 		userRepository:       userRepository,
 		profileRepository:    profileRepository,
 		tokenRepository:      tokenRepository,
+		spaceService:         spaceService,
 		jwtSecret:            jwtSecret,
 		jwtExpiry:            jwtExpiry,
 		tokenMagicLinkExpiry: tokenMagicLinkExpiry,
@@ -214,6 +217,12 @@ func (s *AuthService) SendMagicLink(email string) error {
 				return fmt.Errorf("failed to create profile: %w", err)
 			}
 
+			_, err = s.spaceService.EnsurePersonalSpace(user)
+			if err != nil {
+				// Log the error but don't fail the whole auth flow
+				slog.Error("failed to create personal space for new user", "error", err, "user_id", user.ID)
+			}
+
 			slog.Info("new passwordless user created", "email", email, "user_id", user.ID)
 		} else {
 			// user look up unexpected error

internal/service/space.go

@@ -0,0 +1,81 @@
+package service
+
+import (
+	"fmt"
+	"time"
+
+	"git.juancwu.dev/juancwu/budgit/internal/model"
+	"git.juancwu.dev/juancwu/budgit/internal/repository"
+	"github.com/google/uuid"
+)
+
+const PersonalSpaceName = "Personal Space"
+
+type SpaceService struct {
+	spaceRepo repository.SpaceRepository
+}
+
+func NewSpaceService(spaceRepo repository.SpaceRepository) *SpaceService {
+	return &SpaceService{
+		spaceRepo: spaceRepo,
+	}
+}
+
+// CreateSpace creates a new space and sets the owner.
+func (s *SpaceService) CreateSpace(name string, ownerID string) (*model.Space, error) {
+	if name == "" {
+		return nil, fmt.Errorf("space name cannot be empty")
+	}
+
+	space := &model.Space{
+		ID:        uuid.NewString(),
+		Name:      name,
+		OwnerID:   ownerID,
+		CreatedAt: time.Now(),
+		UpdatedAt: time.Now(),
+	}
+
+	err := s.spaceRepo.Create(space)
+	if err != nil {
+		return nil, fmt.Errorf("failed to create space: %w", err)
+	}
+
+	return space, nil
+}
+
+// EnsurePersonalSpace creates a "Personal Space" for a user if one doesn't exist.
+func (s *SpaceService) EnsurePersonalSpace(user *model.User) (*model.Space, error) {
+	spaces, err := s.spaceRepo.ByUserID(user.ID)
+	if err != nil {
+		return nil, fmt.Errorf("failed to get user spaces: %w", err)
+	}
+
+	// Check if a personal space already exists.
+	// We identify it by the user being the owner and the name being the default.
+	for _, space := range spaces {
+		if space.OwnerID == user.ID && space.Name == PersonalSpaceName {
+			return space, nil // Personal space already exists
+		}
+	}
+
+	// If no personal space, create one.
+	return s.CreateSpace(PersonalSpaceName, user.ID)
+}
+
+// GetSpacesForUser returns all spaces a user is a member of.
+func (s *SpaceService) GetSpacesForUser(userID string) ([]*model.Space, error) {
+	spaces, err := s.spaceRepo.ByUserID(userID)
+	if err != nil {
+		return nil, fmt.Errorf("failed to get spaces for user: %w", err)
+	}
+	return spaces, nil
+}
+
+// IsMember checks if a user is a member of a given space.
+func (s *SpaceService) IsMember(userID, spaceID string) (bool, error) {
+	isMember, err := s.spaceRepo.IsMember(spaceID, userID)
+	if err != nil {
+		return false, fmt.Errorf("failed to check membership: %w", err)
+	}
+	return isMember, nil
+}