/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2023 Mark Roszko <mark.roszko@gmail.com>
 * Copyright The KiCad Developers, see AUTHORS.txt for contributors.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, you may find one here:
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * or you may search the http://www.gnu.org website for the version 2 license,
 * or you may write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
 */

#ifndef KISTATUSBAR_H
#define KISTATUSBAR_H

#include <kicommon.h>
#include <optional>
#include <mutex>
#include <vector>
#include <unordered_map>
#include <widgets/report_severity.h>
#include <wx/statusbr.h>

class wxGauge;
class wxButton;
class wxStaticText;
class BITMAP_BUTTON;

/**
 * KISTATUSBAR is a wxStatusBar suitable for Kicad manager.
 * It displays the fields needed by the caller, and room for 4 other fields (see kistatusbar.cpp)
 * Background text (FIELD_OFFSET_BGJOB_TEXT offset id)
 * Background gauge widget (FIELD_OFFSET_BGJOB_GAUGE offset id)
 * Background background stop button (FIELD_OFFSET_BGJOB_CANCEL offset id)
 * Background notifications button (FIELD_OFFSET_NOTIFICATION_BUTTON  offset id)
 */

/**
 * Structure to store a load message with its severity.
 */
struct LOAD_MESSAGE
{
    wxString  message;
    SEVERITY  severity;
};


class KICOMMON_API KISTATUSBAR : public wxStatusBar
{
public:
    enum STYLE_FLAGS : int
    {
        NONE_STYLE        = 0x00,
        NOTIFICATION_ICON = 0x01,
        CANCEL_BUTTON     = 0x02,
        WARNING_ICON      = 0x04,
    };

    static constexpr auto DEFAULT_STYLE =
            static_cast<STYLE_FLAGS>( NOTIFICATION_ICON | CANCEL_BUTTON );

    KISTATUSBAR( int aNumberFields, wxWindow* parent, wxWindowID id,
                 STYLE_FLAGS aFlags = DEFAULT_STYLE );

    ~KISTATUSBAR();

    /**
     * Show the background progress bar.
     */
    void ShowBackgroundProgressBar( bool aCancellable = false );

    /**
     * Hide the background progress bar.
     */
    void HideBackgroundProgressBar();

    /**
     * Set the current progress of the progress bar.
     */
    void SetBackgroundProgress( int aAmount );

    /**
     * Set the max progress of the progress bar.
     */
    void SetBackgroundProgressMax( int aAmount );

    /**
     * Set the status text that displays next to the progress bar.
     */
    void SetBackgroundStatusText( const wxString& aTxt );

    /**
     * Set the notification count on the notifications button.
     *
     * A value of 0 will hide the count.
     */
    void SetNotificationCount( int aCount );

    /**
     * Clears all warning messages from the given source (or all sources if aSource is empty)
     */
    void ClearWarningMessages( const wxString& aSource = wxEmptyString );

    /**
     * Add warning/error messages (not thread-safe, use the std::vector<LOAD_MESSAGE> variant
     * from other threads)
     */
    void AddWarningMessages( const wxString& aSource, const wxString& aMessages );

    /**
     * Add warning/error messages thread-safely.
     * Can be called from any thread. UI update is deferred to main thread.
     */
    void AddWarningMessages( const wxString& aSource, const std::vector<LOAD_MESSAGE>& aMessages );

    /**
     * Get current message count (thread-safe).
     */
    size_t GetLoadWarningCount() const;

    virtual void SetStatusWidths( int aSize, const int* aWidths ) override;

private:
    void onSize( wxSizeEvent& aEvent );
    void onBackgroundProgressClick( wxMouseEvent& aEvent );
    void onNotificationsIconClick( wxCommandEvent& aEvent );
    void onLoadWarningsIconClick( wxCommandEvent& aEvent );
    void updateWarningUI();  ///< Update warning button visibility and badge (main thread only)
    void updateAuxFieldWidths();
    void updateBackgroundText();
    void layoutControls();

    enum class FIELD
    {
        BGJOB_LABEL,
        BGJOB_GAUGE,
        BGJOB_CANCEL,
        WARNING,
        NOTIFICATION
    };

    std::optional<int> fieldIndex( FIELD aField ) const;

private:
    wxGauge*       m_backgroundProgressBar;
    wxButton*      m_backgroundStopButton;
    wxStaticText*  m_backgroundTxt;
    BITMAP_BUTTON* m_notificationsButton;
    BITMAP_BUTTON* m_warningButton;
    mutable std::mutex m_warningMutex;  ///< Protects m_warningMessages
    std::unordered_map<wxString, std::vector<LOAD_MESSAGE>> m_warningMessages;
    int            m_normalFieldsCount;
    STYLE_FLAGS    m_styleFlags;
    wxString       m_savedStatusText;       ///< Saved text from adjacent field during background jobs
    wxString       m_backgroundRawText;     ///< Unellipsized background status text
    std::vector<int> m_fieldWidths;
};

#endif
