geek-cafe-saas-sdk 0.6.0__py3-none-any.whl → 0.7.1__py3-none-any.whl

geek_cafe_saas_sdk/domains/files/handlers/files/get/app.py

@@ -0,0 +1,62 @@
+"""
+Lambda handler for getting file metadata.
+
+Requires authentication (secure mode).
+"""
+
+from typing import Dict, Any
+from geek_cafe_saas_sdk.lambda_handlers import create_handler
+from geek_cafe_saas_sdk.domains.files.services import FileSystemService
+
+
+# Factory creates handler (defaults to secure auth)
+handler_wrapper = create_handler(
+    service_class=FileSystemService,
+    require_body=False,
+    convert_case=True
+)
+
+
+def handler(event: Dict[str, Any], context: Any, injected_service=None) -> Dict[str, Any]:
+    """
+    Get file metadata by ID.
+    
+    Args:
+        event: Lambda event from API Gateway
+        context: Lambda context
+        injected_service: Optional FileSystemService for testing
+    
+    Path parameters:
+        fileId: File ID
+    
+    Returns 200 with file metadata
+    """
+    return handler_wrapper.execute(event, context, get_file, injected_service)
+
+
+def get_file(
+    event: Dict[str, Any],
+    service: FileSystemService,
+    user_context: Dict[str, str]
+) -> Any:
+    """
+    Business logic for getting file metadata.
+    """
+    tenant_id = user_context.get("tenant_id")
+    user_id = user_context.get("user_id")
+    
+    # Get file ID from path parameters
+    path_params = event.get("pathParameters", {})
+    file_id = path_params.get("fileId") or path_params.get("id")
+    
+    if not file_id:
+        raise ValueError("fileId path parameter is required")
+    
+    # Get file metadata
+    result = service.get_by_id(
+        resource_id=file_id,
+        tenant_id=tenant_id,
+        user_id=user_id
+    )
+    
+    return result

geek_cafe_saas_sdk/domains/files/handlers/files/list/app.py

@@ -0,0 +1,72 @@
+"""
+Lambda handler for listing files.
+
+Requires authentication (secure mode).
+"""
+
+from typing import Dict, Any
+from geek_cafe_saas_sdk.lambda_handlers import create_handler
+from geek_cafe_saas_sdk.domains.files.services import FileSystemService
+
+
+# Factory creates handler (defaults to secure auth)
+handler_wrapper = create_handler(
+    service_class=FileSystemService,
+    require_body=False,
+    convert_case=True
+)
+
+
+def handler(event: Dict[str, Any], context: Any, injected_service=None) -> Dict[str, Any]:
+    """
+    List files.
+    
+    Args:
+        event: Lambda event from API Gateway
+        context: Lambda context
+        injected_service: Optional FileSystemService for testing
+    
+    Query parameters:
+        directoryId: Filter by directory (optional)
+        ownerId: Filter by owner (optional)
+        limit: Max results (optional, default: 100)
+    
+    Returns 200 with list of files
+    """
+    return handler_wrapper.execute(event, context, list_files, injected_service)
+
+
+def list_files(
+    event: Dict[str, Any],
+    service: FileSystemService,
+    user_context: Dict[str, str]
+) -> Any:
+    """
+    Business logic for listing files.
+    """
+    tenant_id = user_context.get("tenant_id")
+    user_id = user_context.get("user_id")
+    
+    # Get query parameters
+    query_params = event.get("queryStringParameters", {}) or {}
+    directory_id = query_params.get("directoryId")
+    owner_id = query_params.get("ownerId") or user_id  # Default to current user
+    limit = int(query_params.get("limit", "100"))
+    
+    # List files by directory or owner
+    if directory_id:
+        result = service.list_files_by_directory(
+            tenant_id=tenant_id,
+            directory_id=directory_id,
+            user_id=user_id,
+            limit=limit
+        )
+    else:
+        result = service.list_files_by_owner(
+            tenant_id=tenant_id,
+            owner_id=owner_id,
+            user_id=user_id,
+            limit=limit
+        )
+    
+    return result

geek_cafe_saas_sdk/domains/files/handlers/lineage/create_derived/app.py

