@ebowwa/glm-daemon 0.1.0

package/src/worktree.ts

@@ -0,0 +1,122 @@
+/**
+ * GLM Daemon - Worktree Manager
+ *
+ * Handles git worktree isolation for parallel task execution.
+ */
+
+import { exec } from "child_process";
+import { promisify } from "util";
+import { join } from "path";
+import type { GLMDaemonConfig } from "./types.js";
+
+const execAsync = promisify(exec);
+
+export class WorktreeManager {
+  private config: GLMDaemonConfig;
+  private worktreePath: string | null = null;
+
+  constructor(config: GLMDaemonConfig) {
+    this.config = config;
+  }
+
+  /**
+   * Create a new worktree for isolation
+   */
+  async createWorktree(): Promise<string> {
+    if (!this.config.enableWorktree) {
+      return this.config.cwd;
+    }
+
+    const timestamp = Date.now();
+    const worktreeName = `glm-${timestamp}`;
+    this.worktreePath = join(this.config.cwd, "worktrees", worktreeName);
+
+    // Create worktrees directory
+    await execAsync(`mkdir -p ${join(this.config.cwd, "worktrees")}`);
+
+    // Create worktree
+    await execAsync(
+      `git worktree add -b ${worktreeName} ${this.worktreePath}`
+    );
+
+    return this.worktreePath;
+  }
+
+  /**
+   * Create a feature branch in the worktree
+   */
+  async createBranch(): Promise<string> {
+    if (!this.worktreePath) {
+      throw new Error("No worktree created");
+    }
+
+    const branchName = `feat/glm-${Date.now()}`;
+    await execAsync(`cd ${this.worktreePath} && git checkout -b ${branchName}`);
+
+    return branchName;
+  }
+
+  /**
+   * Commit changes in the worktree
+   */
+  async commitChanges(options: {
+    message: string;
+    files: string[];
+  }): Promise<string> {
+    const cwd = this.worktreePath || this.config.cwd;
+
+    // Stage files
+    for (const file of options.files) {
+      await execAsync(`cd ${cwd} && git add ${file}`);
+    }
+
+    // Commit
+    const { stdout } = await execAsync(
+      `cd ${cwd} && git commit -m "${options.message}"`
+    );
+
+    return stdout.trim();
+  }
+
+  /**
+   * Create a pull request
+   */
+  async createPullRequest(options: {
+    title: string;
+    body: string;
+  }): Promise<void> {
+    const cwd = this.worktreePath || this.config.cwd;
+
+    // Push branch
+    await execAsync(`cd ${cwd} && git push -u origin HEAD`);
+
+    // Create PR using gh CLI
+    await execAsync(
+      `cd ${cwd} && gh pr create --title "${options.title}" --body "${options.body}"`
+    );
+  }
+
+  /**
+   * Clean up worktree
+   */
+  async cleanup(): Promise<void> {
+    if (!this.worktreePath) {
+      return;
+    }
+
+    // Remove worktree
+    await execAsync(`git worktree remove ${this.worktreePath}`);
+
+    // Delete directory
+    await execAsync(`rm -rf ${this.worktreePath}`);
+
+    this.worktreePath = null;
+  }
+
+  /**
+   * Get worktree path
+   */
+  getPath(): string | null {
+    return this.worktreePath;
+  }
+}