#include <unordered_set>

#include "src/widgets/FilePane.h"
#include "src/widgets/MainFrame.h"

extern MainFrame *Glob_main_frame;

enum
{
	Icon_File,
	Icon_FolderClosed,
	Icon_FolderOpened
};

FilePane::FilePane(
	wxWindow* parent, 
	wxWindowID id, 
	const wxPoint& pos, 
	const wxSize& size, 
	long style, 
	const wxString &name
) : wxTreeListCtrl(parent, id, pos, size, style, name)
{
	this->BindEvents();
	this->InitImageList();
	this->SetImageList(this->img_list);
	
	wxString defaultPath(".");
	wxFileName filename(defaultPath);
	filename.MakeAbsolute(defaultPath);

	this->CreateTree(defaultPath);
	
	this->AppendColumn("", 
		wxCOL_WIDTH_AUTOSIZE, 
		wxALIGN_LEFT, 
		wxCOL_RESIZABLE | wxCOL_SORTABLE);
	
}

FilePane::~FilePane()
{
	delete this->img_list;
}

void FilePane::BindEvents()
{
	this->Bind(wxEVT_TREELIST_ITEM_EXPANDING, &FilePane::OpenFolder, this, wxID_ANY);
	this->Bind(wxEVT_TREELIST_ITEM_ACTIVATED, &FilePane::OpenFileInEditor, this, wxID_ANY);
}

void FilePane::OpenFolder(wxTreeListEvent& event)
{
	
	wxTreeListItem item = event.GetItem();
	const wxString path = this->GetItemText(item, 0);
	
	wxLogDebug("Opening sidebar dir: %s", path);
}

/**
 * Iterates through the specified folder and creates the tree view
 * 
 * @access private
 */
void FilePane::CreateTree(const wxString &path)
{
	wxTreeListItem root = this->GetRootItem();

	auto *files = new wxArrayString();
	wxFileName rootPath(path);
	rootPath.MakeAbsolute();

	this->base_path = rootPath.GetPath();

	wxLogDebug("Base Path for Sidebar: %s", this->base_path);

	wxDir::GetAllFiles(this->base_path, files);

	for (const wxString &file: *files)
	{
		wxFileName fileName(file);

		// If the file is at the root, add it to the tree
		if (fileName.GetDirCount() == rootPath.GetDirCount())
		{
			this->AddDirFiles(root, path);
		}

		auto dir = std::string(fileName.GetPath());

		if (dir.empty() || BaseName(this->base_path) == BaseName(dir))
		{
			continue;
		}

		// Append the folder to the tree
		wxString wdir = wxString(dir);
		this->AddDirToTree(root, BaseName(dir), wxString(""));
	}

	delete files;
}

/**
 * Recursively create directory tree from list of files
 * 
 * @access private
 */
