@ryanfw/prompt-orchestration-pipeline 0.5.0 → 0.7.0

package/src/core/symlink-utils.js

@@ -0,0 +1,94 @@
+import fs from "node:fs/promises";
+import path from "node:path";
+
+/**
+ * Creates an idempotent symlink, safely handling existing files/symlinks.
+ *
+ * @param {string} linkPath - Path where the symlink should be created
+ * @param {string} targetPath - Path that the symlink should point to
+ * @param {'file' | 'dir'} type - Type of symlink to create
+ * @throws {Error} If symlink creation fails on non-POSIX systems
+ */
+export async function ensureSymlink(linkPath, targetPath, type) {
+  try {
+    // Check if linkPath already exists
+    const stats = await fs.lstat(linkPath).catch(() => null);
+
+    if (stats) {
+      if (stats.isSymbolicLink()) {
+        // If it's already a symlink pointing to the correct target, we're done
+        const existingTarget = await fs.readlink(linkPath);
+        if (existingTarget === targetPath) {
+          return;
+        }
+        // If it points to a different target, remove it
+        await fs.unlink(linkPath);
+      } else if (stats.isDirectory()) {
+        // If it's a directory, remove it recursively
+        await fs.rmdir(linkPath, { recursive: true });
+      } else {
+        // If it's a file, remove it
+        await fs.unlink(linkPath);
+      }
+    }
+
+    // Ensure parent directory exists
+    const parentDir = path.dirname(linkPath);
+    await fs.mkdir(parentDir, { recursive: true });
+
+    // Create the symlink
+    await fs.symlink(targetPath, linkPath, type);
+  } catch (error) {
+    // Re-throw with more context
+    throw new Error(
+      `Failed to create symlink from ${linkPath} -> ${targetPath}: ${error.message}`
+    );
+  }
+}
+
+/**
+ * Removes task symlinks from a completed job directory to avoid dangling links.
+ *
+ * @param {string} completedJobDir - Path to the completed job directory (e.g., COMPLETE_DIR/jobId)
+ */
+export async function cleanupTaskSymlinks(completedJobDir) {
+  const tasksDir = path.join(completedJobDir, "tasks");
+
+  try {
+    // Check if tasks directory exists
+    const tasksStats = await fs.lstat(tasksDir).catch(() => null);
+    if (!tasksStats || !tasksStats.isDirectory()) {
+      return; // No tasks directory to clean up
+    }
+
+    // Get all task directories
+    const taskDirs = await fs.readdir(tasksDir, { withFileTypes: true });
+
+    for (const taskDir of taskDirs) {
+      if (!taskDir.isDirectory()) continue;
+
+      const taskPath = path.join(tasksDir, taskDir.name);
+
+      // Remove specific symlinks if they exist and are actually symlinks
+      const symlinksToRemove = ["node_modules", "project", "_task_root"];
+
+      for (const linkName of symlinksToRemove) {
+        const linkPath = path.join(taskPath, linkName);
+
+        try {
+          const stats = await fs.lstat(linkPath);
+          if (stats.isSymbolicLink()) {
+            await fs.unlink(linkPath);
+          }
+        } catch {
+          // Ignore errors (file doesn't exist, permissions, etc.)
+        }
+      }
+    }
+  } catch (error) {
+    // Log but don't fail - cleanup is optional
+    console.warn(
+      `Warning: Failed to cleanup task symlinks in ${completedJobDir}: ${error.message}`
+    );
+  }
+}