9e328bb5b7
File handling APIs are typically considered part of I/O, and we did have most `FileAccess` implementations in `core/io` already.
345 lines
9.9 KiB
C++
345 lines
9.9 KiB
C++
/*************************************************************************/
|
|
/* editor_asset_installer.cpp */
|
|
/*************************************************************************/
|
|
/* This file is part of: */
|
|
/* GODOT ENGINE */
|
|
/* https://godotengine.org */
|
|
/*************************************************************************/
|
|
/* Copyright (c) 2007-2021 Juan Linietsky, Ariel Manzur. */
|
|
/* Copyright (c) 2014-2021 Godot Engine contributors (cf. AUTHORS.md). */
|
|
/* */
|
|
/* 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 "editor_asset_installer.h"
|
|
|
|
#include "core/io/dir_access.h"
|
|
#include "core/io/file_access.h"
|
|
#include "core/io/zip_io.h"
|
|
#include "editor_node.h"
|
|
#include "progress_dialog.h"
|
|
|
|
void EditorAssetInstaller::_update_subitems(TreeItem *p_item, bool p_check, bool p_first) {
|
|
if (p_check) {
|
|
if (p_item->get_custom_color(0) == Color()) {
|
|
p_item->set_checked(0, true);
|
|
}
|
|
} else {
|
|
p_item->set_checked(0, false);
|
|
}
|
|
|
|
if (p_item->get_first_child()) {
|
|
_update_subitems(p_item->get_first_child(), p_check);
|
|
}
|
|
|
|
if (!p_first && p_item->get_next()) {
|
|
_update_subitems(p_item->get_next(), p_check);
|
|
}
|
|
}
|
|
|
|
void EditorAssetInstaller::_uncheck_parent(TreeItem *p_item) {
|
|
if (!p_item) {
|
|
return;
|
|
}
|
|
|
|
bool any_checked = false;
|
|
TreeItem *item = p_item->get_first_child();
|
|
while (item) {
|
|
if (item->is_checked(0)) {
|
|
any_checked = true;
|
|
break;
|
|
}
|
|
item = item->get_next();
|
|
}
|
|
|
|
if (!any_checked) {
|
|
p_item->set_checked(0, false);
|
|
_uncheck_parent(p_item->get_parent());
|
|
}
|
|
}
|
|
|
|
void EditorAssetInstaller::_item_edited() {
|
|
if (updating) {
|
|
return;
|
|
}
|
|
|
|
TreeItem *item = tree->get_edited();
|
|
if (!item) {
|
|
return;
|
|
}
|
|
|
|
String path = item->get_metadata(0);
|
|
|
|
updating = true;
|
|
if (path == String() || item == tree->get_root()) { //a dir or root
|
|
_update_subitems(item, item->is_checked(0), true);
|
|
}
|
|
|
|
if (item->is_checked(0)) {
|
|
while (item) {
|
|
item->set_checked(0, true);
|
|
item = item->get_parent();
|
|
}
|
|
} else {
|
|
_uncheck_parent(item->get_parent());
|
|
}
|
|
updating = false;
|
|
}
|
|
|
|
void EditorAssetInstaller::open(const String &p_path, int p_depth) {
|
|
package_path = p_path;
|
|
Set<String> files_sorted;
|
|
|
|
FileAccess *src_f = nullptr;
|
|
zlib_filefunc_def io = zipio_create_io_from_file(&src_f);
|
|
|
|
unzFile pkg = unzOpen2(p_path.utf8().get_data(), &io);
|
|
if (!pkg) {
|
|
error->set_text(TTR("Error opening package file, not in ZIP format."));
|
|
return;
|
|
}
|
|
|
|
int ret = unzGoToFirstFile(pkg);
|
|
|
|
while (ret == UNZ_OK) {
|
|
//get filename
|
|
unz_file_info info;
|
|
char fname[16384];
|
|
unzGetCurrentFileInfo(pkg, &info, fname, 16384, nullptr, 0, nullptr, 0);
|
|
|
|
String name = fname;
|
|
files_sorted.insert(name);
|
|
|
|
ret = unzGoToNextFile(pkg);
|
|
}
|
|
|
|
Map<String, Ref<Texture2D>> extension_guess;
|
|
{
|
|
extension_guess["png"] = tree->get_theme_icon("ImageTexture", "EditorIcons");
|
|
extension_guess["jpg"] = tree->get_theme_icon("ImageTexture", "EditorIcons");
|
|
extension_guess["atlastex"] = tree->get_theme_icon("AtlasTexture", "EditorIcons");
|
|
extension_guess["scn"] = tree->get_theme_icon("PackedScene", "EditorIcons");
|
|
extension_guess["tscn"] = tree->get_theme_icon("PackedScene", "EditorIcons");
|
|
extension_guess["gdshader"] = tree->get_theme_icon("Shader", "EditorIcons");
|
|
extension_guess["gd"] = tree->get_theme_icon("GDScript", "EditorIcons");
|
|
extension_guess["vs"] = tree->get_theme_icon("VisualScript", "EditorIcons");
|
|
}
|
|
|
|
Ref<Texture2D> generic_extension = tree->get_theme_icon("Object", "EditorIcons");
|
|
|
|
unzClose(pkg);
|
|
|
|
updating = true;
|
|
tree->clear();
|
|
TreeItem *root = tree->create_item();
|
|
root->set_cell_mode(0, TreeItem::CELL_MODE_CHECK);
|
|
root->set_checked(0, true);
|
|
root->set_icon(0, tree->get_theme_icon("folder", "FileDialog"));
|
|
root->set_text(0, "res://");
|
|
root->set_editable(0, true);
|
|
Map<String, TreeItem *> dir_map;
|
|
|
|
for (Set<String>::Element *E = files_sorted.front(); E; E = E->next()) {
|
|
String path = E->get();
|
|
int depth = p_depth;
|
|
bool skip = false;
|
|
while (depth > 0) {
|
|
int pp = path.find("/");
|
|
if (pp == -1) {
|
|
skip = true;
|
|
break;
|
|
}
|
|
path = path.substr(pp + 1, path.length());
|
|
depth--;
|
|
}
|
|
|
|
if (skip || path == String()) {
|
|
continue;
|
|
}
|
|
|
|
bool isdir = false;
|
|
|
|
if (path.ends_with("/")) {
|
|
//a directory
|
|
path = path.substr(0, path.length() - 1);
|
|
isdir = true;
|
|
}
|
|
|
|
int pp = path.rfind("/");
|
|
|
|
TreeItem *parent;
|
|
if (pp == -1) {
|
|
parent = root;
|
|
} else {
|
|
String ppath = path.substr(0, pp);
|
|
ERR_CONTINUE(!dir_map.has(ppath));
|
|
parent = dir_map[ppath];
|
|
}
|
|
|
|
TreeItem *ti = tree->create_item(parent);
|
|
ti->set_cell_mode(0, TreeItem::CELL_MODE_CHECK);
|
|
ti->set_checked(0, true);
|
|
ti->set_editable(0, true);
|
|
if (isdir) {
|
|
dir_map[path] = ti;
|
|
ti->set_text(0, path.get_file() + "/");
|
|
ti->set_icon(0, tree->get_theme_icon("folder", "FileDialog"));
|
|
ti->set_metadata(0, String());
|
|
} else {
|
|
String file = path.get_file();
|
|
String extension = file.get_extension().to_lower();
|
|
if (extension_guess.has(extension)) {
|
|
ti->set_icon(0, extension_guess[extension]);
|
|
} else {
|
|
ti->set_icon(0, generic_extension);
|
|
}
|
|
ti->set_text(0, file);
|
|
|
|
String res_path = "res://" + path;
|
|
if (FileAccess::exists(res_path)) {
|
|
ti->set_custom_color(0, tree->get_theme_color("error_color", "Editor"));
|
|
ti->set_tooltip(0, vformat(TTR("%s (Already Exists)"), res_path));
|
|
ti->set_checked(0, false);
|
|
} else {
|
|
ti->set_tooltip(0, res_path);
|
|
}
|
|
|
|
ti->set_metadata(0, res_path);
|
|
}
|
|
|
|
status_map[E->get()] = ti;
|
|
}
|
|
popup_centered_ratio();
|
|
updating = false;
|
|
}
|
|
|
|
void EditorAssetInstaller::ok_pressed() {
|
|
FileAccess *src_f = nullptr;
|
|
zlib_filefunc_def io = zipio_create_io_from_file(&src_f);
|
|
|
|
unzFile pkg = unzOpen2(package_path.utf8().get_data(), &io);
|
|
if (!pkg) {
|
|
error->set_text(TTR("Error opening package file, not in ZIP format."));
|
|
return;
|
|
}
|
|
|
|
int ret = unzGoToFirstFile(pkg);
|
|
|
|
Vector<String> failed_files;
|
|
|
|
ProgressDialog::get_singleton()->add_task("uncompress", TTR("Uncompressing Assets"), status_map.size());
|
|
|
|
int idx = 0;
|
|
while (ret == UNZ_OK) {
|
|
//get filename
|
|
unz_file_info info;
|
|
char fname[16384];
|
|
ret = unzGetCurrentFileInfo(pkg, &info, fname, 16384, nullptr, 0, nullptr, 0);
|
|
|
|
String name = fname;
|
|
|
|
if (status_map.has(name) && status_map[name]->is_checked(0)) {
|
|
String path = status_map[name]->get_metadata(0);
|
|
if (path == String()) { // a dir
|
|
|
|
String dirpath;
|
|
TreeItem *t = status_map[name];
|
|
while (t) {
|
|
dirpath = t->get_text(0) + dirpath;
|
|
t = t->get_parent();
|
|
}
|
|
|
|
if (dirpath.ends_with("/")) {
|
|
dirpath = dirpath.substr(0, dirpath.length() - 1);
|
|
}
|
|
|
|
DirAccess *da = DirAccess::create(DirAccess::ACCESS_RESOURCES);
|
|
da->make_dir(dirpath);
|
|
memdelete(da);
|
|
|
|
} else {
|
|
Vector<uint8_t> data;
|
|
data.resize(info.uncompressed_size);
|
|
|
|
//read
|
|
unzOpenCurrentFile(pkg);
|
|
unzReadCurrentFile(pkg, data.ptrw(), data.size());
|
|
unzCloseCurrentFile(pkg);
|
|
|
|
FileAccess *f = FileAccess::open(path, FileAccess::WRITE);
|
|
if (f) {
|
|
f->store_buffer(data.ptr(), data.size());
|
|
memdelete(f);
|
|
} else {
|
|
failed_files.push_back(path);
|
|
}
|
|
|
|
ProgressDialog::get_singleton()->task_step("uncompress", path, idx);
|
|
}
|
|
}
|
|
|
|
idx++;
|
|
ret = unzGoToNextFile(pkg);
|
|
}
|
|
|
|
ProgressDialog::get_singleton()->end_task("uncompress");
|
|
unzClose(pkg);
|
|
|
|
if (failed_files.size()) {
|
|
String msg = TTR("The following files failed extraction from package:") + "\n\n";
|
|
for (int i = 0; i < failed_files.size(); i++) {
|
|
if (i > 15) {
|
|
msg += "\n" + vformat(TTR("And %s more files."), itos(failed_files.size() - i));
|
|
break;
|
|
}
|
|
msg += failed_files[i];
|
|
}
|
|
if (EditorNode::get_singleton() != nullptr) {
|
|
EditorNode::get_singleton()->show_warning(msg);
|
|
}
|
|
} else {
|
|
if (EditorNode::get_singleton() != nullptr) {
|
|
EditorNode::get_singleton()->show_warning(TTR("Package installed successfully!"), TTR("Success!"));
|
|
}
|
|
}
|
|
EditorFileSystem::get_singleton()->scan_changes();
|
|
}
|
|
|
|
void EditorAssetInstaller::_bind_methods() {
|
|
}
|
|
|
|
EditorAssetInstaller::EditorAssetInstaller() {
|
|
VBoxContainer *vb = memnew(VBoxContainer);
|
|
add_child(vb);
|
|
|
|
tree = memnew(Tree);
|
|
vb->add_margin_child(TTR("Package Contents:"), tree, true);
|
|
tree->connect("item_edited", callable_mp(this, &EditorAssetInstaller::_item_edited));
|
|
|
|
error = memnew(AcceptDialog);
|
|
add_child(error);
|
|
get_ok_button()->set_text(TTR("Install"));
|
|
set_title(TTR("Package Installer"));
|
|
|
|
updating = false;
|
|
|
|
set_hide_on_ok(true);
|
|
}
|