@runsec/mcp 1.0.1 → 1.0.2

package/dist/index.js

@@ -37,9 +37,21 @@ var import_ignore = __toESM(require("ignore"));
 var import_node_fs = __toESM(require("fs"));
 var import_node_path = __toESM(require("path"));
 var import_js_yaml = __toESM(require("js-yaml"));
-var DATA_DIR = import_node_path.default.resolve(__dirname, "../rules/data");
-var SEMGREP_RULES_DIR = import_node_path.default.join(DATA_DIR, "semgrep-rules");
-var COMPLIANCE_MAP_PATH = import_node_path.default.join(DATA_DIR, "rule-compliance-map.json");
+var cachedDataDirectory = null;
+function getDataDirectory() {
+  if (cachedDataDirectory) return cachedDataDirectory;
+  const prodPath = import_node_path.default.join(__dirname, "data");
+  const fallbackPath = import_node_path.default.join(__dirname, "..", "src", "rules", "data");
+  if (import_node_fs.default.existsSync(prodPath) && import_node_fs.default.statSync(prodPath).isDirectory()) {
+    cachedDataDirectory = import_node_path.default.resolve(prodPath);
+    return cachedDataDirectory;
+  }
+  if (import_node_fs.default.existsSync(fallbackPath) && import_node_fs.default.statSync(fallbackPath).isDirectory()) {
+    cachedDataDirectory = import_node_path.default.resolve(fallbackPath);
+    return cachedDataDirectory;
+  }
+  throw new Error(`Cannot find rules data directory. Looked in: ${prodPath}, ${fallbackPath}`);
+}
 var PCI_CWE = /* @__PURE__ */ new Set(["CWE-798", "CWE-327", "CWE-256", "CWE-89", "CWE-79", "CWE-22", "CWE-287", "CWE-285", "CWE-522"]);
 var SOC2_CWE = /* @__PURE__ */ new Set(["CWE-285", "CWE-306", "CWE-287", "CWE-863", "CWE-16", "CWE-200", "CWE-862"]);
 var HIPAA_CWE = /* @__PURE__ */ new Set(["CWE-532", "CWE-359", "CWE-353", "CWE-345", "CWE-200", "CWE-522"]);
@@ -70,7 +82,8 @@ function extractMetricId(id, message) {
   return id.toUpperCase();
 }
 function readComplianceMap() {
-  const raw = import_node_fs.default.readFileSync(COMPLIANCE_MAP_PATH, "utf-8");
+  const complianceMapPath = import_node_path.default.join(getDataDirectory(), "rule-compliance-map.json");
+  const raw = import_node_fs.default.readFileSync(complianceMapPath, "utf-8");
   return JSON.parse(raw);
 }
 function collectRulePatterns(rule) {
@@ -95,11 +108,12 @@ function collectRulePatterns(rule) {
   return Array.from(new Set(patterns));
 }
 function parseSemgrepRuleFiles() {
-  const files = import_node_fs.default.readdirSync(SEMGREP_RULES_DIR).filter((f) => f.endsWith(".yaml") || f.endsWith(".yml"));
+  const semgrepRulesDir = import_node_path.default.join(getDataDirectory(), "semgrep-rules");
+  const files = import_node_fs.default.readdirSync(semgrepRulesDir).filter((f) => f.endsWith(".yaml") || f.endsWith(".yml"));
   const compliance = readComplianceMap();
   const out = [];
   for (const fileName of files) {
-    const full = import_node_path.default.join(SEMGREP_RULES_DIR, fileName);
+    const full = import_node_path.default.join(semgrepRulesDir, fileName);
     const parsed = import_js_yaml.default.load(import_node_fs.default.readFileSync(full, "utf-8"));
     const rows = Array.isArray(parsed?.rules) ? parsed.rules : [];
     for (const row of rows) {

package/package.json

@@ -1,11 +1,10 @@
 {
   "name": "@runsec/mcp",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "main": "dist/index.js",
   "files": [
     "dist",
-    "README.md",
-    "src/rules/data"
+    "README.md"
   ],
   "bin": {
     "runsec-mcp": "./dist/index.js"
@@ -14,7 +13,7 @@
     "access": "public"
   },
   "scripts": {
-    "build": "tsup src/index.ts --format cjs --clean",
+    "build": "tsup src/index.ts --format cjs --clean && node -e \"require('fs').cpSync('src/rules/data', 'dist/data', {recursive: true})\"",
     "test": "vitest run",
     "simulate:output": "tsx scripts/simulate_output.ts"
   },