From 5259fef743eb899b0ec1ddcc5bb2169f2cbce929 Mon Sep 17 00:00:00 2001
From: FliegendeWurst <2012gdwu@web.de>
Date: Sun, 26 Jul 2020 10:41:50 +0200
Subject: [PATCH] Initial commit

---
 input.conf                 |   3 ++
 ratings-based-shuffle.conf |   2 +
 ratings-based-shuffle.lua  | 102 +++++++++++++++++++++++++++++++++++++
 3 files changed, 107 insertions(+)
 create mode 100644 input.conf
 create mode 100644 ratings-based-shuffle.conf
 create mode 100644 ratings-based-shuffle.lua

diff --git a/input.conf b/input.conf
new file mode 100644
index 0000000..4880dc4
--- /dev/null
+++ b/input.conf
@@ -0,0 +1,3 @@
+Alt+r script-message RBS-init
+Alt+UP script-message RBS-upvote
+Alt+DOWN script-message RBS-downvote
diff --git a/ratings-based-shuffle.conf b/ratings-based-shuffle.conf
new file mode 100644
index 0000000..9849e8e
--- /dev/null
+++ b/ratings-based-shuffle.conf
@@ -0,0 +1,2 @@
+directory=/home/username/Music
+ratings_file=/home/username/.config/mpv/rbs-ratings.txt
diff --git a/ratings-based-shuffle.lua b/ratings-based-shuffle.lua
new file mode 100644
index 0000000..da7db8a
--- /dev/null
+++ b/ratings-based-shuffle.lua
@@ -0,0 +1,102 @@
+utils = require 'mp.utils'
+require 'mp.options'
+
+local options = {
+    directory = ".",
+    ratings_file = "RBS-ratings.txt",
+}
+read_options(options, "ratings-based-shuffle")
+
+all_files = {}
+ratings = {}
+
+function init_playlist()
+    mp.osd_message("Initializing ratings-based shuffle..")
+    mp.set_property("shuffle", "no")
+    mp.set_property("loop-playlist", "no")
+
+    load_ratings(options.ratings_file)
+    load(options.directory)
+
+    mp.commandv('loadfile', all_files[math.random(#all_files)], 'replace')
+
+    mp.register_event("end-file", auto_add_file)
+    auto_add_file(nil)
+    auto_add_file(nil)
+end
+
+function load(path)
+    for idx, name in ipairs(utils.readdir(path, "dirs")) do
+        load(utils.join_path(path, name))
+    end
+    for idx, name in ipairs(utils.readdir(path, "files")) do
+        table.insert(all_files, utils.join_path(path, name))
+    end
+end
+
+function load_ratings(path)
+    info = utils.file_info(path)
+    if info == nil then
+        -- nothing to read
+    elseif info.is_file then
+        file = io.open(path, "r")
+        io.input(file)
+        ratings, err = utils.parse_json(io.read())
+        io.close(file)
+    else
+        msg.warn("could not load ratings")
+    end
+end
+
+function save_ratings(path)
+    info = utils.file_info(path)
+    if info == nil or info.is_file then
+        file = io.open(path, "w")
+        io.output(file)
+        json, err = utils.format_json(ratings)
+        io.write(json)
+        io.close(file)
+    else
+        msg.error("could not save ratings")
+    end
+end
+
+function upvote()
+    file = mp.get_property("path")
+    if ratings[file] == nil then
+        ratings[file] = 1.1
+    else
+        ratings[file] = ratings[file] * 1.1
+    end
+    save_ratings(options.ratings_file)
+end
+
+function downvote()
+    file = mp.get_property("path")
+    if ratings[file] == nil then
+        ratings[file] = 0.9
+    else
+        ratings[file] = ratings[file] * 0.9
+    end
+    save_ratings(options.ratings_file)
+end
+
+function auto_add_file(event)
+    while true do
+        idx = math.random(#all_files)
+        file = all_files[idx]
+        chance = ratings[file]
+        if chance == nil then
+            chance = 1.0
+        end
+        if math.random() < (chance/#all_files) then
+            mp.commandv('loadfile', file, 'append')
+            break
+        end
+    end
+end
+
+math.randomseed(math.sin(os.time())*10000)
+mp.register_script_message("RBS-init", init_playlist)
+mp.register_script_message("RBS-upvote", upvote)
+mp.register_script_message("RBS-downvote", downvote)