stockholm/lass/2configs/mpv.nix

127 lines
3.8 KiB
Nix
Raw Normal View History

{ pkgs, lib, ... }:
2016-04-28 20:55:18 +00:00
let
2020-03-02 19:52:21 +00:00
download_subs = pkgs.writers.writePython3 "download_sub" {
libraries = [ pkgs.python3Packages.subliminal ];
} ''
from subliminal import download_best_subtitles, scan_video
from babelfish import Language
import sys
video_filename = sys.argv[1]
vid = scan_video(video_filename)
2020-04-11 15:31:13 +00:00
try:
sub = download_best_subtitles([vid], {Language('eng')})[vid][0]
2020-03-02 19:52:21 +00:00
2020-04-11 15:31:13 +00:00
filename = '/tmp/' + vid.title + '.srt'
2020-03-02 19:52:21 +00:00
2020-04-11 15:31:13 +00:00
with open(filename, 'wb+') as file:
file.write(sub.content)
2020-03-02 19:52:21 +00:00
2020-04-11 15:31:13 +00:00
print(filename)
except: # noqa
print("/dev/null")
2020-03-02 19:52:21 +00:00
'';
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)
table = { args = {"${download_subs}", mp.get_property('path')} }
result = utils.subprocess(table)
if result.error == nil then
-- remove trailing newline from subtitle filename
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)
'';
2022-05-27 08:22:11 +00:00
mpvInput = pkgs.writeText "mpv.input" ''
: script-binding console/enable
'';
mpvConfig = pkgs.writeText "mpv.conf" ''
2022-05-27 11:42:45 +00:00
osd-font-size=20
2022-05-27 08:22:11 +00:00
'';
2017-06-20 17:30:14 +00:00
mpv = pkgs.symlinkJoin {
name = "mpv";
paths = [
(pkgs.writeDashBin "mpv" ''
2022-05-27 11:42:45 +00:00
set -efu
if [ -n "''${DISPLAY+x}" ]; then
Y_RES=$(${pkgs.xorg.xrandr}/bin/xrandr |
${pkgs.jc}/bin/jc --xrandr |
${pkgs.jq}/bin/jq '.screens[0].current_width'
)
else
Y_RES=1000
fi
2022-05-27 08:22:11 +00:00
# we need to disable sponsorblock local database because of
# https://github.com/po5/mpv_sponsorblock/issues/31
exec ${pkgs.mpv.override {
2022-05-27 11:42:45 +00:00
scripts = with pkgs.mpvScripts; [
sponsorblock
youtube-quality
2022-05-27 08:22:11 +00:00
];
}}/bin/mpv \
2021-10-24 17:53:43 +00:00
-vo=gpu \
--no-config \
2022-05-27 08:22:11 +00:00
--input-conf=${mpvInput} \
--include=${mpvConfig} \
2021-10-24 17:53:43 +00:00
--script=${autosub} \
2022-05-27 11:42:45 +00:00
--ytdl-format="best[height<$Y_RES]" \
2021-10-24 17:53:43 +00:00
--script-opts=ytdl_hook-ytdl_path=${pkgs.yt-dlp}/bin/yt-dlp \
2022-05-27 08:22:11 +00:00
--script-opts-append=sponsorblock-local_database=no \
2021-10-24 17:53:43 +00:00
"$@"
2017-06-20 17:30:14 +00:00
'')
pkgs.mpv
2017-06-20 17:30:14 +00:00
];
};
2016-04-28 20:55:18 +00:00
in {
2017-05-11 16:03:51 +00:00
environment.systemPackages = [
2016-04-28 20:55:18 +00:00
mpv
];
}