xiandie/addons/dialogue_manager/plugin.gd
2025-06-16 16:40:11 +08:00

434 lines
16 KiB
GDScript

@tool
extends EditorPlugin
const MainView = preload("./views/main_view.tscn")
var import_plugin: DMImportPlugin
var export_plugin: DMExportPlugin
var inspector_plugin: DMInspectorPlugin
var translation_parser_plugin: DMTranslationParserPlugin
var main_view
var dialogue_cache: DMCache
func _enable_plugin() -> void:
add_autoload_singleton("DialogueManager", get_plugin_path() + "/dialogue_manager.gd")
func _disable_plugin() -> void:
remove_autoload_singleton("DialogueManager")
func _enter_tree() -> void:
if Engine.is_editor_hint():
Engine.set_meta("DialogueManagerPlugin", self)
DMSettings.prepare()
dialogue_cache = DMCache.new()
Engine.set_meta("DMCache", dialogue_cache)
import_plugin = DMImportPlugin.new()
add_import_plugin(import_plugin)
export_plugin = DMExportPlugin.new()
add_export_plugin(export_plugin)
inspector_plugin = DMInspectorPlugin.new()
add_inspector_plugin(inspector_plugin)
translation_parser_plugin = DMTranslationParserPlugin.new()
add_translation_parser_plugin(translation_parser_plugin)
main_view = MainView.instantiate()
EditorInterface.get_editor_main_screen().add_child(main_view)
_make_visible(false)
main_view.add_child(dialogue_cache)
_update_localization()
EditorInterface.get_file_system_dock().files_moved.connect(_on_files_moved)
EditorInterface.get_file_system_dock().file_removed.connect(_on_file_removed)
add_tool_menu_item("Create copy of dialogue example balloon...", _copy_dialogue_balloon)
# Automatically swap the script on the example balloon depending on if dotnet is being used.
if not FileAccess.file_exists("res://tests/test_basic_dialogue.gd"):
var plugin_path: String = get_plugin_path()
var balloon_file_names: PackedStringArray = ["example_balloon.tscn", "small_example_balloon.tscn"]
for balloon_file_name: String in balloon_file_names:
var balloon_path: String = plugin_path + "/example_balloon/" + balloon_file_name
var balloon_content: String = FileAccess.get_file_as_string(balloon_path)
if "example_balloon.gd" in balloon_content and DMSettings.check_for_dotnet_solution():
balloon_content = balloon_content \
# Replace script path with the C# one
.replace("example_balloon.gd", "ExampleBalloon.cs") \
# Replace script UID with the C# one
.replace(ResourceUID.id_to_text(ResourceLoader.get_resource_uid(plugin_path + "/example_balloon/example_balloon.gd")), ResourceUID.id_to_text(ResourceLoader.get_resource_uid(plugin_path + "/example_balloon/ExampleBalloon.cs")))
var balloon_file: FileAccess = FileAccess.open(balloon_path, FileAccess.WRITE)
balloon_file.store_string(balloon_content)
balloon_file.close()
elif "ExampleBalloon.cs" in balloon_content and not DMSettings.check_for_dotnet_solution():
balloon_content = balloon_content \
# Replace script path with the GDScript one
.replace("ExampleBalloon.cs", "example_balloon.gd") \
# Replace script UID with the GDScript one
.replace(ResourceUID.id_to_text(ResourceLoader.get_resource_uid(plugin_path + "/example_balloon/ExampleBalloon.cs")), ResourceUID.id_to_text(ResourceLoader.get_resource_uid(plugin_path + "/example_balloon/example_balloon.gd")))
var balloon_file: FileAccess = FileAccess.open(balloon_path, FileAccess.WRITE)
balloon_file.store_string(balloon_content)
balloon_file.close()
# Automatically make any changes to the known custom balloon if there is one.
var balloon_path: String = DMSettings.get_setting(DMSettings.BALLOON_PATH, "")
if balloon_path != "" and FileAccess.file_exists(balloon_path):
var is_small_window: bool = ProjectSettings.get_setting("display/window/size/viewport_width") < 400
var example_balloon_file_name: String = "small_example_balloon.tscn" if is_small_window else "example_balloon.tscn"
var example_balloon_path: String = get_plugin_path() + "/example_balloon/" + example_balloon_file_name
var contents: String = FileAccess.get_file_as_string(balloon_path)
var has_changed: bool = false
# Make sure the current balloon has a UID unique from the example balloon's
var example_balloon_uid: String = ResourceUID.id_to_text(ResourceLoader.get_resource_uid(example_balloon_path))
var balloon_uid: String = ResourceUID.id_to_text(ResourceLoader.get_resource_uid(balloon_path))
if example_balloon_uid == balloon_uid:
var new_balloon_uid: String = ResourceUID.id_to_text(ResourceUID.create_id())
contents = contents.replace(example_balloon_uid, new_balloon_uid)
has_changed = true
# Make sure the example balloon copy has the correct renaming of the responses menu
if "reponses" in contents:
contents = contents.replace("reponses", "responses")
has_changed = true
# Save any changes
if has_changed:
var balloon_file: FileAccess = FileAccess.open(balloon_path, FileAccess.WRITE)
balloon_file.store_string(contents)
balloon_file.close()
func _exit_tree() -> void:
remove_import_plugin(import_plugin)
import_plugin = null
remove_export_plugin(export_plugin)
export_plugin = null
remove_inspector_plugin(inspector_plugin)
inspector_plugin = null
remove_translation_parser_plugin(translation_parser_plugin)
translation_parser_plugin = null
if is_instance_valid(main_view):
main_view.queue_free()
Engine.remove_meta("DialogueManagerPlugin")
Engine.remove_meta("DMCache")
EditorInterface.get_file_system_dock().files_moved.disconnect(_on_files_moved)
EditorInterface.get_file_system_dock().file_removed.disconnect(_on_file_removed)
remove_tool_menu_item("Create copy of dialogue example balloon...")
func _has_main_screen() -> bool:
return true
func _make_visible(next_visible: bool) -> void:
if is_instance_valid(main_view):
main_view.visible = next_visible
func _get_plugin_name() -> String:
return "Dialogue"
func _get_plugin_icon() -> Texture2D:
return load(get_plugin_path() + "/assets/icon.svg")
func _handles(object) -> bool:
var editor_settings: EditorSettings = EditorInterface.get_editor_settings()
var external_editor: String = editor_settings.get_setting("text_editor/external/exec_path")
var use_external_editor: bool = editor_settings.get_setting("text_editor/external/use_external_editor") and external_editor != ""
if object is DialogueResource and use_external_editor and DMSettings.get_user_value("open_in_external_editor", false):
var project_path: String = ProjectSettings.globalize_path("res://")
var file_path: String = ProjectSettings.globalize_path(object.resource_path)
OS.create_process(external_editor, [project_path, file_path])
return false
return object is DialogueResource
func _edit(object) -> void:
if is_instance_valid(main_view) and is_instance_valid(object):
main_view.open_resource(object)
func _apply_changes() -> void:
if is_instance_valid(main_view):
main_view.apply_changes()
_update_localization()
func _save_external_data() -> void:
if dialogue_cache != null:
dialogue_cache.reimport_files()
func _build() -> bool:
# If this is the dotnet Godot then we need to check if the solution file exists
DMSettings.check_for_dotnet_solution()
# Ignore errors in other files if we are just running the test scene
if DMSettings.get_user_value("is_running_test_scene", true): return true
if dialogue_cache != null:
dialogue_cache.reimport_files()
var files_with_errors = dialogue_cache.get_files_with_errors()
if files_with_errors.size() > 0:
for dialogue_file in files_with_errors:
push_error("You have %d error(s) in %s" % [dialogue_file.errors.size(), dialogue_file.path])
EditorInterface.edit_resource(load(files_with_errors[0].path))
main_view.show_build_error_dialog()
return false
return true
## Get the shortcuts used by the plugin
func get_editor_shortcuts() -> Dictionary:
var shortcuts: Dictionary = {
toggle_comment = [
_create_event("Ctrl+K"),
_create_event("Ctrl+Slash")
],
delete_line = [
_create_event("Ctrl+Shift+K")
],
move_up = [
_create_event("Alt+Up")
],
move_down = [
_create_event("Alt+Down")
],
save = [
_create_event("Ctrl+Alt+S")
],
close_file = [
_create_event("Ctrl+W")
],
find_in_files = [
_create_event("Ctrl+Shift+F")
],
run_test_scene = [
_create_event("Ctrl+F5")
],
text_size_increase = [
_create_event("Ctrl+Equal")
],
text_size_decrease = [
_create_event("Ctrl+Minus")
],
text_size_reset = [
_create_event("Ctrl+0")
]
}
var paths = EditorInterface.get_editor_paths()
var settings
if FileAccess.file_exists(paths.get_config_dir() + "/editor_settings-4.3.tres"):
settings = load(paths.get_config_dir() + "/editor_settings-4.3.tres")
elif FileAccess.file_exists(paths.get_config_dir() + "/editor_settings-4.tres"):
settings = load(paths.get_config_dir() + "/editor_settings-4.tres")
else:
return shortcuts
for s in settings.get("shortcuts"):
for key in shortcuts:
if s.name == "script_text_editor/%s" % key or s.name == "script_editor/%s" % key:
shortcuts[key] = []
for event in s.shortcuts:
if event is InputEventKey:
shortcuts[key].append(event)
return shortcuts
func _create_event(string: String) -> InputEventKey:
var event: InputEventKey = InputEventKey.new()
var bits = string.split("+")
event.keycode = OS.find_keycode_from_string(bits[bits.size() - 1])
event.shift_pressed = bits.has("Shift")
event.alt_pressed = bits.has("Alt")
if bits.has("Ctrl") or bits.has("Command"):
event.command_or_control_autoremap = true
return event
## Get the editor shortcut that matches an event
func get_editor_shortcut(event: InputEventKey) -> String:
var shortcuts: Dictionary = get_editor_shortcuts()
for key in shortcuts:
for shortcut in shortcuts.get(key, []):
if event.as_text().split(" ")[0] == shortcut.as_text().split(" ")[0]:
return key
return ""
## Get the current version
func get_version() -> String:
var config: ConfigFile = ConfigFile.new()
config.load(get_plugin_path() + "/plugin.cfg")
return config.get_value("plugin", "version")
## Get the current path of the plugin
func get_plugin_path() -> String:
return get_script().resource_path.get_base_dir()
## Update references to a moved file
func update_import_paths(from_path: String, to_path: String) -> void:
dialogue_cache.move_file_path(from_path, to_path)
# Reopen the file if it's already open
if main_view.current_file_path == from_path:
if to_path == "":
main_view.close_file(from_path)
else:
main_view.current_file_path = ""
main_view.open_file(to_path)
# Update any other files that import the moved file
var dependents = dialogue_cache.get_files_with_dependency(from_path)
for dependent in dependents:
dependent.dependencies.remove_at(dependent.dependencies.find(from_path))
dependent.dependencies.append(to_path)
# Update the live buffer
if main_view.current_file_path == dependent.path:
main_view.code_edit.text = main_view.code_edit.text.replace(from_path, to_path)
main_view.open_buffers[main_view.current_file_path].pristine_text = main_view.code_edit.text
# Open the file and update the path
var file: FileAccess = FileAccess.open(dependent.path, FileAccess.READ)
var text = file.get_as_text().replace(from_path, to_path)
file.close()
file = FileAccess.open(dependent.path, FileAccess.WRITE)
file.store_string(text)
file.close()
func _update_localization() -> void:
if not DMSettings.get_setting(DMSettings.UPDATE_POT_FILES_AUTOMATICALLY, true):
return
var dialogue_files = dialogue_cache.get_files()
# Add any new files to POT generation
var files_for_pot: PackedStringArray = ProjectSettings.get_setting("internationalization/locale/translations_pot_files", [])
var files_for_pot_changed: bool = false
for path in dialogue_files:
if not files_for_pot.has(path):
files_for_pot.append(path)
files_for_pot_changed = true
# Remove any POT references that don't exist any more
for i in range(files_for_pot.size() - 1, -1, -1):
var file_for_pot: String = files_for_pot[i]
if file_for_pot.get_extension() == "dialogue" and not dialogue_files.has(file_for_pot):
files_for_pot.remove_at(i)
files_for_pot_changed = true
# Update project settings if POT changed
if files_for_pot_changed:
ProjectSettings.set_setting("internationalization/locale/translations_pot_files", files_for_pot)
ProjectSettings.save()
### Callbacks
func _copy_dialogue_balloon() -> void:
var scale: float = EditorInterface.get_editor_scale()
var directory_dialog: FileDialog = FileDialog.new()
var label: Label = Label.new()
label.text = "Dialogue balloon files will be copied into chosen directory."
directory_dialog.get_vbox().add_child(label)
directory_dialog.file_mode = FileDialog.FILE_MODE_OPEN_DIR
directory_dialog.min_size = Vector2(600, 500) * scale
directory_dialog.dir_selected.connect(func(path):
var plugin_path: String = get_plugin_path()
var is_dotnet: bool = DMSettings.check_for_dotnet_solution()
var balloon_path: String = path + ("/Balloon.tscn" if is_dotnet else "/balloon.tscn")
var balloon_script_path: String = path + ("/DialogueBalloon.cs" if is_dotnet else "/balloon.gd")
# Copy the balloon scene file and change the script reference
var is_small_window: bool = ProjectSettings.get_setting("display/window/size/viewport_width") < 400
var example_balloon_file_name: String = "small_example_balloon.tscn" if is_small_window else "example_balloon.tscn"
var example_balloon_path: String = plugin_path + "/example_balloon/" + example_balloon_file_name
var example_balloon_script_file_name: String = "ExampleBalloon.cs" if is_dotnet else "example_balloon.gd"
var example_balloon_script_uid: String = ResourceUID.id_to_text(ResourceLoader.get_resource_uid(plugin_path + "/example_balloon/example_balloon.gd"))
var example_balloon_uid: String = ResourceUID.id_to_text(ResourceLoader.get_resource_uid(example_balloon_path))
# Copy the script file
var file: FileAccess = FileAccess.open(plugin_path + "/example_balloon/" + example_balloon_script_file_name, FileAccess.READ)
var file_contents: String = file.get_as_text()
if is_dotnet:
file_contents = file_contents.replace("class ExampleBalloon", "class DialogueBalloon")
else:
file_contents = file_contents.replace("class_name DialogueManagerExampleBalloon ", "")
file = FileAccess.open(balloon_script_path, FileAccess.WRITE)
file.store_string(file_contents)
file.close()
var new_balloon_script_uid_raw: int = ResourceUID.create_id()
ResourceUID.add_id(new_balloon_script_uid_raw, balloon_script_path)
var new_balloon_script_uid: String = ResourceUID.id_to_text(new_balloon_script_uid_raw)
# Save the new balloon
file_contents = FileAccess.get_file_as_string(example_balloon_path)
if "example_balloon.gd" in file_contents:
file_contents = file_contents.replace(plugin_path + "/example_balloon/example_balloon.gd", balloon_script_path)
else:
file_contents = file_contents.replace(plugin_path + "/example_balloon/ExampleBalloon.cs", balloon_script_path)
var new_balloon_uid: String = ResourceUID.id_to_text(ResourceUID.create_id())
file_contents = file_contents.replace(example_balloon_uid, new_balloon_uid).replace(example_balloon_script_uid, new_balloon_script_uid)
file = FileAccess.open(balloon_path, FileAccess.WRITE)
file.store_string(file_contents)
file.close()
EditorInterface.get_resource_filesystem().scan()
EditorInterface.get_file_system_dock().call_deferred("navigate_to_path", balloon_path)
DMSettings.set_setting(DMSettings.BALLOON_PATH, balloon_path)
directory_dialog.queue_free()
)
EditorInterface.get_base_control().add_child(directory_dialog)
directory_dialog.popup_centered()
### Signals
func _on_files_moved(old_file: String, new_file: String) -> void:
update_import_paths(old_file, new_file)
DMSettings.move_recent_file(old_file, new_file)
func _on_file_removed(file: String) -> void:
update_import_paths(file, "")
if is_instance_valid(main_view):
main_view.close_file(file)
_update_localization()