Unpack downloaded model file

Author: RandomShaper

addons/sketchfab/ModelDialog.gd

@@ -128,24 +128,46 @@ func _on_Download_pressed():
 	var file_regex = RegEx.new()
 	file_regex.compile("[^/]+?\\.zip")
 	var filename = file_regex.search(url).get_string()
+	var zip_path = "res://sketchfab/%s" % filename
 
 	downloader.connect("download_progressed", self, "_on_download_progressed")
-	downloader.request(path, null, { "download_to": "res://sketchfab/%s" % filename })
+	downloader.request(path, null, { "download_to": zip_path })
 	result = yield(downloader, "completed")
+	if !result:
+		return
 	downloader.term()
 	downloader = null
 
-	download.visible = true
-	progress.visible = false
-	size_label.visible = false
-	if result:
-		if result.ok && result.code == 200:
-			download.text = "Model downloaded!"
-			download.disabled = true
-		else:
-			OS.alert(
-				"Please check your network connectivity, free disk space and try again.",
-				"Download error")
+	if !result.ok || result.code != 200:
+		download.visible = true
+		progress.visible = false
+		size_label.visible = false
+		OS.alert(
+			"Please check your network connectivity, free disk space and try again.",
+			"Download error")
+		return
+
+	# Unpack
+
+	progress.percent_visible = false
+	size_label.text = "    Model downloaded! Unpacking..."
+	yield(get_tree(), "idle_frame")
+	if !get_tree():
+		return
+
+	var out = []
+	OS.execute(OS.get_executable_path(), [
+		"-s", ProjectSettings.globalize_path("res://addons/sketchfab/unzip.gd"),
+		"--zip-to-unpack %s" % ProjectSettings.globalize_path(zip_path),
+		"--no-window",
+		"--quit",
+	], true, out)
+	print(out)
+
+	size_label.text = "    Model unpacked into project!"
+
+	# Trigger import
+	get_tree().get_meta("__editor_interface").get_resource_filesystem().scan()
 
 func _on_download_progressed(bytes, total_bytes):
 	if !get_tree():

addons/sketchfab/plugin.gd

@@ -6,6 +6,7 @@ const Utils = preload("res://addons/sketchfab/Utils.gd")
 var main = preload("res://addons/sketchfab/Main.tscn").instance()
 
 func _enter_tree():
+	get_tree().set_meta("__editor_interface", get_editor_interface())
 	get_editor_interface().get_editor_viewport().add_child(main)
 	main.visible = false
 

addons/sketchfab/unzip.gd

@@ -0,0 +1,59 @@
+extends SceneTree
+
+const ARG_PREFIX = "--zip-to-unpack "
+
+func _init():
+	print("Unzipper started")
+
+	var zip_path
+	for arg in OS.get_cmdline_args():
+		if arg.begins_with(ARG_PREFIX):
+			zip_path = arg.right(ARG_PREFIX.length())
+			break
+
+	if !zip_path:
+		print("No file specified")
+		return
+
+	print("Unpacking %s..." % zip_path)
+
+	if !ProjectSettings.load_resource_pack(zip_path):
+		print("Package file not found")
+		return
+
+	var name_regex = RegEx.new()
+	name_regex.compile("([^/\\\\]+)\\.zip")
+	var base_name = name_regex.search(zip_path).get_string(1)
+
+	var out_path = zip_path.left(zip_path.find(base_name)) + base_name + "/"
+	Directory.new().make_dir(out_path)
+	unpack_dir("res://", out_path)
+
+	print("Done!")
+
+func unpack_dir(src_path, out_path):
+	print("Directory: %s -> %s" % [src_path, out_path])
+
+	var dir = Directory.new()
+	dir.open(src_path)
+	dir.list_dir_begin(true)
+
+	var file_name = dir.get_next()
+	while file_name != "":
+		if dir.current_is_dir():
+			var new_src_path = "%s%s/" % [src_path, file_name]
+			var new_out_path = "%s%s/" % [out_path, file_name]
+			Directory.new().make_dir(new_out_path)
+			unpack_dir(new_src_path, new_out_path)
+		else:
+			var file_src_path = "%s%s" % [src_path, file_name]
+			var file_out_path = "%s%s" % [out_path, file_name]
+			print("File: %s -> %s" % [file_src_path, file_out_path])
+			var file = File.new()
+			file.open(file_src_path, File.READ)
+			var data = file.get_buffer(file.get_len())
+			file.close()
+			file.open(file_out_path, File.WRITE)
+			file.store_buffer(data)
+			file.close()
+		file_name = dir.get_next()