#pragma once

#include <QDialog>
#include <QStringList>
#include <QLineEdit>
#include <QTreeView>
#include <QPlainTextEdit>
#include <QLabel>
#include <QDialogButtonBox>
#include <QModelIndex>
#include <QSplitter>

#include "binaryninjaapi.h"
#include "uitypes.h"

#include <vector>

class ContainerOpenRequest;

class ContainerTreeModel : public QAbstractItemModel
{
	Q_OBJECT

	struct Node
	{
		QString displayName;          // GetFileName() (or synthesized for root)
		QString type;                 // GetTransformName() or "Leaf"/"Root"
		QString breadcrumb;           // human-readable path "a ▸ b ▸ c"
		QStringList pathSegments;     // list of filenames from root to this node
		QString size;
		TransformContextRef ctx;
		Node* parent = nullptr;
		std::vector<std::unique_ptr<Node>> children;
	};

	TransformSessionRef m_session;
	std::unique_ptr<Node> m_root;
	QLocale m_locale;

	const Node* nodeFromIndex(const QModelIndex& index) const;
	static QString joinBreadcrumb(const QStringList& segments);
	void createChildren(Node* parentNode, const std::vector<TransformContextRef>& children, const QStringList& parentSegments = {});

public:
	enum Columns { ColName, ColType, ColSize, ColPath, ColCount };

	ContainerTreeModel(TransformSessionRef session, QObject* parent = nullptr);

	int columnCount(const QModelIndex& parent = {}) const override;
	QModelIndex index(int row, int column, const QModelIndex& parent = {}) const override;
	QModelIndex parent(const QModelIndex& child) const override;
	int rowCount(const QModelIndex& parent = {}) const override;
	QVariant data(const QModelIndex& index, int role) const override;
	QVariant headerData(int section, Qt::Orientation orientation, int role) const override;
	Qt::ItemFlags flags(const QModelIndex& index) const override;

	QString getDisplayName(const QModelIndex& index) const;
	TransformContextRef getTransformContext(const QModelIndex& index) const;

	QStringList pathFor(const QModelIndex& index) const;
	void selectNode(const QModelIndex& index);
	void rebuild();
};


class AllColumnsFilterProxyModel : public QSortFilterProxyModel
{
	Q_OBJECT

public:
	explicit AllColumnsFilterProxyModel(QObject* parent = nullptr);

protected:
	bool filterAcceptsRow(int sourceRow, const QModelIndex& sourceParent) const override;
};


class BINARYNINJAUIAPI ContainerBrowser : public QDialog
{
	Q_OBJECT

	TransformSessionRef m_session;

	ContainerTreeModel* m_model;

	QLineEdit* m_filter = nullptr;
	QTreeView* m_tree = nullptr;
	QPlainTextEdit* m_preview = nullptr;
	QSplitter* m_splitter = nullptr;
	QLabel* m_status = nullptr;
	QLabel* m_extractionStatus = nullptr;
	QDialogButtonBox* m_buttons = nullptr;
	AllColumnsFilterProxyModel* m_proxy = nullptr;

	QPushButton* m_openWithOptionsButton = nullptr;

	QStringList m_pendingSelectionPath;
	QStringList m_selectedPaths;
	int m_lastPreviewSize = 0;
	int m_dialogWidth = 0;
	bool m_openWithOptionsRequested = false;

	void connectSignals();
	void updatePreviewForIndex(const QModelIndex& proxyIndex);
	bool requiresPassword(TransformContextRef context);
	bool requiresExtraction(TransformContextRef context);
	void extractItem(TransformContextRef context);
	void promptForPassword(TransformContextRef context, bool tryCachedPassword = false);
	void showContextMenu(const QPoint& position);
	static QString toHexDump(const QByteArray& data, int bytesPerLine = 16);
	static QString formatMetadata(BinaryNinja::Ref<BinaryNinja::Metadata> metadata, int indent = 0);
	QModelIndex findNodeByPath(const QStringList& path);
	QModelIndex findFirstLeaf();
	QModelIndex findLeafByName(const QString& name);
	void selectNodeByPath(const QStringList& path);
	void selectLeafByName(const QString& name);

public:
	ContainerBrowser(TransformSessionRef session, QWidget* parent = nullptr);

	QStringList selectedPaths() const { return m_selectedPaths; }
	bool openWithOptionsRequested() const { return m_openWithOptionsRequested; }

	static std::vector<TransformContextRef> openContainerFile(const QString& path, bool forceShowDialog = false, bool* outOpenWithOptions = nullptr);

	// Show the container browser dialog for the given open request.
	// Returns the selected contexts, or empty if the user cancelled.
	static std::vector<TransformContextRef> showBrowser(ContainerOpenRequest& request, bool* outOpenWithOptions = nullptr);
};