From 340b98c0abb56ae24d7ee7dee64104583ad8a8c6 Mon Sep 17 00:00:00 2001
From: lordkekz <lordkekz@lkekz.de>
Date: Mon, 3 Jun 2024 20:29:06 +0200
Subject: [PATCH] yazi: Assert that plugins have valid structure

---
 modules/programs/yazi.nix | 23 +++++++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

diff --git a/modules/programs/yazi.nix b/modules/programs/yazi.nix
index f96a9bdf..59443dd3 100644
--- a/modules/programs/yazi.nix
+++ b/modules/programs/yazi.nix
@@ -209,12 +209,31 @@ in {
           else
             "yazi/plugins/${pluginName}.yazi";
 
-        mkPluginLink = pluginName: pluginPackage: {
+        mkPluginLink = pluginName: pluginPackageOrPath: {
           name = ensureSuffix pluginName;
-          value.source = pluginPackage;
+          value.source = pluginPackageOrPath;
         };
 
         pluginLinks = mapAttrs' mkPluginLink cfg.plugins;
       in pluginLinks);
+
+    assertions = (mapAttrsToList (pluginName: pluginPackageOrPath:
+      let
+        isDir = pathIsDirectory "${pluginPackageOrPath}";
+        hasInitLua = pathExists "${pluginPackageOrPath}/init.lua"
+          && !(pathIsDirectory "${pluginPackageOrPath}/init.lua");
+      in {
+        assertion = isDir && hasInitLua;
+        message =
+          "Value at `programs.yazi.plugins.${pluginName}` is not a valid yazi plugin."
+          + (optionalString (!isDir) ''
+
+            The path or package should be a directory, not a single file.'')
+          + (optionalString (!hasInitLua) ''
+
+            The path or package must contain a file `init.lua`.'') + ''
+
+              Evaluated value: `${pluginPackageOrPath}`'';
+      }) cfg.plugins);
   };
 }