Added a significant overhaul to the server manager.

2025-12-29 16:07:30 -05:00
parent 4f4e5276a6
commit 216e18a7b4
796 changed files with 528631 additions and 2671 deletions
@@ -0,0 +1,40 @@
+from fastapi import APIRouter, BackgroundTasks
+from ..comics_manager import ComicsManager
+from ..config import settings
+from ..samba_manager import SambaManager
+
+router = APIRouter()
+
+def run_organize_comics_background():
+    app_settings = settings
+    manager = ComicsManager()
+    samba_manager = SambaManager(
+        app_settings.samba_server_ip,
+        "isolation",
+        app_settings.samba_username,
+        app_settings.samba_password,
+    )
+    manager.organize_comics(samba_manager)
+    samba_manager.close()
+
+@router.post("/organize")
+def organize_comics(background_tasks: BackgroundTasks):
+    background_tasks.add_task(run_organize_comics_background)
+    return {"message": "Comics organization started in the background."}
+
+def run_cleanup_toberead_background():
+    app_settings = settings
+    manager = ComicsManager()
+    samba_manager = SambaManager(
+        app_settings.samba_server_ip,
+        "isolation",
+        app_settings.samba_username,
+        app_settings.samba_password,
+    )
+    manager.cleanup_toberead(samba_manager)
+    samba_manager.close()
+
+@router.delete("/cleanup_toberead")
+def cleanup_toberead(background_tasks: BackgroundTasks):
+    background_tasks.add_task(run_cleanup_toberead_background)
+    return {"message": "Comics cleanup started in the background."}
@@ -0,0 +1,57 @@
+from fastapi import APIRouter, Depends, BackgroundTasks
+from sqlalchemy.orm import Session
+from ..duplicates_manager import DuplicatesManager
+from ..config import settings, Settings
+from ..database import get_db, SessionLocal
+from ..samba_manager import SambaManager
+import os
+
+router = APIRouter()
+
+def run_scan_videos_background():
+    db = SessionLocal()
+    app_settings = Settings()
+    manager = DuplicatesManager(app_settings, db)
+    samba_manager = SambaManager(
+        app_settings.samba_server_ip,
+        "isolation",
+        app_settings.samba_username,
+        app_settings.samba_password,
+    )
+    try:
+        manager.scan_videos(samba_manager)
+    finally:
+        samba_manager.close()
+        db.close()
+
+@router.post("/scan")
+def scan_videos(background_tasks: BackgroundTasks):
+    background_tasks.add_task(run_scan_videos_background)
+    return {"message": "Video scan started in the background."}
+
+@router.post("/scan/pause")
+def pause_scan():
+    with open("logs/duplicates_scan.pause", 'w') as f:
+        pass
+    return {"message": "Video scan paused."}
+
+@router.post("/scan/resume")
+def resume_scan():
+    if os.path.exists("logs/duplicates_scan.pause"):
+        os.remove("logs/duplicates_scan.pause")
+    return {"message": "Video scan resumed."}
+
+@router.get("/scan/status")
+def get_scan_status():
+    if os.path.exists("logs/duplicates_scan.pause"):
+        return {"status": "Paused"}
+    if os.path.exists("logs/duplicates_scan.progress") or os.path.exists("logs/duplicates_scan.state"):
+        return {"status": "Scanning"}
+    return {"status": "Idle"}
+
+
+@router.post("/find")
+def find_duplicates(threshold: float = 0.95, db: Session = Depends(get_db)):
+    manager = DuplicatesManager(settings, db)
+    result = manager.find_duplicates(threshold)
+    return result
@@ -0,0 +1,178 @@
+from fastapi import APIRouter, UploadFile, File, Request, Response
+from fastapi.responses import StreamingResponse
+from pydantic import BaseModel
+from ..samba_manager import SambaManager
+from ..config import settings
+import os
+import io
+import mimetypes
+import zipfile
+
+class RenameRequest(BaseModel):
+    old_path: str
+    new_path: str
+
+router = APIRouter()
+
+
+@router.get("/list/{share_name}")
+def list_files(share_name: str, path: str = "/"):
+    manager = SambaManager(
+        settings.samba_server_ip,
+        share_name,
+        username=settings.samba_username,
+        password=settings.samba_password,
+    )
+    files = manager.list_path(path)
+    manager.close()
+    return files
+
+
+@router.delete("/files/{share_name}")
+def delete_file(share_name: str, path: str):
+    manager = SambaManager(
+        settings.samba_server_ip,
+        share_name,
+        username=settings.samba_username,
+        password=settings.samba_password,
+    )
+    result = manager.delete_file(path)
+    manager.close()
+    return result
+
+
+@router.post("/files/upload/{share_name}")
+def upload_file(share_name: str, path: str, file: UploadFile = File(...)):
+    manager = SambaManager(
+        settings.samba_server_ip,
+        share_name,
+        username=settings.samba_username,
+        password=settings.samba_password,
+    )
+    # The path in storeFile is service_name + path
+    remote_path = os.path.join(path, file.filename)
+    result = manager.upload_file(remote_path, file.file)
+    manager.close()
+    return result
+
+
+@router.get("/files/download/{share_name}")
+def download_file(share_name: str, path: str):
+    manager = SambaManager(
+        settings.samba_server_ip,
+        share_name,
+        username=settings.samba_username,
+        password=settings.samba_password,
+    )
+    file_obj = io.BytesIO()
+    manager.download_file(path, file_obj)
+    manager.close()
+    file_obj.seek(0)
+    return StreamingResponse(file_obj, media_type="application/octet-stream")
+
+
+@router.get("/view/{share_name}")
+def view_file(share_name: str, path: str, request: Request):
+    manager = SambaManager(
+        settings.samba_server_ip,
+        share_name,
+        username=settings.samba_username,
+        password=settings.samba_password,
+    )
+    
+    try:
+        attributes = manager.get_attributes(path)
+        if isinstance(attributes, dict) and "error" in attributes:
+            return Response(status_code=404)
+        file_size = attributes.file_size
+
+        range_header = request.headers.get('Range')
+        
+        media_type, _ = mimetypes.guess_type(path)
+        if media_type is None:
+            media_type = "application/octet-stream"
+
+        if range_header:
+            start_str, end_str = range_header.replace('bytes=', '').split('-')
+            start = int(start_str)
+            end = int(end_str) if end_str else file_size - 1
+
+            if start >= file_size:
+                return Response(status_code=416)
+
+            length = end - start + 1
+            file_obj = io.BytesIO()
+            manager.download_file_range(path, file_obj, offset=start, max_length=length)
+            file_obj.seek(0)
+            
+            headers = {
+                'Content-Range': f'bytes {start}-{end}/{file_size}',
+                'Accept-Ranges': 'bytes',
+                'Content-Length': str(length),
+            }
+            return Response(file_obj.read(), status_code=206, headers=headers, media_type=media_type)
+        else:
+            file_obj = io.BytesIO()
+            manager.download_file(path, file_obj)
+            file_obj.seek(0)
+            return StreamingResponse(file_obj, media_type=media_type, headers={"Accept-Ranges": "bytes", "Content-Length": str(file_size)})
+    finally:
+        manager.close()
+
+
+@router.get("/cbz/list/{share_name}")
+def list_cbz_contents(share_name: str, path: str):
+    manager = SambaManager(
+        settings.samba_server_ip,
+        share_name,
+        username=settings.samba_username,
+        password=settings.samba_password,
+    )
+    file_obj = io.BytesIO()
+    manager.download_file(path, file_obj)
+    manager.close()
+    file_obj.seek(0)
+
+    image_files = []
+    with zipfile.ZipFile(file_obj, 'r') as zip_ref:
+        for file_info in zip_ref.infolist():
+            if file_info.filename.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp', '.webp')):
+                image_files.append(file_info.filename)
+    
+    return sorted(image_files)
+
+@router.get("/cbz/image/{share_name}")
+def get_cbz_image(share_name: str, path: str, image_name: str):
+    manager = SambaManager(
+        settings.samba_server_ip,
+        share_name,
+        username=settings.samba_username,
+        password=settings.samba_password,
+    )
+    file_obj = io.BytesIO()
+    manager.download_file(path, file_obj)
+    manager.close()
+    file_obj.seek(0)
+
+    with zipfile.ZipFile(file_obj, 'r') as zip_ref:
+        with zip_ref.open(image_name) as image_file:
+            image_data = io.BytesIO(image_file.read())
+    
+    media_type, _ = mimetypes.guess_type(image_name)
+    if media_type is None:
+        media_type = "application/octet-stream"
+        
+    return StreamingResponse(image_data, media_type=media_type)
+
+
+@router.put("/files/rename/{share_name}")
+def rename_file(share_name: str, request: RenameRequest):
+    manager = SambaManager(
+        settings.samba_server_ip,
+        share_name,
+        username=settings.samba_username,
+        password=settings.samba_password,
+    )
+    result = manager.rename_file(request.old_path, request.new_path)
+    manager.close()
+    return result