DRY-ify fennel loader

bismuthsoft · Sep 19, 2024 · ab31f16 · ab31f16
1 parent 188f14a
commit ab31f16
Showing 1 changed file with 8 additions and 13 deletions.
diff --git a/src/main.lua b/src/main.lua
@@ -3,19 +3,14 @@
 local fennel = require("lib.fennel")
 local make_love_searcher = function(env)
   return function(module_name)
-    local path = module_name:gsub("%.", "/") .. ".fnl"
-    if love.filesystem.getInfo(path) then
-      return function(...)
-        local code = love.filesystem.read(path)
-        return fennel.eval(code, {env=env, filename=path}, ...)
-      end, path
-    end
-    path = module_name:gsub("%.", "/") .. "/init.fnl"
-    if love.filesystem.getInfo(path) then
-      return function(...)
-        local code = love.filesystem.read(path)
-        return fennel.eval(code, {env=env, filename=path}, ...)
-      end, path
+    for _, filename in ipairs({".fnl", "/init.fnl"}) do
+      local path = module_name:gsub("%.", "/") .. filename
+      if love.filesystem.getInfo(path) then
+        return function(...)
+          local code = love.filesystem.read(path)
+          return fennel.eval(code, {env=env, filename=path}, ...)
+        end
+      end
     end
   end
 end