from flask import Flask, request, jsonify, send_from_directory, redirect
from flask_cors import CORS
from loguru import logger
import sys
from pathlib import Path

from config import Config
from scheduler import ClickScheduler

# 配置日志
Config.ensure_dirs()
logger.remove()
logger.add(
    sys.stdout,
    format="<green>{time:YYYY-MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>",
    level="INFO"
)
logger.add(
    Path(Config.LOG_DIR) / "mip_ad_service_{time}.log",
    rotation="500 MB",
    retention="10 days",
    encoding="utf-8",
    level="DEBUG"
)

# 创建Flask应用
app = Flask(__name__, static_folder='static', static_url_path='')
CORS(app)

# 创建调度器实例
scheduler = ClickScheduler()


@app.route('/')
def index():
    """首页 - 重定向到新的单页应用"""
    return redirect('/app.html')


@app.route('/health', methods=['GET'])
def health():
    """健康检查"""
    return jsonify({'status': 'ok', 'message': '服务运行正常'})


@app.route('/api/urls', methods=['POST'])
def add_urls():
    """添加URL（支持单个或批量）"""
    try:
        data = request.get_json()
        
        if not data:
            return jsonify({'success': False, 'message': '请求数据为空'}), 400
        
        # 支持单个URL或URL列表
        if 'url' in data:
            # 单个URL
            url = data['url']
            if not url:
                return jsonify({'success': False, 'message': 'URL不能为空'}), 400
            
            success = scheduler.add_url(url)
            if success:
                return jsonify({'success': True, 'message': '添加成功'})
            else:
                return jsonify({'success': False, 'message': 'URL已存在或添加失败'}), 400
        
        elif 'urls' in data:
            # 批量URL
            urls = data['urls']
            if not isinstance(urls, list) or not urls:
                return jsonify({'success': False, 'message': 'URLs必须是非空列表'}), 400
            
            count = scheduler.add_urls(urls)
            return jsonify({
                'success': True,
                'message': f'成功添加 {count}/{len(urls)} 个URL',
                'added_count': count,
                'total_count': len(urls)
            })
        
        else:
            return jsonify({'success': False, 'message': '请提供url或urls参数'}), 400
    
    except Exception as e:
        logger.error(f"添加URL异常: {str(e)}")
        return jsonify({'success': False, 'message': f'服务异常: {str(e)}'}), 500


@app.route('/api/urls', methods=['GET'])
def get_urls():
    """获取所有URL列表"""
    try:
        urls = scheduler.data_manager.get_all_urls()
        return jsonify({'success': True, 'data': urls})
    
    except Exception as e:
        logger.error(f"获取URL列表异常: {str(e)}")
        return jsonify({'success': False, 'message': f'服务异常: {str(e)}'}), 500


@app.route('/api/urls/<path:url>', methods=['GET'])
def get_url_detail(url: str):
    """获取URL详细信息"""
    try:
        url_info = scheduler.get_url_detail(url)
        
        if url_info:
            return jsonify({'success': True, 'data': url_info})
        else:
            return jsonify({'success': False, 'message': 'URL不存在'}), 404
    
    except Exception as e:
        logger.error(f"获取URL详情异常: {str(e)}")
        return jsonify({'success': False, 'message': f'服务异常: {str(e)}'}), 500


@app.route('/api/urls/<path:url>', methods=['DELETE'])
def delete_url(url: str):
    """删除URL"""
    try:
        success = scheduler.data_manager.delete_url(url)
        
        if success:
            return jsonify({'success': True, 'message': '删除成功'})
        else:
            return jsonify({'success': False, 'message': 'URL不存在'}), 404
    
    except Exception as e:
        logger.error(f"删除URL异常: {str(e)}")
        return jsonify({'success': False, 'message': f'服务异常: {str(e)}'}), 500


@app.route('/api/urls/<path:url>/reset', methods=['POST'])
def reset_url(url: str):
    """重置URL（重新开始点击）"""
    try:
        success = scheduler.data_manager.reset_url(url)
        
        if success:
            return jsonify({'success': True, 'message': '重置成功'})
        else:
            return jsonify({'success': False, 'message': 'URL不存在'}), 404
    
    except Exception as e:
        logger.error(f"重置URL异常: {str(e)}")
        return jsonify({'success': False, 'message': f'服务异常: {str(e)}'}), 500


