"""
后台任务 API 路由
"""
from flask import Blueprint, jsonify, request
from services.task_service import task_service
from models.database import db
import logging

logger = logging.getLogger('tasks_api')

bp = Blueprint('tasks', __name__, url_prefix='/api/tasks')


@bp.route('/start', methods=['POST'])
def start_task():
    """
    启动后台抓取任务
    
    请求体:
    {
        "results": [{"title": "...", "url": "..."}, ...],
        "auto_save": true,
        "category": "AI模型",
        "keywords": ["关键词1", "关键词2"]
    }
    """
    try:
        data = request.get_json()
        
        if not data or 'results' not in data:
            return jsonify({
                'success': False,
                'error': '缺少 results 参数'
            }), 400
        
        results = data.get('results', [])
        auto_save = data.get('auto_save', False)
        category = data.get('category', '')
        keywords = data.get('keywords', [])
        
        if not results:
            return jsonify({
                'success': False,
                'error': '结果列表为空'
            }), 400
        
        # 启动后台任务
        task_id = task_service.start_fetch_task(
            results=results,
            auto_save=auto_save,
            category=category,
            keywords=keywords
        )
        
        return jsonify({
            'success': True,
            'task_id': task_id,
            'message': f'任务已启动，共 {len(results)} 个URL'
        })
        
    except Exception as e:
        logger.error(f"启动任务失败: {e}")
        return jsonify({
            'success': False,
            'error': str(e)
        }), 500


@bp.route('/<task_id>/status', methods=['GET'])
def get_task_status(task_id):
    """获取任务状态"""
    task = task_service.get_task_status(task_id)
    
    if not task:
        return jsonify({
            'success': False,
            'error': '任务不存在'
        }), 404
    
    return jsonify({
        'success': True,
        'task': task
    })


@bp.route('/<task_id>/stop', methods=['POST'])
def stop_task(task_id):
    """停止任务"""
    task = db.get_task(task_id)
    
    if not task:
        return jsonify({
            'success': False,
            'error': '任务不存在'
        }), 404
    
    if task['status'] not in ('pending', 'running'):
        return jsonify({
            'success': False,
            'error': '任务已完成或已停止'
        }), 400
    
    # 设置停止标志
    if task_service.stop_task(task_id):
        return jsonify({
            'success': True,
            'message': '已请求停止任务'
        })
    else:
        return jsonify({
            'success': False,
            'error': '停止任务失败'
        }), 500


@bp.route('/active', methods=['GET'])
def get_active_tasks():
    """获取活动任务列表"""
    tasks = task_service.get_active_tasks()
    
    return jsonify({
        'success': True,
        'tasks': tasks,
        'count': len(tasks)
    })


@bp.route('/recent', methods=['GET'])
def get_recent_tasks():
    """获取最近的任务"""
    limit = request.args.get('limit', 20, type=int)
    tasks = db.get_recent_tasks(limit)
    
    return jsonify({
        'success': True,
        'tasks': tasks,
        'count': len(tasks)
    })


@bp.route('/<task_id>', methods=['DELETE'])
def delete_task(task_id):
    """删除任务记录"""
    task = db.get_task(task_id)
    
    if not task:
        return jsonify({
            'success': False,
            'error': '任务不存在'
        }), 404
    
    if task['status'] in ('pending', 'running'):
        return jsonify({
            'success': False,
            'error': '任务仍在运行，请先停止'
        }), 400
    
    if db.delete_task(task_id):
        return jsonify({
            'success': True,
            'message': '任务已删除'
        })
    else:
        return jsonify({
            'success': False,
            'error': '删除失败'
        }), 500


@bp.route('/clear', methods=['POST'])
def clear_completed_tasks():
    """清理已完成的任务"""
    count = db.clear_completed_tasks()
    
    return jsonify({
        'success': True,
        'message': f'已清理 {count} 个任务'
    })