#pragma once

#include <QtWidgets/QAbstractScrollArea>

#include <QtWidgets/QComboBox>

#include <QtWidgets/QDialog>

#include <QtWidgets/QLineEdit>

#include <optional>

#include "render.h"

#include "sidebar.h"

#include "uitypes.h"

#define STACK_VIEW_UPDATE_CHECK_INTERVAL 200

class BINARYNINJAUIAPI CreateStackVariableDialog : public QDialog

{

Q_OBJECT

BinaryViewRef m_data;

FunctionRef m_func;

QLineEdit* m_offsetField;

QLineEdit* m_nameField;

QComboBox* m_typeDropdown;

//! Automatically update/sanitize the offset and name fields after input.

void autoFillFields();

protected:

void accept() override;

public:

CreateStackVariableDialog(QWidget* parent, BinaryViewRef data, FunctionRef func, int64_t initialOffset = 0);

};

class StackViewLine

};

enum class Direction

};

class BINARYNINJAUIAPI StackView : public QAbstractScrollArea, public View, public UIContextNotification

{

Q_OBJECT

ViewFrame* m_view;

BinaryViewRef m_data;

FunctionRef m_func;

BinaryNinja::AdvancedFunctionAnalysisDataRequestor m_analysisRequestor;

RenderContext m_renderer;

std::vector<StackViewLine> m_lines;

HighlightTokenState m_highlight;

size_t m_lineIndex;

size_t m_tokenIndex;

bool m_needFirstFocus;

bool m_positionAware;

// Selection state

size_t m_selectionStartLine;

size_t m_selectionStartToken;

bool m_hasSelection;

bool m_isDragging;

//! Bind and register all stack view actions.

void setupActions();

//! Internal refresh method, rebuilds all lines.

void rebuildLines();

//! Refresh the internal highlight token state.

//!

//! \param shouldUpdateStatus Should UIContext::updateStatus() be called?

void refreshHighlight(bool shouldUpdateStatus = true);

//! Find the end of a stack void given a start offset.

int64_t findVoidEnd(int64_t start) const;

//! Get the beginning of the current selection (min position)

void getSelectionRange(size_t& startLine, size_t& startToken, size_t& endLine, size_t& endToken) const;

protected:

void paintEvent(QPaintEvent* event) override;

void mousePressEvent(QMouseEvent* event) override;

void mouseMoveEvent(QMouseEvent* event) override;

void mouseReleaseEvent(QMouseEvent* event) override;

void mouseDoubleClickEvent(QMouseEvent* event) override;

public:

StackView(ViewFrame* view, BinaryViewRef data);

~StackView();

//! Refresh the stack view's content.

void refresh();

//! Move the cursor to approximate clicked position.

void moveCursorToMouse(QMouseEvent* event, bool isSelecting);

//! Ensure the cursor is visible by adjusting the scroll position.

void ensureCursorIsVisible();

//! Move the cursor via the keyboard.

void moveCursor(Direction dir);

//! Ensure the cursor is not selecting any prohibited tokens.

void sanitizeCursor(Direction preference);

//! Get the selected StackViewLine.

const StackViewLine* selectedLine() const;

//! Rename the variable belonging to the selected line.

void renameVariable();

//! Change the type of the variable belonging to the selected line.

void retypeVariable();

//! Undefine the variable belonging to the selected line.

void undefineVariable();

//! Show the "Create Variable" dialog.

void showCreateVariableDialog();

//! Create an integer of the given size at the cursor position. Pass `0`

//! for `size` to cycle through integer sizes automatically.

void quickCreateIntegerAtCursor(size_t size);

//! Toggle the sign of the integer at the cursor position.

void quickInvertIntegerSignAtCursor();

//! Create an float of the given size at the cursor position. Pass `0`

//! for `size` to cycle through float sizes automatically.

void quickCreateFloatAtCursor(size_t size);

//! Create a pointer the cursor.

void quickCreatePointerAtCursor();

//! Create an array at the cursor, spanning until the next stack variable.

void quickCreateArrayAtCursor();

//! Create a new struct at the cursor, spanning until the next stack variable.

void quickCreateStructAtCursor();

//! Show the dialog to switch which base register is used for offsets

void chooseBaseRegister();

//! Show the dialog to pick a constant base offset

void chooseBaseOffset();

//! Set the base offset to a register (or clear on nullopt)

void setBaseRegister(ArchitectureRef arch, std::optional<std::string> regName);

//! Get the base offset

int64_t getBaseOffset(FunctionRef func);

//! Set the base offset to a constant (or clear on nullopt)

void setBaseOffset(FunctionRef func, std::optional<int64_t> offset);

//! Set whether we show only the vars relevant for the current address

void setPositionAware(bool positionAware);

//! Override the default event handler so we can have nice tooltips.

bool event(QEvent* event) override;

//! Get the current base name/offset using the currently selected base register

std::pair<std::string, int64_t> calculateBaseOffset() const;

BinaryViewRef getData() override;

uint64_t getCurrentOffset() override;

bool canCopy() override { return selectedLine() != nullptr; }

bool canCopyAddress() override { return false; };

bool canPaste() override { return false; };

void copy(TransformRef xform = nullptr) override;

void setSelectionOffsets(BNAddressRange range) override;

bool navigate(uint64_t offset) override;

QFont getFont() override;

void updateFonts() override;

virtual void OnNewSelectionForXref(UIContext* context, ViewFrame* frame, View* view, const SelectionInfoForXref& selection) override;

};

class BINARYNINJAUIAPI StackViewSidebarWidget : public SidebarWidget

{

Q_OBJECT

QWidget* m_header;

StackView* m_stackView;

public:

StackViewSidebarWidget(ViewFrame* view, BinaryViewRef data);

QWidget* headerWidget() override { return m_header; }

void refresh();

void focus() override { refresh(); }

void notifyFontChanged() override { m_stackView->updateFonts(); }

};

class BINARYNINJAUIAPI StackViewSidebarWidgetType : public SidebarWidgetType

{

public:

StackViewSidebarWidgetType();

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

virtual SidebarWidgetLocation defaultLocation() const override { return SidebarWidgetLocation::RightContent; }

};