use super::*;
use crate::state::{ClientState, LoggedIn};
use openworm::net::{ClientMsg, NetClientMsg};

pub const SIZE: u32 = 20;

pub fn main_view(client: &mut Client, ui: &mut Ui) -> WidgetRef {
    let ClientState::LoggedIn(state) = &mut client.state else {
        panic!("we ain't logged in buh");
    };
    let msg_panel = msg_panel(ui, state);
    let side_bar = rect(Color::BLACK.brighter(0.05)).width(80);

    (side_bar, msg_panel).span(Dir::RIGHT).add(ui)
}

pub fn msg_widget(username: &str, content: &str) -> impl WidgetRet {
    let content = wtext(content)
        .editable(false)
        .size(SIZE)
        .wrap(true)
        .attr::<Selectable>(());
    let header = wtext(username).size(SIZE);
    (
        image(include_bytes!("../assets/sungals.png"))
            .sized((70, 70))
            .align(Align::TOP),
        (header, content)
            .span(Dir::DOWN)
            .gap(10)
            .width(rest(1))
            .align(Align::TOP),
    )
        .span(Dir::RIGHT)
        .gap(10)
        .to_any()
}

pub fn msg_panel(ui: &mut Ui, state: &mut LoggedIn) -> WidgetRef {
    let msg_area = Span::empty(Dir::DOWN).gap(15).add(ui);
    state.channel = Some(msg_area.clone());

    let send_text = wtext("")
        .editable(false)
        .size(SIZE)
        .wrap(true)
        .hint(hint("send message"))
        .add(ui);

    (
        msg_area
            .clone()
            .scroll()
            .pad(Padding::x(15).with_top(15))
            .height(rest(1)),
        send_text
            .clone()
            .on(Submit, move |ctx| {
                let ClientState::LoggedIn(state) = &mut ctx.state.state else {
                    panic!("we ain't logged in buh");
                };
                let content = ctx.widget.get_mut().take();
                let msg = NetClientMsg {
                    content: content.clone(),
                };
                state.network.send(ClientMsg::SendMsg(msg.clone()));
                let msg = msg_widget(&state.username, &content).add(ctx.ui);
                msg_area.get_mut().children.push(msg);
            })
            .pad(15)
            .attr::<Selector>(send_text)
            .scroll()
            .masked()
            .background(rect(Color::BLACK.brighter(0.05)).radius(15))
            .pad(15)
            .max_height(rel(0.5))
            .layer_offset(1),
    )
        .span(Dir::DOWN)
        .width(rest(1))
        .add(ui)
}