use std::{cell::RefCell, rc::Rc};

use gpui::{
    Context, IntoElement, MouseButton, ParentElement as _, Render, Styled as _, TestAppContext,
    VisualTestContext, Window, div, point, px,
};
use gpui_native_core::context_menu::ContextMenu;
use gpui_native_core::menu::MenuItem;
use gpui_native_theme::Theme;

type Actions = Rc<RefCell<Vec<&'static str>>>;

struct Harness {
    actions: Actions,
}

impl Render for Harness {
    fn render(&mut self, _window: &mut Window, _cx: &mut Context<Self>) -> impl IntoElement {
        let actions = self.actions.clone();
        div().size_full().p_4().child(
            ContextMenu::new("ctx")
                .target(|_, _| div().h(px(40.)).w(px(200.)).child("Row").into_any_element())
                .items({
                    let actions = actions.clone();
                    move |_, _| {
                        vec![
                            MenuItem::new("Rename").handler({
                                let actions = actions.clone();
                                move |_, _| actions.borrow_mut().push("rename")
                            }),
                            MenuItem::new("Delete").handler({
                                let actions = actions.clone();
                                move |_, _| actions.borrow_mut().push("delete")
                            }),
                        ]
                    }
                }),
        )
    }
}

/// The target row rides the `p_4` band: `16..216` horizontally, `16..56`
/// vertically, so right-clicks land near `(100, 30)`.
fn context_window() -> (TestAppContext, VisualTestContext, Actions) {
    let mut cx = TestAppContext::single();
    cx.update(Theme::init);
    let actions = Rc::new(RefCell::new(Vec::new()));
    let harness_actions = actions.clone();
    let window = cx.add_window(move |_, _| Harness {
        actions: harness_actions,
    });
    let mut visual = VisualTestContext::from_window(window.into(), &cx);
    visual.update(|window, cx| {
        window.draw(cx).clear(cx);
    });
    (cx, visual, actions)
}

fn right_click(visual: &mut VisualTestContext, position: gpui::Point<gpui::Pixels>) {
    visual.simulate_mouse_down(position, MouseButton::Right, gpui::Modifiers::default());
    visual.simulate_mouse_up(position, MouseButton::Right, gpui::Modifiers::default());
}

fn click(visual: &mut VisualTestContext, position: gpui::Point<gpui::Pixels>) {
    visual.simulate_mouse_down(position, MouseButton::Left, gpui::Modifiers::default());
    visual.simulate_mouse_up(position, MouseButton::Left, gpui::Modifiers::default());
}

fn draw(visual: &mut VisualTestContext) {
    visual.update(|window, cx| {
        window.draw(cx).clear(cx);
    });
}

#[test]
fn right_click_opens_the_menu_and_items_activate() {
    let (_cx, mut visual, actions) = context_window();

    right_click(&mut visual, point(px(100.), px(30.)));
    draw(&mut visual);
    // The panel claims focus, so the first item is keyboard-highlighted.
    visual.simulate_keystrokes("enter");
    draw(&mut visual);
    assert_eq!(*actions.borrow(), vec!["rename"]);

    // The menu closed with the choice, so Enter reports nothing more.
    visual.simulate_keystrokes("enter");
    draw(&mut visual);
    assert_eq!(*actions.borrow(), vec!["rename"]);

    // Right-click again reopens; Down reaches the second item.
    right_click(&mut visual, point(px(100.), px(30.)));
    draw(&mut visual);
    visual.simulate_keystrokes("down");
    visual.simulate_keystrokes("enter");
    draw(&mut visual);
    assert_eq!(*actions.borrow(), vec!["rename", "delete"]);
}

#[test]
fn a_left_click_on_the_target_does_not_open_the_menu() {
    let (_cx, mut visual, actions) = context_window();

    click(&mut visual, point(px(100.), px(30.)));
    draw(&mut visual);
    visual.simulate_keystrokes("enter");
    draw(&mut visual);
    assert_eq!(*actions.borrow(), Vec::<&str>::new());
}

#[test]
fn a_click_outside_closes_the_menu() {
    let (_cx, mut visual, actions) = context_window();

    right_click(&mut visual, point(px(100.), px(30.)));
    draw(&mut visual);
    click(&mut visual, point(px(450.), px(300.)));
    draw(&mut visual);
    visual.simulate_keystrokes("enter");
    draw(&mut visual);
    assert_eq!(*actions.borrow(), Vec::<&str>::new());

    // A right-click outside the target never opens it either.
    right_click(&mut visual, point(px(450.), px(300.)));
    draw(&mut visual);
    visual.simulate_keystrokes("enter");
    draw(&mut visual);
    assert_eq!(*actions.borrow(), Vec::<&str>::new());
}