@app.route('/api/statistics', methods=['GET'])
def get_statistics():
    """获取统计数据"""
    try:
        stats = scheduler.get_statistics()
        return jsonify({'success': True, 'data': stats})
    
    except Exception as e:
        logger.error(f"获取统计数据异常: {str(e)}")
        return jsonify({'success': False, 'message': f'服务异常: {str(e)}'}), 500


@app.route('/api/scheduler/start', methods=['POST'])
def start_scheduler():
    """启动调度器"""
    try:
        scheduler.start_scheduler()
        return jsonify({'success': True, 'message': '调度器已启动'})
    
    except Exception as e:
        logger.error(f"启动调度器异常: {str(e)}")
        return jsonify({'success': False, 'message': f'服务异常: {str(e)}'}), 500


@app.route('/api/scheduler/stop', methods=['POST'])
def stop_scheduler():
    """停止调度器"""
    try:
        scheduler.stop_scheduler()
        return jsonify({'success': True, 'message': '调度器已停止'})
    
    except Exception as e:
        logger.error(f"停止调度器异常: {str(e)}")
        return jsonify({'success': False, 'message': f'服务异常: {str(e)}'}), 500


@app.route('/api/scheduler/status', methods=['GET'])
def get_scheduler_status():
    """获取调度器状态"""
    try:
        status = 'running' if scheduler.running else 'stopped'
        return jsonify({'success': True, 'data': {'status': status}})
    
    except Exception as e:
        logger.error(f"获取调度器状态异常: {str(e)}")
        return jsonify({'success': False, 'message': f'服务异常: {str(e)}'}), 500


# AdsPower 接口调试
@app.route('/api/adspower/groups', methods=['GET'])
def adspower_list_groups():
    """查询分组列表"""
    try:
        from adspower_client import AdsPowerClient
        
        group_name = request.args.get('group_name')
        page = request.args.get('page', 1, type=int)
        page_size = request.args.get('page_size', 2000, type=int)
        
        client = AdsPowerClient()
        result = client.list_groups(group_name=group_name, page=page, page_size=page_size)
        
        if result:
            return jsonify({'success': True, 'data': result})
        else:
            return jsonify({'success': False, 'message': '查询分组列表失败'}), 500
    except Exception as e:
        logger.error(f"AdsPower查询分组异常: {str(e)}")
        return jsonify({'success': False, 'message': str(e)}), 500


@app.route('/api/adspower/group/env', methods=['GET'])
def adspower_get_group_by_env():
    """根据当前运行环境获取对应的分组ID"""
    try:
        from adspower_client import AdsPowerClient
        
        client = AdsPowerClient()
        group_id = client.get_group_by_env()
        
        if group_id:
            return jsonify({'success': True, 'data': {'group_id': group_id, 'env': Config.ENV}})
        else:
            return jsonify({'success': False, 'message': f'未找到对应环境的分组'}), 404
    except Exception as e:
        logger.error(f"AdsPower获取环境分组异常: {str(e)}")
        return jsonify({'success': False, 'message': str(e)}), 500


@app.route('/api/adspower/profiles', methods=['GET'])
def adspower_list_profiles():
    """查询Profile列表（支持多个查询参数）"""
    try:
        from adspower_client import AdsPowerClient
        import json as json_module
        
        # 获取查询参数
        group_id = request.args.get('group_id')
        page = request.args.get('page', 1, type=int)
        limit = request.args.get('limit', type=int)  # 可选
        page_size = request.args.get('page_size', type=int)  # 可选
        
        # 数组参数（JSON格式）
        profile_id = request.args.get('profile_id')
        profile_no = request.args.get('profile_no')
        
        # 解析JSON数组
        if profile_id:
            try:
                profile_id = json_module.loads(profile_id)
            except:
                profile_id = None
        
        if profile_no:
            try:
                profile_no = json_module.loads(profile_no)
            except:
                profile_no = None
        
        # 排序参数
        sort_type = request.args.get('sort_type')
        sort_order = request.args.get('sort_order')
        
        # 如果没有指定group_id，尝试根据环境自动获取
        client = AdsPowerClient()
        if not group_id:
            group_id = client.get_group_by_env()
            if group_id:
                logger.info(f"自动获取到分组ID: {group_id}")
        
        # 查询Profile列表
        result = client.list_profiles(
            group_id=group_id,
            page=page,
            page_size=page_size if page_size else 100,
            profile_id=profile_id,
            profile_no=profile_no,
            limit=limit,
            sort_type=sort_type,
            sort_order=sort_order
        )
        
        if result:
            return jsonify({'success': True, 'data': result})
        else:
            return jsonify({'success': False, 'message': '查询Profile列表失败'}), 500
    except Exception as e:
        logger.error(f"AdsPower查询Profile异常: {str(e)}")
        return jsonify({'success': False, 'message': str(e)}), 500


