import json
import os
import panel
from datetime import datetime
from typing import Any

from pydantic import BaseModel

from watt42_viewlib import attach_w42_state

panel.extension('echarts', 'ace', 'jsoneditor')

SYSTEM_ID = os.environ.get("WATT42_SYSTEM_ID", "invalid-system-id")
API_TOKEN = os.environ.get("WATT42_API_TOKEN", "invalid-api-token")

w42_state = panel.rx(None)

attach_w42_state(rx_var=w42_state, system_id=SYSTEM_ID, token=API_TOKEN)

class PvForecast(BaseModel):
    at: datetime = datetime.now()
    slots: list[float] = []

class LoadForecast(BaseModel):
    at: datetime = datetime.now()
    slots: list[float] = []


class InverterForecast(BaseModel):
    levels: list[float] = []
    
class SystemState(BaseModel):
    pv_forecast: PvForecast = PvForecast()
    load_forecast: LoadForecast = LoadForecast()
    value: int = 42
    inverter: InverterForecast = InverterForecast()

def state_to_text(state: dict[str, Any]) -> str:
    return f"Watt42 State:\n\n```\n{json.dumps(state, indent=2)}\n```\n\nReplace this with some awesome visuals"

state_as_text = panel.bind(state_to_text, w42_state)

state_pane = panel.pane.Markdown(state_as_text, sizing_mode='stretch_width')

def chart1(state: SystemState) -> dict:
    """ Return an ECharts option dict visualizing some aspect of the state."""

    now = state.load_forecast.at
    pv_slots = state.pv_forecast.slots
    load_slots = state.load_forecast.slots
    hours = list(range(len(pv_slots)))
    option = {
        "title": {"text": f"PV and Load Forecast at {now.strftime('%Y-%m-%d %H:%M')}"},
        "tooltip": {"trigger": "axis"},
        "legend": {"data": ["PV Forecast", "Load Forecast", "Inverter Level"]},
        "xAxis": {"type": "category", "data": hours, "name": "Hours"},
        "yAxis": {"type": "value", "name": "Power (kW)"},
        "series": [
            {
                "name": "PV Forecast",
                "type": "line",
                "data": pv_slots,
            },
            {
                "name": "Load Forecast",
                "type": "line",
                "data": load_slots,
            },
            {
                "name": "Inverter Level",
                "type": "line",
                "data": state.inverter.levels,
            }
        ],
    }
    return option

chart1_rx = panel.rx(chart1)(panel.rx(lambda s: SystemState.model_validate(s) if s else SystemState())(w42_state))


sidebar_content = """
Demonstrates how to visualize the Watt42 "dumb inverter simulation" system state.

Find instructions on [how to use this example
here](https://source.c3.uber5.com/watt42-public/watt42_viewlib/src/branch/main/README.md#how-to-use).

"""

_ = panel.template.FastListTemplate(
    title="Watt42: Dumb Inverter Visualization Example",
    sidebar=[panel.pane.Markdown(sidebar_content, sizing_mode='stretch_width')],
    main=[
        panel.pane.ECharts(chart1_rx, sizing_mode='stretch_width', height=400, theme='light'),
        state_pane,
        w42_state,
    ],
).servable()