void FilePane::AddDirToTree(wxTreeListItem &root, const wxString &path, const wxString &parent)
{
	auto pathBase = BaseName(path);
	auto fullPath = this->base_path;

	if ( ! parent.empty())
	{
		fullPath += "/";

		auto par = parent.Clone();
		par.Replace((wxString)this->base_path, "");

		fullPath += par.ToStdString();
	}

	if ( ! fullPath.Contains(pathBase))
	{
		fullPath += "/";
		fullPath += pathBase;
	}

	wxFileName parentDir(fullPath);
	parentDir.MakeAbsolute();
	parentDir.Normalize();

	auto parentDirs = parentDir.GetDirs();

	wxString wFullPath(fullPath);

	wxLogInfo("Rendering Dir Tree for %s, full path: %s", path, fullPath);

	auto *files = new wxArrayString();
	wxDir::GetAllFiles(fullPath, files);

	for (const wxString &item: *files)
	{
		wxFileName fileName(item);
		fileName.MakeRelativeTo(fullPath);
		auto dir = std::string(fileName.GetPath());

		wxLogDebug("File %s in %s", fileName.GetFullName(), fullPath);
		wxLogDebug("Dir %s in %s", dir, fullPath);

		// Add files
		if (fileName.GetDirCount() == 1)
		{
			this->AddDirFiles(root, fullPath);
		}

		// Stop early if folder exists
		/*auto it = this->dir_set.find(std::string(dir));
		if (it != this->dir_set.end())
		{
			continue;
		}

		if (fileName.GetDirCount() < parentDir.GetDirCount() || ( ! item.Contains(wFullPath)))
		{
			continue;
		}

		auto dirs = fileName.GetDirs();

		// Remove the last folder from the filename to pass as the parent dir
		fileName.RemoveLastDir();
		auto parentPath = fileName.GetPath();
		parentPath.Replace((wxString)this->base_path, "");
		parentPath.Replace("/", "");

		if (parentDirs.GetCount() == dirs.GetCount())
		{
			continue;
		}

		for (auto i = 0; i < dirs.GetCount(); i++)
		{
			if (dirs[i] == "")
			{
				break;
			}

			// Skip path segments that already exist
			if (parentDirs.GetCount() > i && parentDirs[i] == dirs[i])
			{
				continue;
			}

			 if (parentDirs.GetCount() > i && parentDirs[i] != dirs[i])
			{
				wxLogWarning("Wat?! Where'd this path segment come from? :%s not in %s", dirs[i], parentPath);
				continue;
			}

			wxLogDebug("Path segment to Add: %s, Base Dir: %s", dirs[i], fullPath);

			auto fileData = new wxStringClientData();
			fileData->SetData(dir);

			auto dir_label = BaseName(dir);
			auto dir_node = this->AppendItem(root, dir_label, Icon_FolderClosed, Icon_FolderOpened, fileData);

			// wxLogDebug("Recursing for dir %s, from parent %s", parentPath, fullPath);

			// this->AddDirToTree(dir_node, dir_label, parentPath);

			break;
		} */

		/* auto dir = std::string(fileName.GetPath());

		// Remove the last folder from the filename to pass as the parent dir
		fileName.RemoveLastDir();
		auto parentPath = fileName.GetPath();
		parentPath.Replace((wxString)this->base_path, "");

		wxString wdir(dir);
		wxFileName dirName(dir);

		if (( ! wdir.Contains((wxString) fullPath)) || dirName.GetDirCount() != parentDir.GetDirCount())
		{
			continue;
		}

		// Stop early if folder exists
		auto it = this->dir_set.find(std::string(dir));
		if (it != this->dir_set.end())
		{
			continue;
		}

		auto fileData = new wxStringClientData();
		fileData->SetData(dir);

		auto dir_label = BaseName(dir);
		auto dir_node = this->AppendItem(root, dir_label, Icon_FolderClosed, Icon_FolderOpened, fileData);

		this->dir_set.insert(std::string(dir));

		wxLogDebug("Recursing for dir %s, from parent %s", parentPath, fullPath);

		this->AddDirToTree(dir_node, dir_label, parentPath); */
	}

	// delete files;
}

void FilePane::AddDirFiles(wxTreeListItem &root, const wxString &path)
{
	auto *files = new wxArrayString();
	wxDir::GetAllFiles(path, files, wxEmptyString, wxDIR_FILES);

	wxFileName rootPath(path);
	rootPath.MakeAbsolute();

	for (const wxString &item: *files)
	{
		wxFileName fileName(item);
		fileName.MakeAbsolute();

		// If the file is in another folder, don't add it here!
		if (fileName.GetDirCount() != rootPath.GetDirCount())
		{
			continue;
		}

		auto it = this->file_set.find(std::string(fileName.GetFullPath()));
		if (it != this->file_set.end())
		{
			continue;
		}

		auto fileData = new wxStringClientData();
		fileData->SetData(fileName.GetFullPath());

		auto fileLabel = BaseName(fileName.GetFullName());

		this->AppendItem(root, fileLabel, Icon_File, Icon_File, fileData);
		this->file_set.insert(std::string(fileName.GetFullPath()));
	}

	delete files;
}

/**
 * Open a file you double-click on the file list
 */
void FilePane::OpenFileInEditor(wxTreeListEvent& event)
{
	wxTreeListItem item = event.GetItem();
	auto data = (wxStringClientData*)this->GetItemData(item);
	const wxString& path = data->GetData();

	wxLogDebug("Opening file from sidebar: %s", path);

	wxString path_arr [1] = { path };
	auto files = wxArrayString(1, *path_arr);
	Glob_main_frame->OpenFiles(files);
}

/**
 * Create the image list object for the file pane widget
 */
void FilePane::InitImageList()
{
	wxSize iconSize = wxArtProvider::GetSizeHint(wxART_LIST);
	
	if (iconSize == wxDefaultSize)
	{
		iconSize = wxSize(16, 16);
	}
	
	this->img_list = new wxImageList(iconSize.x, iconSize.y);
	
	static const char* icons[] = 
	{
		wxART_NORMAL_FILE,
		wxART_FOLDER,
		wxART_FOLDER_OPEN
	};
	
	for (auto icon : icons)
	{
		this->img_list->Add(
			wxArtProvider::GetIcon(icon, wxART_LIST, iconSize)
		);
	}
}