#ifndef IMGUI_DISK_MANIPULATOR_HH

#define IMGUI_DISK_MANIPULATOR_HH

#include "ImGuiPart.hh"

#include "DiskManipulator.hh"

#include "SectorAccessibleDisk.hh"

#include <ctime>

#include <map>

#include <memory>

#include <optional>

#include <string>

#include <string_view>

#include <variant>

#include <vector>

namespace openmsx {

class DiskContainer;

class DiskPartition;

class MSXtar;

class ImGuiDiskManipulator final : public ImGuiPart

{

public:

	using ImGuiPart::ImGuiPart;

	void paint(MSXMotherBoard* motherBoard) override;

public:

	bool show = false;

private:

	struct FileInfo {

		std::string filename;

		size_t size = 0;

		time_t modified = 0;

		MSXDirEntry::AttribValue attrib{0};

		bool isDirectory = false;

		bool isSelected = false;

	};

	struct Nop {};

	struct ChangeDir { std::string_view name; };

	struct Delete { std::string_view name; };

	struct Rename { std::string_view name; };

	using Action = std::variant<Nop, ChangeDir, Delete, Rename>;

	struct DrivePartitionTar {

		DiskContainer* drive;

		std::unique_ptr<DiskPartition> disk; // will often be the full disk

		std::unique_ptr<MSXtar> tar;

	};

	[[nodiscard]] DiskContainer* getDrive();

	[[nodiscard]] std::optional<DiskManipulator::DriveAndPartition> getDriveAndDisk();

	[[nodiscard]] std::optional<DrivePartitionTar> getMsxStuff();

	[[nodiscard]] bool isValidMsxDirectory(DrivePartitionTar& stuff, const std::string& dir) const;

	[[nodiscard]] std::string getDiskImageName();

	[[nodiscard]] std::vector<FileInfo> dirMSX(DrivePartitionTar& stuff);

	void refreshMsx(DrivePartitionTar& stuff);

	void refreshHost();

	void checkSort(std::vector<FileInfo>& files, bool& forceSort) const;

	[[nodiscard]] Action drawTable(std::vector<FileInfo>& files, int& lastClickIdx, bool& forceSort, bool msxSide) const;

	void insertMsxDisk();

	void exportDiskImage();

	void msxParentDirectory();

	void hostParentDirectory();

	void msxRefresh();

	void hostRefresh();

	[[nodiscard]] bool setupTransferHostToMsx(DrivePartitionTar& stuff);

	void executeTransferHostToMsx(DrivePartitionTar& stuff);

	void transferMsxToHost(DrivePartitionTar& stuff);

private:

	struct PartitionSize {

		unsigned count;

		enum Unit : int { KB, MB };

		int unit = KB;

		[[nodiscard]] unsigned asSectorCount() const {

			return count * (((unit == KB) ? 1024 : (1024 * 1024)) / SectorAccessibleDisk::SECTOR_SIZE);

		}

	};

	std::vector<FileInfo> msxFileCache;

	std::vector<FileInfo> hostFileCache;

	std::string selectedDrive = "virtual_drive";

	std::string msxDir = "/", editMsxDir = "/";

	std::string hostDir, editHostDir;

	std::string editModal;

	std::string renameFrom;

	int msxLastClick = -1;

	int hostLastClick = -1;

	bool hostNeedRefresh = true;

	bool hostForceSort = false;

	enum TransferHostToMsxPhase : uint8_t {

		IDLE,

		CHECK,

		EXECUTE_PRESERVE,

		EXECUTE_OVERWRITE,

	} transferHostToMsxPhase = IDLE;

	std::vector<FileInfo> existingFiles;

	std::vector<FileInfo> existingDirs;

	// std::less<> is better, but runs into a gcc-16 bug (fixed 2026/09/04):

	//   https://gcc.gnu.org/bugzilla/show_bug.cgi?id=126716

	//std::map<std::string, std::vector<FileInfo>, std::less<>> duplicateEntries;

	std::map<std::string, std::vector<FileInfo>> duplicateEntries;

	enum NewDiskType : int { UNPARTITIONED = 0, PARTITIONED = 1 };

	int newDiskType = UNPARTITIONED;

	int bootType = static_cast<int>(MSXBootSectorType::DOS2);

	PartitionSize unpartitionedSize = {.count = 720, .unit = PartitionSize::KB};

	std::vector<PartitionSize> partitionSizes;

};

} // namespace openmsx

#endif