#pragma once

#include <QtWidgets/QAbstractScrollArea>

#include <QtWidgets/QComboBox>

#include <QtWidgets/QDialog>

#include <QtWidgets/QLineEdit>

#include <QtWidgets/QTableWidget>

#include <QtWidgets/QCheckBox>

#include <QtWidgets/QPushButton>

#include <QtWidgets/QStyledItemDelegate>

#include <QtCore/QSortFilterProxyModel>

#include "binaryninjaapi.h"

#include "notificationsdispatcher.h"

#include "render.h"

#include "sidebar.h"

#include "uitypes.h"

#include "fontsettings.h"

#include "viewframe.h"

class BINARYNINJAUIAPI MemoryMapItemDelegate : public QStyledItemDelegate

{

MemoryMapItemDelegate(QObject* parent = nullptr): QStyledItemDelegate(parent) {};

virtual void paint(QPainter* painter, const QStyleOptionViewItem& option, const QModelIndex& index) const override;

};

class BINARYNINJAUIAPI MemoryRegionDialog : public QDialog

{

QPushButton* m_selectFileButton;

QPushButton* m_acceptButton;

QPushButton* m_cancelButton;

QLineEdit* m_nameField;

QLineEdit* m_startField;

QLineEdit* m_lengthField;

QLineEdit* m_sourceOffsetField;

QLineEdit* m_sourceLengthField;

QLineEdit* m_patternField;

QLineEdit* m_fillByteField;

QCheckBox* m_flagRead;

QCheckBox* m_flagWrite;

QCheckBox* m_flagExec;

QComboBox* m_sourceSelection;

QLabel* m_fileContentsLabel;

QLabel* m_lengthLabel;

QLabel* m_sourceOffsetLabel;

QLabel* m_sourceLengthLabel;

QLabel* m_fileLabel;

QLabel* m_fileLabelText;

QLabel* m_patternLabel;

QLabel* m_fillByteLabel;

BinaryViewRef m_data;

SegmentRef m_segment;

std::optional<std::string> m_filePath;

void SelectFile();

void Submit();

MemoryRegionDialog(QWidget* parent, BinaryViewRef data, SegmentRef associatedSegment = nullptr);

};

class BINARYNINJAUIAPI SegmentDialog : public QDialog

{

QPushButton* m_acceptButton;

QPushButton* m_cancelButton;

QLineEdit* m_startField;

QLineEdit* m_lengthField;

QLineEdit* m_dataOffsetField;

QLineEdit* m_dataLengthField;

QCheckBox* m_flagRead;

QCheckBox* m_flagWrite;

QCheckBox* m_flagExec;

BinaryViewRef m_data;

SegmentRef m_segment;

void Submit();

SegmentDialog(QWidget* parent, BinaryViewRef data, SegmentRef segment = nullptr);

};

class BINARYNINJAUIAPI SectionDialog : public QDialog

{

QPushButton* m_acceptButton;

QPushButton* m_cancelButton;

QLineEdit* m_nameField;

QLineEdit* m_startField;

QLineEdit* m_lengthField;

QComboBox* m_semanticsField;

BinaryViewRef m_data;

SectionRef m_section;

void Submit();

void SetDefaultValues(const QString& name, uint64_t start, uint64_t length, BNSectionSemantics semantics);

SectionDialog(QWidget* parent, BinaryViewRef data, SectionRef section = nullptr);

};

enum class SegmentColumn : int {

};

class BINARYNINJAUIAPI SegmentModel : public QAbstractItemModel

{

BinaryViewRef m_data;

std::vector<SegmentRef> m_segments;

QHash<QPersistentModelIndex, bool> m_highlightedIndices;

SegmentModel(BinaryViewRef data, QObject* parent = nullptr);

~SegmentModel();

int columnCount(const QModelIndex& parent = QModelIndex()) const override;

int rowCount(const QModelIndex& parent = QModelIndex()) const override;

QModelIndex index(int row, int column, const QModelIndex& parent = QModelIndex()) const override;

QModelIndex parent(const QModelIndex& child) const override;

bool hasChildren(const QModelIndex&) const override;

QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const override;

bool setData(const QModelIndex& index, const QVariant& value, int role = Qt::EditRole) override;

QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override;

void updateSegments(std::vector<SegmentRef>&& segments);

};

class BINARYNINJAUIAPI SegmentWidget : public QWidget

