@desplega.ai/wts 0.1.2 → 0.1.4

package/dist/index.js

@@ -1877,7 +1877,7 @@ var {
 // package.json
 var package_default = {
   name: "@desplega.ai/wts",
-  version: "0.1.2",
+  version: "0.1.4",
   description: "Git worktree manager with tmux integration",
   type: "module",
   bin: {
@@ -3046,14 +3046,19 @@ async function runSetupScript(worktreePath, scriptPath, gitRoot) {
   }
   const ext = scriptPath.split(".").pop()?.toLowerCase();
   try {
+    const env2 = {
+      ...process.env,
+      WTS_WORKTREE_PATH: worktreePath,
+      WTS_GIT_ROOT: gitRoot
+    };
     if (ext === "ts") {
-      await Bun.$`bun ${fullScriptPath}`.cwd(worktreePath).env({ ...process.env, WTS_WORKTREE_PATH: worktreePath });
+      await Bun.$`bun ${fullScriptPath}`.cwd(worktreePath).env(env2);
     } else if (ext === "sh") {
-      await Bun.$`bash ${fullScriptPath}`.cwd(worktreePath).env({ ...process.env, WTS_WORKTREE_PATH: worktreePath });
+      await Bun.$`bash ${fullScriptPath}`.cwd(worktreePath).env(env2);
     } else if (ext === "js" || ext === "mjs") {
-      await Bun.$`bun ${fullScriptPath}`.cwd(worktreePath).env({ ...process.env, WTS_WORKTREE_PATH: worktreePath });
+      await Bun.$`bun ${fullScriptPath}`.cwd(worktreePath).env(env2);
     } else {
-      await Bun.$`${fullScriptPath}`.cwd(worktreePath).env({ ...process.env, WTS_WORKTREE_PATH: worktreePath });
+      await Bun.$`${fullScriptPath}`.cwd(worktreePath).env(env2);
     }
     console.log(source_default.green("Setup script completed"));
   } catch (error) {
@@ -3408,6 +3413,104 @@ var prCommand = new Command2("pr").description("Create a pull request from a wor
   }
 });
 
+// src/commands/setup.ts
+import { access as access2, chmod } from "node:fs/promises";
+import { join as join5 } from "node:path";
+var SHELL_TEMPLATE = `#!/bin/bash
+# wts setup script - runs after worktree creation
+#
+# Environment variables:
+#   WTS_WORKTREE_PATH - path to the new worktree (also the working directory)
+#   WTS_GIT_ROOT      - path to the main repository root
+#
+# Example: copy .env from main repo to worktree
+#   cp "$WTS_GIT_ROOT/.env" "$WTS_WORKTREE_PATH/.env"
+
+set -e
+
+echo "Setting up worktree..."
+
+# Install dependencies
+# npm install
+# bun install
+
+# Copy files from main repo
+# cp "$WTS_GIT_ROOT/.env" .env
+# cp -r "$WTS_GIT_ROOT/node_modules" .
+
+echo "Setup complete!"
+`;
+var TS_TEMPLATE = `#!/usr/bin/env bun
+// wts setup script - runs after worktree creation
+//
+// Environment variables:
+//   WTS_WORKTREE_PATH - path to the new worktree (also the working directory)
+//   WTS_GIT_ROOT      - path to the main repository root
+
+const worktreePath = process.env.WTS_WORKTREE_PATH!;
+const gitRoot = process.env.WTS_GIT_ROOT!;
+
+console.log("Setting up worktree...");
+
+// Install dependencies
+// await Bun.$\`npm install\`;
+// await Bun.$\`bun install\`;
+
+// Copy files from main repo
+// await Bun.$\`cp \${gitRoot}/.env .env\`;
+// await Bun.$\`cp -r \${gitRoot}/node_modules .\`;
+
+console.log("Setup complete!");
+`;
+var setupCommand = new Command2("setup").description("Generate a setup script template for worktree initialization").option("--sh", "Generate a shell script (.wts-setup.sh)").option("--ts", "Generate a TypeScript script (.wts-setup.ts)").option("--no-config", "Don't update .wts-config.json").action(async (options) => {
+  const gitRoot = await getGitRoot();
+  if (!gitRoot) {
+    console.error(source_default.red("Error: Not in a git repository"));
+    process.exit(1);
+  }
+  let scriptType;
+  if (options.sh && options.ts) {
+    console.error(source_default.red("Error: Cannot specify both --sh and --ts"));
+    process.exit(1);
+  } else if (options.sh) {
+    scriptType = "sh";
+  } else if (options.ts) {
+    scriptType = "ts";
+  } else {
+    scriptType = await select("Which type of setup script would you like?", [
+      { value: "sh", label: "Shell script (.wts-setup.sh)" },
+      { value: "ts", label: "TypeScript (.wts-setup.ts) - requires bun" }
+    ], "sh");
+  }
+  const filename = `.wts-setup.${scriptType}`;
+  const filepath = join5(gitRoot, filename);
+  try {
+    await access2(filepath);
+    const overwrite = await confirm(`${filename} already exists. Overwrite?`, false);
+    if (!overwrite) {
+      console.log(source_default.dim("Cancelled"));
+      return;
+    }
+  } catch {}
+  const template = scriptType === "sh" ? SHELL_TEMPLATE : TS_TEMPLATE;
+  await Bun.write(filepath, template);
+  if (scriptType === "sh") {
+    await chmod(filepath, 493);
+  }
+  console.log(source_default.green(`Created ${filename}`));
+  if (options.config !== false) {
+    const localConfig = await loadLocalConfig(gitRoot) ?? {};
+    localConfig.setupScript = filename;
+    await saveLocalConfig(gitRoot, localConfig);
+    console.log(source_default.dim(`Updated .wts-config.json with setupScript: "${filename}"`));
+  }
+  console.log(source_default.dim(`
+Next steps:`));
+  console.log(source_default.dim(`  1. Edit ${filename} to add your setup commands`));
+  console.log(source_default.dim(`  2. Run 'wts create <alias>' - setup script will run automatically`));
+  console.log(source_default.dim(`  3. Use '--no-setup' flag to skip the script when needed`));
+});
+
 // src/commands/switch.ts
 var switchCommand = new Command2("switch").description("Switch to a worktree (interactive or by alias)").argument("[alias]", "Worktree alias to switch to").option("--tmux", "Open in a new tmux window").option("--claude", "Launch Claude Code in the tmux window").action(async (alias, options) => {
   const gitRoot = await getGitRoot();
@@ -3485,5 +3588,6 @@ program3.addCommand(deleteCommand);
 program3.addCommand(cdCommand);
 program3.addCommand(switchCommand);
 program3.addCommand(prCommand);
+program3.addCommand(setupCommand);
 program3.addCommand(cleanupCommand);
 program3.parse();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@desplega.ai/wts",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "Git worktree manager with tmux integration",
   "type": "module",
   "bin": {