local utils = require("mp.utils")

-- The directory the script will work on
-- Leaving it blank will make it work on every video you watch with mpv
-- You can still update manually via Ctrl+A
-- Setting a directory will only work if the path of the video contains this directory
DIRECTORY = ""

-- Example: DIRECTORY = "D:/Torrents" or "D:/Anime"

function callback(success, result, error)
	if result.status == 0 then
		mp.osd_message("Updated anime correctly.", 2)
	end
end

local function get_python_command()
	-- Determine OS type
	local os_name = package.config:sub(1, 1)

	-- Get the path to this Lua file
	local info = debug.getinfo(1, "S")
	local script_path = info.source:sub(info.source:find("@") == 1 and 2 or 1)

	-- Extract the directory containing this file
	local script_dir
	if os_name == "\\" then
		-- Windows
		script_dir = script_path:match("(.+)\\[^\\]+$") or "."
	else
		-- Unix
		script_dir = script_path:match("(.+)/[^/]+$") or "."
	end

	-- Build absolute path to the Python executable in the virtual environment
	if os_name == "\\" then
		-- Windows
		return script_dir .. "\\env\\Scripts\\python.exe"
	else
		-- Unix
		return script_dir .. "/env/bin/python3"
	end
end

local python_command = get_python_command()

-- Make sure it doesnt trigger twice in 1 video
local triggered = false

-- Function to check if we've reached 85% of the video
function check_progress()
	if triggered then
		return
	end

	local percent_pos = mp.get_property_number("percent-pos")

	if percent_pos then
		if percent_pos >= 85 then
			update_anilist("update")
			triggered = true
		end
	end
end

-- Function to launch the .py script
function update_anilist(action)
	if action == "launch" then
		mp.osd_message("Launching AniList", 2)
	end
	local script_dir = debug.getinfo(1).source:match("@?(.*/)")
	local directory = mp.get_property("working-directory")
	-- It seems like in Linux working-directory sometimes returns it without a "/" at the end
	directory = (directory:sub(-1) == "/" or directory:sub(-1) == "\\") and directory or directory .. "/"
	-- For some reason, "path" sometimes returns the absolute path, sometimes it doesn't.
	local file_path = mp.get_property("path")
	local path = utils.join_path(directory, file_path)

	local table = {}
	table.name = "subprocess"
	table.args = { python_command, script_dir .. "anilistUpdater.py", path, action }
	local cmd = mp.command_native_async(table, callback)
end

mp.observe_property("percent-pos", "number", check_progress)

-- Reset triggered
mp.register_event("file-loaded", function()
	triggered = false
	if DIRECTORY ~= "" then
		local directory = mp.get_property("working-directory")
		directory = (directory:sub(-1) == "/" or directory:sub(-1) == "\\") and directory or directory .. "/"
		local file_path = mp.get_property("path")
		local path = utils.join_path(directory, file_path)
		path = path:gsub("\\", "/")

		if string.find(path, DIRECTORY) ~= 1 then
			mp.unobserve_property(check_progress)
		end
	end
end)

-- Keybinds, modify as you please
mp.add_key_binding("ctrl+a", "update_anilist", function()
	update_anilist("update")
end)

mp.add_key_binding("ctrl+b", "launch_anilist", function()
	update_anilist("launch")
end)

-- Open the folder that the video is
function open_folder()
	local path = mp.get_property("path")
	local directory

	if not path then
		mp.msg.warn("No file is currently playing.")
		return
	end

	if path:find("\\") then
		directory = path:match("(.*)\\")
	elseif path:find("\\\\") then
		directory = path:match("(.*)\\\\")
	else
		directory = mp.get_property("working-directory")
	end

	-- Use the system command to open the folder in File Explorer
	local args
	if package.config:sub(1, 1) == "\\" then
		-- Windows
		args = { "explorer", directory }
	elseif os.getenv("XDG_CURRENT_DESKTOP") or os.getenv("WAYLAND_DISPLAY") or os.getenv("DISPLAY") then
		-- Linux (assume a desktop environment like GNOME, KDE, etc.)
		args = { "xdg-open", directory }
	elseif package.config:sub(1, 1) == "/" then
		-- macOS
		args = { "open", directory }
	end

	mp.command_native({ name = "subprocess", args = args, detach = true })
end

mp.add_key_binding("ctrl+d", "open_folder", open_folder)