Merge pull request #36 from vordgi/cm-35

cm-35 add experimental freedNamesPolicy option
vordgi · Jun 27, 2024 · e856f6c · e856f6c
2 parents 02de8b5 + 3c14c19
commit e856f6c
Show file tree

Hide file tree

Showing 5 changed files with 60 additions and 18 deletions.
diff --git a/src/ClassnamesMinifier.ts b/src/ClassnamesMinifier.ts
@@ -1,22 +1,46 @@
 // eslint-disable-next-line @typescript-eslint/no-unused-vars
 import type { LoaderContext } from "webpack";
+import fs from "fs";
 import path from "path";
 import type { Config } from "./lib/types/plugin";
+import { CODE_VERSION } from "./lib/constants/configuration";
 import validateConfig from "./lib/validateConfig";
 import ConverterMinified from "./lib/ConverterMinified";
 import validateDist from "./lib/validateDist";
+import rmDist from "./lib/rmDist";
 
 class ClassnamesMinifier {
     converterMinified: ConverterMinified;
 
     constructor(config: Config) {
         validateConfig(config);
 
-        if (config.cacheDir) {
-            validateDist(config);
-        }
-
         this.converterMinified = new ConverterMinified(config);
+
+        if (!config.cacheDir || !config.distDir) {
+            console.log(
+                "classnames-minifier: Failed to check the dist folder because cacheDir or distDir is not specified",
+            );
+        } else {
+            const manifestDir = path.join(config.cacheDir, "ncm-meta");
+            const manifestPath = path.join(manifestDir, "manifest.json");
+
+            if (config.cacheDir) {
+                validateDist(config, manifestPath);
+            }
+
+            const { freedNamesPolicy = "transmit", blockLimit = 100000 } = config.experimental || {};
+            if (
+                freedNamesPolicy === "block" &&
+                this.converterMinified.freeClasses.length > blockLimit &&
+                config.distDir
+            ) {
+                rmDist(config.distDir, `Freed names exceeds the limit (${blockLimit})`);
+                this.converterMinified.clean();
+            }
+            if (!fs.existsSync(manifestDir)) fs.mkdirSync(manifestDir, { recursive: true });
+            fs.writeFileSync(manifestPath, JSON.stringify({ ...config, version: CODE_VERSION }), { encoding: "utf-8" });
+        }
     }
 
     get getLocalIdent() {

diff --git a/src/lib/ConverterMinified.ts b/src/lib/ConverterMinified.ts
@@ -19,7 +19,7 @@ class ConverterMinified {
 
     private reservedNames: string[];
 
-    dirtyСache: CacheType = {};
+    private freedNamesPolicy: "transmit" | "block";
 
     private symbols: string[] = [
         "a",
@@ -86,7 +86,9 @@ class ConverterMinified {
         "9",
     ];
 
-    private freeClasses: string[] = [];
+    dirtyСache: CacheType = {};
+
+    freeClasses: string[] = [];
 
     lastIndex = 0;
 
@@ -96,13 +98,23 @@ class ConverterMinified {
 
     private nameMap = [0];
 
-    constructor({ cacheDir, prefix = "", reservedNames = [] }: Config) {
+    constructor({ cacheDir, prefix = "", reservedNames = [], experimental }: Config) {
         this.prefix = prefix;
         this.reservedNames = reservedNames;
+        this.freedNamesPolicy = experimental?.freedNamesPolicy || "transmit";
 
         if (cacheDir) this.recycleCache(path.join(cacheDir, "ncm"));
     }
 
+    clean = () => {
+        this.dirtyСache = {};
+        this.freeClasses = [];
+        this.lastIndex = 0;
+        this.nextLoopEndsWith = 26;
+        this.currentLoopLength = 0;
+        this.nameMap = [0];
+    };
+
     private recycleCache = (cacheDir: string) => {
         this.cacheDir = cacheDir;
         if (!existsSync(cacheDir)) mkdirSync(cacheDir, { recursive: true });
@@ -185,7 +197,7 @@ class ConverterMinified {
 
     private getTargetClassName(origName: string) {
         let targetClassName: string;
-        if (this.freeClasses.length) {
+        if (this.freeClasses.length && this.freedNamesPolicy === "transmit") {
             targetClassName = this.freeClasses.shift() as string;
         } else {
             targetClassName = this.generateClassName();

diff --git a/src/lib/rmDist.ts b/src/lib/rmDist.ts
@@ -0,0 +1,9 @@
+import { rmSync } from "fs";
+
+const rmDist = (distDir: string, message: string) => {
+    console.log(`classnames-minifier: ${message}Cleaning the dist folder...`);
+    rmSync(distDir, { recursive: true, force: true });
+    console.log("classnames-minifier: Dist folder cleared");
+};
+
+export default rmDist;
diff --git a/src/lib/types/plugin.ts b/src/lib/types/plugin.ts
@@ -4,4 +4,8 @@ export type Config = {
     prefix?: string;
     reservedNames?: string[];
     disableDistDeletion?: boolean;
+    experimental?: {
+        freedNamesPolicy?: "transmit" | "block";
+        blockLimit?: number;
+    };
 };
diff --git a/src/lib/validateDist.ts b/src/lib/validateDist.ts
@@ -1,7 +1,7 @@
-import path from "path";
 import fs from "fs";
 import type { Config } from "./types/plugin";
 import { CODE_VERSION } from "./constants/configuration";
+import rmDist from "./rmDist";
 
 const readManifest = (manifestPath: string) => {
     try {
@@ -12,7 +12,7 @@ const readManifest = (manifestPath: string) => {
     }
 };
 
-const validateDist = (pluginOptions: Config) => {
+const validateDist = (pluginOptions: Config, manifestPath: string) => {
     const { cacheDir, distDir, prefix, reservedNames, disableDistDeletion } = pluginOptions;
 
     if (!cacheDir || !distDir) {
@@ -21,9 +21,6 @@ const validateDist = (pluginOptions: Config) => {
         );
         return;
     }
-
-    const manifestDir = path.join(cacheDir, "ncm-meta");
-    const manifestPath = path.join(manifestDir, "manifest.json");
     let configurationError = "";
 
     if (fs.existsSync(manifestPath)) {
@@ -60,15 +57,11 @@ const validateDist = (pluginOptions: Config) => {
     }
     if (configurationError) {
         if (!disableDistDeletion) {
-            console.log(`classnames-minifier: ${configurationError}Cleaning the dist folder...`);
-            fs.rmSync(distDir, { recursive: true, force: true });
-            console.log("classnames-minifier: Dist folder cleared");
+            rmDist(distDir, configurationError);
         } else {
             console.log(`classnames-minifier: ${configurationError}"disableDistDeletion" option was set to true`);
         }
     }
-    if (!fs.existsSync(manifestDir)) fs.mkdirSync(manifestDir, { recursive: true });
-    fs.writeFileSync(manifestPath, JSON.stringify({ ...pluginOptions, version: CODE_VERSION }), { encoding: "utf-8" });
 };
 
 export default validateDist;