1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
{ pkgs, lib, ... }:
let
dl_subs = pkgs.writers.writeDashBin "dl_subs" ''
filename=$1
${pkgs.subdl}/bin/subdl --output='/tmp/{m}.{M}.sub' "$filename" 1>&2
echo "/tmp/$(basename "$filename").sub"
'';
autosub = pkgs.writeText "autosub.lua" ''
-- Requires Subliminal version 1.0 or newer
-- Make sure to specify your system's Subliminal location below:
local utils = require 'mp.utils'
-- Log function: log to both terminal and mpv OSD (On-Screen Display)
function log(string, secs)
secs = secs or 2 -- secs defaults to 2 when the secs parameter is absent
mp.msg.warn(string) -- This logs to the terminal
mp.osd_message(string, secs) -- This logs to mpv screen
end
function download()
log('Searching subtitles ...', 10)
path = mp.get_property('path')
result = utils.subprocess({ args = {"${dl_subs}/bin/dl_subs", path} })
if result.error == nil then
filename = string.gsub(result.stdout, "\n", "")
log(filename)
mp.commandv('sub_add', filename)
log('Subtitles ready!')
else
log('Subtitles failed downloading')
end
end
-- Control function: only download if necessary
function control_download()
duration = tonumber(mp.get_property('duration'))
if duration < 900 then
mp.msg.warn('Video is less than 15 minutes\n', '=> NOT downloading any subtitles')
return
end
-- There does not seem to be any documentation for the 'sub' property,
-- but it works on both internally encoded as well as external subtitle files!
-- -> sub = '1' when subtitles are present
-- -> sub = 'no' when subtitles are not present
-- -> sub = 'auto' when called before the 'file-loaded' event is triggered
sub = mp.get_property('sub')
if sub == '1' then
mp.msg.warn('Sub track is already present\n', '=> NOT downloading other subtitles')
return
end
mp.msg.warn('No sub track was detected\n', '=> Proceeding to download subtitles:')
download()
end
mp.add_key_binding('S', "download_subs", download)
'';
mpvInput = pkgs.writeText "mpv.input" ''
: script-binding console/enable
x add audio-delay -0.050
X add audio-delay 0.050
'';
mpvConfig = pkgs.writeText "mpv.conf" ''
osd-font-size=20
'';
mpv = pkgs.symlinkJoin {
name = "mpv";
paths = [
(pkgs.writeDashBin "mpv" ''
set -efu
Y_RES=1081
# we need to disable sponsorblock local database because of
# https://github.com/po5/mpv_sponsorblock/issues/31
exec ${pkgs.mpv.override {
scripts = with pkgs.mpvScripts; [
sponsorblock
youtube-quality
];
}}/bin/mpv \
--no-config \
--input-conf=${mpvInput} \
--include=${mpvConfig} \
--script=${autosub} \
--ytdl-format="best[height<$Y_RES]" \
--script-opts=ytdl_hook-ytdl_path=${pkgs.yt-dlp}/bin/yt-dlp \
--script-opts-append=sponsorblock-local_database=no \
"$@"
'')
pkgs.mpv
];
};
in {
environment.systemPackages = [
mpv
dl_subs
];
}
|