7128ca727b
- Added various sound files to the soundboard, including insults, quotes, and sound effects. - Introduced new dependencies: ffmpeg-python (version 0.2.0) and pydub (version 0.25.1) for audio processing. - Updated the lock file to reflect the new packages and their respective versions. - Added .gitignore files in the sounds/stream and sounds/temp directories to exclude unnecessary files.
158 lines
4.5 KiB
Python
158 lines
4.5 KiB
Python
"""Main routes for the application."""
|
|
|
|
from flask import Blueprint, request
|
|
|
|
from app.services.decorators import (
|
|
get_current_user,
|
|
require_auth,
|
|
require_credits,
|
|
require_role,
|
|
)
|
|
from app.services.scheduler_service import scheduler_service
|
|
from app.services.sound_normalizer_service import SoundNormalizerService
|
|
from app.services.sound_scanner_service import SoundScannerService
|
|
|
|
bp = Blueprint("main", __name__)
|
|
|
|
|
|
@bp.route("/")
|
|
def index() -> dict[str, str]:
|
|
"""Root endpoint that returns API status."""
|
|
return {"message": "API is running", "status": "ok"}
|
|
|
|
|
|
@bp.route("/protected")
|
|
@require_auth
|
|
def protected() -> dict[str, str]:
|
|
"""Protected endpoint that requires authentication."""
|
|
user = get_current_user()
|
|
return {
|
|
"message": f"Hello {user['name']}, this is a protected endpoint!",
|
|
"user": user,
|
|
}
|
|
|
|
|
|
@bp.route("/api-protected")
|
|
@require_auth
|
|
def api_protected() -> dict[str, str]:
|
|
"""Protected endpoint that accepts JWT or API token authentication."""
|
|
user = get_current_user()
|
|
return {
|
|
"message": f"Hello {user['name']}, you accessed this via {user['provider']}!",
|
|
"user": user,
|
|
}
|
|
|
|
|
|
@bp.route("/admin")
|
|
@require_auth
|
|
@require_role("admin")
|
|
def admin_only() -> dict[str, str]:
|
|
"""Admin-only endpoint to demonstrate role-based access."""
|
|
user = get_current_user()
|
|
return {
|
|
"message": f"Hello admin {user['name']}, you have admin access!",
|
|
"user": user,
|
|
"admin_info": "This endpoint is only accessible to admin users",
|
|
}
|
|
|
|
|
|
@bp.route("/health")
|
|
def health() -> dict[str, str]:
|
|
"""Health check endpoint."""
|
|
return {"status": "ok"}
|
|
|
|
|
|
@bp.route("/use-credits/<int:amount>")
|
|
@require_auth
|
|
@require_credits(5)
|
|
def use_credits(amount: int) -> dict[str, str]:
|
|
"""Test endpoint that costs 5 credits to use."""
|
|
user = get_current_user()
|
|
return {
|
|
"message": f"Successfully used endpoint! You requested amount: {amount}",
|
|
"user": user["email"],
|
|
"remaining_credits": user["credits"]
|
|
- 5, # Note: credits already deducted by decorator
|
|
}
|
|
|
|
|
|
@bp.route("/expensive-operation")
|
|
@require_auth
|
|
@require_credits(10)
|
|
def expensive_operation() -> dict[str, str]:
|
|
"""Test endpoint that costs 10 credits to use."""
|
|
user = get_current_user()
|
|
return {
|
|
"message": "Expensive operation completed successfully!",
|
|
"user": user["email"],
|
|
"operation_cost": 10,
|
|
}
|
|
|
|
|
|
@bp.route("/admin/scheduler/status")
|
|
@require_auth
|
|
@require_role("admin")
|
|
def scheduler_status() -> dict:
|
|
"""Get scheduler status (admin only)."""
|
|
return scheduler_service.get_scheduler_status()
|
|
|
|
|
|
@bp.route("/admin/credits/refill", methods=["POST"])
|
|
@require_auth
|
|
@require_role("admin")
|
|
def manual_credit_refill() -> dict:
|
|
"""Manually trigger credit refill for all users (admin only)."""
|
|
return scheduler_service.trigger_credit_refill_now()
|
|
|
|
|
|
@bp.route("/admin/sounds/scan", methods=["POST"])
|
|
@require_auth
|
|
@require_role("admin")
|
|
def manual_sound_scan() -> dict:
|
|
"""Manually trigger sound directory scan (admin only)."""
|
|
return scheduler_service.trigger_sound_scan_now()
|
|
|
|
|
|
@bp.route("/admin/sounds/stats")
|
|
@require_auth
|
|
@require_role("admin")
|
|
def sound_statistics() -> dict:
|
|
"""Get sound database statistics (admin only)."""
|
|
return SoundScannerService.get_scan_statistics()
|
|
|
|
|
|
@bp.route("/admin/sounds/normalize/<int:sound_id>", methods=["POST"])
|
|
@require_auth
|
|
@require_role("admin")
|
|
def normalize_sound(sound_id: int) -> dict:
|
|
"""Normalize a specific sound file (admin only)."""
|
|
overwrite = request.args.get("overwrite", "false").lower() == "true"
|
|
return SoundNormalizerService.normalize_sound(sound_id, overwrite)
|
|
|
|
|
|
@bp.route("/admin/sounds/normalize-all", methods=["POST"])
|
|
@require_auth
|
|
@require_role("admin")
|
|
def normalize_all_sounds() -> dict:
|
|
"""Normalize all soundboard files (admin only)."""
|
|
overwrite = request.args.get("overwrite", "false").lower() == "true"
|
|
limit_str = request.args.get("limit")
|
|
limit = int(limit_str) if limit_str else None
|
|
return SoundNormalizerService.normalize_all_sounds(overwrite, limit)
|
|
|
|
|
|
@bp.route("/admin/sounds/normalization-status")
|
|
@require_auth
|
|
@require_role("admin")
|
|
def normalization_status() -> dict:
|
|
"""Get normalization status statistics (admin only)."""
|
|
return SoundNormalizerService.get_normalization_status()
|
|
|
|
|
|
@bp.route("/admin/sounds/ffmpeg-check")
|
|
@require_auth
|
|
@require_role("admin")
|
|
def ffmpeg_check() -> dict:
|
|
"""Check ffmpeg availability and capabilities (admin only)."""
|
|
return SoundNormalizerService.check_ffmpeg_availability()
|