@app.route('/api/adspower/browser/start', methods=['POST'])
def adspower_start_browser():
    """启动浏览器"""
    try:
        from adspower_client import AdsPowerClient
        data = request.get_json() or {}
        user_id = data.get('user_id')
        
        client = AdsPowerClient()
        result = client.start_browser(user_id=user_id)
        return jsonify({'success': True, 'data': result})
    except Exception as e:
        logger.error(f"AdsPower启动浏览器异常: {str(e)}")
        return jsonify({'success': False, 'message': str(e)}), 500


@app.route('/api/adspower/browser/stop', methods=['POST'])
def adspower_stop_browser():
    """停止浏览器"""
    try:
        from adspower_client import AdsPowerClient
        data = request.get_json() or {}
        user_id = data.get('user_id')
        
        client = AdsPowerClient()
        result = client.stop_browser(user_id=user_id)
        return jsonify({'success': True, 'data': result})
    except Exception as e:
        logger.error(f"AdsPower停止浏览器异常: {str(e)}")
        return jsonify({'success': False, 'message': str(e)}), 500


@app.route('/api/adspower/proxy/damai', methods=['GET'])
def adspower_get_damai_proxy():
    """获取大麦IP代理"""
    try:
        from adspower_client import AdsPowerClient
        client = AdsPowerClient()
        result = client.get_damai_proxy()
        return jsonify({'success': True, 'data': result})
    except Exception as e:
        logger.error(f"获取大麦IP异常: {str(e)}")
        return jsonify({'success': False, 'message': str(e)}), 500


@app.route('/api/adspower/proxy/create', methods=['POST'])
def adspower_create_proxy():
    """创建代理"""
    try:
        from adspower_client import AdsPowerClient
        data = request.get_json() or {}
        proxy_config = data.get('proxy_config')
        
        if not proxy_config:
            return jsonify({'success': False, 'message': '缺少代理配置'}), 400
        
        client = AdsPowerClient()
        proxy_id = client.create_proxy(proxy_config)
        return jsonify({'success': True, 'data': {'proxy_id': proxy_id}})
    except Exception as e:
        logger.error(f"AdsPower创建代理异常: {str(e)}")
        return jsonify({'success': False, 'message': str(e)}), 500


@app.route('/api/adspower/proxy/list', methods=['GET'])
def adspower_list_proxies():
    """查询代理列表"""
    try:
        from adspower_client import AdsPowerClient
        
        page = request.args.get('page', 1, type=int)
        limit = request.args.get('limit', 100, type=int)
        
        client = AdsPowerClient()
        result = client.list_proxies(page=page, limit=limit)
        
        if result is None:
            return jsonify({'success': False, 'message': '查询代理列表失败，请检查AdsPower是否运行'}), 500
        
        return jsonify({'success': True, 'data': result})
    except Exception as e:
        logger.error(f"AdsPower查询代理列表异常: {str(e)}")
        return jsonify({'success': False, 'message': str(e)}), 500


