-
-
Notifications
You must be signed in to change notification settings - Fork 8
/
youtube-upnext.lua
1389 lines (1242 loc) · 49.8 KB
/
youtube-upnext.lua
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
-- youtube-upnext.lua
--
-- Fetch upnext/recommended videos from youtube
-- This is forked/based on https://github.com/jgreco/mpv-youtube-quality
--
-- Diplays a menu that lets you load the upnext/recommended video from youtube
-- that appear on the right side on the youtube website.
-- If auto_add is set to true (default), the 'up next' video is automatically
-- appended to the current playlist
--
-- Bound to ctrl-u by default.
--
-- Requires curl/curl.exe or wget/wget.exe in PATH. On Windows with wget you may need
-- to set check_certificate to false, otherwise wget.exe might not be able to
-- download the youtube website.
local opts = {
--key bindings
toggle_menu_binding = "ctrl+u",
up_binding = "UP",
down_binding = "DOWN",
select_binding = "ENTER",
append_binding = "SPACE",
close_binding = "ESC",
--auto fetch recommended videos when opening a url
fetch_on_start = true,
--auto load and add the "upnext" video to the playlist
auto_add = true,
--formatting / cursors
cursor_selected = "● ",
cursor_unselected = "○ ",
cursor_appended = "▷ ",
cursor_appended_selected = "▶ ",
--font size scales by window, if false requires larger font and padding sizes
scale_playlist_by_window = false,
--playlist ass style overrides inside curly brackets, \keyvalue is one field, extra \ for escape in lua
--example {\\fnUbuntu\\fs10\\b0\\bord1} equals: font=Ubuntu, size=10, bold=no, border=1
--read http://docs.aegisub.org/3.2/ASS_Tags/ for reference of tags
--undeclared tags will use default osd settings
--these styles will be used for the whole playlist. More specific styling will need to be hacked in
--
--(a monospaced font is recommended but not required e.g. {\\fnmonospace\\fs25} )
style_ass_tags = "{\\fs25}",
--paddings for top left corner
text_padding_x = 5,
text_padding_y = 5,
menu_timeout = 10,
--Screen dim when menu is open 0.0 - 1.0 (0 is no dim, 1 is black)
curtain_opacity = 0.7,
youtube_url = "https://www.youtube.com/watch?v=%s",
-- Fallback Invidious instance, see https://api.invidious.io/ for alternatives
invidious_instance = "https://inv.tux.pizza",
-- Keep the width of the window the same when the next video is played
restore_window_width = false,
-- On Windows wget.exe may not be able to check SSL certificates for HTTPS, so you can disable checking here
check_certificate = true,
-- Use a cookies file
-- (Same as youtube-dl --cookies or curl -b or wget --load-cookies)
-- If you don't set this, the script will try to create a temporary cookies file for you
-- On Windows you need to use a double blackslash or a single fordwardslash
-- For example "C:\\Users\\Username\\cookies.txt"
-- Or "C:/Users/Username/cookies.txt"
-- Alternatively you can set this from the command line with --ytdl-raw-options=cookies=file.txt
-- or --ytdl-raw-options-append=cookies=file.txt
-- or in mpv.conf with ytdl-raw-options-append=cookies=file.txt
-- If you want to use cookies from your browser you need to set both
-- cookies and cookies-from-browser in mpv.conf or the command line:
-- ytdl-raw-options-append=cookies=file.txt
-- ytdl-raw-options-append=cookies-from-browser=edge
cookies = "",
-- When a video is selected from the menu, the new video can be appended to the playlist
-- or the playlist can be cleared and replaced with only the selected video.
-- If true, the video will be appended to the playlist. If false, the playlist will be cleared.
keep_playlist_on_select = true,
-- What should happen if a video recommendation in uosc menu is clicked? Options are:
-- 'submenu' -- show a submenu with play/upnext/append option
-- 'append' -- append the video to the playlist
-- 'insert' -- play the video after the current video
-- 'play' -- append the video to the playlist and play it
-- 'replace' -- play the video and clear the playlist
uosc_entry_action = "submenu",
-- Should the uosc menu stay open after clicking a video recommendation
uosc_keep_menu_open = false,
-- Don't play/append videos that are shorter than this time. Format is "HH:MM:SS" or "MM:SS"
skip_shorter_than = "",
-- Don't play/append videos that are longer than this time. Format is "HH:MM:SS" or "MM:SS"
skip_longer_than = "",
-- Don't show the videos that are too short or too long in the menu
hide_skipped_videos = false,
-- Limit the number of suggested videos in the menu
suggestions_limit = 0,
}
local mp = require "mp"
local utils = require "mp.utils"
local msg = require "mp.msg"
local assdraw = require "mp.assdraw"
local input_import, input = pcall(require, "mp.input")
if not input_import then
-- If mp.input is not available, use an empty implementation
input = {
get = function(foo) end,
terminate = function()
mp.osd_message("", 1)
-- Print media info (in case the input box is closed without playing a new video)
local media_title = mp.get_property("media-title")
local metadata = mp.get_property_native("metadata")
if metadata then
if metadata["uploader"] then
mp.commandv("print-text", " Uploader: " .. metadata["uploader"])
end
if metadata["channel_url"] then
mp.commandv("print-text", " Channel URL: " .. metadata["channel_url"])
end
end
if media_title then
mp.commandv("print-text", " Title: " .. media_title)
end
end,
set_log = function(lines)
local text = ""
for i = 1, #lines do
if type(lines[i]) == 'table' then
text = text .. "\n" .. lines[i].terminal_style .. lines[i].text .. "\027[0m"
else
text = text .. "\n" .. tostring(lines[i])
end
end
mp.osd_message(text, 999)
end
}
end
(require "mp.options").read_options(opts, "youtube-upnext")
-- Command line options
if opts.cookies == nil or opts.cookies == "" then
local raw_options = mp.get_property_native("options/ytdl-raw-options")
for param, arg in pairs(raw_options) do
if (param == "cookies") and (arg ~= "") then
opts.cookies = arg
end
end
end
local script_name = mp.get_script_name()
local destroyer = nil
local redraw_menu = nil
local handled_cursor = 0
local upnext_cache = {}
local prefered_win_width = nil
local last_dheight = nil
local last_dwidth = nil
local watched_ids = {}
local appended_to_playlist = {}
local terminal_id = 0
local function table_size(t)
local s = 0
for _, _ in ipairs(t) do
s = s + 1
end
return s
end
local function exec(args, capture_stdout, capture_stderr)
local ret =
mp.command_native(
{
name = "subprocess",
playback_only = false,
capture_stdout = capture_stdout,
capture_stderr = capture_stderr,
args = args
}
)
return ret.status, ret.stdout, ret.stderr, ret
end
local function url_encode(s)
local function repl(x)
return string.format("%%%02X", string.byte(x))
end
return string.gsub(s, "([^0-9a-zA-Z!'()*._~-])", repl)
end
local function parse_yt_time(hour_min_second_string)
local hour, min, sec = string.match(hour_min_second_string, "(%d+):(%d+):(%d+)")
if hour == nil then
min, sec = string.match(hour_min_second_string, "(%d+):(%d+)")
end
if min == nil then
sec = string.match(hour_min_second_string, "(%d+)")
end
return (hour or 0) * 3600 + (min or 0) * 60 + (sec or 0)
end
local function create_yt_time(seconds)
local hour = math.floor(seconds / 3600)
local min = math.floor((seconds - hour * 3600) / 60)
local sec = math.floor(seconds - hour * 3600 - min * 60)
if hour > 0 then
return string.format("%02d:%02d:%02d", hour, min, sec)
end
return string.format("%02d:%02d", min, sec)
end
local function extract_videoid(url)
local video_id = nil
if string.find(url, "youtu") ~= nil then
-- extract vidoe id from https://www.youtube.com/watch?v=abcd_1234-ef
local s, e = string.find(url, "v=[^#?!&]+")
if s ~= nil then
video_id = string.sub(url, s + 2, e)
else
-- extract from https://youtu.be/abcd_1234-ef
local s2, e2 = string.find(url, "youtu.be/[^#?!&]+")
if s2 ~= nil then
video_id = string.sub(url, s2 + 9, e2)
end
end
end
return video_id
end
local skip_shorter_than = -1
if opts.skip_shorter_than ~= nil and opts.skip_shorter_than ~= "" then
skip_shorter_than = parse_yt_time(opts.skip_shorter_than)
end
local skip_longer_than = -1
if opts.skip_longer_than ~= nil and opts.skip_longer_than ~= "" then
skip_longer_than = parse_yt_time(opts.skip_longer_than)
end
if skip_longer_than > -1 and skip_shorter_than > -1 and skip_longer_than < skip_shorter_than then
msg.error("skip_longer_than must be greater than skip_shorter_than")
skip_longer_than = -1
skip_shorter_than = -1
end
local function download_upnext(url, post_data)
if opts.fetch_on_start or opts.auto_add then
msg.debug("fetching 'up next' with curl...")
else
mp.osd_message("fetching 'up next' with curl...", 60)
end
local command = { "curl", "--silent", "--location" }
if post_data then
table.insert(command, "--request")
table.insert(command, "POST")
table.insert(command, "--data")
table.insert(command, post_data)
end
if opts.cookies ~= nil and opts.cookies ~= "" then
table.insert(command, "--cookie-jar")
table.insert(command, opts.cookies)
table.insert(command, "--cookie")
table.insert(command, opts.cookies)
end
table.insert(command, url)
local es, s, _, _ = exec(command, true)
if (es ~= 0) or (s == nil) or (s == "") then
if es == -1 or es == -3 or es == 127 or es == 9009 then
-- MP_SUBPROCESS_EINIT is -3 which can mean the command was not found:
-- https://github.com/mpv-player/mpv/blob/24dcb5d167ba9580119e0b9cc26f79b1d155fcdc/osdep/subprocess-posix.c#L335-L336
msg.debug("curl not found, trying wget")
local command_wget = { "wget", "-q", "-O", "-" }
if not opts.check_certificate then
table.insert(command_wget, "--no-check-certificate")
end
if post_data then
table.insert(command_wget, "--post-data")
table.insert(command_wget, post_data)
end
if opts.cookies then
table.insert(command_wget, "--load-cookies")
table.insert(command_wget, opts.cookies)
table.insert(command_wget, "--save-cookies")
table.insert(command_wget, opts.cookies)
table.insert(command_wget, "--keep-session-cookies")
end
table.insert(command_wget, url)
es, s, _, _ = exec(command, true)
if (es ~= 0) or (s == nil) or (s == "") then
mp.osd_message("upnext failed: curl was not found, wget failed", 10)
return "{}"
end
else
mp.osd_message("upnext failed: error=" .. tostring(es), 10)
msg.error("failed to get upnext list: error=" .. tostring(es))
msg.error("s: " .. tostring(s))
msg.debug("exec (async): " .. table.concat(command, " "))
return "{}"
end
end
local consent_pos = s:find('action="https://consent.youtube.com/s"')
if consent_pos ~= nil then
-- Accept cookie consent form
msg.debug("Need to accept cookie consent form")
s = s:sub(s:find(">", consent_pos + 1, true), s:find("</form", consent_pos + 1, true))
local post_str = ""
for k, v in string.gmatch(s, 'name="([^"]+)" value="([^"]*)"') do
msg.debug("name=" .. tostring(k) .. " value=" .. tostring(v))
post_str = post_str .. url_encode(k) .. "=" .. url_encode(v) .. "&"
end
msg.debug("post-data=" .. tostring(post_str))
if opts.cookies == nil or opts.cookies == "" then
local temp_dir = os.getenv("TEMP")
if temp_dir == nil or temp_dir == "" then
temp_dir = os.getenv("XDG_RUNTIME_DIR")
end
if temp_dir == nil or temp_dir == "" then
opts.cookies = os.tmpname()
else
opts.cookies = temp_dir .. "/youtube-upnext.cookies"
end
msg.warn(
'Created a cookies jar file at "' ..
tostring(opts.cookies) .. '". To hide this warning, set a cookies file in the script configuration'
)
end
return download_upnext("https://consent.youtube.com/s", post_str)
end
local pos1 = string.find(s, "ytInitialData =", 1, true)
if pos1 == nil then
mp.osd_message("upnext failed, no upnext data found err01", 10)
msg.error("failed to find json position 01: pos1=nil")
return "{}"
end
local pos2 = string.find(s, ";%s*</script>", pos1 + 1)
if pos2 ~= nil then
s = string.sub(s, pos1 + 15, pos2 - 1)
return s
else
msg.error("failed to find json position 02")
end
mp.osd_message("upnext failed, no upnext data found err03", 10)
msg.error("failed to get upnext data: pos1=" .. tostring(pos1) .. " pos2=" .. tostring(pos2))
return "{}"
end
local function get_invidious(url)
-- convert to invidious API call
url = string.gsub(url, "https://youtube%.com/watch%?v=", opts.invidious_instance .. "/api/v1/videos/")
url = string.gsub(url, "https://www%.youtube%.com/watch%?v=", opts.invidious_instance .. "/api/v1/videos/")
url = string.gsub(url, "https://youtu%.be/", opts.invidious_instance .. "/api/v1/videos/")
msg.debug("Invidious url:" .. url)
local command = { "curl", "--silent", "--location" }
if not opts.check_certificate then
table.insert(command, "--no-check-certificate")
end
table.insert(command, url)
local es, s, _, _ = exec(command, true)
if (es ~= 0) or (s == nil) or (s == "") then
if es == -1 or es == -3 or es == 127 or es == 9009 then
msg.debug("curl not found, trying wget")
local command_wget = { "wget", "-q", "-O", "-" }
if not opts.check_certificate then
table.insert(command_wget, "--no-check-certificate")
end
table.insert(command_wget, url)
es, s, _, _ = exec(command_wget, true)
if (es ~= 0) or (s == nil) or (s == "") then
mp.osd_message("upnext failed: curl was not found, wget failed", 10)
return {}
end
else
mp.osd_message("upnext failed: error=" .. tostring(es), 10)
msg.error("failed to get invidious: error=" .. tostring(es))
return {}
end
end
local data, err = utils.parse_json(s)
if data == nil then
mp.osd_message("upnext fetch failed (Invidious): JSON decode failed", 10)
msg.error("parse_json failed (Invidious): " .. err)
return {}
end
if data.recommendedVideos then
local res = {}
msg.debug("downloaded and decoded json successfully (Invidious)")
for i, v in ipairs(data.recommendedVideos) do
local duration = -1
if v.lengthSeconds ~= nil then
duration = tonumber(v.lengthSeconds)
end
table.insert(
res,
{
index = i,
label = v.title .. " - " .. v.author,
file = string.format(opts.youtube_url, v.videoId),
length = duration
}
)
end
mp.osd_message("upnext fetch from Invidious succeeded", 10)
return res
elseif data.error then
mp.osd_message("upnext fetch failed (Invidious): " .. data.error, 10)
msg.error("Invidious error: " .. data.error)
else
mp.osd_message("upnext: No recommended videos! (Invidious)", 10)
msg.error("No recommended videos! (Invidious)")
end
return {}
end
local function parse_upnext(json_str, current_video_url)
if json_str == "{}" then
return {}, 0
end
local data, err = utils.parse_json(json_str)
if data == nil then
mp.osd_message("upnext failed: JSON decode failed", 10)
msg.error("parse_json failed: " .. tostring(err))
msg.debug("Corrupted JSON:\n" .. json_str .. "\n")
return {}, 0
end
local skipped_results = {}
local res = {}
msg.debug("downloaded and decoded json successfully")
local index = 1
local autoplay_id = nil
if
data.playerOverlays and data.playerOverlays.playerOverlayRenderer and
data.playerOverlays.playerOverlayRenderer.autoplay and
data.playerOverlays.playerOverlayRenderer.autoplay.playerOverlayAutoplayRenderer
then
local playerOverlayAutoplayRenderer = data.playerOverlays.playerOverlayRenderer.autoplay
.playerOverlayAutoplayRenderer
local title = playerOverlayAutoplayRenderer.videoTitle.simpleText
local video_id = playerOverlayAutoplayRenderer.videoId
local duration = -1
if playerOverlayAutoplayRenderer.thumbnailOverlays and playerOverlayAutoplayRenderer.thumbnailOverlays[1] and
playerOverlayAutoplayRenderer.thumbnailOverlays[1].thumbnailOverlayTimeStatusRenderer and
playerOverlayAutoplayRenderer.thumbnailOverlays[1].thumbnailOverlayTimeStatusRenderer.text
then
duration = parse_yt_time(playerOverlayAutoplayRenderer.thumbnailOverlays[1]
.thumbnailOverlayTimeStatusRenderer.text.simpleText)
end
if watched_ids[video_id] == nil then -- Skip if the video was already watched
autoplay_id = video_id
table.insert(
res,
{
index = index,
label = title,
file = string.format(opts.youtube_url, video_id),
length = duration
}
)
index = index + 1
else
table.insert(
skipped_results,
{
index = index,
label = title,
file = string.format(opts.youtube_url, video_id),
length = duration
}
)
index = index + 1
end
end
if
data.playerOverlays and data.playerOverlays.playerOverlayRenderer and
data.playerOverlays.playerOverlayRenderer.endScreen and
data.playerOverlays.playerOverlayRenderer.endScreen.watchNextEndScreenRenderer and
data.playerOverlays.playerOverlayRenderer.endScreen.watchNextEndScreenRenderer.results
then
local n = table_size(data.playerOverlays.playerOverlayRenderer.endScreen.watchNextEndScreenRenderer.results)
for i, v in ipairs(data.playerOverlays.playerOverlayRenderer.endScreen.watchNextEndScreenRenderer.results) do
if v.endScreenVideoRenderer and v.endScreenVideoRenderer.title and v.endScreenVideoRenderer.title.simpleText then
local title = v.endScreenVideoRenderer.title.simpleText
local video_id = v.endScreenVideoRenderer.videoId
local duration = -1
if v.endScreenVideoRenderer.lengthText then
duration = parse_yt_time(v.endScreenVideoRenderer.lengthText.simpleText)
end
if video_id ~= autoplay_id and watched_ids[video_id] == nil then
table.insert(
res,
{
index = index + i,
label = title,
file = string.format(opts.youtube_url, video_id),
length = duration
}
)
elseif watched_ids[video_id] ~= nil then
table.insert(
skipped_results,
{
index = index + i,
label = title,
file = string.format(opts.youtube_url, video_id),
length = duration
}
)
end
end
end
index = index + n
end
if
data.contents and data.contents.twoColumnWatchNextResults and
data.contents.twoColumnWatchNextResults.secondaryResults
then
local secondaryResults = data.contents.twoColumnWatchNextResults.secondaryResults
if secondaryResults.secondaryResults then
secondaryResults = secondaryResults.secondaryResults
end
for i, v in ipairs(secondaryResults.results) do
local compactVideoRenderer = nil
local watchnextindex = index
if
v.compactAutoplayRenderer and v.compactAutoplayRenderer and v.compactAutoplayRenderer.contents and
v.compactAutoplayRenderer.contents.compactVideoRenderer
then
compactVideoRenderer = v.compactAutoplayRenderer.contents.compactVideoRenderer
watchnextindex = 0
elseif v.compactVideoRenderer then
compactVideoRenderer = v.compactVideoRenderer
end
if
compactVideoRenderer and compactVideoRenderer.videoId and compactVideoRenderer.title and
compactVideoRenderer.title.simpleText
then
local title = compactVideoRenderer.title.simpleText
local video_id = compactVideoRenderer.videoId
local duration = -1
if compactVideoRenderer.lengthText then
duration = parse_yt_time(compactVideoRenderer.lengthText.simpleText)
end
local video_url = string.format(opts.youtube_url, video_id)
local duplicate = false
for _, entry in ipairs(res) do
if video_url == entry.file then
duplicate = true
end
end
if watched_ids[video_id] ~= nil then
if not duplicate then
table.insert(
skipped_results,
{
index = watchnextindex + i,
label = title,
file = video_url,
length = duration
}
)
end
duplicate = true
end
if not duplicate then
table.insert(
res,
{
index = watchnextindex + i,
label = title,
file = video_url,
length = duration
}
)
end
end
end
end
-- all results where already watched, reset watched videos and use skipped_results
if table_size(res) == 0 and table_size(skipped_results) > 0 then
msg.debug("All upnext videos are already watched. Watched video list will be reset!")
res = skipped_results
watched_ids = {}
end
table.sort(
res,
function(a, b)
return a.index < b.index
end
)
-- Limit amount of suggestions
if opts.suggestions_limit ~= nil and opts.suggestions_limit > 0 and table_size(res) > opts.suggestions_limit then
local new_res = {}
for i = 1, opts.suggestions_limit do
new_res[i] = res[i]
end
res = new_res
end
upnext_cache[current_video_url] = res
return res, table_size(res)
end
local function load_upnext()
local url = mp.get_property("path")
if url == nil then
url = ""
end
url = string.gsub(url, "ytdl://", "") -- Strip possible ytdl:// prefix.
url = string.gsub(url, "/shorts/", "/watch?v=") -- Convert shorts to watch?v=.
url = string.gsub(url, "//.*/watch%?v=", "//youtube.com/watch?v=") -- Account for alternative frontends.
url = string.gsub(url, "%?feature=share", "") -- Strip possible ?feature=share suffix.
if string.find(url, "//youtu.be/") == nil and string.find(url, "//youtube.com/") == nil then
-- SVP calls mpv like this:
-- mpv '--player-operation-mode=pseudo-gui'
-- '--input-ipc-server=mpvpipe' '--no-ytdl' '--audio-file=https://rr3---sn-4g5ednsd.googlevideo.com/videopl....'
-- '--force-media-title=Dog Years'
-- '--http-header-fields=Referer: https://www.youtube.com/watch?v=AbCdEf_Gh,User-Agent: Mozilla/5.0 ...'
-- 'https://rr3---sn-4g5ednsd.googlevideo.com/videoplaybac....'
-- We can extract the url from the header field:
local headers = mp.get_property("http-header-fields")
if headers ~= nil then
local i = headers:find("Referer: ")
if i ~= nil then
i = i + #"Referer: "
local j = headers:find(",", i + 15)
if j ~= nil then
url = headers:sub(i, j - 1)
end
end
end
if string.find(url, "//youtu.be/") == nil and string.find(url, "//youtube.com/") == nil then
-- Neither path nor Referer are a youtube link
return {}, 0
else
-- Disable the '--no-ytdl' option from SVP
mp.set_property_bool("no-ytdl", false)
mp.set_property_bool("ytdl", true)
mp.set_property_bool("autoload-files", false)
end
end
-- don't fetch the website if it's already cached
if upnext_cache[url] ~= nil then
local res = upnext_cache[url]
return res, table_size(res)
end
local res, n = parse_upnext(download_upnext(url, nil), url)
-- Fallback to Invidious API
if n == 0 and opts.invidious_instance and opts.invidious_instance ~= "" then
res = get_invidious(url)
n = table_size(res)
end
return res, n
end
local function add_to_playlist(path, title, length, flag)
if length ~= nil or length < 0 then
length = 0
end
local playlist = "memory://#EXTM3U\n#EXTINF:" .. tostring(length) .. "," .. title .. "\n" .. path
mp.commandv("loadlist", playlist, flag)
end
local function on_file_start(_)
local url = mp.get_property("path")
url = string.gsub(url, "ytdl://", "") -- Strip possible ytdl:// prefix.
url = string.gsub(url, "/shorts/", "/watch?v=") -- Convert shorts to watch?v=.
url = string.gsub(url, "//.*/watch%?v=", "//youtube.com/watch?v=") -- Account for alternative frontends.
url = string.gsub(url, "%?feature=share", "") -- Strip possible ?feature=share suffix.
if string.find(url, "youtu") ~= nil then
-- Try to add current video ID to watched list
-- extract from https://www.youtube.com/watch?v=abcd_1234-ef
local video_id = extract_videoid(url)
if video_id ~= nil then
watched_ids[video_id] = true
msg.debug("Adding to watched_ids: " .. tostring(video_id))
end
local upnext, num_upnext = load_upnext()
if num_upnext > 0 then
if skip_shorter_than > -1 or skip_longer_than > -1 then
-- Append first video that is not too long or too short
for _, v in ipairs(upnext) do
local skip_it = false
if v ~= nil then
if v.length ~= nil and v.length > 0 then
if skip_shorter_than > -1 and v.length < skip_shorter_than then
skip_it = true
end
if skip_longer_than > -1 and v.length > skip_longer_than then
skip_it = true
end
if not skip_it then
-- Append first video
add_to_playlist(v.file, v.label, v.length, "append")
appended_to_playlist[v.file] = true
return
end
end
end
end
msg.warn("No video between " .. opts.skip_shorter_than .. " and " .. opts.skip_longer_than .. " found")
end
-- Append first video
add_to_playlist(upnext[1].file, upnext[1].label, upnext[1].length, "append")
appended_to_playlist[upnext[1].file] = true
end
end
end
local function show_menu()
local upnext, num_upnext = load_upnext()
if num_upnext == 0 then
return
end
mp.osd_message("", 1)
local timeout = nil
local no_video = not mp.get_property("current-vo")
local selected = 1
local function choose_prefix(i, already_appended)
if i == selected and already_appended then
return opts.cursor_appended_selected
elseif i == selected then
return opts.cursor_selected
end
if i ~= selected and already_appended then
return opts.cursor_appended
elseif i ~= selected then
return opts.cursor_unselected
end
return "> " --shouldn't get here
end
local function choose_style(i, already_appended)
if i == selected and already_appended then
return 7
elseif i == selected then
return 7
end
if i ~= selected and already_appended then
return 2
elseif i ~= selected then
return 0
end
return 0 --shouldn't get here
end
local function selected_move(amt)
selected = selected + amt
if selected < 1 then
selected = num_upnext
elseif selected > num_upnext then
selected = 1
end
if timeout ~= nil then
timeout:kill()
timeout:resume()
end
if redraw_menu ~= nil then
redraw_menu()
end
end
local function end_terminal_menu()
if destroyer ~= nil then
destroyer()
end
end
local function terminal_submit(text, text_override)
if text_override ~= nil then
text = text_override
else
text = text:sub(handled_cursor)
end
if text:sub(handled_cursor) == "q" then
end_terminal_menu()
return
end
local number = tonumber(text:sub(handled_cursor))
if number ~= nil and number > 0 and number <= num_upnext then
selected = number
end
if text:sub(-1) == " " then
-- Append video to playlist
-- prevent appending the same video twice
if appended_to_playlist[upnext[selected].file] == true then
msg.info("Already in playlist: " .. upnext[selected].label)
if timeout ~= nil then
timeout:kill()
timeout:resume()
end
return
else
msg.info("Appending " .. upnext[selected].label .. " to playlist")
add_to_playlist(upnext[selected].file, upnext[selected].label, upnext[selected].length, "append")
appended_to_playlist[upnext[selected].file] = true
selected_move(1)
end
elseif opts.keep_playlist_on_selectthen then
-- Play (append to playlist)
msg.info("Playing " .. tostring(upnext[selected].label))
add_to_playlist(upnext[selected].file, upnext[selected].label, upnext[selected].length, "append-play")
local playlist_index_current = tonumber(mp.get_property("playlist-current-pos", "1"))
local playlist_index_newfile = tonumber(mp.get_property("playlist-count", "1")) - 1
mp.commandv("playlist-move", playlist_index_newfile, playlist_index_current + 1)
mp.commandv("playlist-play-index", playlist_index_current + 1)
appended_to_playlist[upnext[selected].file] = true
end_terminal_menu()
else
-- Play (replace playlist)
msg.info("Playing " .. tostring(upnext[selected].label))
add_to_playlist(upnext[selected].file, upnext[selected].label, upnext[selected].length, "replace")
end_terminal_menu()
end
end
local function terminal_edited(text)
local number = tonumber(text:sub(handled_cursor))
if number ~= nil and number > 0 and number <= num_upnext then
selected = number
if redraw_menu ~= nil then
redraw_menu()
end
end
-- Submit "append" action on space
if text:sub(-1) == " " then
terminal_submit("", tostring(selected) .. " ")
handled_cursor = #text + 1
end
end
local function terminal_closed()
if destroyer ~= nil then
destroyer()
end
handled_cursor = 0
end
local function draw_menu()
local ass = assdraw.ass_new()
local w, h = mp.get_osd_size()
if opts.curtain_opacity ~= nil and opts.curtain_opacity ~= 0 and opts.curtain_opacity < 1.0 then
-- From https://github.com/christoph-heinrich/mpv-quality-menu/blob/501794bfbef468ee6a61e54fc8821fe5cd72c4ed/quality-menu.lua#L699-L707
local alpha = 255 - math.ceil(255 * opts.curtain_opacity)
ass.text = string.format("{\\pos(0,0)\\rDefault\\an7\\1c&H000000&\\alpha&H%X&}", alpha)
ass:draw_start()
ass:rect_cw(0, 0, w, h)
ass:draw_stop()
ass:new_event()
end
ass:pos(opts.text_padding_x, opts.text_padding_y)
ass:append(opts.style_ass_tags)
local terminal_lines = {}
-- Print media info
local media_title = mp.get_property("media-title")
local metadata = mp.get_property_native("metadata")
if metadata then
if metadata["uploader"] then
table.insert(terminal_lines, " Uploader: " .. metadata["uploader"])
end
if metadata["channel_url"] then
table.insert(terminal_lines, " Channel URL: " .. metadata["channel_url"])
end
end
if media_title then
table.insert(terminal_lines, " Title: " .. media_title)
end
table.insert(terminal_lines, {
text = "░░░░░░░░░░░░░░░░░░░░░░░░ Up Next ░░░░░░░░░░░░░░░░░░░░░░░░",
terminal_style = "\027[1m",
})
local skipped = 0
local entries = 0
for i, v in ipairs(upnext) do
local skip_it = false
if v ~= nil then
local duration = ""
if v.length ~= nil and v.length > 0 then
duration = " " .. create_yt_time(v.length)
if opts.hide_skipped_videos then
if skip_shorter_than > -1 and v.length < skip_shorter_than then
skipped = skipped + 1
skip_it = true
end
if skip_longer_than > -1 and v.length > skip_longer_than then
skipped = skipped + 1
skip_it = true
end
end
end
if not skip_it then
ass:append(choose_prefix(i, appended_to_playlist[v.file] ~= nil) .. v.label .. duration .. "\\N")
entries = entries + 1
local padded_label = v.label .. string.rep(" ", 17 - #v.label)
local number = tostring(entries)
number = string.rep(" ", 2 - #number) .. number .. " "
table.insert(terminal_lines, {
text = number ..
choose_prefix(i, appended_to_playlist[v.file] ~= nil) .. duration .. " " .. padded_label,
terminal_style = "\027[" .. choose_style(i, appended_to_playlist[v.file] ~= nil) .. "m",
})
end
end
end
if entries == 0 and skipped > 0 then
if skip_shorter_than > -1 and skip_longer_than > -1 then
ass:append("No videos between " ..
opts.skip_shorter_than .. " and " .. opts.skip_longer_than .. " found\\N")
elseif skip_shorter_than > -1 then
ass:append("No videos shorter than " .. opts.skip_shorter_than .. " found\\N")
else
ass:append("No videos longer than " .. opts.skip_longer_than .. " found\\N")
end
end
if opts.scale_playlist_by_window then
w, h = 0, 0
end
mp.set_osd_ass(w, h, ass.text)
if no_video then
input.set_log(terminal_lines)
-- Open terminal input
terminal_id = terminal_id + 1
input.get({
prompt = "Select next video:",
submit = terminal_submit,
edited = terminal_edited,
closed = terminal_closed,
id = "upnext" .. tostring(terminal_id),