Initial commit: Open sourcing all of the Maple Open Technologies code.

cloud/maplefile-backend/internal/repo/filemetadata/count.go

@@ -0,0 +1,138 @@
+// monorepo/cloud/maplefile-backend/internal/maplefile/repo/filemetadata/count.go
+package filemetadata
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/gocql/gocql"
+	"go.uber.org/zap"
+
+	dom_file "codeberg.org/mapleopentech/monorepo/cloud/maplefile-backend/internal/domain/file"
+)
+
+// CountFilesByUser counts all active files accessible to the user
+// accessibleCollectionIDs should include all collections the user owns or has access to
+func (impl *fileMetadataRepositoryImpl) CountFilesByUser(ctx context.Context, userID gocql.UUID, accessibleCollectionIDs []gocql.UUID) (int, error) {
+	if len(accessibleCollectionIDs) == 0 {
+		// No accessible collections, return 0
+		impl.Logger.Debug("no accessible collections provided for file count",
+			zap.String("user_id", userID.String()))
+		return 0, nil
+	}
+
+	// Create a map for efficient collection access checking
+	accessibleCollections := make(map[gocql.UUID]bool)
+	for _, cid := range accessibleCollectionIDs {
+		accessibleCollections[cid] = true
+	}
+
+	// Query files for the user using the user sync table
+	query := `SELECT id, collection_id, state FROM maplefile.files_by_user
+		WHERE user_id = ?`
+
+	iter := impl.Session.Query(query, userID).WithContext(ctx).Iter()
+
+	count := 0
+	var fileID, collectionID gocql.UUID
+	var state string
+
+	for iter.Scan(&fileID, &collectionID, &state) {
+		// Only count files from accessible collections
+		if !accessibleCollections[collectionID] {
+			continue
+		}
+
+		// Only count active files
+		if state != dom_file.FileStateActive {
+			continue
+		}
+
+		count++
+	}
+
+	if err := iter.Close(); err != nil {
+		impl.Logger.Error("failed to count files by user",
+			zap.String("user_id", userID.String()),
+			zap.Int("accessible_collections_count", len(accessibleCollectionIDs)),
+			zap.Error(err))
+		return 0, fmt.Errorf("failed to count files by user: %w", err)
+	}
+
+	impl.Logger.Debug("counted files by user successfully",
+		zap.String("user_id", userID.String()),
+		zap.Int("accessible_collections_count", len(accessibleCollectionIDs)),
+		zap.Int("file_count", count))
+
+	return count, nil
+}
+
+// CountFilesByOwner counts all active files owned by the user (alternative approach)
+func (impl *fileMetadataRepositoryImpl) CountFilesByOwner(ctx context.Context, ownerID gocql.UUID) (int, error) {
+	// Query files owned by the user using the owner table
+	query := `SELECT id, state FROM maplefile.files_by_owner
+		WHERE owner_id = ?`
+
+	iter := impl.Session.Query(query, ownerID).WithContext(ctx).Iter()
+
+	count := 0
+	var fileID gocql.UUID
+	var state string
+
+	for iter.Scan(&fileID, &state) {
+		// Only count active files
+		if state != dom_file.FileStateActive {
+			continue
+		}
+
+		count++
+	}
+
+	if err := iter.Close(); err != nil {
+		impl.Logger.Error("failed to count files by owner",
+			zap.String("owner_id", ownerID.String()),
+			zap.Error(err))
+		return 0, fmt.Errorf("failed to count files by owner: %w", err)
+	}
+
+	impl.Logger.Debug("counted files by owner successfully",
+		zap.String("owner_id", ownerID.String()),
+		zap.Int("file_count", count))
+
+	return count, nil
+}
+
+// CountFilesByCollection counts active files in a specific collection
+func (impl *fileMetadataRepositoryImpl) CountFilesByCollection(ctx context.Context, collectionID gocql.UUID) (int, error) {
+	// Query files in the collection using the collection table
+	query := `SELECT id, state FROM maplefile.files_by_collection
+		WHERE collection_id = ?`
+
+	iter := impl.Session.Query(query, collectionID).WithContext(ctx).Iter()
+
+	count := 0
+	var fileID gocql.UUID
+	var state string
+
+	for iter.Scan(&fileID, &state) {
+		// Only count active files
+		if state != dom_file.FileStateActive {
+			continue
+		}
+
+		count++
+	}
+
+	if err := iter.Close(); err != nil {
+		impl.Logger.Error("failed to count files by collection",
+			zap.String("collection_id", collectionID.String()),
+			zap.Error(err))
+		return 0, fmt.Errorf("failed to count files by collection: %w", err)
+	}
+
+	impl.Logger.Debug("counted files by collection successfully",
+		zap.String("collection_id", collectionID.String()),
+		zap.Int("file_count", count))
+
+	return count, nil
+}