steamutils 1.4.40 → 1.4.41

package/create_proto.js

@@ -1,47 +1,96 @@
-import helpers, { loadProfos } from "./helpers/protos.js";
-import path from "path";
-import { fileURLToPath } from "url";
-import Protobuf from "protobufjs";
-import fs from "fs";
-
-const __dirname = path.dirname(fileURLToPath(import.meta.url));
-
-function main() {
-  const rootDir = `${__dirname}/protos/`;
-  const Protos = helpers([
-    {
-      name: "csgo",
-      protos: loadProfos(rootDir),
-    },
-  ]);
-
-  const fileName = "steamproto.js";
-  const content = fs.readFileSync("_steamproto.js", { encoding: "utf8", flag: "r" });
-
-  fs.writeFileSync(fileName, `${content}\nexport const SteamProtoType = {};export const ESteamProto = {};`);
-
-  for (const key in Protos.csgo) {
-    let csgo = Protos.csgo[key];
-    if (typeof csgo === "function" || typeof csgo === "boolean" || typeof csgo === "string" || Array.isArray(csgo) || csgo === null) {
-      continue;
-    }
-
-    if (csgo instanceof Protobuf.Type) {
-      const fields = {};
-      for (const fieldsKey in csgo.fields) {
-        fields[fieldsKey] = fieldsKey;
-      }
-
-      const filename = path.relative(rootDir, csgo.filename);
-      fs.appendFileSync(fileName, `SteamProtoType.${key} = ${JSON.stringify({ name: csgo.name, filename })};`);
-    } else {
-      for (const csgoKey in csgo) {
-        if (csgo[csgoKey] instanceof Protobuf.Enum) {
-          fs.appendFileSync(fileName, `ESteamProto.${csgoKey} = ${JSON.stringify(csgo[csgoKey].values)};`);
-        }
-      }
-    }
-  }
-}
-
-main();
+import helpers, { loadProfos } from "./helpers/protos.js";
+import path from "path";
+import { fileURLToPath } from "url";
+import Protobuf from "protobufjs";
+import fs from "fs";
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+const shortfileName = "steamproto.js";
+const fullfileName = "full_steamproto.js";
+
+function main() {
+  const rootDir = `${__dirname}/protos/`;
+  const Protos = helpers([
+    {
+      name: "csgo",
+      protos: loadProfos(rootDir),
+    },
+  ]);
+
+  const content = fs.readFileSync("_steamproto.js", { encoding: "utf8", flag: "r" });
+
+  fs.writeFileSync(shortfileName, `${content}\nexport const SteamProtoType = {};export const ESteamProto = {};`);
+  fs.writeFileSync(fullfileName, `${content}\nexport const SteamProtoType = {};export const ESteamProto = {};`);
+
+  for (const key in Protos.csgo) {
+    let csgo = Protos.csgo[key];
+    if (typeof csgo === "function" || typeof csgo === "boolean" || typeof csgo === "string" || Array.isArray(csgo) || csgo === null) {
+      continue;
+    }
+
+    if (csgo instanceof Protobuf.Type) {
+      const fields = {};
+      for (const fieldsKey in csgo.fields) {
+        fields[fieldsKey] = fieldsKey;
+      }
+
+      const filename = path.relative(rootDir, csgo.filename);
+      const text = `SteamProtoType.${key} = ${JSON.stringify({ name: csgo.name, filename })};`;
+      fs.appendFileSync(fullfileName, text);
+      if (hasKeyDir(key, __dirname)) {
+        fs.appendFileSync(shortfileName, text);
+      }
+    } else {
+      for (const csgoKey in csgo) {
+        if (csgo[csgoKey] instanceof Protobuf.Enum) {
+          const values = csgo[csgoKey].values;
+          fs.appendFileSync(fullfileName, `ESteamProto.${csgoKey} = ${JSON.stringify(values)};`);
+
+          for (const valuesKey in values) {
+            if (!hasKeyDir(`${csgoKey}.${valuesKey}`, __dirname)) {
+              delete values[valuesKey];
+            }
+          }
+          if (Object.keys(values).length) {
+            fs.appendFileSync(shortfileName, `ESteamProto.${csgoKey} = ${JSON.stringify(values)};`);
+          }
+        }
+      }
+    }
+  }
+}
+
+const ignoreFiles = ["create_remote_file.js", shortfileName, fullfileName];
+
+export function hasKeyDir(key, dir) {
+  const currentDir = fs.readdirSync(dir);
+  for (const currentFile of currentDir) {
+    if (!["SteamClient.js", "index.js"].includes(currentFile)) {
+      continue;
+    }
+
+    const _dir = path.join(dir, currentFile);
+    const isDirectory = fs.lstatSync(_dir).isDirectory();
+    if (isDirectory) {
+      if (currentFile !== "node_modules") {
+        if (hasKeyDir(key, _dir)) {
+          return true;
+        }
+      }
+    } else {
+      if (currentFile.endsWith(".js") && !ignoreFiles.includes(currentFile)) {
+        if (hasKeyFile(key, _dir)) {
+          return true;
+        }
+      }
+    }
+  }
+  return false;
+}
+
+export function hasKeyFile(key, file) {
+  const content = fs.readFileSync(file);
+  return content.includes(key);
+}
+
+main();