"""
AI模型 CRUD API
"""
import uuid
import json
from datetime import datetime
from flask import Blueprint, request, jsonify, Response
from config import MODELS_FILE
from utils import load_data, save_data, parse_date_to_timestamp, safe_sort_key

models_bp = Blueprint('api_models', __name__)


def is_review_required():
    """检查是否需要审核"""
    try:
        import app as app_module
        return getattr(app_module, 'REQUIRE_REVIEW', False)
    except:
        return False


@models_bp.route('/api/models')
def api_models():
    models = load_data(MODELS_FILE)
    hide_hidden = request.args.get('all', '0') == '0'
    if hide_hidden:
        models = [m for m in models if m.get('visible', True)]
    keyword = request.args.get('q', '').strip().lower()
    if keyword:
        models = [m for m in models if keyword in m.get('name', '').lower() or
                  keyword in m.get('organization', '').lower()]
    sort_by = request.args.get('sort', 'default')
    reverse = request.args.get('order', 'desc') == 'desc'
    if sort_by == 'default':
        models = sorted(models, key=lambda x: (
            not x.get('is_pinned', False),
            -(parse_date_to_timestamp(x.get('publish_date', '')) or
              parse_date_to_timestamp(x.get('created_at', '')) or 0)
        ))
    elif sort_by in ['name', 'parameters', 'context_length', 'mmlu', 'created_at',
                      'publish_date', 'views', 'updated_at']:
        models = sorted(models, key=lambda x: safe_sort_key(x, sort_by), reverse=reverse)
    return jsonify(models)


@models_bp.route('/api/models/<model_id>')
def api_model_detail(model_id):
    models = load_data(MODELS_FILE)
    model = next((m for m in models if m['id'] == model_id), None)
    if not model:
        return jsonify({'error': 'Model not found'}), 404
    return jsonify(model)


@models_bp.route('/api/models', methods=['POST'])
def api_create_model():
    data = request.get_json()
    
    # 审核模式：提交到审核队列
    if is_review_required():
        from modules.routes.api_reviews import submit_for_review
        from modules.routes.api_notifications import create_notification
        
        review = submit_for_review('ai-models', data, source='web')
        product_name = data.get('name', '未知')
        create_notification(
            title='新产品待审核',
            message=f'有新的AI模型"{product_name}"待审核',
            level='warning',
            category='review',
            data={'review_id': review['id'], 'category': 'ai-models'}
        )
        return jsonify({'success': True, 'message': '已提交审核，请等待管理员确认', 'review_id': review['id']})
    
    # 直接创建模式
    models = load_data(MODELS_FILE)
    data['id'] = uuid.uuid4().hex[:12]
    data['created_at'] = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
    data['visible'] = data.get('visible', True)
    data['publish_date'] = data.get('publish_date', '')
    data['views'] = data.get('views', 0)
    data['is_pinned'] = data.get('is_pinned', False)
    models.append(data)
    save_data(MODELS_FILE, models)
    return jsonify(data)


@models_bp.route('/api/models/<model_id>', methods=['PUT'])
def api_update_model(model_id):
    data = request.get_json()
    models = load_data(MODELS_FILE)
    model = next((m for m in models if m['id'] == model_id), None)
    if not model:
        return jsonify({'error': 'Model not found'}), 404
    model.update(data)
    model['updated_at'] = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
    save_data(MODELS_FILE, models)
    return jsonify(model)


@models_bp.route('/api/models/<model_id>', methods=['DELETE'])
def api_delete_model(model_id):
    models = load_data(MODELS_FILE)
    models = [m for m in models if m['id'] != model_id]
    save_data(MODELS_FILE, models)
    return jsonify({'success': True})


@models_bp.route('/api/models/<model_id>/visible', methods=['POST'])
def api_toggle_model_visible(model_id):
    models = load_data(MODELS_FILE)
    model = next((m for m in models if m['id'] == model_id), None)
    if not model:
        return jsonify({'error': 'Model not found'}), 404
    model['visible'] = not model.get('visible', True)
    save_data(MODELS_FILE, models)
    return jsonify({'success': True, 'visible': model['visible']})


# ─── 导出导入 ───────────────────────────────────────────────────────

@models_bp.route('/api/models/export', methods=['GET'])
def api_export_models():
    """导出所有模型数据"""
    try:
        models = load_data(MODELS_FILE)
        export_data = {
            'type': 'models',
            'items': models,
            'count': len(models),
            'export_time': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
            'version': '1.0'
        }
        json_str = json.dumps(export_data, ensure_ascii=False, indent=2)
        response = Response(
            json_str,
            mimetype='application/json',
            headers={'Content-Disposition': f'attachment; filename=models-export-{datetime.now().strftime("%Y%m%d%H%M%S")}.json'}
        )
        return response
    except Exception as e:
        return jsonify({'error': str(e)}), 500


@models_bp.route('/api/models/import', methods=['GET', 'POST'])
def api_import_models():
    """导入模型数据"""
    try:
        if request.method == 'GET':
            return jsonify({'endpoint': '/api/models/import', 'method': 'POST', 'params': {'mode': 'merge 或 replace'}})
        
        import_data = request.get_json()
        if not import_data or 'items' not in import_data:
            return jsonify({'error': '缺少 items 字段'}), 400
        
        mode = request.args.get('mode', 'merge')
        models = load_data(MODELS_FILE)
        imported_items = import_data['items']
        
        result = {'success': True, 'imported': 0, 'updated': 0, 'skipped': []}
        
        for item in imported_items:
            existing = next((m for m in models if m['id'] == item['id']), None)
            if existing:
                if mode == 'replace':
                    existing.update(item)
                    result['updated'] += 1
                else:
                    result['skipped'].append(item['id'])
            else:
                models.append(item)
                result['imported'] += 1
        
        save_data(MODELS_FILE, models)
        return jsonify(result)
    except Exception as e:
        return jsonify({'error': str(e)}), 500