lula2 0.7.5 → 0.8.4

package/dist/cli/server/server.js

@@ -1650,6 +1650,7 @@ var fileStore_exports = {};
 __export(fileStore_exports, {
   FileStore: () => FileStore
 });
+import { createHash as createHash2 } from "crypto";
 import {
   existsSync as existsSync2,
   promises as fs,
@@ -1662,7 +1663,6 @@ import {
 } from "fs";
 import * as yaml2 from "js-yaml";
 import { join as join2 } from "path";
-import { createHash as createHash2 } from "crypto";
 var FileStore;
 var init_fileStore = __esm({
   "cli/server/infrastructure/fileStore.ts"() {
@@ -1682,6 +1682,40 @@ var init_fileStore = __esm({
           this.refreshControlsCache();
         }
       }
+      /**
+       * Update a single mapping in place, preserving file order
+       */
+      async updateMapping(oldCompositeKey, updatedMapping) {
+        const mappingFiles = this.getAllMappingFiles();
+        for (const file of mappingFiles) {
+          try {
+            const content = readFileSync2(file, "utf8");
+            let mappings = yaml2.load(content) || [];
+            let changed = false;
+            mappings = mappings.map((m) => {
+              const hash = createHash2("sha256").update(JSON.stringify(m)).digest("hex");
+              if (`${m.control_id}:${hash}` === oldCompositeKey) {
+                const clean = { ...updatedMapping };
+                delete clean.hash;
+                changed = true;
+                return clean;
+              }
+              return m;
+            });
+            if (changed) {
+              const yamlContent = yaml2.dump(mappings, {
+                indent: 2,
+                lineWidth: -1,
+                noRefs: true
+              });
+              writeFileSync(file, yamlContent, "utf8");
+              return;
+            }
+          } catch (error) {
+            console.error(`Error processing mapping file ${file}:`, error);
+          }
+        }
+      }
       /**
        * Get simple filename from control ID
        */
@@ -1704,8 +1738,8 @@ var init_fileStore = __esm({
         if (!this.baseDir || this.baseDir === "." || this.baseDir === process.cwd()) {
           return;
         }
-        const lulaConfigPath2 = join2(this.baseDir, "lula.yaml");
-        if (!existsSync2(lulaConfigPath2)) {
+        const lulaConfigPath = join2(this.baseDir, "lula.yaml");
+        if (!existsSync2(lulaConfigPath)) {
           return;
         }
         if (!existsSync2(this.controlsDir)) {
@@ -1864,10 +1898,10 @@ var init_fileStore = __esm({
           return [];
         }
         let controlOrder = null;
+        const lulaConfigPath = join2(this.baseDir, "lula.yaml");
         try {
-          const lulaConfigPath2 = join2(this.baseDir, "lula.yaml");
-          if (existsSync2(lulaConfigPath2)) {
-            const content = readFileSync2(lulaConfigPath2, "utf8");
+          if (existsSync2(lulaConfigPath)) {
+            const content = readFileSync2(lulaConfigPath, "utf8");
             const metadata = yaml2.load(content);
             controlOrder = metadata?.controlOrder || null;
           }
@@ -5116,12 +5150,12 @@ import { fileURLToPath } from "url";
 import { readFileSync as readFileSync4 } from "fs";
 init_debug();
 init_controlHelpers();
-init_serverState();
 init_gitHistory();
+init_serverState();
 import * as yaml5 from "js-yaml";
+import crypto2 from "node:crypto";
 import { join as join5 } from "path";
 import { WebSocket, WebSocketServer } from "ws";
-import crypto2 from "node:crypto";
 var WebSocketManager = class {
   wss = null;
   clients = /* @__PURE__ */ new Set();
@@ -5162,6 +5196,62 @@ var WebSocketManager = class {
           }
           break;
         }
+        case "update-mapping": {
+          const state = getServerState();
+          if (payload && payload.old_composite_key && payload.mapping) {
+            const oldCompositeKey = payload.old_composite_key;
+            const existing = state.mappingsCache.get(oldCompositeKey);
+            if (!existing) {
+              console.error("Mapping not found for update:", oldCompositeKey);
+              break;
+            }
+            const incoming = payload.mapping;
+            const updated = {
+              ...existing,
+              ...incoming,
+              control_id: incoming.control_id || existing.control_id,
+              uuid: incoming.uuid || existing.uuid
+            };
+            if (!updated.hash || updated.hash === "") {
+              updated.hash = crypto2.createHash("sha256").update(JSON.stringify({ ...updated, hash: void 0 })).digest("hex");
+            }
+            const oldHash = existing.hash;
+            const oldControlId = existing.control_id;
+            const oldFamily = oldControlId.split("-")[0];
+            const newHash = updated.hash;
+            const newControlId = updated.control_id;
+            const newFamily = newControlId.split("-")[0];
+            const newCompositeKey = `${newControlId}:${newHash}`;
+            await state.fileStore.updateMapping(oldCompositeKey, updated);
+            const entries = Array.from(state.mappingsCache.entries());
+            const oldIndex = entries.findIndex(([key]) => key === oldCompositeKey);
+            if (oldIndex === -1) {
+              state.mappingsCache.delete(oldCompositeKey);
+              state.mappingsCache.set(newCompositeKey, updated);
+            } else {
+              entries[oldIndex] = [newCompositeKey, updated];
+              state.mappingsCache = new Map(entries);
+            }
+            state.mappingsByFamily.get(oldFamily)?.delete(oldHash);
+            state.mappingsByControl.get(oldControlId)?.delete(oldHash);
+            if (!state.mappingsByFamily.has(newFamily)) {
+              state.mappingsByFamily.set(newFamily, /* @__PURE__ */ new Set());
+            }
+            state.mappingsByFamily.get(newFamily).add(newHash);
+            if (!state.mappingsByControl.has(newControlId)) {
+              state.mappingsByControl.set(newControlId, /* @__PURE__ */ new Set());
+            }
+            state.mappingsByControl.get(newControlId).add(newHash);
+            ws.send(
+              JSON.stringify({
+                type: "mapping-updated",
+                payload: { uuid: updated.uuid, success: true }
+              })
+            );
+            this.broadcastState();
+          }
+          break;
+        }
         case "refresh-controls": {
           const state = getServerState();
           state.controlsCache.clear();

package/dist/cli/server/serverState.js

@@ -11,6 +11,7 @@ function debug(...args) {
 }
 
 // cli/server/infrastructure/fileStore.ts
+import { createHash } from "crypto";
 import {
   existsSync as existsSync2,
   promises as fs,
@@ -89,7 +90,6 @@ function getControlId(control, baseDir) {
 }
 
 // cli/server/infrastructure/fileStore.ts
-import { createHash } from "crypto";
 var FileStore = class {
   baseDir;
   controlsDir;
@@ -104,6 +104,40 @@ var FileStore = class {
       this.refreshControlsCache();
     }
   }
+  /**
+   * Update a single mapping in place, preserving file order
+   */
+  async updateMapping(oldCompositeKey, updatedMapping) {
+    const mappingFiles = this.getAllMappingFiles();
+    for (const file of mappingFiles) {
+      try {
+        const content = readFileSync2(file, "utf8");
+        let mappings = yaml2.load(content) || [];
+        let changed = false;
+        mappings = mappings.map((m) => {
+          const hash = createHash("sha256").update(JSON.stringify(m)).digest("hex");
+          if (`${m.control_id}:${hash}` === oldCompositeKey) {
+            const clean = { ...updatedMapping };
+            delete clean.hash;
+            changed = true;
+            return clean;
+          }
+          return m;
+        });
+        if (changed) {
+          const yamlContent = yaml2.dump(mappings, {
+            indent: 2,
+            lineWidth: -1,
+            noRefs: true
+          });
+          writeFileSync(file, yamlContent, "utf8");
+          return;
+        }
+      } catch (error) {
+        console.error(`Error processing mapping file ${file}:`, error);
+      }
+    }
+  }
   /**
    * Get simple filename from control ID
    */
@@ -126,8 +160,8 @@ var FileStore = class {
     if (!this.baseDir || this.baseDir === "." || this.baseDir === process.cwd()) {
       return;
     }
-    const lulaConfigPath2 = join2(this.baseDir, "lula.yaml");
-    if (!existsSync2(lulaConfigPath2)) {
+    const lulaConfigPath = join2(this.baseDir, "lula.yaml");
+    if (!existsSync2(lulaConfigPath)) {
       return;
     }
     if (!existsSync2(this.controlsDir)) {
@@ -286,10 +320,10 @@ var FileStore = class {
       return [];
     }
     let controlOrder = null;
+    const lulaConfigPath = join2(this.baseDir, "lula.yaml");
     try {
-      const lulaConfigPath2 = join2(this.baseDir, "lula.yaml");
-      if (existsSync2(lulaConfigPath2)) {
-        const content = readFileSync2(lulaConfigPath2, "utf8");
+      if (existsSync2(lulaConfigPath)) {
+        const content = readFileSync2(lulaConfigPath, "utf8");
         const metadata = yaml2.load(content);
         controlOrder = metadata?.controlOrder || null;
       }