From 4f0509d30e17e3336de6bca7564bf5bb674cf6a0 Mon Sep 17 00:00:00 2001
From: MCorange <mcorange@mcorangehq.xyz>
Date: Sun, 25 Aug 2024 22:03:37 +0300
Subject: [PATCH] :3

---
 rtmc.lua |  2 ++
 x.py     | 24 +++++++++++++++---------
 2 files changed, 17 insertions(+), 9 deletions(-)

diff --git a/rtmc.lua b/rtmc.lua
index 101c5e4..8e0e4db 100644
--- a/rtmc.lua
+++ b/rtmc.lua
@@ -3,6 +3,8 @@ local __DEFAULT_IMPORT = require
 local require = function(path)
     if __BUNDLER_FILES[path] then
         return __BUNDLER_FILES[path]()
+    elseif __BUNDLER_FILES[path .. ".lua"] then
+        return __BUNDLER_FILES[path .. ".lua"]()
     else
         return __DEFAULT_IMPORT(path)
     end
diff --git a/x.py b/x.py
index 928aba2..57936c0 100755
--- a/x.py
+++ b/x.py
@@ -39,17 +39,23 @@ def minimise(buf: str) -> str:
 #    hasher.update(bytes(buf, "utf-8"));
 #    return hasher.hexdigest();
 
+REQ_HEADER="""local __BUNDLER_FILES = {}
+local __DEFAULT_IMPORT = require
+local require = function(path)
+    if __BUNDLER_FILES[path] then
+        return __BUNDLER_FILES[path]()
+    elseif __BUNDLER_FILES[path .. ".lua"] then
+        return __BUNDLER_FILES[path .. ".lua"]()
+    else
+        return __DEFAULT_IMPORT(path)
+    end
+end
+"""
+
+
 def main():
     buf = ""
-    buf += "local __BUNDLER_FILES = {}\n";
-    buf += "local __DEFAULT_IMPORT = require\n";
-    buf += "local require = function(path)\n";
-    buf += "    if __BUNDLER_FILES[path] then\n";
-    buf += "        return __BUNDLER_FILES[path]()\n";
-    buf += "    else\n";
-    buf += "        return __DEFAULT_IMPORT(path)\n";
-    buf += "    end\n";
-    buf += "end\n";
+    buf += REQ_HEADER;
     # buf += "local __UPDATE_HASH = __BUNDLER_REPLACE_HASH__\n";
     for file in FILES:
         print(f"=== FILE: {file}");