@tool
extends EditorPlugin
class_name FormatOnSavePlugin

const FORMAT_ACTION := "simple_format_on_save/format_code"
const SETTING_PREFIX = "text_editor/format_on_save/"
var format_key: InputEventKey
var formatter: Formatter


## Create an editor setting so users can toggle features
func _set_default_setting(setting:String,default_value:Variant):
	var _EditorSettings = EditorInterface.get_editor_settings()
	if not _EditorSettings.has_setting(SETTING_PREFIX+setting):
		_EditorSettings.set_setting(SETTING_PREFIX+setting, default_value)
		_EditorSettings.set_initial_value(SETTING_PREFIX+setting, default_value,false)


func _enter_tree():
	_set_default_setting("blank_after_dedent",false)

	add_tool_menu_item("Format (Ctrl+Alt+L)", _on_format_code)

	if InputMap.has_action(FORMAT_ACTION):
		InputMap.erase_action(FORMAT_ACTION)

	InputMap.add_action(FORMAT_ACTION)
	format_key = InputEventKey.new()
	format_key.keycode = KEY_L
	format_key.ctrl_pressed = true
	format_key.alt_pressed = true
	InputMap.action_add_event(FORMAT_ACTION, format_key)
	resource_saved.connect(on_resource_saved)


func _exit_tree():
	remove_tool_menu_item("Format (Ctrl+Alt+L)")
	InputMap.erase_action(FORMAT_ACTION)
	resource_saved.disconnect(on_resource_saved)


# Return true if formatted code != original code
func _on_format_code() -> bool:
	var current_editor := EditorInterface.get_script_editor().get_current_editor()

	if not (current_editor and current_editor.is_class("ScriptTextEditor")):
		return false

	var text_edit = current_editor.get_base_editor()
	var code = text_edit.text

	if not formatter:
		formatter = Formatter.new()

	var formatted_code = formatter.format_code(code)

	if not formatted_code:
		return false

	if code.length() == formatted_code.length() and code == formatted_code:
		return false

	var scroll_horizontal = text_edit.scroll_horizontal
	var scroll_vertical = text_edit.scroll_vertical
	var caret_column = text_edit.get_caret_column(0)
	var caret_line = text_edit.get_caret_line(0)

	text_edit.text = formatted_code
	text_edit.set_caret_line(caret_line)
	text_edit.set_caret_column(caret_column)
	text_edit.scroll_horizontal = scroll_horizontal
	text_edit.scroll_vertical = scroll_vertical

	return true


func _shortcut_input(event: InputEvent) -> void:
	if event is InputEventKey and event.is_pressed():
		if Input.is_action_pressed(FORMAT_ACTION):
			_on_format_code()


# CALLED WHEN A SCRIPT IS SAVED
func on_resource_saved(resource: Resource):
	if resource is Script:
		var script: Script = resource
		var current_script = get_editor_interface().get_script_editor().get_current_script()

		# Prevents other unsaved scripts from overwriting the active one
		if current_script == script:
			var is_modified: bool = _on_format_code()

			#if is_modified:
				#print_rich("[color=#636363]Auto formatted code[/color]")