/* Copyright (c) 2013-2016, Stanislaw Halik <sthalik@misaki.pl>

 * Permission to use, copy, modify, and/or distribute this
 * software for any purpose with or without fee is hereby granted,
 * provided that the above copyright notice and this permission
 * notice appear in all copies.
 */

#pragma once

#include "api/plugin-support.hpp"
#include "gui/mapping-dialog.hpp"
#include "gui/settings.hpp"
#include "gui/process_detector.h"
#include "logic/main-settings.hpp"
#include "logic/pipeline.hpp"
#include "logic/shortcuts.h"
#include "logic/work.hpp"
#include "logic/state.hpp"
#include "options/options.hpp"
#include "compat/qt-signal.hpp"

#include <QApplication>
#include <QMainWindow>
#include <QKeySequence>
#include <QShortcut>
#include <QPixmap>
#include <QTimer>
#include <QSystemTrayIcon>
#include <QString>
#include <QMenu>
#include <QAction>
#include <QEvent>
#include <QCloseEvent>

#include <tuple>
#include <memory>

#include "ui_main-window.h"

class main_window final : public QMainWindow, private State
{
    Q_DECLARE_TR_FUNCTIONS(main_window)

    Ui::main_window ui;

    Shortcuts global_shortcuts;
    module_settings m;
    std::unique_ptr<QSystemTrayIcon> tray;
    QMenu tray_menu;
    QTimer pose_update_timer { this };
    QTimer det_timer;
    QTimer config_list_timer;
    std::unique_ptr<options_dialog> options_widget;
    std::unique_ptr<mapping_dialog> mapping_widget;
    QShortcut kbd_quit { QKeySequence("Ctrl+Q"), this };
    std::unique_ptr<IFilterDialog> pFilterDialog;
    std::unique_ptr<IProtocolDialog> pProtocolDialog;
    std::unique_ptr<ITrackerDialog> pTrackerDialog;

    process_detector_worker det;
    QMenu profile_menu;

    QAction menu_action_header   { &tray_menu },
            menu_action_show     { &tray_menu },
            menu_action_exit     { &tray_menu },
            menu_action_tracker  { &tray_menu },
            menu_action_filter   { &tray_menu },
            menu_action_proto    { &tray_menu },
            menu_action_options  { &tray_menu },
            menu_action_mappings { &tray_menu };

    bool exiting_already { false };

    using dylib_ptr = Modules::dylib_ptr;
    using dylib_list = Modules::dylib_list;

    dylib_ptr current_tracker();
    dylib_ptr current_protocol();
    dylib_ptr current_filter();

    static std::tuple<dylib_ptr, int> module_by_name(const QString& name, Modules::dylib_list& list);

    void update_button_state(bool running, bool inertialp);
    void display_pose(const double* mapped, const double* raw);
    void set_title(const QString& game_title = QString());
    [[nodiscard]] static bool get_new_config_name_from_dialog(QString &ret);
    void set_profile_in_registry(const QString& profile);
    void register_shortcuts();
    void set_keys_enabled(bool flag);
    bool config_listed(const QString& name);

    void init_dylibs();
    void init_tray_menu();
    void init_profiles();
    void init_shortcuts();
    void init_buttons();

#if !defined _WIN32
    void annoy_if_root();
#endif

    void changeEvent(QEvent* e) override;
    bool event(QEvent *event) override;
    bool maybe_hide_to_tray(QEvent* e);

    void closeEvent(QCloseEvent *event) override;

    void die_on_config_not_writable();
    bool is_tray_enabled();
    bool start_in_tray();

    void refresh_config_list();

    void make_empty_config();
    void make_copied_config();
    void open_config_directory();

    void show_tracker_settings();
    void show_proto_settings();
    void show_filter_settings();
    void show_options_dialog();
    void show_mapping_window();
    void show_pose();

    void start_tracker_();
    void stop_tracker_();
    void restart_tracker_();
    void toggle_tracker_();

    void set_profile(const QString& new_name, bool migrate = true);
    void maybe_start_profile_from_executable();

    void ensure_tray();
    void toggle_restore_from_tray(QSystemTrayIcon::ActivationReason e);

    void save_modules();
    void exit(int status = EXIT_SUCCESS);

    static void set_working_directory();

    qt_sig::nullary start_tracker;
    qt_sig::nullary stop_tracker;
    qt_sig::nullary toggle_tracker;
    qt_sig::nullary restart_tracker;

public:
    main_window();
    ~main_window() override;
};