{

Q_OBJECT

BinaryViewRef m_data;

QTableView* m_table;

SegmentModel* m_model;

QSortFilterProxyModel* m_proxyModel;

std::mutex m_updateMutex;

//void updateInfo();

void showContextMenu(const QPoint& point);

QMenu* createHeaderContextMenu(const QPoint& p);

void restoreDefaults();

void addMemoryRegion(SegmentRef segment);

void addSegment();

void editSegment(SegmentRef segment);

void disableSegment(SegmentRef segment);

void removeSegment(SegmentRef segment);

SegmentWidget(BinaryViewRef data, QWidget* parent = nullptr);

virtual ~SegmentWidget();

SegmentModel* model() { return m_model; }

void updateFont();

void highlightRelatedSegments(SectionRef section);

void currentRowChanged(const QModelIndex& current, const QModelIndex& previous);

Q_SIGNALS:

void currentSegmentChanged(SegmentRef current);

void addressDoubleClicked(uint64_t address);

void rawAddressDoubleClicked(uint64_t address);

};

enum class SectionColumn: int

};

class BINARYNINJAUIAPI SectionModel : public QAbstractItemModel

{

BinaryViewRef m_data;

std::vector<SectionRef> m_sections;

QHash<QPersistentModelIndex, bool> m_highlightedIndices;

SectionModel(BinaryViewRef data, QObject* parent = nullptr);

~SectionModel();

int columnCount(const QModelIndex& parent = QModelIndex()) const override;

int rowCount(const QModelIndex& parent = QModelIndex()) const override;

QModelIndex index(int row, int column, const QModelIndex& parent = QModelIndex()) const override;

QModelIndex parent(const QModelIndex& child) const override;

bool hasChildren(const QModelIndex&) const override;

QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const override;

bool setData(const QModelIndex& index, const QVariant& value, int role = Qt::EditRole) override;

QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override;

void updateSections(std::vector<SectionRef>&& sections);

};

class BINARYNINJAUIAPI SectionWidget : public QWidget

{

Q_OBJECT

BinaryViewRef m_data;

QTableView* m_table;

SectionModel* m_model;

QSortFilterProxyModel* m_proxyModel;

std::mutex m_updateMutex;

void showContextMenu(const QPoint& point);

void addSection();

void editSection(SectionRef section);

void removeSection(SectionRef section);

SectionWidget(BinaryViewRef data, QWidget* parent = nullptr);

virtual ~SectionWidget();

SectionModel* model() { return m_model; }

void updateFont();

void highlightRelatedSections(SegmentRef segment);

void currentRowChanged(const QModelIndex& current, const QModelIndex& previous);

Q_SIGNALS:

void currentSectionChanged(SectionRef current);

void addressDoubleClicked(uint64_t address);

};

class MemoryMapContainer;

class MemoryMapSidebarWidget;

class BINARYNINJAUIAPI MemoryMapView : public QWidget, public View

{

Q_OBJECT

BinaryViewRef m_data;

std::unique_ptr<NotificationsDispatcher> m_dispatcher = nullptr;

MemoryMapContainer* m_container;

SectionWidget* m_sectionWidget;

SegmentWidget* m_segmentWidget;

uint64_t m_currentOffset;

void navigateToAddress(uint64_t address);

void navigateToRawAddress(uint64_t address);

MemoryMapView(BinaryViewRef data, MemoryMapContainer* container);

virtual ~MemoryMapView();

BinaryViewRef getData() override { return m_data; }

uint64_t getCurrentOffset() override;

BNAddressRange getSelectionOffsets() override;

void setSelectionOffsets(BNAddressRange range) override;

bool navigate(uint64_t offset) override;

QFont getFont() override { return getMonospaceFont(this); }

void setCurrentOffset(uint64_t offset);

void updateFonts() override;

};

class BINARYNINJAUIAPI MemoryMapContainer : public QWidget, public ViewContainer

{

Q_OBJECT

friend class StringsView;

MemoryMapView* m_memoryMap;

MemoryMapSidebarWidget* m_widget;

UIActionHandler m_actionHandler;

MemoryMapContainer(BinaryViewRef data, MemoryMapSidebarWidget* parent);

virtual View* getView() override { return m_memoryMap; }

MemoryMapView* getMemoryMapView() { return m_memoryMap; }

virtual void focusInEvent(QFocusEvent* event) override;

void bindNavigationShortcuts();

void navigateBack();

void navigateForward();

};

class MemoryMapViewType : public ViewType

};

class BINARYNINJAUIAPI MemoryMapSidebarWidget : public SidebarWidget

{

Q_OBJECT

friend class MemoryMapView;

MemoryMapContainer* m_container;

MemoryMapSidebarWidget(BinaryViewRef data);

void focus() override;

};

class BINARYNINJAUIAPI MemoryMapSidebarWidgetType : public SidebarWidgetType

{

MemoryMapSidebarWidgetType();

SidebarWidgetLocation defaultLocation() const override { return SidebarWidgetLocation::LeftContent; }

SidebarContextSensitivity contextSensitivity() const override { return PerViewTypeSidebarContext; }

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

virtual bool canUseAsPane(SplitPaneWidget*, BinaryViewRef) const override { return true; }

virtual Pane* createPane(SplitPaneWidget* panes, BinaryViewRef data) override;

};