mpv: init

2 files changed, 74 insertions, 1 deletions

diff --git a/lib/util.lua b/lib/util.lua
index d962961..15221dd 100644
--- a/lib/util.lua
+++ b/lib/util.lua
@@ -37,7 +37,7 @@ local zip_ext_first = function (zip, out)
 
 	rc = os.execute('unzip -qq ' .. zip .. ' -d ' .. dir)
 	srt = io.popen('find ' .. dir .. ' -type f -name *.srt'):read('*l')
-	os.rename(srt, out)
+	os.execute("mv '" .. srt .. "' '"  .. out .. "'")
 	os.remove(dir)
 
 	return rc
diff --git a/main.lua b/main.lua
new file mode 100644
index 0000000..463a412
--- /dev/null
+++ b/main.lua
@@ -0,0 +1,73 @@
+local mutil = require 'mp.utils'
+local subscene = require 'server/subscene'
+
+local notify = function (str, sec)
+    mp.osd_message(str, sec)
+end
+
+local mkdir = function (path)
+    local info = mutil.file_info(path)
+    if info and not info.is_dir then
+        os.remove(path)
+    end
+
+    return os.execute('mkdir -p ' .. path)
+end
+
+local sub_needed = function ()
+    local duration, isvideo
+
+    duration = tonumber(mp.get_property('duration'))
+    if duration < 900 then -- duration is less than 15 minutes
+        return false
+    end
+
+    for _, v in pairs(mp.get_property_native('track-list')) do
+        if v['type'] == 'sub' then
+            return false
+        end
+        if v['type'] == 'video' then
+            isvideo = true
+        end
+    end
+    if not isvideo then
+        return false
+    end
+
+    return true
+end
+
+local sub_setup = function ()
+    local out, name
+
+    notify('fetching subtitles')
+
+    out = mp.get_property_native('sub-file-paths')[1]
+    if out then
+        out = out:gsub('^~/', os.getenv('HOME') .. '/')
+    else
+        out = os.getenv('HOME') .. '.local/share/subs'
+        mp.set_property('sub-file-paths', out)
+    end
+
+    mkdir(out)
+    name = mp.get_property_native('path')
+    out = out .. '/' .. name .. '.srt'
+
+    if subscene.search(name, out) then
+        mp.commandv('rescan_external_files')
+        notify('fetch success')
+    else
+        notify('fetch failure')
+    end
+end
+
+local file_listener = function ()
+    if not sub_needed() then
+        return
+    end
+
+    sub_setup()
+end
+
+mp.register_event('file-loaded', file_listener)