Add post_create_script config option (#20)

Author: blipinsk

CLAUDE.md

@@ -97,6 +97,7 @@ KNOWN_ISSUES.md         # tracked issues not yet fixed
   "base_branch": "main",
   "repo_root": "/Users/you/repos/android",
   "mobs_dir": "/Users/you/repos/.codemob/android/mobs",
+  "post_create_script": "",
   "mobs": [
     {
       "name": "fix-auth-bug",

README.md

@@ -203,6 +203,30 @@ codemob purge                    # remove all
 
 No patches, no plugins, no monkey-patching. Just a shell function pretending to be `claude` and skimming a few arguments off the top.
 
+## Hooks
+
+### `post_create_script`
+
+Run a shell script automatically after every `codemob new`. The script runs with `cwd` set to the new worktree - before the agent launches.
+
+Set it in `.codemob/config.json` (absolute or relative to repo root):
+
+```json
+{
+  "post_create_script": "./scripts/mob-setup.sh"
+}
+```
+
+Example - install dependencies so the agent starts in a working environment:
+
+```sh
+#!/bin/sh
+npm install --silent
+bundle install --quiet
+```
+
+If the script exits non-zero, the mob is cleaned up and the agent is not launched.
+
 ## Under the hood
 
 【🌕】Each mob is a git worktree under `.codemob/mobs/`. Agents are launched as child processes. When you queue an action from inside an agent (via slash command), codemob detects it, terminates the agent, and launches the next session automatically.

cmd/root.go

@@ -316,6 +316,24 @@ func createMob(root string, cfg *mob.Config, name, agent string) (string, error)
 	return worktreePath, nil
 }
 
+func runPostCreateScript(cfg *mob.Config, worktreePath string) error {
+	if cfg.PostCreateScript == "" {
+		return nil
+	}
+	dim := "\033[2m"
+	reset := "\033[0m"
+	mobStatus("Running post-create script...")
+	fmt.Printf("  %s╭─%s\n", dim, reset)
+	err := mob.RunPostCreateScript(cfg, worktreePath)
+	fmt.Printf("  %s╰─%s\n", dim, reset)
+	if err != nil {
+		mobStatus("Post-create script failed")
+		return err
+	}
+	mobStatus("Post-create script completed")
+	return nil
+}
+
 // removeMob handles the full mob-removal sequence: worktree removal, branch deletion,
 // config update, save, and session file cleanup.
 func removeMob(root string, cfg *mob.Config, m *mob.Mob, force bool) error {
@@ -378,6 +396,13 @@ func cmdNew(args []string) error {
 		return err
 	}
 
+	if err := runPostCreateScript(cfg, worktreePath); err != nil {
+		if m := mob.FindMob(cfg, filepath.Base(worktreePath)); m != nil {
+			_ = removeMob(root, cfg, m, true)
+		}
+		return err
+	}
+
 	if !noLaunch {
 		return launchAgent(root, agent, worktreePath, false)
 	}
@@ -772,6 +797,12 @@ func resolveNextAction(root string, next *mob.QueuedAction) (workdir, agent stri
 		if err != nil {
 			return "", "", false, err
 		}
+		if err := runPostCreateScript(cfg, worktreePath); err != nil {
+			if m := mob.FindMob(cfg, filepath.Base(worktreePath)); m != nil {
+				_ = removeMob(root, cfg, m, true)
+			}
+			return "", "", false, err
+		}
 		return worktreePath, agent, false, nil
 
 	case "remove":
@@ -876,6 +907,7 @@ func launchAgent(root, agent, workdir string, resume bool) error {
 			}
 		}
 
+		fmt.Print("\033[2K")
 		mobStatus(fmt.Sprintf("Session ended - mob '%s'", filepath.Base(workdir)))
 
 		// Always check for queued action, regardless of how the agent exited

internal/mob/init.go

@@ -681,6 +681,7 @@ func setupRepo(reprompt bool) string {
 	fullyConfigured := cfg.RepoRoot != ""
 
 	if !isNew && !reprompt && fullyConfigured {
+		_ = SaveConfig(root, cfg)
 		info(fmt.Sprintf("Repo already initialized at %s", root))
 		return root
 	}

internal/mob/integration_test.go

@@ -172,6 +172,15 @@ func runCoreExpectError(t *testing.T, bin, dir string, args ...string) string {
 	return string(out)
 }
 
+// patchConfig reads the config, applies a mutation, and writes it back.
+func patchConfig(t *testing.T, repoPath string, mutate func(map[string]interface{})) {
+	t.Helper()
+	cfg := readConfig(t, repoPath)
+	mutate(cfg)
+	data, _ := json.MarshalIndent(cfg, "", "  ")
+	os.WriteFile(filepath.Join(repoPath, ".codemob", "config.json"), append(data, '\n'), 0644)
+}
+
 // readConfig reads and parses .codemob/config.json from the repo.
 func readConfig(t *testing.T, repoPath string) map[string]interface{} {
 	t.Helper()
@@ -205,6 +214,9 @@ func TestInit(t *testing.T) {
 	if cfg["base_branch"] != "main" {
 		t.Errorf("expected base_branch=main, got %v", cfg["base_branch"])
 	}
+	if _, ok := cfg["post_create_script"]; !ok {
+		t.Error("expected post_create_script field to be present in config")
+	}
 
 	// then -> .codemob/mobs/ dir should exist
 	if _, err := os.Stat(filepath.Join(repoPath, ".codemob", "mobs")); err != nil {
@@ -334,6 +346,30 @@ func TestInitIdempotent(t *testing.T) {
 	}
 }
 
+func TestInitAddsNewFieldsToExistingConfig(t *testing.T) {
+	bin := buildCore(t)
+	_, repoPath := setupTestRepo(t)
+	initRepo(t, bin, repoPath)
+
+	// given -> remove post_create_script from config (simulates pre-feature config)
+	patchConfig(t, repoPath, func(cfg map[string]interface{}) {
+		delete(cfg, "post_create_script")
+	})
+	cfg := readConfig(t, repoPath)
+	if _, ok := cfg["post_create_script"]; ok {
+		t.Fatal("precondition: post_create_script should have been removed")
+	}
+
+	// when -> run init again (not reinit)
+	initRepo(t, bin, repoPath)
+
+	// then -> field should be present
+	cfg = readConfig(t, repoPath)
+	if _, ok := cfg["post_create_script"]; !ok {
+		t.Error("expected init to add post_create_script to existing config")
+	}
+}
+
 func TestNewMob(t *testing.T) {
 	bin := buildCore(t)
 	_, repoPath := setupTestRepo(t)
@@ -1519,6 +1555,151 @@ func TestPurgeExternalThenListAndNew(t *testing.T) {
 	}
 }
 
+func TestPostCreateScriptRuns(t *testing.T) {
+	bin := buildCore(t)
+	_, repoPath := setupTestRepo(t)
+	initRepo(t, bin, repoPath)
+
+	// given -> a post_create_script that creates a marker file
+	scriptPath := filepath.Join(repoPath, "setup.sh")
+	os.WriteFile(scriptPath, []byte("#!/bin/sh\ntouch .setup-done\n"), 0755)
+	patchConfig(t, repoPath, func(cfg map[string]interface{}) {
+		cfg["post_create_script"] = scriptPath
+	})
+
+	// when -> create a mob
+	runCore(t, bin, repoPath, "new", "script-test", "--no-launch")
+
+	// then -> marker file should exist in the worktree
+	worktreePath := filepath.Join(repoPath, ".codemob", "mobs", "script-test")
+	if _, err := os.Stat(filepath.Join(worktreePath, ".setup-done")); err != nil {
+		t.Errorf("post_create_script did not run: marker file missing: %v", err)
+	}
+}
+
+func TestPostCreateScriptRelativePath(t *testing.T) {
+	bin := buildCore(t)
+	_, repoPath := setupTestRepo(t)
+	initRepo(t, bin, repoPath)
+
+	// given -> a post_create_script with a relative path
+	scriptPath := filepath.Join(repoPath, "setup.sh")
+	os.WriteFile(scriptPath, []byte("#!/bin/sh\ntouch .setup-done\n"), 0755)
+	patchConfig(t, repoPath, func(cfg map[string]interface{}) {
+		cfg["post_create_script"] = "setup.sh"
+	})
+
+	// when -> create a mob
+	runCore(t, bin, repoPath, "new", "rel-path-test", "--no-launch")
+
+	// then -> marker file should exist in the worktree
+	worktreePath := filepath.Join(repoPath, ".codemob", "mobs", "rel-path-test")
+	if _, err := os.Stat(filepath.Join(worktreePath, ".setup-done")); err != nil {
+		t.Errorf("post_create_script with relative path did not run: %v", err)
+	}
+}
+
+func TestPostCreateScriptFailureBlocksMob(t *testing.T) {
+	bin := buildCore(t)
+	_, repoPath := setupTestRepo(t)
+	initRepo(t, bin, repoPath)
+
+	// given -> a post_create_script that fails
+	scriptPath := filepath.Join(repoPath, "bad-setup.sh")
+	os.WriteFile(scriptPath, []byte("#!/bin/sh\nexit 1\n"), 0755)
+	patchConfig(t, repoPath, func(cfg map[string]interface{}) {
+		cfg["post_create_script"] = scriptPath
+	})
+
+	// when -> create a mob
+	out := runCoreExpectError(t, bin, repoPath, "new", "fail-test", "--no-launch")
+
+	// then -> error message should mention the script
+	if !strings.Contains(out, "post_create_script failed") {
+		t.Errorf("expected error about post_create_script, got: %s", out)
+	}
+
+	// then -> worktree should be cleaned up
+	worktreePath := filepath.Join(repoPath, ".codemob", "mobs", "fail-test")
+	if _, err := os.Stat(worktreePath); err == nil {
+		t.Error("worktree should have been removed after script failure")
+	}
+
+	// then -> mob should not exist in config
+	cfg := readConfig(t, repoPath)
+	mobs, _ := cfg["mobs"].([]interface{})
+	for _, m := range mobs {
+		entry, _ := m.(map[string]interface{})
+		if entry["name"] == "fail-test" {
+			t.Error("mob should have been removed from config after script failure")
+		}
+	}
+}
+
+func TestPostCreateScriptMissingFileError(t *testing.T) {
+	bin := buildCore(t)
+	_, repoPath := setupTestRepo(t)
+	initRepo(t, bin, repoPath)
+
+	// given -> post_create_script pointing to a nonexistent file
+	patchConfig(t, repoPath, func(cfg map[string]interface{}) {
+		cfg["post_create_script"] = "/nonexistent/setup.sh"
+	})
+
+	// when -> create a mob
+	out := runCoreExpectError(t, bin, repoPath, "new", "missing-script", "--no-launch")
+
+	// then -> error message should mention "not found"
+	if !strings.Contains(out, "not found") {
+		t.Errorf("expected 'not found' error, got: %s", out)
+	}
+}
+
+func TestPostCreateScriptNotExecutableError(t *testing.T) {
+	bin := buildCore(t)
+	_, repoPath := setupTestRepo(t)
+	initRepo(t, bin, repoPath)
+
+	// given -> a post_create_script without the executable bit
+	scriptPath := filepath.Join(repoPath, "setup.sh")
+	os.WriteFile(scriptPath, []byte("#!/bin/sh\ntouch .setup-done\n"), 0644)
+	patchConfig(t, repoPath, func(cfg map[string]interface{}) {
+		cfg["post_create_script"] = scriptPath
+	})
+
+	// when -> create a mob
+	out := runCoreExpectError(t, bin, repoPath, "new", "noexec-test", "--no-launch")
+
+	// then -> error message should mention "not executable"
+	if !strings.Contains(out, "not executable") {
+		t.Errorf("expected 'not executable' error, got: %s", out)
+	}
+}
+
+func TestPostCreateScriptEmptyIsNoop(t *testing.T) {
+	bin := buildCore(t)
+	_, repoPath := setupTestRepo(t)
+	initRepo(t, bin, repoPath)
+
+	// given -> no post_create_script configured (empty string default)
+	// when -> create a mob
+	runCore(t, bin, repoPath, "new", "no-script-test", "--no-launch")
+
+	// then -> mob should be created successfully
+	cfg := readConfig(t, repoPath)
+	mobs, _ := cfg["mobs"].([]interface{})
+	found := false
+	for _, m := range mobs {
+		mob, _ := m.(map[string]interface{})
+		if mob["name"] == "no-script-test" {
+			found = true
+		}
+	}
+	if !found {
+		t.Error("mob should have been created when post_create_script is empty")
+	}
+}
+
 func TestSlashCommandsCopiedToExternalWorktree(t *testing.T) {
 	bin := buildCore(t)
 	_, repoPath := setupTestRepo(t)