profilarr/backend/app/arr/__init__.py

from flask import Blueprint, request, jsonify
from flask_cors import cross_origin
import logging
from .status.ping import ping_service
from .manager import (save_arr_config, get_all_arr_configs, get_arr_config,
                      update_arr_config, delete_arr_config)

logger = logging.getLogger(__name__)
logger.setLevel(logging.ERROR)

bp = Blueprint('arr', __name__)


@bp.route('/ping', methods=['POST', 'OPTIONS'])
@cross_origin()
def ping():
    if request.method == 'OPTIONS':
        return jsonify({}), 200
    data = request.get_json()
    url = data.get('url')
    api_key = data.get('apiKey')
    arr_type = data.get('type')

    if not url or not api_key or not arr_type:
        return jsonify({
            'success': False,
            'error': 'URL, API key, and type are required'
        }), 400

    logger.error(f"Attempting to ping URL: {url} of type: {arr_type}")
    success, message = ping_service(url, api_key, arr_type)
    logger.error(f"Ping result - Success: {success}, Message: {message}")

    return jsonify({
        'success': success,
        'message': message
    }), 200 if success else 400


@bp.route('/config', methods=['POST', 'OPTIONS'])
@cross_origin()
def add_config():
    if request.method == 'OPTIONS':
        return jsonify({}), 200

    try:
        config = request.json
        
        # Validate sync_interval if schedule method
        if config.get('sync_method') == 'schedule':
            sync_interval = config.get('sync_interval', 0)
            if sync_interval < 60 or sync_interval > 43200:
                return jsonify({
                    'success': False,
                    'error': 'Sync interval must be between 60 minutes (1 hour) and 43200 minutes (1 month)'
                }), 400
        
        result = save_arr_config(config)

        # Handle the conflict case first
        if not result['success'] and result.get('status_code') == 409:
            return jsonify({'success': False, 'error': result['error']}), 409

        # Handle other failure cases
        if not result['success']:
            return jsonify(result), 400

        return jsonify(result), 200
    except Exception as e:
        logger.error(f"Error saving arr config: {str(e)}")
        return jsonify({'success': False, 'error': str(e)}), 400


@bp.route('/config', methods=['GET', 'OPTIONS'])
@cross_origin()
def get_configs():
    if request.method == 'OPTIONS':
        return jsonify({}), 200

    try:
        configs = get_all_arr_configs()
        logger.debug(f"Retrieved {len(configs)} arr configs")
        return jsonify(configs), 200
    except Exception as e:
        logger.error(f"Error getting arr configs: {str(e)}")
        return jsonify({'success': False, 'error': str(e)}), 400


@bp.route('/config/<int:id>', methods=['GET', 'PUT', 'DELETE', 'OPTIONS'])
@cross_origin()
def handle_config(id):
    if request.method == 'OPTIONS':
        return jsonify({}), 200

    try:
        if request.method == 'GET':
            config = get_arr_config(id)
            if config:
                logger.debug(f"Retrieved arr config: {id}")
                return jsonify({'success': True, 'data': config}), 200
            logger.debug(f"Arr config not found: {id}")
            return jsonify({
                'success': False,
                'error': 'Config not found'
            }), 404

        elif request.method == 'PUT':
            config = request.json
            
            # Validate sync_interval if schedule method
            if config.get('sync_method') == 'schedule':
                sync_interval = config.get('sync_interval', 0)
                if sync_interval < 60 or sync_interval > 43200:
                    return jsonify({
                        'success': False,
                        'error': 'Sync interval must be between 60 minutes (1 hour) and 43200 minutes (1 month)'
                    }), 400
            
            result = update_arr_config(id, config)

            # Handle the conflict case first
            if not result['success'] and result.get('status_code') == 409:
                return jsonify({
                    'success': False,
                    'error': result['error']
                }), 409

            # Handle other failure cases
            if not result['success']:
                logger.debug(f"Arr config not found for update: {id}")
                return jsonify({
                    'success': False,
                    'error': 'Config not found'
                }), 404

            logger.debug(f"Updated arr config: {id}")
            return jsonify({'success': True}), 200

        elif request.method == 'DELETE':
            success = delete_arr_config(id)
            if success:
                logger.debug(f"Deleted arr config: {id}")
                return jsonify({'success': True}), 200
            logger.debug(f"Arr config not found for deletion: {id}")
            return jsonify({
                'success': False,
                'error': 'Config not found'
            }), 404

    except Exception as e:
        logger.error(f"Error handling arr config {id}: {str(e)}")
        return jsonify({'success': False, 'error': str(e)}), 400


@bp.route('/config/<int:id>/sync', methods=['POST', 'OPTIONS'])
@cross_origin()
def trigger_sync(id):
    if request.method == 'OPTIONS':
        return jsonify({}), 200

    try:
        # Get the config first
        config_result = get_arr_config(id)
        if not config_result.get('success'):
            logger.error(f"Config not found for sync: {id}")
            return jsonify({
                'success': False,
                'error': 'Configuration not found'
            }), 404

        config_data = config_result.get('data')
        if not config_data:
            logger.error(f"Invalid config data for sync: {id}")
            return jsonify({
                'success': False,
                'error': 'Invalid configuration data'
            }), 400

        # Run the import
        from ..importer import handle_pull_import
        handle_pull_import(id)

        logger.debug(f"Manual sync triggered for arr config: {id}")
        return jsonify({'success': True}), 200

    except Exception as e:
        logger.error(f"Error triggering sync for arr config {id}: {str(e)}")
        return jsonify({'success': False, 'error': str(e)}), 400