pandemonium_engine/editor/filesystem_dock.h

#ifndef FILESYSTEM_DOCK_H
#define FILESYSTEM_DOCK_H

/*************************************************************************/
/*  filesystem_dock.h                                                    */
/*************************************************************************/
/*                         This file is part of:                         */
/*                          PANDEMONIUM ENGINE                           */
/*             https://github.com/Relintai/pandemonium_engine            */
/*************************************************************************/
/* Copyright (c) 2022-present Péter Magyar.                              */
/* Copyright (c) 2014-2022 Godot Engine contributors (cf. AUTHORS.md).   */
/* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur.                 */
/*                                                                       */
/* Permission is hereby granted, free of charge, to any person obtaining */
/* a copy of this software and associated documentation files (the       */
/* "Software"), to deal in the Software without restriction, including   */
/* without limitation the rights to use, copy, modify, merge, publish,   */
/* distribute, sublicense, and/or sell copies of the Software, and to    */
/* permit persons to whom the Software is furnished to do so, subject to */
/* the following conditions:                                             */
/*                                                                       */
/* The above copyright notice and this permission notice shall be        */
/* included in all copies or substantial portions of the Software.       */
/*                                                                       */
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,       */
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF    */
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY  */
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,  */
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE     */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                */
/*************************************************************************/

#include "scene/gui/box_container.h"

#include "scene/gui/split_container.h"

#include "core/containers/list.h"
#include "core/containers/rb_map.h"
#include "core/containers/rb_set.h"
#include "core/containers/vector.h"
#include "core/math/vector2.h"
#include "core/object/object.h"
#include "core/object/reference.h"
#include "core/os/dir_access.h"
#include "core/os/thread.h"
#include "core/string/string_name.h"
#include "core/string/ustring.h"
#include "core/variant/variant.h"

class EditorNode;
class Button;
class ConfirmationDialog;
class Control;
class CreateDialog;
class DependencyEditor;
class DependencyEditorOwners;
class DependencyRemoveDialog;
class EditorDirDialog;
class EditorFileSystemDirectory;
class InputEvent;
class ItemList;
class LineEdit;
class MenuButton;
class PopupMenu;
class ProgressBar;
class ScriptCreateDialog;
class Texture;
class TextureRect;
class Tree;
class TreeItem;
class DirectoryCreateDialog;
class ToolButton;

class FileSystemDock : public VBoxContainer {
	GDCLASS(FileSystemDock, VBoxContainer);

public:
	enum FileListDisplayMode {
		FILE_LIST_DISPLAY_THUMBNAILS,
		FILE_LIST_DISPLAY_LIST
	};

	enum DisplayMode {
		DISPLAY_MODE_TREE_ONLY,
		DISPLAY_MODE_SPLIT,
	};

	enum SplitMode {
		SPLIT_MODE_HORIZONTAL = 0,
		SPLIT_MODE_VERTICAL,
	};

	enum DockMode {
		DOCK_MODE_DOCK,
		DOCK_MODE_BOTTOM_BAR,
	};

	enum FileSortOption {
		FILE_SORT_NAME = 0,
		FILE_SORT_NAME_REVERSE,
		FILE_SORT_TYPE,
		FILE_SORT_TYPE_REVERSE,
		FILE_SORT_MODIFIED_TIME,
		FILE_SORT_MODIFIED_TIME_REVERSE,
		FILE_SORT_MAX,
	};

private:
	enum FileMenu {
		FILE_OPEN = 0,
		FILE_INHERIT,
		FILE_MAIN_SCENE,
		FILE_INSTANCE,
		FILE_ADD_FAVORITE,
		FILE_REMOVE_FAVORITE,
		FILE_DEPENDENCIES,
		FILE_OWNERS,
		FILE_MOVE,
		FILE_RENAME,
		FILE_REMOVE,
		FILE_DUPLICATE,
		FILE_REIMPORT,
		FILE_INFO,
		FILE_NEW_FOLDER,
		FILE_NEW_SCRIPT,
		FILE_NEW_SCENE,
		FILE_SHOW_IN_EXPLORER,
		FILE_COPY_PATH,
		FILE_NEW_RESOURCE,
		FOLDER_EXPAND_ALL,
		FOLDER_COLLAPSE_ALL,
		FILE_NEW_CUSTOM_ENTRY_START,
	};

	FileSortOption file_sort;

	ToolButton *bottom_panel_tool_button;

	VBoxContainer *scanning_vb;
	ProgressBar *scanning_progress;

	CSplitContainer *split_box;

	VBoxContainer *file_list_vb;

