226 lines
8.2 KiB
Python
226 lines
8.2 KiB
Python
from fastapi import APIRouter, HTTPException, Request
|
|
from fastapi.responses import JSONResponse
|
|
import json
|
|
import logging
|
|
import os
|
|
from typing import Any
|
|
|
|
# Import the centralized config getters that handle file creation and defaults
|
|
from routes.utils.celery_config import (
|
|
get_config_params as get_main_config_params,
|
|
DEFAULT_MAIN_CONFIG,
|
|
CONFIG_FILE_PATH as MAIN_CONFIG_FILE_PATH,
|
|
)
|
|
from routes.utils.watch.manager import (
|
|
get_watch_config as get_watch_manager_config,
|
|
DEFAULT_WATCH_CONFIG,
|
|
CONFIG_FILE_PATH as WATCH_CONFIG_FILE_PATH,
|
|
)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
# Flag for config change notifications
|
|
config_changed = False
|
|
last_config: dict[str, Any] = {}
|
|
|
|
# Define parameters that should trigger notification when changed
|
|
NOTIFY_PARAMETERS = [
|
|
"maxConcurrentDownloads",
|
|
"service",
|
|
"fallback",
|
|
"spotifyQuality",
|
|
"deezerQuality",
|
|
]
|
|
|
|
|
|
# Helper to get main config (uses the one from celery_config)
|
|
def get_config():
|
|
"""Retrieves the main configuration, creating it with defaults if necessary."""
|
|
return get_main_config_params()
|
|
|
|
|
|
# Helper to save main config
|
|
def save_config(config_data):
|
|
"""Saves the main configuration data to main.json."""
|
|
try:
|
|
MAIN_CONFIG_FILE_PATH.parent.mkdir(parents=True, exist_ok=True)
|
|
# Load current or default config
|
|
existing_config = {}
|
|
if MAIN_CONFIG_FILE_PATH.exists():
|
|
with open(MAIN_CONFIG_FILE_PATH, "r") as f_read:
|
|
existing_config = json.load(f_read)
|
|
else: # Should be rare if get_config_params was called
|
|
existing_config = DEFAULT_MAIN_CONFIG.copy()
|
|
|
|
# Update with new data
|
|
for key, value in config_data.items():
|
|
existing_config[key] = value
|
|
|
|
# Ensure all default keys are still there
|
|
for default_key, default_value in DEFAULT_MAIN_CONFIG.items():
|
|
if default_key not in existing_config:
|
|
existing_config[default_key] = default_value
|
|
|
|
with open(MAIN_CONFIG_FILE_PATH, "w") as f:
|
|
json.dump(existing_config, f, indent=4)
|
|
logger.info(f"Main configuration saved to {MAIN_CONFIG_FILE_PATH}")
|
|
return True, None
|
|
except Exception as e:
|
|
logger.error(f"Error saving main configuration: {e}", exc_info=True)
|
|
return False, str(e)
|
|
|
|
|
|
# Helper to get watch config (uses the one from watch/manager.py)
|
|
def get_watch_config_http(): # Renamed to avoid conflict with the imported get_watch_config
|
|
"""Retrieves the watch configuration, creating it with defaults if necessary."""
|
|
return get_watch_manager_config()
|
|
|
|
|
|
# Helper to save watch config
|
|
def save_watch_config_http(watch_config_data): # Renamed
|
|
"""Saves the watch configuration data to watch.json."""
|
|
try:
|
|
WATCH_CONFIG_FILE_PATH.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
# Similar logic to save_config: merge with defaults/existing
|
|
existing_config = {}
|
|
if WATCH_CONFIG_FILE_PATH.exists():
|
|
with open(WATCH_CONFIG_FILE_PATH, "r") as f_read:
|
|
existing_config = json.load(f_read)
|
|
else: # Should be rare if get_watch_manager_config was called
|
|
existing_config = DEFAULT_WATCH_CONFIG.copy()
|
|
|
|
for key, value in watch_config_data.items():
|
|
existing_config[key] = value
|
|
|
|
for default_key, default_value in DEFAULT_WATCH_CONFIG.items():
|
|
if default_key not in existing_config:
|
|
existing_config[default_key] = default_value
|
|
|
|
with open(WATCH_CONFIG_FILE_PATH, "w") as f:
|
|
json.dump(existing_config, f, indent=4)
|
|
logger.info(f"Watch configuration saved to {WATCH_CONFIG_FILE_PATH}")
|
|
return True, None
|
|
except Exception as e:
|
|
logger.error(f"Error saving watch configuration: {e}", exc_info=True)
|
|
return False, str(e)
|
|
|
|
|
|
@router.get("/config")
|
|
async def handle_config():
|
|
"""Handles GET requests for the main configuration."""
|
|
try:
|
|
config = get_config()
|
|
return config
|
|
except Exception as e:
|
|
logger.error(f"Error in GET /config: {e}", exc_info=True)
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail={"error": "Failed to retrieve configuration", "details": str(e)}
|
|
)
|
|
|
|
|
|
@router.post("/config")
|
|
@router.put("/config")
|
|
async def update_config(request: Request):
|
|
"""Handles POST/PUT requests to update the main configuration."""
|
|
try:
|
|
new_config = await request.json()
|
|
if not isinstance(new_config, dict):
|
|
raise HTTPException(status_code=400, detail={"error": "Invalid config format"})
|
|
|
|
# Preserve the explicitFilter setting from environment
|
|
explicit_filter_env = os.environ.get("EXPLICIT_FILTER", "false").lower()
|
|
new_config["explicitFilter"] = explicit_filter_env in ("true", "1", "yes", "on")
|
|
|
|
success, error_msg = save_config(new_config)
|
|
if success:
|
|
# Return the updated config
|
|
updated_config_values = get_config()
|
|
if updated_config_values is None:
|
|
# This case should ideally not be reached if save_config succeeded
|
|
# and get_config handles errors by returning a default or None.
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail={"error": "Failed to retrieve configuration after saving"}
|
|
)
|
|
|
|
return updated_config_values
|
|
else:
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail={"error": "Failed to update configuration", "details": error_msg}
|
|
)
|
|
except json.JSONDecodeError:
|
|
raise HTTPException(status_code=400, detail={"error": "Invalid JSON data"})
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
logger.error(f"Error in POST/PUT /config: {e}", exc_info=True)
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail={"error": "Failed to update configuration", "details": str(e)}
|
|
)
|
|
|
|
|
|
@router.get("/config/check")
|
|
async def check_config_changes():
|
|
# This endpoint seems more related to dynamically checking if config changed
|
|
# on disk, which might not be necessary if settings are applied on restart
|
|
# or by a dedicated manager. For now, just return current config.
|
|
try:
|
|
config = get_config()
|
|
return {"message": "Current configuration retrieved.", "config": config}
|
|
except Exception as e:
|
|
logger.error(f"Error in GET /config/check: {e}", exc_info=True)
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail={"error": "Failed to check configuration", "details": str(e)}
|
|
)
|
|
|
|
|
|
@router.get("/config/watch")
|
|
async def handle_watch_config():
|
|
"""Handles GET requests for the watch configuration."""
|
|
try:
|
|
watch_config = get_watch_config_http()
|
|
return watch_config
|
|
except Exception as e:
|
|
logger.error(f"Error in GET /config/watch: {e}", exc_info=True)
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail={"error": "Failed to retrieve watch configuration", "details": str(e)}
|
|
)
|
|
|
|
|
|
@router.post("/config/watch")
|
|
@router.put("/config/watch")
|
|
async def update_watch_config(request: Request):
|
|
"""Handles POST/PUT requests to update the watch configuration."""
|
|
try:
|
|
new_watch_config = await request.json()
|
|
if not isinstance(new_watch_config, dict):
|
|
raise HTTPException(status_code=400, detail={"error": "Invalid watch config format"})
|
|
|
|
success, error_msg = save_watch_config_http(new_watch_config)
|
|
if success:
|
|
return {"message": "Watch configuration updated successfully"}
|
|
else:
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail={"error": "Failed to update watch configuration", "details": error_msg}
|
|
)
|
|
except json.JSONDecodeError:
|
|
raise HTTPException(status_code=400, detail={"error": "Invalid JSON data for watch config"})
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
logger.error(f"Error in POST/PUT /config/watch: {e}", exc_info=True)
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail={"error": "Failed to update watch configuration", "details": str(e)}
|
|
)
|