@@ -0,0 +1,99 @@
+"""
+Lambda handler for creating a derived file from a main file.
+
+Requires authentication (secure mode).
+"""
+
+from typing import Dict, Any
+from geek_cafe_saas_sdk.lambda_handlers import create_handler
+from geek_cafe_saas_sdk.domains.files.services import FileLineageService
+import base64
+
+
+# Factory creates handler (defaults to secure auth)
+handler_wrapper = create_handler(
+    service_class=FileLineageService,
+    require_body=True,
+    convert_case=True
+)
+
+
+def handler(event: Dict[str, Any], context: Any, injected_service=None) -> Dict[str, Any]:
+    """
+    Create a derived file from a main file (e.g., data cleaning).
+    
+    Args:
+        event: Lambda event from API Gateway
+        context: Lambda context
+        injected_service: Optional FileLineageService for testing
+    
+    Expected body (camelCase from frontend):
+    {
+        "mainFileId": "file-456",  # Main file ID
+        "fileName": "data_clean_v1.csv",
+        "fileData": "base64_encoded_content",
+        "transformationOperation": "data_cleaning_v1",
+        "transformationMetadata": {
+            "cleaning_version": 1,
+            "operations": ["remove_nulls", "normalize_units"],
+            "rows_processed": 1000
+        },
+        "directoryId": "dir-789"  # Optional
+    }
+    
+    Returns 201 with created derived file
+    """
+    return handler_wrapper.execute(event, context, create_derived_file, injected_service)
+
+
+def create_derived_file(
+    event: Dict[str, Any],
+    service: FileLineageService,
+    user_context: Dict[str, str]
+) -> Any:
+    """
+    Business logic for creating derived file.
+    """
+    payload = event["parsed_body"]
+    
+    tenant_id = user_context.get("tenant_id")
+    user_id = user_context.get("user_id")
+    
+    # Extract required fields
+    main_file_id = payload.get("main_file_id")
+    file_name = payload.get("file_name")
+    file_data_b64 = payload.get("file_data")
+    transformation_operation = payload.get("transformation_operation")
+    
+    if not main_file_id:
+        raise ValueError("main_file_id is required")
+    if not file_name:
+        raise ValueError("file_name is required")
+    if not file_data_b64:
+        raise ValueError("file_data is required")
+    if not transformation_operation:
+        raise ValueError("transformation_operation is required")
+    
+    # Decode base64 file data
+    try:
+        file_data = base64.b64decode(file_data_b64)
+    except Exception as e:
+        raise ValueError(f"Invalid base64 file_data: {str(e)}")
+    
+    # Extract optional fields
+    transformation_metadata = payload.get("transformation_metadata", {})
+    directory_id = payload.get("directory_id")
+    
+    # Create derived file
+    result = service.create_derived_file(
+        tenant_id=tenant_id,
+        user_id=user_id,
+        main_file_id=main_file_id,
+        file_name=file_name,
+        file_data=file_data,
+        transformation_operation=transformation_operation,
+        transformation_metadata=transformation_metadata,
+        directory_id=directory_id
+    )
+    
+    return result

geek_cafe_saas_sdk/domains/files/handlers/lineage/create_main/app.py