	EditorNode *editor;
	RBSet<String> favorites;

	Button *button_toggle_display_mode;
	Button *button_reload;
	Button *button_file_list_display_mode;
	Button *button_hist_next;
	Button *button_hist_prev;
	LineEdit *current_path;

	HBoxContainer *toolbar2_hbc;
	LineEdit *tree_search_box;
	MenuButton *tree_button_sort;

	LineEdit *file_list_search_box;
	MenuButton *file_list_button_sort;

	String searched_string;
	Vector<String> uncollapsed_paths_before_search;

	TextureRect *search_icon;
	HBoxContainer *path_hb;

	FileListDisplayMode file_list_display_mode;
	DisplayMode display_mode;
	DisplayMode old_display_mode;

	DockMode dock_mode;
	DockMode applied_dock_mode;

	SplitMode split_mode;

	PopupMenu *file_list_popup;
	PopupMenu *tree_popup;

	DependencyEditor *deps_editor;
	DependencyEditorOwners *owners_editor;
	DependencyRemoveDialog *remove_dialog;

	EditorDirDialog *move_dialog;
	ConfirmationDialog *rename_dialog;
	LineEdit *rename_dialog_text;
	ConfirmationDialog *duplicate_dialog;
	LineEdit *duplicate_dialog_text;
	DirectoryCreateDialog *make_dir_dialog;
	ConfirmationDialog *make_scene_dialog;
	LineEdit *make_scene_dialog_text;
	ConfirmationDialog *overwrite_dialog;
	ScriptCreateDialog *make_script_dialog;
	CreateDialog *new_resource_dialog;

	bool always_show_folders;

	class FileOrFolder {
	public:
		String path;
		bool is_file;

		FileOrFolder() :
				path(""),
				is_file(false) {}
		FileOrFolder(const String &p_path, bool p_is_file) :
				path(p_path),
				is_file(p_is_file) {}
	};

	FileOrFolder to_rename;
	FileOrFolder to_duplicate;
	Vector<FileOrFolder> to_move;
	String to_move_path;

	Vector<String> history;
	int history_pos;
	int history_max_size;

	String path;

	bool initialized;

	bool updating_tree;
	int tree_update_id;
	Tree *tree;
	ItemList *files;
	bool import_dock_needs_update;

	Ref<Texture> _get_tree_item_icon(bool p_is_valid, String p_file_type);
	bool _create_tree(TreeItem *p_parent, EditorFileSystemDirectory *p_dir, Vector<String> &uncollapsed_paths, bool p_select_in_favorites, bool p_unfold_path = false);
	Vector<String> _compute_uncollapsed_paths();
	void _update_tree(const Vector<String> &p_uncollapsed_paths = Vector<String>(), bool p_uncollapse_root = false, bool p_select_in_favorites = false, bool p_unfold_path = false);
	void _navigate_to_path(const String &p_path, bool p_select_in_favorites = false);

	void _file_list_gui_input(Ref<InputEvent> p_event);
	void _tree_gui_input(Ref<InputEvent> p_event);

	void _update_file_list(bool p_keep_selection);
	void _toggle_file_display();
	void _set_file_display(bool p_active);
	void _fs_changed();

	void _select_file(const String &p_path, bool p_select_in_favorites = false);
	void _tree_activate_file();
	void _file_list_activate_file(int p_idx);
	void _file_multi_selected(int p_index, bool p_selected);
	void _tree_multi_selected(Object *p_item, int p_column, bool p_selected);

	void _update_import_dock();

	void _get_all_items_in_dir(EditorFileSystemDirectory *efsd, Vector<String> &files, Vector<String> &folders) const;
	void _find_remaps(EditorFileSystemDirectory *efsd, const RBMap<String, String> &renames, Vector<String> &to_remaps) const;
	void _try_move_item(const FileOrFolder &p_item, const String &p_new_path, RBMap<String, String> &p_file_renames, RBMap<String, String> &p_folder_renames);
	void _try_duplicate_item(const FileOrFolder &p_item, const String &p_new_path) const;
	void _update_dependencies_after_move(const RBMap<String, String> &p_renames) const;
	void _update_resource_paths_after_move(const RBMap<String, String> &p_renames) const;
	void _save_scenes_after_move(const RBMap<String, String> &p_renames) const;
	void _update_favorites_list_after_move(const RBMap<String, String> &p_files_renames, const RBMap<String, String> &p_folders_renames) const;
	void _update_project_settings_after_move(const RBMap<String, String> &p_renames) const;

	void _file_removed(String p_file);
	void _folder_removed(String p_folder);

