@desplega.ai/wts 0.1.2 → 0.1.3

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.3",
   description: "Git worktree manager with tmux integration",
   type: "module",
   bin: {
@@ -3408,6 +3408,100 @@ 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
+# Working directory: the new worktree
+# Environment: WTS_WORKTREE_PATH contains the worktree path
+
+set -e
+
+echo "Setting up worktree..."
+
+# Install dependencies
+# npm install
+# bun install
+
+# Copy environment files
+# cp .env.example .env
+
+# Run any other setup tasks
+# ...
+
+echo "Setup complete!"
+`;
+var TS_TEMPLATE = `#!/usr/bin/env bun
+// wts setup script - runs after worktree creation
+// Working directory: the new worktree
+// Environment: WTS_WORKTREE_PATH contains the worktree path
+
+const worktreePath = process.env.WTS_WORKTREE_PATH;
+
+console.log("Setting up worktree...");
+
+// Install dependencies
+// await Bun.$\`npm install\`;
+// await Bun.$\`bun install\`;
+
+// Copy environment files
+// await Bun.$\`cp .env.example .env\`;
+
+// Run any other setup tasks
+// ...
+
+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 +3579,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.3",
   "description": "Git worktree manager with tmux integration",
   "type": "module",
   "bin": {