import nonebot
import psutil
from dash import Dash, Input, Output, dcc, html
from starlette.middleware.wsgi import WSGIMiddleware

from liteyuki.utils.language import Language
from liteyuki.utils.tools import convert_size

app = nonebot.get_app()


def get_system_info():
    cpu_percent = psutil.cpu_percent()
    memory_info = psutil.virtual_memory()
    memory_percent = memory_info.percent
    return {
            "cpu_percent"   : cpu_percent,
            "memory_percent": memory_percent
    }


@app.get("/system_info")
async def system_info():
    return get_system_info()


lang = Language()
dash_app = Dash(__name__)
dash_app.layout = dash_app.layout = html.Div(children=[
        html.H1(children=lang.get("main.monitor.title"), style={
                "textAlign": "center"
        }),

        dcc.Graph(id="live-update-graph"),
        dcc.Interval(
            id="interval-component",
            interval=1 * 1000,  # in milliseconds
            n_intervals=0
        )
])


@dash_app.callback(Output("live-update-graph", "figure"),
                   [Input("interval-component", "n_intervals")])
def update_graph_live(n):
    lang = Language()
    system_inf = get_system_info()
    dash_app.layout = html.Div(children=[
            html.H1(children=lang.get("main.monitor.title"), style={
                    "textAlign": "center"
            }),

            dcc.Graph(id="live-update-graph"),
            dcc.Interval(
                id="interval-component",
                interval=2 * 1000,  # in milliseconds
                n_intervals=0
            )
    ])
    mem = psutil.virtual_memory()
    cpu_f = psutil.cpu_freq()
    figure = {
            "data"  : [
                    {
                            "x"   : [f"{cpu_f.current / 1000:.2f}GHz {psutil.cpu_count(logical=False)}c{psutil.cpu_count()}t"],
                            "y"   : [system_inf["cpu_percent"]],
                            "type": "bar",
                            "name": f"{lang.get('main.monitor.cpu')} {lang.get('main.monitor.usage')}"

                    },
                    {
                            "x"   : [f"{convert_size(mem.used, add_unit=False)}/{convert_size(mem.total)}({mem.used / mem.total * 100:.2f}%)"],
                            "y"   : [system_inf["memory_percent"]],
                            "type": "bar",
                            "name": f"{lang.get('main.monitor.memory')} {lang.get('main.monitor.usage')}"
                    },
            ],
            "layout": {
                    "title": lang.get("main.monitor.description"),
                    # "xaxis": {
                    #         "range": [0, 10]
                    #         },  # 设置x轴的范围
                    "yaxis": {
                            "range": [0, 100]
                    },  # 设置y轴的范围
            }
    }
    return figure


app.mount("/", WSGIMiddleware(dash_app.server))