@@ -0,0 +1,104 @@
+"""
+Lambda handler for creating a main file from an original file.
+
+Requires authentication (secure mode).
+"""
+
+from typing import Dict, Any
+from geek_cafe_saas_sdk.lambda_handlers import create_handler
+from geek_cafe_saas_sdk.domains.files.services import FileLineageService
+import base64
+
+
+# Factory creates handler (defaults to secure auth)
+handler_wrapper = create_handler(
+    service_class=FileLineageService,
+    require_body=True,
+    convert_case=True
+)
+
+
+def handler(event: Dict[str, Any], context: Any, injected_service=None) -> Dict[str, Any]:
+    """
+    Create a main file from an original file (e.g., XLS → CSV conversion).
+    
+    Args:
+        event: Lambda event from API Gateway
+        context: Lambda context
+        injected_service: Optional FileLineageService for testing
+    
+    Expected body (camelCase from frontend):
+    {
+        "originalFileId": "file-123",  # Original file ID
+        "fileName": "data.csv",
+        "fileData": "base64_encoded_content",
+        "mimeType": "text/csv",
+        "transformationOperation": "xls_to_csv",
+        "transformationMetadata": {
+            "source_format": "xls",
+            "target_format": "csv",
+            "converter_version": "1.0"
+        },
+        "directoryId": "dir-456"  # Optional
+    }
+    
+    Returns 201 with created main file
+    """
+    return handler_wrapper.execute(event, context, create_main_file, injected_service)
+
+
+def create_main_file(
+    event: Dict[str, Any],
+    service: FileLineageService,
+    user_context: Dict[str, str]
+) -> Any:
+    """
+    Business logic for creating main file.
+    """
+    payload = event["parsed_body"]
+    
+    tenant_id = user_context.get("tenant_id")
+    user_id = user_context.get("user_id")
+    
+    # Extract required fields
+    original_file_id = payload.get("original_file_id")
+    file_name = payload.get("file_name")
+    file_data_b64 = payload.get("file_data")
+    mime_type = payload.get("mime_type")
+    transformation_operation = payload.get("transformation_operation")
+    
+    if not original_file_id:
+        raise ValueError("original_file_id is required")
+    if not file_name:
+        raise ValueError("file_name is required")
+    if not file_data_b64:
+        raise ValueError("file_data is required")
+    if not mime_type:
+        raise ValueError("mime_type is required")
+    if not transformation_operation:
+        raise ValueError("transformation_operation is required")
+    
+    # Decode base64 file data
+    try:
+        file_data = base64.b64decode(file_data_b64)
+    except Exception as e:
+        raise ValueError(f"Invalid base64 file_data: {str(e)}")
+    
+    # Extract optional fields
+    transformation_metadata = payload.get("transformation_metadata", {})
+    directory_id = payload.get("directory_id")
+    
+    # Create main file
+    result = service.create_main_file(
+        tenant_id=tenant_id,
+        user_id=user_id,
+        original_file_id=original_file_id,
+        file_name=file_name,
+        file_data=file_data,
+        mime_type=mime_type,
+        transformation_operation=transformation_operation,
+        transformation_metadata=transformation_metadata,
+        directory_id=directory_id
+    )
+    
+    return result

geek_cafe_saas_sdk/domains/files/handlers/lineage/download_bundle/app.py

@@ -0,0 +1,99 @@
+"""
+Lambda handler for downloading complete lineage bundle with file content.
+
+Requires authentication (secure mode).
+"""
+
+from typing import Dict, Any
+from geek_cafe_saas_sdk.lambda_handlers import create_handler
+from geek_cafe_saas_sdk.domains.files.services import FileLineageService
+import base64
+
+
+# Factory creates handler (defaults to secure auth)
+handler_wrapper = create_handler(
+    service_class=FileLineageService,
+    require_body=False,
+    convert_case=True
+)
+
+
+def handler(event: Dict[str, Any], context: Any, injected_service=None) -> Dict[str, Any]:
+    """
+    Download complete lineage bundle with file content.
+    
+    Args:
+        event: Lambda event from API Gateway
+        context: Lambda context
+        injected_service: Optional FileLineageService for testing
+    
+    Path parameters:
+        fileId: File ID to bundle
+    
+    Returns 200 with bundle including file data (base64 encoded):
+    {
+        "selected": {
+            "file": {file object},
+            "data": "base64_encoded_content"
+        },
+        "main": {
+            "file": {file object},
+            "data": "base64_encoded_content"
+        },
+        "original": {
+            "file": {file object},
+            "data": "base64_encoded_content"
+        },
+        "metadata": {transformation chain}
+    }
+    """
+    return handler_wrapper.execute(event, context, download_bundle, injected_service)
+
+
+def download_bundle(
+    event: Dict[str, Any],
+    service: FileLineageService,
+    user_context: Dict[str, str]
+) -> Dict[str, Any]:
+    """
+    Business logic for downloading lineage bundle.
+    """
+    tenant_id = user_context.get("tenant_id")
+    user_id = user_context.get("user_id")
+    
+    # Get file ID from path parameters
+    path_params = event.get("pathParameters", {})
+    file_id = path_params.get("fileId") or path_params.get("id")
+    
+    if not file_id:
+        raise ValueError("fileId path parameter is required")
+    
+    # Download bundle
+    result = service.download_lineage_bundle(
+        selected_file_id=file_id,
+        tenant_id=tenant_id,
+        user_id=user_id
+    )
+    
+    if result.success:
+        bundle = result.data
+        
+        # Encode file data as base64 for JSON response
+        if bundle.get('selected') and bundle['selected'].get('data'):
+            bundle['selected']['data'] = base64.b64encode(
+                bundle['selected']['data']
+            ).decode('utf-8')
+        
+        if bundle.get('main') and bundle['main'].get('data'):
+            bundle['main']['data'] = base64.b64encode(
+                bundle['main']['data']
+            ).decode('utf-8')
+        
+        if bundle.get('original') and bundle['original'].get('data'):
+            bundle['original']['data'] = base64.b64encode(
+                bundle['original']['data']
+            ).decode('utf-8')
+        
+        return bundle
+    
+    return result

