119 lines
2.9 KiB
Lua
119 lines
2.9 KiB
Lua
local utils = require 'mp.utils'
|
|
require 'mp.options'
|
|
|
|
options = {}
|
|
options.down_dir = ""
|
|
options.sub_language = "eng"
|
|
options.subselect_path = utils.join_path(mp.get_script_directory(), "subselect.py")
|
|
options.providers_auth = "{}"
|
|
|
|
if package.config:sub(1,1) == "/" then
|
|
ops = "unix"
|
|
else
|
|
ops = "win"
|
|
end
|
|
|
|
function fixsub(path)
|
|
f = io.open(path, "r")
|
|
if f == nil then
|
|
return
|
|
end
|
|
content = f:read("*all")
|
|
f:close()
|
|
write = false
|
|
content = content:gsub("%s*%-%->%s*",
|
|
function(w)
|
|
if w:len() < 5 then
|
|
write = true
|
|
return " --> "
|
|
end
|
|
end, 1)
|
|
if not write then
|
|
return
|
|
end
|
|
f = io.open(path, "w")
|
|
f:write(content)
|
|
f:flush()
|
|
f:close()
|
|
end
|
|
|
|
function set_down_dir(options)
|
|
ddir = options.down_dir
|
|
if ddir == "" then
|
|
if mp.get_property_native("path", ""):find("://") ~= nil then
|
|
if ops == "win" then
|
|
ddir = utils.join_path(os.getenv("USERPROFILE"), "Downloads")
|
|
else
|
|
ddir = os.getenv("HOME")
|
|
end
|
|
else
|
|
ddir = utils.split_path(mp.get_property_native("path", ""))
|
|
if ops == "win" then
|
|
if ddir:find("^%a:") == nil then
|
|
ddir = utils.join_path(os.getenv("USERPROFILE"), "Downloads")
|
|
end
|
|
else
|
|
if ddir:find("^/") == nil then
|
|
ddir = os.getenv("HOME")
|
|
end
|
|
end
|
|
end
|
|
end
|
|
options.down_dir = ddir
|
|
end
|
|
|
|
function get_python_binary()
|
|
python = nil
|
|
python_error = ""
|
|
python_version = mp.command_native({
|
|
name = "subprocess",
|
|
args = { "python", "--version"},
|
|
capture_stdout = true
|
|
})
|
|
|
|
if python_version.error_string ~= "" then
|
|
python_error = "python not found"
|
|
else
|
|
if python_version.stdout:find("3%.") ~= nil then
|
|
python = "python"
|
|
else
|
|
python_version = mp.command_native({
|
|
name = "subprocess",
|
|
args = { "python3", "--version" }
|
|
})
|
|
|
|
if python_version.error_string ~= "" then
|
|
python_error = "python3 not installed"
|
|
else
|
|
python = "python3"
|
|
end
|
|
end
|
|
end
|
|
return python, python_error
|
|
end
|
|
|
|
read_options(options)
|
|
|
|
function search_subs()
|
|
set_down_dir(options)
|
|
video = mp.get_property_native("filename/no-ext", "")
|
|
python, python_error = get_python_binary()
|
|
if python ~= nil then
|
|
ret = mp.command_native({
|
|
name = "subprocess",
|
|
args = { python, options.subselect_path, video, options.down_dir, options.sub_language, options.providers_auth },
|
|
capture_stdout = true
|
|
})
|
|
else
|
|
mp.osd_message(python_error)
|
|
return
|
|
end
|
|
if string.find(ret.stdout, ".") ~= nil then
|
|
mp.osd_message("loading subtitle: "..ret.stdout)
|
|
subtitle_path = utils.join_path(ddir, ret.stdout)
|
|
fixsub(subtitle_path)
|
|
mp.commandv("sub-add", subtitle_path)
|
|
end
|
|
end
|
|
|
|
mp.add_key_binding("alt+u", "subselect", search_subs) |