#pragma once

#include <gtkmm.h>

#include <vector>

#include <string>

#include "boost/filesystem.hpp"

#include <thread>

#include <mutex>

#include <atomic>

#include <unordered_map>

#include <unordered_set>

#include "git.h"

#include "dispatcher.h"

class Directories : public Gtk::ListViewText {

class DirectoryData {

public:

Gtk::TreeModel::Row row;

Glib::RefPtr<Gio::FileMonitor> monitor;

std::shared_ptr<Git::Repository> repository;

std::shared_ptr<sigc::connection> connection;

};

enum class PathType {KNOWN, UNKNOWN};

class TreeStore : public Gtk::TreeStore {

protected:

TreeStore()=default;

bool row_drop_possible_vfunc(const Gtk::TreeModel::Path &path, const Gtk::SelectionData &selection_data) const override;

bool drag_data_received_vfunc(const TreeModel::Path &path, const Gtk::SelectionData &selection_data) override;

bool drag_data_delete_vfunc (const Gtk::TreeModel::Path &path) override;

public:

class ColumnRecord : public Gtk::TreeModel::ColumnRecord {

public:

ColumnRecord() {

add(id);

add(name);

add(markup);

add(path);

add(type);

}

Gtk::TreeModelColumn<std::string> id;

Gtk::TreeModelColumn<std::string> name;

Gtk::TreeModelColumn<Glib::ustring> markup;

Gtk::TreeModelColumn<boost::filesystem::path> path;

Gtk::TreeModelColumn<PathType> type;

};

static Glib::RefPtr<TreeStore> create() {return Glib::RefPtr<TreeStore>(new TreeStore());}

};

Directories();

public:

static Directories &get() {

static Directories singleton;

return singleton;

}

~Directories() override;

void open(const boost::filesystem::path &dir_path="");

void update();

void on_save_file(const boost::filesystem::path &file_path);

void select(const boost::filesystem::path &path);

boost::filesystem::path path;

protected:

bool on_button_press_event(GdkEventButton *event) override;

private:

void add_or_update_path(const boost::filesystem::path &dir_path, const Gtk::TreeModel::Row &row, bool include_parent_paths);

void remove_path(const boost::filesystem::path &dir_path);

void colorize_path(boost::filesystem::path dir_path_, bool include_parent_paths);

Glib::RefPtr<Gtk::TreeStore> tree_store;

TreeStore::ColumnRecord column_record;

std::unordered_map<std::string, DirectoryData> directories;

Dispatcher dispatcher;

Gtk::Menu menu;

Gtk::MenuItem menu_item_new_file;

Gtk::MenuItem menu_item_new_folder;

Gtk::SeparatorMenuItem menu_item_separator;

Gtk::MenuItem menu_item_rename;

Gtk::MenuItem menu_item_delete;

Gtk::Menu menu_root;

Gtk::MenuItem menu_root_item_new_file;

Gtk::MenuItem menu_root_item_new_folder;

boost::filesystem::path menu_popup_row_path;

};