// Copyright (c) 2011-2019 The Bitcoin Core developers
// Copyright (c) 2020-2023 The Bitcoin developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.

#pragma once

#if defined(HAVE_CONFIG_H)
#include <config/bitcoin-config.h>
#endif

#include <qt/optionsdialog.h>

#include <amount.h>

#include <QLabel>
#include <QMainWindow>
#include <QMap>
#include <QPoint>
#include <QSystemTrayIcon>

#ifdef Q_OS_MAC
#include <qt/macos_appnap.h>
#endif

#include <memory>

class ClientModel;
class NetworkStyle;
class Notificator;
class OptionsModel;
class PlatformStyle;
class RPCConsole;
class SendCoinsRecipient;
class UnitDisplayStatusBarControl;
class WalletController;
class WalletFrame;
class WalletModel;
class HelpMessageDialog;
class ModalOverlay;

class Config;

namespace interfaces {
class Handler;
class Node;
} // namespace interfaces

QT_BEGIN_NAMESPACE
class QAction;
class QComboBox;
class QMenu;
class QProgressBar;
class QProgressDialog;
QT_END_NAMESPACE

namespace GUIUtil {
class ClickableLabel;
class ClickableProgressBar;
} // namespace GUIUtil

/**
 * Bitcoin GUI main class. This class represents the main window of the Bitcoin
 * UI. It communicates with both the client and wallet models to give the user
 * an up-to-date view of the current core state.
 */
class BitcoinGUI : public QMainWindow {
    Q_OBJECT

public:
    static const std::string DEFAULT_UIPLATFORM;

    explicit BitcoinGUI(interfaces::Node &node, const Config *,
                        const PlatformStyle *platformStyle,
                        const NetworkStyle *networkStyle,
                        QWidget *parent = nullptr);
    ~BitcoinGUI();

    /**
     * Set the client model.
     * The client model represents the part of the core that communicates with
     * the P2P network, and is wallet-agnostic.
     */
    void setClientModel(ClientModel *clientModel);
#ifdef ENABLE_WALLET
    void setWalletController(WalletController *wallet_controller);
#endif

#ifdef ENABLE_WALLET
    /**
     * Set the wallet model.
     * The wallet model represents a bitcoin wallet, and offers access to the
     * list of transactions, address book and sending functionality.
     */
    void addWallet(WalletModel *walletModel);
    void removeWallet(WalletModel *walletModel);
    void removeAllWallets();
#endif // ENABLE_WALLET
    bool enableWallet = false;


    /** Disconnect core signals from GUI client */
    void unsubscribeFromCoreSignals();

    bool isPrivacyModeActivated() const;

protected:
    void changeEvent(QEvent *e) override;
    void closeEvent(QCloseEvent *event) override;
    void showEvent(QShowEvent *event) override;
    void dragEnterEvent(QDragEnterEvent *event) override;
    void dropEvent(QDropEvent *event) override;
    bool eventFilter(QObject *object, QEvent *event) override;

private:
    interfaces::Node &m_node;
    WalletController *m_wallet_controller{nullptr};
    std::unique_ptr<interfaces::Handler> m_handler_message_box;
    std::unique_ptr<interfaces::Handler> m_handler_question;
    ClientModel *clientModel = nullptr;
    WalletFrame *walletFrame = nullptr;

    UnitDisplayStatusBarControl *unitDisplayControl = nullptr;
    QLabel *labelWalletEncryptionIcon = nullptr;
    QLabel *labelWalletHDStatusIcon = nullptr;
    GUIUtil::ClickableLabel *labelProxyIcon = nullptr;
    GUIUtil::ClickableLabel *connectionsControl = nullptr;
    GUIUtil::ClickableLabel *labelBlocksIcon = nullptr;
    QLabel *progressBarLabel = nullptr;
    GUIUtil::ClickableProgressBar *progressBar = nullptr;
    QProgressDialog *progressDialog = nullptr;

