#pragma once

#include <QtWidgets/QAbstractScrollArea>
#include <QtCore/QTimer>
#include "binaryninjaapi.h"
#include "viewframe.h"
#include "render.h"
#include "commentdialog.h"
#include "menus.h"
#include "uicontext.h"

/*!

	\defgroup tokenizedtextview TokenizedTextView
 	\ingroup uiapi
*/

/*!

    \ingroup tokenizedtextview
*/
class BINARYNINJAUIAPI TokenizedTextViewHistoryEntry : public HistoryEntry
{
	size_t m_topLine, m_cursorLine;
	HighlightTokenState m_highlight;

  public:
	size_t getTopLine() const { return m_topLine; }
	size_t getCursorLine() const { return m_cursorLine; }
	const HighlightTokenState& getHighlightTokenState() const { return m_highlight; }

	void setTopLine(size_t line) { m_topLine = line; }
	void setCursorLine(size_t line) { m_cursorLine = line; }
	void setHighlightTokenState(const HighlightTokenState& state) { m_highlight = state; }

	virtual QString getDescription() const override;
	virtual Json::Value serialize() const override;
	virtual bool deserialize(const Json::Value& value) override;
};

/*!

    \ingroup tokenizedtextview
*/
class BINARYNINJAUIAPI TokenizedTextView :
    public QAbstractScrollArea,
    public View,
    public BinaryNinja::BinaryDataNotification
{
	Q_OBJECT

	BinaryViewRef m_data;
	FunctionRef m_function;

	RenderContext m_render;
	int m_cols, m_rows;
	int m_wheelDelta;
	bool m_updatingScrollBar;

	bool m_updatesRequired;

	int m_cursorLine;
	HighlightTokenState m_highlight;
	uint64_t m_navByRefTarget;
	bool m_navByRef = false;

	std::vector<BinaryNinja::LinearDisassemblyLine> m_lines;
	DisassemblySettingsRef m_settings;

	QTimer* m_updateTimer;

	ContextMenuManager* m_contextMenuManager;
	QPointer<CommentDialog> m_commentDialog;

	void adjustSize(int width, int height);

	void scrollLines(int count);

	void bindActions();
	void bindDynamicActions();
	void getHexDumpLineBytes(const BinaryNinja::LinearDisassemblyLine& line, size_t& skippedBytes, size_t& totalBytes, size_t& totalCols);

	void setSectionSemantics(const std::string& name, BNSectionSemantics semantics);

	void viewInHexEditor();
	void viewInGraph();
	void goToAddress();
	void defineNameAtAddr(uint64_t addr);
	void defineName();
	void undefine();
	void createFunc(const UIActionContext& context);
	void defineFuncName();
	void editFunctionProperties();
	void reanalyze();
	void comment();
	void commentAccepted();

	void convertToNop();
	void alwaysBranch();
	void invertBranch();
	void skipAndReturnZero();
	void skipAndReturnValue();

	void makeInt8();
	void makeInt16();
	void makeInt32();
	void makeInt64();
	void toggleIntSize();
	void toggleIntSign();
	void makeFloat32();
	void makeFloat64();
	void toggleFloatSize();
	void makePtr();
	void makeString(size_t charSize = 1);
	void changeType();
	void inferStructureType(const UIActionContext& context);

	//! Get the length of of the string (if there is one) starting at the
	//! given address. String type is assumed to be UTF-8 by default, but the
	//! `charSize` parameter can be set to 2 or 4 to look for UTF-16 or
	//! UTF-32 string, respectively.
	//!
	//! Returns the length of the string in bytes, NOT the number of characters.
	size_t getStringLength(uint64_t startAddr, size_t charSize = 1);

	void setInstructionHighlight(BNHighlightColor color);
	void setBlockHighlight(BNHighlightColor color);

  private Q_SLOTS:
	void scrollBarMoved(int value);
	void scrollBarAction(int action);
	void updateTimerEvent();

  public:
	explicit TokenizedTextView(QWidget* parent, BinaryViewRef data,
	    const std::vector<BinaryNinja::LinearDisassemblyLine>& lines =
	        std::vector<BinaryNinja::LinearDisassemblyLine>());
	virtual ~TokenizedTextView();

	virtual BinaryViewRef getData() override { return m_data; }
	virtual uint64_t getCurrentOffset() override;
	virtual BNAddressRange getSelectionOffsets() override;
	virtual SelectionInfoForXref getSelectionForXref() override;
	virtual void setSelectionOffsets(BNAddressRange range) override;
	virtual FunctionRef getCurrentFunction() override;
	virtual BasicBlockRef getCurrentBasicBlock() override;
	virtual ArchitectureRef getCurrentArchitecture() override;
	virtual bool navigate(uint64_t pos) override;

	virtual BinaryNinja::Ref<HistoryEntry> getHistoryEntry() override;
	void populateDefaultHistoryEntry(TokenizedTextViewHistoryEntry* entry);
	virtual void navigateToHistoryEntry(BinaryNinja::Ref<HistoryEntry> entry) override;
	virtual bool canDisplayAs(const UIActionContext& context, const BNIntegerDisplayType displayType) override;

	virtual void OnBinaryDataWritten(BinaryNinja::BinaryView* data, uint64_t offset, size_t len) override;
	virtual void OnBinaryDataInserted(BinaryNinja::BinaryView* data, uint64_t offset, size_t len) override;
	virtual void OnBinaryDataRemoved(BinaryNinja::BinaryView* data, uint64_t offset, uint64_t len) override;
	virtual void OnAnalysisFunctionAdded(BinaryNinja::BinaryView* view, BinaryNinja::Function* func) override;
	virtual void OnAnalysisFunctionRemoved(BinaryNinja::BinaryView* view, BinaryNinja::Function* func) override;
	virtual void OnAnalysisFunctionUpdated(BinaryNinja::BinaryView* view, BinaryNinja::Function* func) override;
	virtual void OnDataVariableAdded(BinaryNinja::BinaryView* view, const BinaryNinja::DataVariable& var) override;
	virtual void OnDataVariableRemoved(BinaryNinja::BinaryView* view, const BinaryNinja::DataVariable& var) override;
	virtual void OnDataVariableUpdated(BinaryNinja::BinaryView* view, const BinaryNinja::DataVariable& var) override;
	virtual void OnDataMetadataUpdated(BinaryNinja::BinaryView* view, uint64_t offset) override;
	virtual void OnTagUpdated(BinaryNinja::BinaryView* data, const BinaryNinja::TagReference& tagRef) override;

	virtual void updateFonts() override;

	virtual void followPointer();

	virtual void cut() override;
	virtual void copy(TransformRef xform = nullptr) override;
	virtual void paste(TransformRef xform = nullptr) override;
	virtual void copyAddress() override;

	virtual HighlightTokenState getHighlightTokenState() override { return m_highlight; }

	virtual bool goToReference(FunctionRef func, uint64_t source, uint64_t target) override;
	QFont getFont() override { return m_render.getFont(); }

	static void registerActions();

	virtual void updateLines();
	void setLines(const std::vector<BinaryNinja::LinearDisassemblyLine>& lines);
	void setUpdatedLines(const std::vector<BinaryNinja::LinearDisassemblyLine>& lines);

	void setFunction(FunctionRef func);

  protected:
	virtual void resizeEvent(QResizeEvent* event) override;
	virtual void paintEvent(QPaintEvent* event) override;
	virtual void wheelEvent(QWheelEvent* event) override;
	virtual void mousePressEvent(QMouseEvent* event) override;
	virtual void mouseDoubleClickEvent(QMouseEvent* event) override;
};