# -*- coding: utf-8 -*-
from datetime import datetime 
from flask import Flask, render_template
from flask_socketio import SocketIO, emit
from scriptBase.comon import *
import pyautogui
import base64
import threading
from dongri_task import *
from collections import deque
import json
from concurrent.futures import ThreadPoolExecutor

# 全局线程池，限制最大线程数为1
executor = ThreadPoolExecutor(max_workers=1)

app = Flask(__name__)
socketio = SocketIO(app, cors_allowed_origins="*")
event = threading.Event()
g_status_list = []
last_time = 0.0
task_queue = deque()
last_process = ''
isGameBegin = True
autoTask = None
isReset = False

def thread_runTask():
    global last_process
    global task_queue,isReset
    while True:
        if event.is_set():
            task_queue.clear()
        if len(task_queue) != 0:
            task = task_queue[-1]
            task_queue.pop()
            last_process = task.name
            task.run()
            myTimeSleep_small()
        else:
            myTimeSleep_big()
        if isReset:
            isReset = False
            restart_game()


@app.route('/')
def index():
    return render_template('index_dongri.html')

@socketio.on('connect')
def handle_connect():
    print('Client connected')

@socketio.on('disconnect')
def handle_disconnect():
    print('Client disconnected')

def send_hint(msg):#数组信息
    emit('processing_hint', msg)

def send_status(msg):#软件执行状态
    global g_status_list
    try:
        if not msg == "":
            # 添加新的状态消息和时间到列表
            timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')  # 获取当前时间
            status_entry = {'msg': msg, 'time': timestamp}  # 存储消息和时间
            g_status_list.append(status_entry)

            # 如果列表超过 5 条，移除最早的一条
            if len(g_status_list) > 5:
                g_status_list.pop(0)
        else:
            sendStr = ''
            for item in g_status_list:
                sendStr = sendStr + f"{item['time']}-{item['msg']}<br>" 

            #print(sendStr)
            emit('processing_status', sendStr)
        
        

        # 如果消息是 "结束"，发送所有状态并清空列表
        if msg == "结束":
            g_status_list = []  # 清空列表
            event.clear()
    except Exception as e:
        print(f"Error in send_status: {e}")
        return


@socketio.on('monitor_begin')
def monitor_begin():
    global last_time, last_process
    current_time = time.time()
    elapsed_time = current_time - last_time
    if elapsed_time < 0.5:
        return
    last_time = current_time
    regionRet, regionPos = game_region()
    screenshot = pyautogui.screenshot(region=regionPos)
    #binary_img = binarize_image(screenshot)
    compressed_data = compress_image(screenshot)
    
    image_data_base64 = base64.b64encode(compressed_data).decode('utf-8')
    socketio.emit('image_data', image_data_base64)
    task_arr = []
    if not event.is_set():
        task_arr.append(last_process)
    for item in reversed(task_queue):
        task_arr.append(item.name)
    send_hint(json.dumps(task_arr, ensure_ascii=False))
    send_status('')
    #print("send img")

@socketio.on('end_script')
def handle_end_script():
    event.set()

@socketio.on('end_game')
def handle_end_game():
    event.set()
    task_close_game()
    send_status("结束2")
    event.clear()

@socketio.on('get_title')
def handle_get_title():
    str = task_getComputerName()
    dst = str + ' machine'
    emit('processing_title', dst)

@socketio.on('reset_script')
def handle_reset_script():
    python = sys.executable
    while '--reset' in sys.argv:
    # 从 sys.argv 列表中删除 --reset 参数
        sys.argv.remove('--reset')
    os.execl(python, python, *sys.argv)

@socketio.on('restart_game')
def handle_restart_game():
    python = sys.executable
    os.execl(python, python, *sys.argv, '--reset')

@socketio.on('read_cfg')
def handle_read_cfg():
    cfg = read_cfg()
    emit('processing_cfg', cfg)

def restart_game():
    global isGameBegin
    isGameBegin = False
    while True:
        task_close_game()
        if True == task_start_game():
            break
        else:
            send_status("启动失败")
    isGameBegin = True
    send_status("结束")
    config = read_cfg()
    print("config", config)
    auto_task(config)

def auto_participate():
    task_queue.appendleft(task_returnAllLine())

    timeout = 40 * 60
    start_time = time.time()  # 记录开始时间


    while not event.is_set():
        if len(task_queue) < 4:
            task_queue.appendleft(task_paticipateInTeam())
            task_queue.appendleft(task_paticipateInTeam())
            task_queue.appendleft(task_paticipateInTeam())
            task_queue.appendleft(task_checkHelp(True))
        myTimeSleep_big()
        # 每次循环检查已用时间
        current_time = time.time()
        elapsed_time = current_time - start_time
        
        if elapsed_time >= timeout:
            handle_restart_game()
            break