@app.route('/api/adspower/profile/update', methods=['POST'])
def adspower_update_profile():
    """更新Profile代理（API v2方式）"""
    try:
        from adspower_client import AdsPowerClient
        data = request.get_json() or {}
        profile_id = data.get('profile_id')
        proxy_id = data.get('proxy_id')
        
        if not profile_id or not proxy_id:
            return jsonify({'success': False, 'message': '缺少profile_id或proxy_id'}), 400
        
        client = AdsPowerClient()
        result = client.update_profile_proxy(profile_id, proxy_id)
        return jsonify({'success': True, 'data': {'updated': result}})
    except Exception as e:
        logger.error(f"AdsPower更新Profile异常: {str(e)}")
        return jsonify({'success': False, 'message': str(e)}), 500


@app.route('/api/adspower/profile/update-v1', methods=['POST'])
def adspower_update_profile_v1():
    """更新Profile代理（API v1方式，直接传入proxy_config）"""
    try:
        from adspower_client import AdsPowerClient
        data = request.get_json() or {}
        profile_id = data.get('profile_id')
        proxy_config = data.get('proxy_config')
        
        if not profile_id or not proxy_config:
            return jsonify({'success': False, 'message': '缺少profile_id或proxy_config'}), 400
        
        # 验证必要字段
        if 'proxy_host' not in proxy_config or 'proxy_port' not in proxy_config:
            return jsonify({'success': False, 'message': 'proxy_config中缺少proxy_host或proxy_port'}), 400
        
        client = AdsPowerClient()
        result = client.update_profile_proxy_v1(profile_id, proxy_config)
        return jsonify({'success': True, 'data': {'updated': result}})
    except Exception as e:
        logger.error(f"AdsPower更新Profile异常 (v1): {str(e)}")
        return jsonify({'success': False, 'message': str(e)}), 500


# 数据库查询接口
@app.route('/api/clicks', methods=['GET'])
def get_all_clicks():
    """获取所有点击记录"""
    try:
        from db_manager import ClickManager
        
        site_id = request.args.get('site_id', type=int)
        limit = request.args.get('limit', 100, type=int)
        
        click_mgr = ClickManager()
        
        if site_id:
            clicks = click_mgr.get_clicks_by_site(site_id, limit=limit)
        else:
            # 获取所有点击记录
            conn = click_mgr.get_connection()
            cursor = conn.cursor()
            cursor.execute(f"""
                SELECT * FROM ai_mip_click 
                ORDER BY click_time DESC 
                LIMIT ?
            """, (limit,))
            rows = cursor.fetchall()
            conn.close()
            clicks = [click_mgr._dict_from_row(row) for row in rows]
        
        return jsonify({'success': True, 'data': clicks})
    except Exception as e:
        logger.error(f"查询点击记录异常: {str(e)}")
        return jsonify({'success': False, 'message': str(e)}), 500


@app.route('/api/interactions', methods=['GET'])
def get_all_interactions():
    """获取所有互动记录"""
    try:
        from db_manager import InteractionManager
        
        site_id = request.args.get('site_id', type=int)
        limit = request.args.get('limit', 100, type=int)
        
        interaction_mgr = InteractionManager()
        
        if site_id:
            interactions = interaction_mgr.get_interactions_by_site(site_id, limit=limit)
        else:
            # 获取所有互动记录
            conn = interaction_mgr.get_connection()
            cursor = conn.cursor()
            cursor.execute(f"""
                SELECT * FROM ai_mip_interaction 
                ORDER BY interaction_time DESC 
                LIMIT ?
            """, (limit,))
            rows = cursor.fetchall()
            conn.close()
            interactions = [interaction_mgr._dict_from_row(row) for row in rows]
        
        return jsonify({'success': True, 'data': interactions})
    except Exception as e:
        logger.error(f"查询互动记录异常: {str(e)}")
        return jsonify({'success': False, 'message': str(e)}), 500


if __name__ == '__main__':
    logger.info(f"启动MIP广告点击服务 - 环境: {Config.ENV}")
    logger.info(f"服务地址: http://{Config.SERVER_HOST}:{Config.SERVER_PORT}")
    logger.info(f"调试模式: {Config.DEBUG}")
    
    # 自动启动调度器
    scheduler.start_scheduler()
    
    # 启动Flask应用
    app.run(
        host=Config.SERVER_HOST,
        port=Config.SERVER_PORT,
        debug=Config.DEBUG
    )