    QMenuBar *appMenuBar = nullptr;
    QToolBar *appToolBar = nullptr;
    QAction *overviewAction = nullptr;
    QAction *historyAction = nullptr;
    QAction *quitAction = nullptr;
    QAction *sendCoinsAction = nullptr;
    QAction *sendCoinsMenuAction = nullptr;
    QAction *usedSendingAddressesAction = nullptr;
    QAction *usedReceivingAddressesAction = nullptr;
    QAction *signMessageAction = nullptr;
    QAction *verifyMessageAction = nullptr;
    QAction *aboutAction = nullptr;
    QAction *receiveCoinsAction = nullptr;
    QAction *receiveCoinsMenuAction = nullptr;
    QAction *optionsAction = nullptr;
    QAction *toggleHideAction = nullptr;
    QAction *encryptWalletAction = nullptr;
    QAction *backupWalletAction = nullptr;
    QAction *changePassphraseAction = nullptr;
    QAction *aboutQtAction = nullptr;
    QAction *openRPCConsoleAction = nullptr;
    QAction *openAction = nullptr;
    QAction *showHelpMessageAction = nullptr;
    QAction *m_wallet_selector_label_action = nullptr;
    QAction *m_wallet_selector_action = nullptr;
    QAction *m_mask_values_action{nullptr};

    /** Only set to non-nullptr in constructor on OSX */
    QAction *m_main_window_action = nullptr;

    QVector<QAction *> m_node_actions;

    QLabel *m_wallet_selector_label = nullptr;
    QComboBox *m_wallet_selector = nullptr;

    QSystemTrayIcon *trayIcon = nullptr;
    const std::unique_ptr<QMenu> trayIconMenu;
    Notificator *notificator = nullptr;
    RPCConsole *rpcConsole = nullptr;
    HelpMessageDialog *helpMessageDialog = nullptr;
    ModalOverlay *modalOverlay = nullptr;

#ifdef Q_OS_MAC
    CAppNapInhibitor *m_app_nap_inhibitor = nullptr;
#endif

    /** Keep track of previous number of blocks, to detect progress */
    int prevBlocks = 0;
    int spinnerFrame = 0;

    const Config *config;
    const PlatformStyle *platformStyle;
    const NetworkStyle *const m_network_style;

    /** Create the main UI actions. */
    void createActions();
    /** Create the menu bar and sub-menus. */
    void createMenuBar();
    /** Create the toolbars */
    void createToolBars();
    /** Create system tray icon and notification */
    void createTrayIcon();
    /** Create system tray menu (or setup the dock menu) */
    void createTrayIconMenu();

    /** Enable or disable all wallet-related actions */
    void setWalletActionsEnabled(bool enabled);

    /** @brief Enable or disable all the main window-related actions */
    void setWindowActionsEnabled(bool enabled);

     /** Connect core signals to GUI client */
    void subscribeToCoreSignals();

    /** Update UI with latest network info from model. */
    void updateNetworkState();

    void updateHeadersSyncProgressLabel();

    /** Open the OptionsDialog on the specified tab index */
    void openOptionsDialogWithTab(OptionsDialog::Tab tab);

Q_SIGNALS:
    /** Signal raised when a URI was entered or dragged to the GUI */
    void receivedURI(const QString &uri);
    /** Signal raised when RPC console shown */
    void consoleShown(RPCConsole *console);
    void setPrivacy(bool privacy);

public Q_SLOTS:
    /** Set number of connections shown in the UI */
    void setNumConnections(int count);
    /** Set network state shown in the UI */
    void setNetworkActive(bool networkActive);
    /** Set number of blocks, last block date and last block hash shown in the UI */
    void setNumBlocks(int count, const QDateTime& blockDate, const QString& blockHash, double nVerificationProgress, bool headers);

    /** Notify the user of an event from the core network or transaction
       handling code.
       @param[in] title     the message box / notification title
       @param[in] message   the displayed text
       @param[in] style     modality and style definitions (icon and used
       buttons - buttons only for message boxes)
                            @see CClientUIInterface::MessageBoxFlags
       @param[in] ret       pointer to a bool that will be modified to whether
       Ok was clicked (modal only)
     */
    void message(const QString &title, const QString &message,
                 unsigned int style, bool *ret = nullptr);

#ifdef ENABLE_WALLET
    void setCurrentWallet(WalletModel *wallet_model);
    void setCurrentWalletBySelectorIndex(int index);
    /** Set the UI status indicators based on the currently selected wallet.
     */
    void updateWalletStatus();

private:
    /** Set the encryption status as shown in the UI.
       @param[in] status            current encryption status
       @see WalletModel::EncryptionStatus
     */
    void setEncryptionStatus(int status);