def add_auto_task(isMaxCollect, isJina, isSimple = False, isAddStrengh = False, activity = 'None', isAutoParticipate = True):
    collectArr = [int(x) for x in isMaxCollect.split(",")]
    print("collectArr", collectArr)
    times = 0
    while not event.is_set():
        task_queue.appendleft(task_information())
        if activity == 'lianmeng':
            task_queue.appendleft(task_activity_lianmeng())
        #if not isSimple:
        #    task_queue.appendleft(task_paticipateInTeam())
        task_queue.appendleft(check_buildOrResearch())
        
        if not isSimple:
            if isJina == 'jina':
                task_queue.appendleft(task_fight_jina(isAddStrengh))
            elif isJina == 'yongbing':
                task_queue.appendleft(task_fight_yongbing(isAddStrengh))
            elif isJina == 'monster':
                task_queue.appendleft(task_fightMonster(isAddStrengh, False, isSimple))
            elif isJina == 'big_monster':
                task_queue.appendleft(task_fightMonster(isAddStrengh, True, isSimple))
        task_queue.appendleft(task_cure())
        task_queue.appendleft(task_checkStoreRoom())
        task_queue.appendleft(task_collect(collectArr, isSimple, isAddStrengh))
        task_queue.appendleft(task_train(False))
        if not isAddStrengh: # 如果不是添加体力，则添加一次采集
            task_queue.appendleft(task_collect(collectArr, isSimple, isAddStrengh))
        

        task_queue.appendleft(task_cure())
        if isSimple:
            task_queue.appendleft(check_buildOrResearch())
        else:
            if isJina == 'monster':
                task_queue.appendleft(task_fightMonster(isAddStrengh, False, isSimple))
            else:
                task_queue.appendleft(task_collect(collectArr, isSimple, isAddStrengh))
        

        #task_queue.appendleft(task_waitTime())
        times += 1
        if times % 3 == 0:
            task_queue.appendleft(task_checkAdventure())
            task_queue.appendleft(task_train(False))
            task_queue.appendleft(task_checkDonata())
            task_queue.appendleft(task_useAnnimalSkill())
            task_queue.appendleft(task_checkHelp(False))
            if not auto_participate:
                task_queue.appendleft(task_checkConfilits())
        if times == 10:
            handle_end_game()
            myTimeSleep(random.randint(1000, 2000), send_status)
            handle_restart_game()
        else:
            if isJina == 'monster':
                myTimeSleep(random.randint(400, 500), send_status)
            else:
                myTimeSleep(random.randint(600, 1000), send_status)
            
        
    task_queue.clear()
    send_status(f'自动模式结束')
    event.clear()
        

def write_cfg(config):
    with open('config.json', 'w') as config_file:
        json.dump(config, config_file, indent=4)
def read_cfg():
    try:
        with open('config.json', 'r') as config_file:
            config = json.load(config_file)
            return config
    except FileNotFoundError:
        print("配置文件不存在，请检查文件路径。")
        return None
    except PermissionError:
        print("没有权限读取配置文件。")
        return None
    except json.JSONDecodeError:
        print("配置文件格式错误，请检查文件内容是否为有效的 JSON。")
        return None

@socketio.on('begin_auto')
def handle_auto(data):
    write_cfg(data)
    config = read_cfg()
    print("config", config)
    auto_task(config)

def auto_task(data):
    global autoTask
    if data == None:
        isMaxCollect = '4,3,2,1'
        isSimple = False
        isJina = 'jina'
        isAddStrengh = False
        activity = 'none'
        participateJijie = False
    else:
        isMaxCollect = data['maxCollect']
        isSimple = data['simple']
        isJina = data['jina']
        isAddStrengh = data['add_strength']
        activity = data['activity']
        participateJijie = data['participate_jijie']
    send_status(f'开始自动模式')
    executor.submit(add_auto_task, isMaxCollect, isJina, isSimple, isAddStrengh, activity, participateJijie)


@socketio.on('begin_auto_participate')
def handle_auto_participate():
    global autoTask
    send_status(f'开始自动集结模式')
    executor.submit(auto_participate)


if __name__ == '__main__':
    init()
    if '--reset' in sys.argv:
        isReset = True
        print("需要重启游戏")
    runTask = threading.Thread(target=thread_runTask)#启动线程往里面添加任务
    runTask.daemon = True
    runTask.start()
    
    socketio.run(app, host= '0.0.0.0', debug=True)