storm-lua-minify 0.1.1 → 0.1.2

package/dist/ast2lua.js

@@ -2,7 +2,7 @@
 // based on "luamin": Copyright Mathias Bynens <https://mathiasbynens.be/>
 // SPDX-License-Identifier: MIT
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Minifier = void 0;
+exports.MinifyFile = void 0;
 const source_map_1 = require("source-map");
 const PRECEDENCE = {
     or: 1,
@@ -89,8 +89,6 @@ const IDENTIFIER_PARTS = [
     "Z",
     "_",
 ];
-const identifierMap = new Map();
-const identifiersInUse = new Set();
 function wrapArray(obj) {
     if (Array.isArray(obj)) {
         return obj;
@@ -206,25 +204,27 @@ function prependWithSeparator(val, prepending, separator = " ") {
 function insertSeparator(a, b, separator = " ") {
     return isNeedSeparator(a.toString(), b.toString()) ? separator : undefined;
 }
-class Minifier {
+class MinifyFile {
     fileName;
     ast;
-    requireHelper;
-    constructor(fileName, ast, requireHelper) {
+    minifier;
+    mode;
+    constructor(fileName, ast, minifier, mode) {
         this.fileName = fileName;
         this.ast = ast;
-        this.requireHelper = requireHelper;
-        ast.globals?.map((v) => identifiersInUse.add(v.name));
+        this.minifier = minifier;
+        this.mode = mode;
     }
-    parse() {
+    parse(noComment) {
         const body = this.formatStatementList(this.ast.body);
-        /*if (this.ast.comments) {
-          const comments = this.ast.comments as Comment[];
-          comments.reverse().filter(v => v.raw.includes("--#") || v.raw.includes("[[#")).forEach(comment => {
-            body.prepend(this.sourceNodeHelper(comment, comment.raw));
-          })
-          return body;
-        } else*/ {
+        if (!noComment && this.ast.comments) {
+            const comments = this.ast.comments;
+            comments.reverse().filter(v => v.raw.includes("--#") || v.raw.includes("[[#")).forEach(comment => {
+                body.prepend([this.sourceNodeHelper(comment, comment.raw), "\n"]);
+            });
+            return body;
+        }
+        else {
             return body;
         }
     }
@@ -501,11 +501,21 @@ class Minifier {
         }
         else if (expression.type == "CallExpression") {
             // requireの展開モードは2種類: SLモード-その場に読み込み, FLモード-require相当の関数で呼び出し
-            if (this.formatBase(expression.base).toString() == "require" &&
+            const callExpr = this.formatBase(expression.base).toString();
+            if ((callExpr === "require" || callExpr === "dofile") &&
                 expression?.arguments[0].type === "StringLiteral") {
-                const res = this.requireHelper(expression.arguments[0].raw.replaceAll('"', ""));
-                if (res != undefined) {
-                    return res;
+                const moduleName = expression.arguments[0].raw
+                    .replaceAll('"', "")
+                    .replaceAll("'", "");
+                const res = this.minifier.parseModule(moduleName);
+                if (this.mode.moduleLikeLua) {
+                    if (callExpr === "dofile") {
+                        return (res || this.sourceNodeHelper(undefined, ""));
+                    }
+                    // requireならスキップ
+                }
+                else {
+                    return (res || this.sourceNodeHelper(undefined, ""));
                 }
             }
             const args = expression.arguments
@@ -632,7 +642,7 @@ class Minifier {
         if (nameItem.name === "self") {
             return this.sourceNodeHelper(nameItem, "self", "self");
         }
-        const defined = identifierMap.get(nameItem.name);
+        const defined = this.minifier.identifierMap.get(nameItem.name);
         if (defined) {
             return this.sourceNodeHelper(nameItem, defined, nameItem.name); // 第3引数は要調査
         }
@@ -649,21 +659,21 @@ class Minifier {
                         IDENTIFIER_PARTS[index + 1] +
                         generateZeroes(length - (position + 1));
                 if (isKeyword(this.currentIdentifier) ||
-                    identifiersInUse.has(this.currentIdentifier)) {
+                    this.minifier.identifiersInUse.has(this.currentIdentifier)) {
                     return this.generateIdentifier(nameItem, nested);
                 }
-                identifierMap.set(nameItem.name, this.currentIdentifier);
+                this.minifier.identifierMap.set(nameItem.name, this.currentIdentifier);
                 return this.generateIdentifier(nameItem, nested);
             }
             --position;
         }
         this.currentIdentifier = "a" + generateZeroes(length);
-        if (identifiersInUse.has(this.currentIdentifier)) {
+        if (this.minifier.identifiersInUse.has(this.currentIdentifier)) {
             return this.generateIdentifier(nameItem, nested);
         }
-        identifierMap.set(nameItem.name, this.currentIdentifier);
+        this.minifier.identifierMap.set(nameItem.name, this.currentIdentifier);
         return this.generateIdentifier(nameItem, nested);
         //    return this.sourceNodeHelper(nameItem, nameItem.name, nested ? nameItem.name : undefined);
     }
 }
-exports.Minifier = Minifier;
+exports.MinifyFile = MinifyFile;

package/dist/cli.js

@@ -7,10 +7,12 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const commander_1 = require("commander");
-const luaparse_1 = __importDefault(require("luaparse"));
-const ast2lua_1 = require("./ast2lua");
+const minifier_1 = require("./minifier");
 const program = new commander_1.Command();
-program.version("0.1.0").description("A Lua minifier also outputs source map");
+program
+    .version("0.1.2")
+    .description("A Lua minifier also outputs source map")
+    .option("-m, --module-like-lua", "require・dofileの動作を実際のLuaに近づけます");
 program.parse(process.argv);
 const luaFiles = program.args;
 const luaparseSetting = {
@@ -19,48 +21,25 @@ const luaparseSetting = {
     ranges: true,
     scope: true,
 };
+const mode = program.opts();
 luaFiles.forEach((fileName) => {
-    const includes = new Map();
     const parsedFileName = path_1.default.parse(fileName);
-    function requireHelper(recursiveFilePath) {
-        const resolvePath = path_1.default.relative(parsedFileName.dir, path_1.default.join(parsedFileName.dir, recursiveFilePath.replaceAll(".", path_1.default.sep) + ".lua"));
-        const fullResolvePath = path_1.default.join(parsedFileName.dir, resolvePath);
-        if (!includes.has(resolvePath) && fs_1.default.existsSync(fullResolvePath)) {
-            const code = fs_1.default.readFileSync(fullResolvePath).toString();
-            const ast = luaparse_1.default.parse(code, luaparseSetting);
-            if ("globals" in ast) {
-                includes.set(resolvePath, code);
-                return new ast2lua_1.Minifier(resolvePath, ast, requireHelper).parse();
-            }
-            includes.set(resolvePath, "");
-            return undefined;
-        }
-    }
     if (fs_1.default.existsSync(fileName)) {
-        const code = fs_1.default.readFileSync(fileName).toString();
-        const ast = luaparse_1.default.parse(code, luaparseSetting);
-        if ("globals" in ast) {
-            includes.set(parsedFileName.base, code);
-            const map = new ast2lua_1.Minifier(parsedFileName.base, ast, requireHelper).parse();
-            includes.forEach((v, k) => {
-                console.log(k);
-                map.setSourceContent(k, v);
-            });
-            const minFileName = path_1.default.format({
-                dir: parsedFileName.dir,
-                name: parsedFileName.name + ".min",
-                ext: ".lua",
-            });
-            const mapFileName = path_1.default.format({
-                dir: parsedFileName.dir,
-                name: parsedFileName.name,
-                ext: parsedFileName.ext + ".map",
-            });
-            map.add("\n--[[\n//# sourceMappingURL=" + mapFileName + "\n]]");
-            const sourceAndMap = map.toStringWithSourceMap();
-            fs_1.default.writeFileSync(minFileName, sourceAndMap.code);
-            fs_1.default.writeFileSync(mapFileName, JSON.stringify(sourceAndMap.map));
-        }
+        const map = new minifier_1.Minifier(fileName, luaparseSetting, mode).parse();
+        const minFileName = path_1.default.format({
+            dir: parsedFileName.dir,
+            name: parsedFileName.name + ".min",
+            ext: ".lua",
+        });
+        const mapFileName = path_1.default.format({
+            dir: parsedFileName.dir,
+            name: parsedFileName.name,
+            ext: parsedFileName.ext + ".map",
+        });
+        map.add("\n--[[\n//# sourceMappingURL=" + mapFileName + "\n]]");
+        const sourceAndMap = map.toStringWithSourceMap();
+        fs_1.default.writeFileSync(minFileName, sourceAndMap.code);
+        fs_1.default.writeFileSync(mapFileName, JSON.stringify(sourceAndMap.map));
     }
     else {
         console.error("No such file: " + fileName);

package/dist/minifier.js

@@ -0,0 +1,94 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Minifier = void 0;
+const luaparse_1 = __importDefault(require("luaparse"));
+const path_1 = __importDefault(require("path"));
+const fs_1 = __importDefault(require("fs"));
+const source_map_1 = require("source-map");
+const ast2lua_1 = require("./ast2lua");
+class Minifier {
+    identifierMap;
+    identifiersInUse;
+    moduleSourceText;
+    moduleSourceNode;
+    moduleAST;
+    dir;
+    entryModule;
+    mode;
+    luaParseSettings;
+    constructor(entryFilePath, luaParseSettings, mode) {
+        this.identifierMap = new Map();
+        this.identifiersInUse = new Set();
+        this.moduleSourceText = new Map();
+        this.moduleSourceNode = new Map();
+        this.moduleAST = new Map();
+        this.luaParseSettings = luaParseSettings;
+        this.mode = mode;
+        const pn = path_1.default.parse(entryFilePath);
+        this.dir = pn.dir;
+        this.entryModule = pn.name;
+    }
+    parse() {
+        const sn = this.parseModule(this.entryModule);
+        if (this.mode.moduleLikeLua) {
+            // require関数を作成し、流し込む
+            /*
+            function require(m,r)
+              package=package or {loaded={}}
+              if package.loaded[m] then return package.loaded[m] end
+              if m=="<MODULE_NAME>" then r=(function()[[MODULE]]end)() end
+              package.loaded[m]=package.loaded[m] or r or true;return package.loaded[m]
+            end
+            */
+            sn.prepend("package.loaded[m]=package.loaded[m]or r or true;return package.loaded[m]end\n");
+            this.moduleSourceNode.forEach((v, k) => {
+                if (k !== this.entryModule) {
+                    sn.prepend(["if m==\"", k, "\"then r=(function() ", v, " end)()end\n"]);
+                }
+            });
+            sn.prepend("function require(m,r)package=package or{loaded={}};if package.loaded[m]then return package.loaded[m]end\n");
+        }
+        // コメントの流し込み
+        const comments = this.moduleAST.get(this.entryModule)?.comments;
+        if (comments) {
+            comments
+                .reverse()
+                .filter((v) => v.raw.includes("--#") || v.raw.includes("[[#"))
+                .forEach((comment) => {
+                sn.prepend([
+                    new source_map_1.SourceNode(comment.loc?.start.line || null, comment.loc?.start.column || null, this.entryModule, // 本当に自分のファイル名でよいかは要検討
+                    comment.raw),
+                    "\n"
+                ]);
+            });
+        }
+        return sn;
+    }
+    parseModule(moduleName) {
+        const resolvePath = moduleName.replaceAll(".", path_1.default.sep) + ".lua";
+        const fullResolvePath = path_1.default.join(this.dir, resolvePath);
+        if (this.moduleSourceNode.has(moduleName)) {
+            const res = this.moduleSourceNode.get(moduleName);
+            if (res) {
+                return res;
+            }
+        }
+        else if (fs_1.default.existsSync(fullResolvePath)) {
+            const code = fs_1.default.readFileSync(fullResolvePath).toString();
+            const ast = luaparse_1.default.parse(code, this.luaParseSettings);
+            if ("globals" in ast) {
+                ast.globals?.map((v) => this.identifiersInUse.add(v.name));
+                const sourceNode = new ast2lua_1.MinifyFile(resolvePath, ast, this, this.mode).parse(resolvePath === this.entryModule);
+                this.moduleSourceText.set(moduleName, code);
+                this.moduleAST.set(moduleName, ast);
+                this.moduleSourceNode.set(moduleName, sourceNode);
+                return sourceNode;
+            }
+        }
+        throw new Error(moduleName + " is not found");
+    }
+}
+exports.Minifier = Minifier;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "storm-lua-minify",
   "description": "A Lua minifier also outputs source map",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "dependencies": {
     "commander": "^12.1.0",
     "luamin": "^1.0.4",

package/src/cli.ts

@@ -9,7 +9,7 @@ import { Minifier, MinifierMode } from "./minifier";
 const program = new Command();
 
 program
-  .version("0.1.1")
+  .version("0.1.2")
   .description("A Lua minifier also outputs source map")
   .option(
     "-m, --module-like-lua",