    /** Set the hd-enabled status as shown in the UI.
     @param[in] status            current hd enabled status
     @see WalletModel::EncryptionStatus
     */
    void setHDStatus(int hdEnabled);

public Q_SLOTS:
    bool handlePaymentRequest(const SendCoinsRecipient &recipient);

    /** Show incoming transaction notification for new transactions. */
    void incomingTransaction(const QString &date, int unit, const Amount amount,
                             const QString &type, const QString &address,
                             const QString &label, const QString &walletName);
#endif // ENABLE_WALLET

private:
    /** Set the proxy-enabled icon as shown in the UI. */
    void updateProxyIcon();
    void updateWindowTitle();

public Q_SLOTS:
#ifdef ENABLE_WALLET
    /** Switch to overview (home) page */
    void gotoOverviewPage();
    /** Switch to history (transactions) page */
    void gotoHistoryPage();
    /** Switch to receive coins page */
    void gotoReceiveCoinsPage();
    /** Switch to send coins page */
    void gotoSendCoinsPage(QString addr = "");

    /** Show Sign/Verify Message dialog and switch to sign message tab */
    void gotoSignMessageTab(QString addr = "");
    /** Show Sign/Verify Message dialog and switch to verify message tab */
    void gotoVerifyMessageTab(QString addr = "");

    /** Show open dialog */
    void openClicked();
#endif // ENABLE_WALLET
    /** Show configuration dialog */
    void optionsClicked();
    /** Show about dialog */
    void aboutClicked();
    /** Show debug window */
    void showDebugWindow();
    /** Show debug window and set focus to the console */
    void showDebugWindowActivateConsole();
    /** Show help message dialog */
    void showHelpMessageClicked();
#ifndef Q_OS_MAC
    /** Handle tray icon clicked */
    void trayIconActivated(QSystemTrayIcon::ActivationReason reason);
#else
    /** Handle macOS Dock icon clicked */
    void macosDockIconActivated();
#endif

    /** Show window if hidden, unminimize when minimized, rise when obscured or
     * show if hidden and fToggleHidden is true */
    void showNormalIfMinimized() { showNormalIfMinimized(false); }
    void showNormalIfMinimized(bool fToggleHidden);
    /** Simply calls showNormalIfMinimized(true) for use in SLOT() macro */
    void toggleHidden();

    /** called by a timer to check if ShutdownRequested() has been set **/
    void detectShutdown();

    /** Show progress dialog e.g. for verifychain
     *  Calling this method for values <100 will create a new dialog if it
     *  doesn't exist, or update the existing dialog if it does.
     *  When nProgress >= 100 any existing dialog is closed & deleted. */
    void showProgress(const QString &title, int nProgress);

    /** When hideTrayIcon setting is changed in OptionsModel hide or show the
     * icon accordingly. */
    void setTrayIconVisible(bool);

    void showModalOverlay();
};

class UnitDisplayStatusBarControl : public QLabel {
    Q_OBJECT

public:
    explicit UnitDisplayStatusBarControl(const PlatformStyle *platformStyle);
    /** Lets the control know about the Options Model (and its signals) */
    void setOptionsModel(OptionsModel *optionsModel);

protected:
    /** So that it responds to left-button clicks */
    void mousePressEvent(QMouseEvent *event) override;

private:
    OptionsModel *optionsModel;
    QMenu *menu;

    /** Shows context menu with Display Unit options by the mouse coordinates */
    void onDisplayUnitsClicked(const QPoint &point);
    /** Creates context menu, its actions, and wires up all the relevant signals
     * for mouse events. */
    void createContextMenu();

private Q_SLOTS:
    /** When Display Units are changed on OptionsModel it will refresh the
     * display text of the control on the status bar */
    void updateDisplayUnit(int newUnits);
    /** Tells underlying optionsModel to update its current display unit. */
    void onMenuSelection(QAction *action);
};