	void _resource_created() const;
	void _make_scene_confirm();
	void _rename_operation_confirm();
	void _duplicate_operation_confirm();
	void _move_with_overwrite();
	Vector<String> _check_existing();
	void _move_operation_confirm(const String &p_to_path, bool overwrite = false);

	void _tree_rmb_option(int p_option);
	void _file_list_rmb_option(int p_option);
	void _file_option(int p_option, const Vector<String> &p_selected);

	void _fw_history();
	void _bw_history();
	void _update_history();
	void _push_to_history();

	void _set_scanning_mode();
	void _rescan();

	void _toggle_split_mode(bool p_active);

	void _focus_current_search_box();
	void _search_changed(const String &p_text, const Control *p_from);

	MenuButton *_create_file_menu_button();
	void _file_sort_popup(int p_id);

	void _file_and_folders_fill_popup(PopupMenu *p_popup, Vector<String> p_paths, bool p_display_path_dependent_options = true);
	void _tree_rmb_select(const Vector2 &p_pos);
	void _tree_rmb_empty(const Vector2 &p_pos);
	void _file_list_rmb_select(int p_item, const Vector2 &p_pos);
	void _file_list_rmb_pressed(const Vector2 &p_pos);
	void _tree_empty_selected();

	struct FileInfo {
		String name;
		String path;
		StringName type;
		Vector<String> sources;
		bool import_broken;
		uint64_t modified_time;

		bool operator<(const FileInfo &fi) const {
			return NaturalNoCaseComparator()(name, fi.name);
		}
	};

	struct FileInfoTypeComparator;
	struct FileInfoModifiedTimeComparator;

	void _sort_file_info_list(List<FileSystemDock::FileInfo> &r_file_list);

	void _search(EditorFileSystemDirectory *p_path, List<FileInfo> *matches, int p_max_items);

	void _set_current_path_text(const String &p_path);

	Variant get_drag_data_fw(const Point2 &p_point, Control *p_from);
	bool can_drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from) const;
	void drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from);
	void _get_drag_target_folder(String &target, bool &target_favorites, const Point2 &p_point, Control *p_from) const;

	void _preview_invalidated(const String &p_path);
	void _file_list_thumbnail_done(const String &p_path, const Ref<Texture> &p_preview, const Ref<Texture> &p_small_preview, const Variant &p_udata);
	void _tree_thumbnail_done(const String &p_path, const Ref<Texture> &p_preview, const Ref<Texture> &p_small_preview, const Variant &p_udata);

	void _update_display_mode(bool p_force = false);

	Vector<String> _tree_get_selected(bool remove_self_inclusion = true);

	Vector<String> _remove_self_included_paths(Vector<String> selected_strings);

	struct CustomPopupCreationEntry {
		String entry_text;
		String theme_icon_name;
		String theme_icon_category;
		int id;
	};

	Vector<CustomPopupCreationEntry> _custom_popup_creation_entries;

protected:
	void _notification(int p_what);
	static void _bind_methods();

public:
	String get_selected_path() const;

	String get_current_path() const;
	void navigate_to_path(const String &p_path);
	void focus_on_filter();

	void fix_dependencies(const String &p_for_file);

	int get_split_offset() {
		return split_box->get_split_offset();
	}
	void set_split_offset(int p_offset) {
		split_box->set_split_offset(p_offset);
	}
	void select_file(const String &p_file);

	void set_display_mode(DisplayMode p_display_mode);
	DisplayMode get_display_mode() { return display_mode; }

	void set_file_sort(FileSortOption p_file_sort);
	FileSortOption get_file_sort() { return file_sort; }

	void set_file_list_display_mode(FileListDisplayMode p_mode);
	FileListDisplayMode get_file_list_display_mode() { return file_list_display_mode; }

	void set_dock_mode(DockMode p_dock_mode);
	DockMode get_dock_mode() { return dock_mode; }

	void set_applied_dock_mode(DockMode p_dock_mode);
	DockMode get_applied_dock_mode() { return applied_dock_mode; }

	void set_split_mode(SplitMode p_split_mode);
	SplitMode get_split_mode();

	int add_custom_popup_creation_entry(const String &entry_text, const String &theme_icon_name = "", const String &theme_icon_category = "");
	void remove_custom_popup_creation_entry(const int id);

	void ensure_visible();
	void set_bottom_panel_tool_button(ToolButton *fs_button);

	void on_editor_save_and_restart();

	FileSystemDock(EditorNode *p_editor);
	~FileSystemDock();
};

#endif // SCENES_DOCK_H