From 18fa9f323d8adbb0b7b8b98a8488db308210ed93 Mon Sep 17 00:00:00 2001
From: Austin Horstman <khaneliman12@gmail.com>
Date: Sat, 1 Feb 2025 16:19:26 -0600
Subject: [PATCH] yazi: add main.lua support to plugins (#6394)

Yazi is deprecating init.lua for main.lua usages. This adds support for
main.lua in plugins.
---
 modules/programs/yazi.nix | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/modules/programs/yazi.nix b/modules/programs/yazi.nix
index f7970ef4..27737f86 100644
--- a/modules/programs/yazi.nix
+++ b/modules/programs/yazi.nix
@@ -263,8 +263,12 @@ in {
                 toString missingFiles
               }";
             singularOpt = removeSuffix "s" opt;
+            isPluginValid = opt == "plugins"
+              && (any (file: pathExists "${value}/${file}") requiredFiles);
+            isValid =
+              if opt == "plugins" then isPluginValid else missingFiles == [ ];
           in {
-            assertion = isDir && missingFiles == [ ];
+            assertion = isDir && isValid;
             message = ''
               Value at `programs.yazi.${opt}.${name}` is not a valid yazi ${singularOpt}.
               ${msgNotDir}
@@ -279,6 +283,6 @@ in {
       "preview.png"
       "LICENSE"
       "LICENSE-tmtheme"
-    ]) ++ (mkAsserts "plugins" [ "init.lua" ]);
+    ]) ++ (mkAsserts "plugins" [ "init.lua" "main.lua" ]);
   };
 }