From f0294e40d89df9e29e7453afc61cf7edda25d2fc Mon Sep 17 00:00:00 2001
From: Marko Korhonen <marko@korhonen.cc>
Date: Tue, 31 May 2022 17:08:07 +0300
Subject: [PATCH] mpv: fix playing yle areena with plugin, update mimeo
 association

---
 home/.config/mimeo/associations.txt     |  6 +-
 home/.config/mpv/scripts/yledl_hook.lua | 76 +++++++++++++++++++++++++
 2 files changed, 79 insertions(+), 3 deletions(-)
 create mode 100644 home/.config/mpv/scripts/yledl_hook.lua

diff --git a/home/.config/mimeo/associations.txt b/home/.config/mimeo/associations.txt
index da50c637..d9a37f2d 100644
--- a/home/.config/mimeo/associations.txt
+++ b/home/.config/mimeo/associations.txt
@@ -34,6 +34,9 @@ mpvqueue %U
 mpvqueue %U
   ^https?://v.redd.it/.*
 
+mpvqueue %U
+  ^https?://areena.yle.fi.*
+
 imageviewer %U
   ^https?://(?:[a-z0-9\-]+\.)+[a-z]{2,6}(?:/[^/#?]+)+\.(?:jpg|jpeg|png|svg)
 
@@ -45,6 +48,3 @@ alacritty -e rtv "%U"
 
 alacritty -e rtv "%U"
   ^https?://redd.it.*
-
-mpvqueue $(yle-dl --showurl %U) # TODO Command substition not working in mimeo
-  ^https?://areena.yle.fi.*
diff --git a/home/.config/mpv/scripts/yledl_hook.lua b/home/.config/mpv/scripts/yledl_hook.lua
new file mode 100644
index 00000000..1ff15ff1
--- /dev/null
+++ b/home/.config/mpv/scripts/yledl_hook.lua
@@ -0,0 +1,76 @@
+-- Copied from https://github.com/pekkarr/mpv-yledl
+-- Copyright 2021 Pekka Ristola
+
+-- This program is free software: you can redistribute it and/or modify
+-- it under the terms of the GNU General Public License as published by
+-- the Free Software Foundation, either version 3 of the License, or
+-- (at your option) any later version.
+--
+-- This program is distributed in the hope that it will be useful,
+-- but WITHOUT ANY WARRANTY; without even the implied warranty of
+-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+-- GNU General Public License for more details.
+--
+-- You should have received a copy of the GNU General Public License
+-- along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+local msg = require 'mp.msg'
+local utils = require 'mp.utils'
+
+local function add_series(data)
+  local playlist = { "#EXTM3U" }
+  for _, episode in pairs(data) do
+    table.insert(playlist, episode["webpage"])
+  end
+  mp.set_property("stream-open-filename", "memory://" .. table.concat(playlist, "\n"))
+end
+
+local function add_single_video(data)
+  local flavors = data["flavors"]
+  table.sort(flavors, function(a, b) return a["bitrate"] > b["bitrate"] end)
+  local best = flavors[1]
+  mp.set_property("stream-open-filename", best["url"])
+  mp.set_property("file-local-options/force-media-title", data["title"])
+  for _, sub in ipairs(data["subtitles"]) do
+    local lang = sub["language"]
+    msg.verbose("Adding subtitles for " .. lang)
+    mp.commandv("sub-add", sub["url"], "auto", sub["category"], lang)
+  end
+end
+
+mp.add_hook("on_load", 9, function()
+  msg.verbose('yle-dl hook')
+  local url = mp.get_property("stream-open-filename", "")
+  if (url:find("https?://%a+%.yle%.fi/") == 1) or (url:find("https?://yle%.fi/") == 1) then
+    local start_time = os.clock()
+    local command = { "yle-dl", "--showmetadata", url }
+    msg.debug("Running: " .. table.concat(command, ' '))
+    local ret = mp.command_native({name = "subprocess",
+                                   args = command,
+                                   capture_stdout = true,
+                                   capture_stderr = false})
+    if ret.killed_by_us then
+      return
+    end
+    if (ret.status < 0) or (ret.stdout == nil) or (ret.stdout == "") then
+      msg.error("yle-dl failed to parse url")
+      return
+    end
+
+    local json, err = utils.parse_json(ret.stdout)
+    if (json == nil) then
+      msg.error("failed to parse JSON: " .. err)
+      return
+    end
+    msg.verbose("yle-dl succeeded")
+    msg.debug("running yle-dl took " .. os.clock() - start_time .. " seconds")
+
+    if #json == 1 then
+      add_single_video(json[1])
+    else
+      add_series(json)
+    end
+  else
+    msg.verbose('not an areena url')
+  end
+end)