gh-christianlouis-docuelevate/app/views/general.py

"""
General routes for the application homepage and basic pages.
"""

from datetime import date, datetime, timezone
from pathlib import Path

from fastapi import Depends, HTTPException, Request
from fastapi.responses import FileResponse, RedirectResponse
from sqlalchemy import func
from sqlalchemy.orm import Session

from app.utils.config_validator import get_provider_status, validate_storage_configs
from app.views.base import APIRouter, get_db, logger, require_login, templates

router = APIRouter()

# Date format constant
_DATE_DISPLAY_FORMAT = "%B %d, %Y"


@router.get("/", include_in_schema=False)
async def serve_index(request: Request, db: Session = Depends(get_db)):
    """
    Serve the index/home page.

    If the system requires initial setup, redirect to the setup wizard.
    """
    # Check if setup wizard is needed
    from app.utils.settings_service import get_setting_from_db
    from app.utils.setup_wizard import is_setup_required

    # Check if setup was explicitly skipped
    setup_skipped = get_setting_from_db(db, "_setup_wizard_skipped")

    # Check setup completion query param
    setup_complete = request.query_params.get("setup") == "complete"

    if not setup_skipped and not setup_complete and is_setup_required():
        logger.info("System requires initial setup, redirecting to wizard")
        return RedirectResponse(url="/setup?step=1", status_code=303)

    # Get provider information from config validator
    providers = get_provider_status()

    # Count configured providers
    configured_providers = sum(1 for provider in providers.values() if provider["configured"])

    # Count different types of storage targets
    storage_issues = validate_storage_configs()
    configured_storage_targets = sum(
        1
        for provider, issues in storage_issues.items()
        if not issues
        and provider in ["dropbox", "nextcloud", "sftp", "s3", "ftp", "webdav", "google_drive", "onedrive"]
    )

    from app.models import FileRecord

    today = datetime.now(timezone.utc).date()

    # Global file counts (or per-user in multi-user mode)
    from app.config import settings

    user = request.session.get("user") or {}
    is_admin = user.get("is_admin", False)

    try:
        total_files: int = db.query(func.count(FileRecord.id)).scalar() or 0

        files_today: int = (
            db.query(func.count(FileRecord.id)).filter(func.date(FileRecord.created_at) == today).scalar() or 0
        )

        files_month: int = (
            db.query(func.count(FileRecord.id))
            .filter(func.strftime("%Y-%m", FileRecord.created_at) == today.strftime("%Y-%m"))
            .scalar()
            or 0
        )

        files_with_ocr: int = db.query(func.count(FileRecord.id)).filter(FileRecord.ocr_text.isnot(None)).scalar() or 0

        unique_users: int = (
            db.query(func.count(func.distinct(FileRecord.owner_id))).filter(FileRecord.owner_id.isnot(None)).scalar()
            or 0
        )
    except Exception as e:
        logger.error(f"Error computing dashboard stats: {e}")
        total_files = files_today = files_month = files_with_ocr = unique_users = 0

    # Per-user usage for the subscription widget (multi-user only)
    user_usage = None
    user_tier = None
    if settings.multi_user_enabled:
        owner_id: str = user.get("username") or user.get("email") or user.get("sub") or ""
        if owner_id:
            try:
                from app.utils.subscription import get_tier, get_user_tier_id, get_user_usage

                tier_id = get_user_tier_id(db, owner_id)
                user_tier = get_tier(tier_id)
                user_usage = get_user_usage(db, owner_id)
            except Exception as e:
                logger.error(f"Error fetching subscription info: {e}")

    stats = {
        "processed_files": total_files,
        "files_today": files_today,
        "files_month": files_month,
        "files_with_ocr": files_with_ocr,
        "unique_users": unique_users,
        "active_integrations": configured_providers,
        "storage_targets": configured_storage_targets,
    }

    return templates.TemplateResponse(
        "index.html",
        {
            "request": request,
            "stats": stats,
            "user_usage": user_usage,
            "user_tier": user_tier,
            "multi_user_enabled": settings.multi_user_enabled,
            "is_admin": is_admin,
            "allow_signup": settings.multi_user_enabled and settings.allow_local_signup,
        },
    )


@router.get("/about", include_in_schema=False)
async def serve_about(request: Request):
    """Serve the about page."""
    return templates.TemplateResponse("about.html", {"request": request})


@router.get("/privacy", include_in_schema=False)
async def serve_privacy(request: Request):
    """Serve the privacy policy page."""
    # Pass the current date for the "Last Updated" field
    current_date = date.today().strftime(_DATE_DISPLAY_FORMAT)
    return templates.TemplateResponse("privacy.html", {"request": request, "current_date": current_date})


@router.get("/imprint", include_in_schema=False)
async def serve_imprint(request: Request):
    """Serve the imprint/impressum page."""
    return templates.TemplateResponse("imprint.html", {"request": request})


@router.get("/upload", include_in_schema=False)
@require_login
async def serve_upload(request: Request):
    """Serve the upload page."""
    from app.config import settings

    return templates.TemplateResponse(
        "upload.html",
        {
            "request": request,
            "upload_concurrency": settings.upload_concurrency,
            "upload_queue_delay_ms": settings.upload_queue_delay_ms,
        },
    )


@router.get("/favicon.ico", include_in_schema=False)
def favicon():
    """Serve the favicon."""
    favicon_path = Path(__file__).parent.parent.parent / "frontend" / "static" / "favicon.ico"
    if not favicon_path.exists():
        # If favicon doesn't exist, return a 404
        raise HTTPException(status_code=404, detail="Favicon not found")
    return FileResponse(favicon_path)


@router.get("/license", include_in_schema=False)
async def serve_license(request: Request):
    """Serve the license page."""
    # Try multiple possible locations for the license file
    possible_locations = [
        Path(__file__).parent.parent.parent / "LICENSE",  # Repository root
        Path("/app/LICENSE"),  # Docker container path
        Path.home() / "LICENSE",  # Home directory (fallback)
    ]

    license_text = None

    # Try to read from any of the possible locations
    for path in possible_locations:
        try:
            with open(path, "r") as f:
                license_text = f.read()
                break  # File found and read, exit loop
        except (FileNotFoundError, PermissionError):
            continue  # Try next location

    # If license text is still None, use embedded text
    if license_text is None:
        license_text = """
Apache License
Version 2.0, January 2004
http://www.apache.org/licenses/

This software is licensed under the Apache License 2.0.
The full license text could not be located on this system.
Please visit http://www.apache.org/licenses/LICENSE-2.0 for the complete license text.
"""

    return templates.TemplateResponse("license.html", {"request": request, "license_text": license_text})


@router.get("/cookies", include_in_schema=False)
async def serve_cookies(request: Request):
    """Serve the cookie policy page."""
    current_date = date.today().strftime(_DATE_DISPLAY_FORMAT)
    return templates.TemplateResponse("cookies.html", {"request": request, "current_date": current_date})


@router.get("/terms", include_in_schema=False)
async def serve_terms(request: Request):
    """Serve the terms of service page."""
    current_date = date.today().strftime(_DATE_DISPLAY_FORMAT)
    return templates.TemplateResponse("terms.html", {"request": request, "current_date": current_date})