l 2 mpv: refactor, add "mark as good" key

1 files changed, 23 insertions, 4 deletions

diff --git a/lass/2configs/mpv.nix b/lass/2configs/mpv.nix
index 71f1409e0..ff5698e4e 100644
--- a/lass/2configs/mpv.nix
+++ b/lass/2configs/mpv.nix
@@ -1,15 +1,34 @@
-{ pkgs, ... }:
+{ pkgs, lib, ... }:
 
 let
 
   mpv-config = pkgs.writeText "mpv-config" ''
-    script=${deleteCurrentFile}
+    script=${lib.concatStringsSep "," [
+      good
+      delete
+    ]}
   '';
   mpv = pkgs.writeDashBin "mpv" ''
-    exec ${pkgs.mpv}/bin/mpv --no-config --include=${mpv-config} $@
+    exec ${pkgs.mpv}/bin/mpv --no-config --include=${mpv-config} "$@"
   '';
 
-  deleteCurrentFile = pkgs.writeText "delete.lua" ''
+  moveToDir = key: dir: pkgs.writeText "move-with-${key}.lua" ''
+    tmp_dir = "${dir}"
+
+    function move_current_track_${key}()
+      track = mp.get_property("path")
+      os.execute("mkdir -p '" .. tmp_dir .. "'")
+      os.execute("mv '" .. track .. "' '" .. tmp_dir .. "'")
+      print("moved '" .. track .. "' to " .. tmp_dir)
+    end
+
+    mp.add_key_binding("${key}", "move_current_track_${key}", move_current_track_${key})
+  '';
+
+  good = moveToDir "G" "./.good";
+  delete = moveToDir "D" "./.graveyard";
+
+  deleteCurrentTrack = pkgs.writeText "delete.lua" ''
     deleted_tmp = "./.graveyard"
 
     -- Delete the current track by moving it to the `deleted_tmp` location.