#include "app_core/quick_ui.h"
#include "test_harness.h"

namespace {

void slot_click_selects_or_opens_popup(pp::tests::Harness& harness)
{
    const auto select_brush = pp::app::plan_quick_slot_click(pp::app::QuickUiSlotKind::brush, 0, 2, 3);
    PP_EXPECT(harness, select_brush);
    if (select_brush) {
        PP_EXPECT(harness, select_brush.value().operation == pp::app::QuickUiOperation::select_slot);
        PP_EXPECT(harness, select_brush.value().slot_kind == pp::app::QuickUiSlotKind::brush);
        PP_EXPECT(harness, select_brush.value().slot_index == 2);
        PP_EXPECT(harness, select_brush.value().previous_index == 0);
        PP_EXPECT(harness, select_brush.value().updates_selection);
        PP_EXPECT(harness, select_brush.value().invokes_change_callback);
        PP_EXPECT(harness, select_brush.value().mutates_quick_state);
        PP_EXPECT(harness, !select_brush.value().opens_brush_popup);
    }

    const auto open_brush = pp::app::plan_quick_slot_click(pp::app::QuickUiSlotKind::brush, 1, 1, 3);
    PP_EXPECT(harness, open_brush);
    if (open_brush) {
        PP_EXPECT(harness, open_brush.value().operation == pp::app::QuickUiOperation::open_slot_popup);
        PP_EXPECT(harness, open_brush.value().opens_brush_popup);
        PP_EXPECT(harness, !open_brush.value().opens_color_picker);
        PP_EXPECT(harness, !open_brush.value().updates_selection);
        PP_EXPECT(harness, !open_brush.value().mutates_quick_state);
    }

    const auto open_color = pp::app::plan_quick_slot_click(pp::app::QuickUiSlotKind::color, 0, 0, 3);
    PP_EXPECT(harness, open_color);
    if (open_color) {
        PP_EXPECT(harness, open_color.value().opens_color_picker);
        PP_EXPECT(harness, !open_color.value().opens_brush_popup);
    }
}

void slot_click_rejects_invalid_indices(pp::tests::Harness& harness)
{
    PP_EXPECT(harness, !pp::app::plan_quick_slot_click(pp::app::QuickUiSlotKind::brush, -1, 0, 3));
    PP_EXPECT(harness, !pp::app::plan_quick_slot_click(pp::app::QuickUiSlotKind::brush, 0, 3, 3));
    PP_EXPECT(harness, !pp::app::plan_quick_slot_click(pp::app::QuickUiSlotKind::color, 0, 0, 0));
}

void restore_and_reset_validate_state(pp::tests::Harness& harness)
{
    const auto restore = pp::app::plan_quick_state_restore(2, 1, 3, true);
    PP_EXPECT(harness, restore);
    if (restore) {
        PP_EXPECT(harness, restore.value().operation == pp::app::QuickUiOperation::restore_state);
        PP_EXPECT(harness, restore.value().slot_count == 3);
        PP_EXPECT(harness, restore.value().fire_event);
        PP_EXPECT(harness, restore.value().restores_slots);
        PP_EXPECT(harness, restore.value().redraws_brush_previews);
        PP_EXPECT(harness, restore.value().invokes_change_callback);
    }

    const auto reset = pp::app::plan_quick_state_reset(3, false);
    PP_EXPECT(harness, reset);
    if (reset) {
        PP_EXPECT(harness, reset.value().operation == pp::app::QuickUiOperation::reset_state);
        PP_EXPECT(harness, reset.value().resets_slots);
        PP_EXPECT(harness, reset.value().redraws_brush_previews);
        PP_EXPECT(harness, !reset.value().invokes_change_callback);
        PP_EXPECT(harness, reset.value().mutates_quick_state);
    }

    PP_EXPECT(harness, !pp::app::plan_quick_state_restore(3, 0, 3, false));
    PP_EXPECT(harness, !pp::app::plan_quick_state_restore(0, -1, 3, false));
    PP_EXPECT(harness, !pp::app::plan_quick_state_reset(0, false));
}

} // namespace

int main()
{
    pp::tests::Harness harness;
    harness.run("slot click selects or opens popup", slot_click_selects_or_opens_popup);
    harness.run("slot click rejects invalid indices", slot_click_rejects_invalid_indices);
    harness.run("restore and reset validate state", restore_and_reset_validate_state);
    return harness.finish();
}