"""
AI模型 CRUD API
"""
import uuid
from datetime import datetime
from flask import Blueprint, request, jsonify
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__)


@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()
    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']})