#pragma once

#include <QAction>
#include <QToolBar>
#include <QPushButton>
#include <QIcon>
#include <QLabel>
#include <QPropertyAnimation>
#include <QTabWidget>
#include "binaryninjaapi.h"
#include "flowgraphwidget.h"
#include "fontsettings.h"
#include "pane.h"
#include "sidebarwidget.h"
#include "uicontext.h"
#include "viewframe.h"


class BINARYNINJAUIAPI WorkflowMonitorWidget : public QWidget
{
	Q_OBJECT
	Q_PROPERTY(qreal dotBrightness READ dotBrightness WRITE setDotBrightness)

	BinaryViewRef m_data;
	FunctionRef m_function;
	WorkflowRef m_workflow;

	Menu m_menu;
	ContextMenuManager* m_contextMenuManager;
	UIActionHandler m_actionHandler;

	QToolBar* m_toolbar;
	QTabWidget* m_tabs;
	FlowGraphWidget* m_flowGraphWidget;

	QAction* m_startAction;
	QAction* m_haltAction;
	QAction* m_stepAction;
	QAction* m_resetAction;
	QAction* m_toggleSuspendAction;
	QAction* m_toggleLogAction;
	QPushButton* m_contextButton;
	QLabel* m_contextLabel;

	std::map<std::string, QIcon> m_iconCache;
	std::map<std::string, QColor> m_colorCache;
	std::string m_lastState;
	QColor m_lastStatusColor;
	QColor m_labelColor;
	std::string m_breakRequest;

	QLabel* m_currentActivity;
	QLabel* m_statusIndicator;
	QPropertyAnimation* m_dotAnimation;
	qreal m_dotBrightness;
	bool m_animationRunning;

	void updateVisualState(const BinaryNinja::WorkflowMachine::Status& status, bool force);
	void updateButtonStates(const BinaryNinja::WorkflowMachine::Status& status);
	void updateToolbarActions(bool force = false);
	void updateToolbarIcons();
	void setupToolbar();
	void setupActions();
	void updateStatusIndicator();

	qreal dotBrightness() const;
	void setDotBrightness(qreal brightness);

public:
	WorkflowMonitorWidget(BinaryViewRef data, FunctionRef function = nullptr);
	~WorkflowMonitorWidget();

	FunctionRef getCurrentFunction() const { return m_function; }

	void notifyRefresh();
	void notifyFontChanged();
	void notifyThemeChanged();
	void notifyViewLocationChanged(View* view, const ViewLocation& viewLocation);

	void startDotAnimation();
	void stopDotAnimation();
};


class BINARYNINJAUIAPI WorkflowMonitorSidebarWidget : public SidebarWidget
{
	Q_OBJECT

	WorkflowMonitorWidget* m_widget;

public:
	WorkflowMonitorSidebarWidget(BinaryViewRef data);

	void notifyRefresh() override { m_widget->notifyRefresh(); }
	void notifyFontChanged() override { m_widget->notifyFontChanged(); }
	void notifyThemeChanged() override { m_widget->notifyThemeChanged(); }
	void notifyViewLocationChanged(View* view, const ViewLocation& viewLocation) override { m_widget->notifyViewLocationChanged(view, viewLocation); }
};


class BINARYNINJAUIAPI WorkflowMonitorSidebarWidgetType : public SidebarWidgetType
{
public:
	WorkflowMonitorSidebarWidgetType();

	SidebarWidgetLocation defaultLocation() const override { return SidebarWidgetLocation::LeftReference; }
	SidebarContextSensitivity contextSensitivity() const override { return PerViewTypeSidebarContext; }
	SidebarIconVisibility defaultIconVisibility() const override { return AlwaysHideSidebarIcon; }

	SidebarWidget* createWidget(ViewFrame* frame, BinaryViewRef data) override;

	bool canUseAsPane(SplitPaneWidget* panes, BinaryViewRef data) const override { return true; }
	Pane* createPane(SplitPaneWidget* panes, BinaryViewRef data) override;
};


class BINARYNINJAUIAPI WorkflowView : public QWidget, public View
{
	Q_OBJECT

	BinaryViewRef m_data;
	uint64_t m_currentOffset = 0;
	WorkflowMonitorWidget* m_widget;

public:
	WorkflowView(BinaryViewRef data, ViewFrame* frame);

	void notifyRefresh() override { m_widget->notifyRefresh(); }
	BinaryViewRef getData() override { return m_data; }
	uint64_t getCurrentOffset() override { return m_currentOffset; }
	void setSelectionOffsets(BNAddressRange range) override { m_currentOffset = range.start; }
	bool navigate(uint64_t offset) override;
	QFont getFont() override { return getMonospaceFont(m_widget); }
	FunctionRef getCurrentFunction() override { return m_widget->getCurrentFunction(); }
};


class BINARYNINJAUIAPI WorkflowViewType : public ViewType
{
	static WorkflowViewType* m_instance;

public:
	WorkflowViewType();
	virtual int getPriority(BinaryViewRef data, const QString& filename);
	virtual QWidget* create(BinaryViewRef data, ViewFrame* viewFrame);
	static void init();
};