geek_cafe_saas_sdk/domains/files/handlers/lineage/get_lineage/app.py

@@ -0,0 +1,68 @@
+"""
+Lambda handler for getting file lineage.
+
+Requires authentication (secure mode).
+"""
+
+from typing import Dict, Any
+from geek_cafe_saas_sdk.lambda_handlers import create_handler
+from geek_cafe_saas_sdk.domains.files.services import FileLineageService
+
+
+# Factory creates handler (defaults to secure auth)
+handler_wrapper = create_handler(
+    service_class=FileLineageService,
+    require_body=False,
+    convert_case=True
+)
+
+
+def handler(event: Dict[str, Any], context: Any, injected_service=None) -> Dict[str, Any]:
+    """
+    Get complete lineage for a file.
+    
+    Args:
+        event: Lambda event from API Gateway
+        context: Lambda context
+        injected_service: Optional FileLineageService for testing
+    
+    Path parameters:
+        fileId: File ID
+    
+    Returns 200 with lineage information:
+    {
+        "selected": {file object},
+        "main": {file object or null},
+        "original": {file object or null},
+        "allDerived": [{file objects}]  # If viewing main file
+    }
+    """
+    return handler_wrapper.execute(event, context, get_file_lineage, injected_service)
+
+
+def get_file_lineage(
+    event: Dict[str, Any],
+    service: FileLineageService,
+    user_context: Dict[str, str]
+) -> Any:
+    """
+    Business logic for getting file lineage.
+    """
+    tenant_id = user_context.get("tenant_id")
+    user_id = user_context.get("user_id")
+    
+    # Get file ID from path parameters
+    path_params = event.get("pathParameters", {})
+    file_id = path_params.get("fileId") or path_params.get("id")
+    
+    if not file_id:
+        raise ValueError("fileId path parameter is required")
+    
+    # Get lineage
+    result = service.get_lineage(
+        file_id=file_id,
+        tenant_id=tenant_id,
+        user_id=user_id
+    )
+    
+    return result

geek_cafe_saas_sdk/domains/files/handlers/lineage/prepare_bundle/app.py

@@ -0,0 +1,76 @@
+"""
+Lambda handler for preparing lineage bundle.
+
+Requires authentication (secure mode).
+"""
+
+from typing import Dict, Any
+from geek_cafe_saas_sdk.lambda_handlers import create_handler
+from geek_cafe_saas_sdk.domains.files.services import FileLineageService
+
+
+# Factory creates handler (defaults to secure auth)
+handler_wrapper = create_handler(
+    service_class=FileLineageService,
+    require_body=False,
+    convert_case=True
+)
+
+
+def handler(event: Dict[str, Any], context: Any, injected_service=None) -> Dict[str, Any]:
+    """
+    Prepare lineage bundle for a file (metadata only, no file content).
+    
+    Args:
+        event: Lambda event from API Gateway
+        context: Lambda context
+        injected_service: Optional FileLineageService for testing
+    
+    Path parameters:
+        fileId: File ID to bundle
+    
+    Returns 200 with bundle information:
+    {
+        "selectedFile": {file object},
+        "mainFile": {file object or null},
+        "originalFile": {file object or null},
+        "metadata": {
+            "selectedFileId": "...",
+            "selectedFileName": "...",
+            "transformationChain": [
+                {"step": 1, "type": "original", "fileId": "...", "fileName": "..."},
+                {"step": 2, "type": "convert", "fileId": "...", "fileName": "...", "operation": "..."},
+                {"step": 3, "type": "clean", "fileId": "...", "fileName": "...", "operation": "..."}
+            ]
+        }
+    }
+    """
+    return handler_wrapper.execute(event, context, prepare_bundle, injected_service)
+
+
+def prepare_bundle(
+    event: Dict[str, Any],
+    service: FileLineageService,
+    user_context: Dict[str, str]
+) -> Any:
+    """
+    Business logic for preparing lineage bundle.
+    """
+    tenant_id = user_context.get("tenant_id")
+    user_id = user_context.get("user_id")
+    
+    # Get file ID from path parameters
+    path_params = event.get("pathParameters", {})
+    file_id = path_params.get("fileId") or path_params.get("id")
+    
+    if not file_id:
+        raise ValueError("fileId path parameter is required")
+    
+    # Prepare bundle
+    result = service.prepare_lineage_bundle(
+        selected_file_id=file_id,
+        tenant_id=tenant_id,
+        user_id=user_id
+    )
+    
+    return result

