add downlod endpoint

Author: ix-56h

src/server/models.py

@@ -69,6 +69,8 @@ class IngestSuccessResponse(BaseModel):
         Short form of repository URL (user/repo).
     summary : str
         Summary of the ingestion process including token estimates.
+    ingest_id : str
+        Ingestion id used to download full context.
     tree : str
         File tree structure of the repository.
     content : str
@@ -85,6 +87,7 @@ class IngestSuccessResponse(BaseModel):
     repo_url: str = Field(..., description="Original repository URL")
     short_repo_url: str = Field(..., description="Short repository URL (user/repo)")
     summary: str = Field(..., description="Ingestion summary with token estimates")
+    ingest_id: str = Field(..., description="Ingestion id used to download full context")
     tree: str = Field(..., description="File tree structure")
     content: str = Field(..., description="Processed file content")
     default_max_file_size: int = Field(..., description="File size slider position used")

src/server/query_processor.py

@@ -122,6 +122,7 @@ async def process_query(
         repo_url=input_text,
         short_repo_url=short_repo_url,
         summary=summary,
+        ingest_id=query.id,
         tree=tree,
         content=content,
         default_max_file_size=slider_position,

src/server/routers/ingest.py

@@ -1,8 +1,9 @@
 """Ingest endpoint for the API."""
 
-from fastapi import APIRouter, Request, status
-from fastapi.responses import JSONResponse
+from fastapi import APIRouter, HTTPException, Request, status
+from fastapi.responses import FileResponse, JSONResponse
 
+from gitingest.config import TMP_BASE_PATH
 from server.models import IngestErrorResponse, IngestRequest, IngestSuccessResponse
 from server.query_processor import process_query
 from server.server_config import MAX_DISPLAY_SIZE
@@ -157,3 +158,46 @@ async def api_ingest_get(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
             content=error_response.model_dump(),
         )
+
+
+@router.get("/api/download/file/{ingest_id}", response_class=FileResponse)
+async def download_ingest(ingest_id: str) -> FileResponse:
+    """Return the first ``*.txt`` file produced for ``ingest_id`` as a download.
+
+    Parameters
+    ----------
+    ingest_id : str
+        Identifier that the ingest step emitted (also the directory name that stores the artefacts).
+
+    Returns
+    -------
+    FileResponse
+        Streamed response with media type ``text/plain`` that prompts the browser to download the file.
+
+    Raises
+    ------
+    HTTPException
+        **404** - digest directory is missing or contains no ``*.txt`` file.
+        **403** - the process lacks permission to read the directory or file.
+
+    """
+    directory = TMP_BASE_PATH / ingest_id
+
+    if not directory.is_dir():
+        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"Digest {ingest_id!r} not found")
+
+    try:
+        first_txt_file = next(directory.glob("*.txt"))
+    except StopIteration as exc:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail=f"No .txt file found for digest {ingest_id!r}",
+        ) from exc
+
+    try:
+        return FileResponse(path=first_txt_file, media_type="text/plain", filename=first_txt_file.name)
+    except PermissionError as exc:
+        raise HTTPException(
+            status_code=status.HTTP_403_FORBIDDEN,
+            detail=f"Permission denied for {first_txt_file}",
+        ) from exc

src/static/js/utils.js

@@ -172,6 +172,9 @@ function handleSuccessfulResponse(data) {
     // Show results section
     showResults();
 
+    // Store the ingest_id for download functionality
+    window.currentIngestId = data.ingest_id;
+
     // Set plain text content for summary, tree, and content
     document.getElementById('result-summary').value = data.summary || '';
     document.getElementById('directory-structure-content').value = data.tree || '';
@@ -268,33 +271,36 @@ function copyFullDigest() {
 }
 
 function downloadFullDigest() {
-    const summary = document.getElementById('result-summary').value;
-    const directoryStructure = document.getElementById('directory-structure-content').value;
-    const filesContent = document.querySelector('.result-text').value;
+    // Check if we have an ingest_id
+    if (!window.currentIngestId) {
+        console.error('No ingest_id available for download');
 
-    // Create the full content with all three sections
-    const fullContent = `${summary}\n${directoryStructure}\n${filesContent}`;
+        return;
+    }
 
-    // Create a blob with the content
-    const blob = new Blob([fullContent], { type: 'text/plain' });
+    // Show feedback on the button
+    const button = document.querySelector('[onclick="downloadFullDigest()"]');
+    const originalText = button.innerHTML;
 
-    // Create a download link
-    const url = window.URL.createObjectURL(blob);
+    button.innerHTML = `
+        <svg class="w-4 h-4 mr-2" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+            <path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M4 4v5h.582m15.356 2A8.001 8.001 0 004.582 9m0 0H9m11 11v-5h-.581m0 0a8.003 8.003 0 01-15.357-2m15.357 2H15"></path>
+        </svg>
+        Downloading...
+    `;
+
+    // Create a download link to the server endpoint
     const a = document.createElement('a');
 
-    a.href = url;
+    a.href = `/api/download/file/${window.currentIngestId}`;
     a.download = 'digest.txt';
     document.body.appendChild(a);
     a.click();
 
     // Clean up
-    window.URL.revokeObjectURL(url);
     document.body.removeChild(a);
 
-    // Show feedback on the button
-    const button = document.querySelector('[onclick="downloadFullDigest()"]');
-    const originalText = button.innerHTML;
-
+    // Update button to show success
     button.innerHTML = `
         <svg class="w-4 h-4 mr-2" fill="none" stroke="currentColor" viewBox="0 0 24 24">
             <path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M5 13l4 4L19 7"></path>