/*
 * Copyright (c) 2020, the SerenityOS developers.
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include "ShutdownDialog.h"
#include <AK/DeprecatedString.h>
#include <AK/Vector.h>
#include <LibGUI/BoxLayout.h>
#include <LibGUI/Button.h>
#include <LibGUI/ImageWidget.h>
#include <LibGUI/Label.h>
#include <LibGUI/RadioButton.h>
#include <LibGUI/Widget.h>
#include <LibGfx/Font/Font.h>
#include <LibGfx/Font/FontDatabase.h>

struct Option {
    DeprecatedString title;
    ShutdownDialog::Command command;
    bool enabled;
    bool default_action;
};

static Array const options = {
    Option { "Power off computer", { "/bin/shutdown"sv, { "--now" } }, true, true },
    Option { "Reboot", { "/bin/reboot"sv, {} }, true, false },
    Option { "Log out", { "/bin/logout"sv, {} }, true, false },
};

Optional<ShutdownDialog::Command const&> ShutdownDialog::show()
{
    auto dialog = ShutdownDialog::construct();
    auto rc = dialog->exec();
    if (rc == ExecResult::OK && dialog->m_selected_option != -1)
        return options[dialog->m_selected_option].command;
    return {};
}

ShutdownDialog::ShutdownDialog()
    : Dialog(nullptr)
{
    auto widget = set_main_widget<GUI::Widget>().release_value_but_fixme_should_propagate_errors();
    widget->set_fill_with_background_color(true);
    widget->set_layout<GUI::VerticalBoxLayout>(GUI::Margins {}, 0);

    auto& banner_image = widget->add<GUI::ImageWidget>();
    banner_image.load_from_file("/res/graphics/brand-banner.png"sv);

    auto& content_container = widget->add<GUI::Widget>();
    content_container.set_layout<GUI::HorizontalBoxLayout>();

    auto& left_container = content_container.add<GUI::Widget>();
    left_container.set_fixed_width(60);
    left_container.set_layout<GUI::VerticalBoxLayout>(GUI::Margins { 12, 0, 0 });

    auto& icon_wrapper = left_container.add<GUI::Widget>();
    icon_wrapper.set_fixed_size(32, 48);
    icon_wrapper.set_layout<GUI::VerticalBoxLayout>();

    auto& icon_image = icon_wrapper.add<GUI::ImageWidget>();
    icon_image.set_bitmap(Gfx::Bitmap::load_from_file("/res/icons/32x32/shutdown.png"sv).release_value_but_fixme_should_propagate_errors());

    auto& right_container = content_container.add<GUI::Widget>();
    right_container.set_layout<GUI::VerticalBoxLayout>(GUI::Margins { 12, 12, 8, 0 });

    auto& label = right_container.add<GUI::Label>("What would you like to do?"_string.release_value_but_fixme_should_propagate_errors());
    label.set_text_alignment(Gfx::TextAlignment::CenterLeft);
    label.set_fixed_height(22);
    label.set_font(Gfx::FontDatabase::default_font().bold_variant());

    for (size_t i = 0; i < options.size(); i++) {
        auto action = options[i];
        auto& radio = right_container.add<GUI::RadioButton>();
        radio.set_enabled(action.enabled);
        radio.set_text(String::from_deprecated_string(action.title).release_value_but_fixme_should_propagate_errors());

        radio.on_checked = [this, i](auto) {
            m_selected_option = i;
        };

        if (action.default_action) {
            radio.set_checked(true);
            m_selected_option = i;
        }
    }

    right_container.add_spacer().release_value_but_fixme_should_propagate_errors();

    auto& button_container = right_container.add<GUI::Widget>();
    button_container.set_fixed_height(23);
    button_container.set_layout<GUI::HorizontalBoxLayout>(GUI::Margins {}, 5);
    button_container.add_spacer().release_value_but_fixme_should_propagate_errors();
    auto& ok_button = button_container.add<GUI::Button>("OK"_short_string);
    ok_button.set_fixed_size(80, 23);
    ok_button.on_click = [this](auto) {
        done(ExecResult::OK);
    };
    ok_button.set_default(true);
    auto& cancel_button = button_container.add<GUI::Button>("Cancel"_short_string);
    cancel_button.set_fixed_size(80, 23);
    cancel_button.on_click = [this](auto) {
        done(ExecResult::Cancel);
    };

    resize(413, 235);
    center_on_screen();
    set_resizable(false);
    set_title("Exit SerenityOS");
    set_icon(Gfx::Bitmap::load_from_file("/res/icons/16x16/power.png"sv).release_value_but_fixme_should_propagate_errors());

    // Request WindowServer to re-update us on the current theme as we might've not been alive for the last notification.
    refresh_system_theme();
}