geek_cafe_saas_sdk/domains/files/models/__init__.py

@@ -0,0 +1,17 @@
+"""File models.
+
+Geek Cafe, LLC
+MIT License. See Project Root for the license information.
+"""
+
+from .file import File
+from .directory import Directory
+from .file_version import FileVersion
+from .file_share import FileShare
+
+__all__ = [
+    "File",
+    "Directory",
+    "FileVersion",
+    "FileShare",
+]

geek_cafe_saas_sdk/domains/files/models/directory.py

@@ -21,12 +21,12 @@ class Directory(BaseModel):
     Moving a file updates its directory_id, not its S3 location.
     
     Access Patterns (DynamoDB Keys):
-    - pk: DIRECTORY#{tenant_id}#{directory_id}
-    - sk: METADATA
-    - gsi1_pk: TENANT#{tenant_id}
-    - gsi1_sk: PATH#{full_path}
-    - gsi2_pk: DIRECTORY#{tenant_id}#{parent_id}
-    - gsi2_sk: NAME#{directory_name}
+    - pk: directory#{tenant_id}#{directory_id}
+    - sk: metadata
+    - gsi1_pk: tenant#{tenant_id}
+    - gsi1_sk: path#{full_path}
+    - gsi2_pk: directory#{tenant_id}#{parent_id}
+    - gsi2_sk: name#{directory_name}
     """
 
     def __init__(self):
@@ -61,6 +61,42 @@ class Directory(BaseModel):
         
         # Timestamps (inherited from BaseModel)
         # created_utc_ts, updated_utc_ts
+        
+        # CRITICAL: Call _setup_indexes() as LAST line in __init__
+        self._setup_indexes()
+    
+    def _setup_indexes(self):
+        """Setup DynamoDB indexes for directory queries."""
+        
+        # Primary index: Directory by ID
+        primary = DynamoDBIndex()
+        primary.name = "primary"
+        primary.partition_key.attribute_name = "pk"
+        primary.partition_key.value = lambda: DynamoDBKey.build_key(("directory", self.id))
+        primary.sort_key.attribute_name = "sk"
+        primary.sort_key.value = lambda: "metadata"
+        self.indexes.add_primary(primary)
+        
+        # GSI1: Directories by path
+        gsi = DynamoDBIndex()
+        gsi.name = "gsi1"
+        gsi.partition_key.attribute_name = f"{gsi.name}_pk"
+        gsi.partition_key.value = lambda: DynamoDBKey.build_key(("tenant", self.tenant_id))
+        gsi.sort_key.attribute_name = f"{gsi.name}_sk"
+        gsi.sort_key.value = lambda: DynamoDBKey.build_key(("path", self.full_path))
+        self.indexes.add_secondary(gsi)
+        
+        # GSI2: Subdirectories by parent
+        gsi = DynamoDBIndex()
+        gsi.name = "gsi2"
+        gsi.partition_key.attribute_name = f"{gsi.name}_pk"
+        if self.parent_id:
+            gsi.partition_key.value = lambda: DynamoDBKey.build_key(("directory", self.tenant_id), ("parent", self.parent_id))
+        else:
+            gsi.partition_key.value = lambda: DynamoDBKey.build_key(("directory", self.tenant_id), ("root", "root"))
+        gsi.sort_key.attribute_name = f"{gsi.name}_sk"
+        gsi.sort_key.value = lambda: DynamoDBKey.build_key(("name", self.directory_name))
+        self.indexes.add_secondary(gsi)
     
     # Properties - Identity
     @property