from flask import Flask, render_template
from flask_socketio import SocketIO
import toml
from loguru import logger
import logging
from multiprocessing import Process, Queue
import threading
import multiprocessing
import os
import time
import subprocess
import signal
import importlib
from main_upper import main_func
server_command = [
    {"path": "/home/evan/Workplace/project_capture/build/", "script": "./capture"},
    {"path": "/home/evan/Workplace/project_infer/lane_server/", "script": "lane_infer_server.py"},
    {"path": "/home/evan/Workplace/project_infer/lane_server/", "script": "lane_infer_server1.py"},
    {"path": "/home/evan/Workplace/project_infer/yolo_server/", "script": "yolo_infer_server.py"},

]
processes = []

time_record = None

# 日志队列
queue = Queue()
# 跳过任务 干预任务调度
skip_task_queue = Queue()

app = Flask(__name__)
app.jinja_env.variable_start_string = '[('
app.jinja_env.variable_end_string = ')]'
app.config['SECRET_KEY'] = 'secret!'
socketio = SocketIO(app, allow_unsafe_werkzeug=True)

server_process = None
task_process = None

class WebSocketHandler(logging.Handler):
    def emit(self, record):
        log_entry = self.format(record)
        socketio.emit('log', {'level': record.levelname.lower(), 'content': log_entry})

# 设置日志
logger.remove()
handler = WebSocketHandler()
logger.add(handler, format="{time:MM-DD HH:mm:ss} {message}", level="DEBUG")

fileOptions_path = '/home/evan/Workplace/project_main'
fileOptions_list = ['cfg_args.toml','cfg_main.toml', 'cfg_subtask.toml']

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

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

@socketio.on('operate')
def operate_handle(data):
    global server_process
    global task_process
    global processes
    global time_record
    if data['type'] == 'save_config':
        f = open(os.path.join(fileOptions_path,data['file_name']), 'w')
        ret = toml.dump(data['content'], f)
        logger.info(f"保存成功 {data['file_name']}")
        f.close()
    elif data['type'] == 'operate_server':
        logger.info(data)
        if data['content'] == 'run':
            log_file = "server_processes.log"
            log = open(log_file, "w")
            time.sleep(2)
            # 启动所有脚本
            for i, env_info in enumerate(server_command):
                env_path = env_info["path"]
                script = env_info["script"]
                env = os.environ.copy() 
                if i == 0:
                    process = subprocess.Popen([script], cwd=env_path, env=env, stdout=log, stderr=subprocess.STDOUT)
                    processes.append(process)
                    time.sleep(2)
                
                process = subprocess.Popen(['python', script], cwd=env_path, env=env, stdout=log, stderr=subprocess.STDOUT)
                processes.append(process)
            logger.info("开启 server")
        elif data['content'] == 'stop':
            for process in processes:
                logger.error(process.pid)
                os.kill(process.pid, signal.SIGINT)
            logger.info("关闭 server")
        elif data['content'] == 'restart':
            
            logger.info("重启 server")
    elif data['type'] == 'operate_task':
        # 任务函数
        if data['content'] == 'run':
            if task_process != None:
                task_process.terminate()
            time_record = time.perf_counter()
            task_process = Process(target=main_func, args=(queue,skip_task_queue))
            task_process.start()
            logger.info("开启 task")
        elif data['content'] == 'stop':
            task_process.terminate()
            logger.info(f"任务结束 用时{time.perf_counter() - time_record}s")
            logger.info("关闭  task")
        elif data['content'] == 'restart':
            if task_process != None:
                task_process.terminate()
            task_process = Process(target=main_func, args=(queue,skip_task_queue))
            task_process.start()
    elif data['type'] == 'show_server_log':
        content = ''
        try:
            with open("server_processes.log", 'r') as file:
                content = file.read()
        except:
            pass
        socketio.emit('server_log', {'type': 'server_log', 'content': content})
    elif data['type'] == 'skip_task':
        logger.info(data)
        skip_task_queue.put(1)

@socketio.on('connect')
def test_connect():
    logger.info('Client connected')
    socketio.emit('config_data', {'type': 'fileOptions', 'content': fileOptions_list})
    config_data = {}
    for item in fileOptions_list:
        config_data[item] = toml.load(os.path.join(fileOptions_path,item))
    socketio.emit('config_data', {'type': 'config_data', 'content': config_data})
def thread_function():
    global queue
    while True:
        try:
            log = queue.get()
            socketio.emit('log', log)
        except multiprocessing.Queue.Empty:
            pass

    
if __name__ == '__main__':

    log_file = "server_processes.log"
    log = open(log_file, "w")
    time.sleep(2)
    # 启动所有脚本
    for i, env_info in enumerate(server_command):
        env_path = env_info["path"]
        script = env_info["script"]
        env = os.environ.copy() 
        if i == 0:
            process = subprocess.Popen([script], cwd=env_path, env=env, stdout=log, stderr=subprocess.STDOUT)
            processes.append(process)
            time.sleep(2)
        
        process = subprocess.Popen(['python', script], cwd=env_path, env=env, stdout=log, stderr=subprocess.STDOUT)
        processes.append(process)

    thread1 = threading.Thread(target=thread_function, daemon = True)
    thread1.start()
    socketio.run(app, host='0.0.0.0', port=5001, allow_unsafe_werkzeug=True)
    if server_process != None:
        server_process.terminate()
    if task_process != None:
        task_process.terminate()
    for process in processes:
        logger.error(process.pid)
        os.kill(process.pid, signal.SIGINT)