fix ask add alias output for 1a1731dc-1f11-42bd-be12-4c1af7f7e673

3 files changed, 45 insertions, 5 deletions

diff --git a/internal/askcli/command_info_add.go b/internal/askcli/command_info_add.go
index 5332e71..4fa0941 100644
--- a/internal/askcli/command_info_add.go
+++ b/internal/askcli/command_info_add.go
@@ -104,7 +104,12 @@ func (d Dispatcher) handleAdd(ctx context.Context, args []string, stdout, stderr
 		io.WriteString(stderr, "error: could not parse UUID from task creation output\n")
 		return 1, nil
 	}
-	io.WriteString(stdout, uuid+"\n")
+	aliases, err := ensureTaskAliasesForUUIDs([]string{uuid})
+	if err != nil {
+		fmt.Fprintf(stderr, "error: failed to assign task alias: %v\n", err)
+		return 1, nil
+	}
+	io.WriteString(stdout, displayTaskAlias(uuid, aliases)+"\n")
 	return 0, nil
 }
 
diff --git a/internal/askcli/command_info_add_test.go b/internal/askcli/command_info_add_test.go
index bd95de4..11bc451 100644
--- a/internal/askcli/command_info_add_test.go
+++ b/internal/askcli/command_info_add_test.go
@@ -211,7 +211,14 @@ func TestHandleInfo_MissingUUID_MultipleStartedTasks(t *testing.T) {
 }
 
 func TestHandleAdd_Success(t *testing.T) {
-	// With rc.verbose=new-uuid, task add outputs "Created task <uuid>." directly.
+	now := useIsolatedTaskAliasCache(t)
+	writeTaskAliasCacheForTest(t, taskAliasCache{
+		NextID: 1,
+		Entries: []taskAliasCacheEntry{
+			{UUID: "existing-uuid", Alias: "0", CreatedAt: now},
+		},
+	})
+
 	d := NewDispatcher(&spyRunner{runFn: func(ctx context.Context, args []string, stdin io.Reader, stdout, stderr io.Writer) (int, error) {
 		io.WriteString(stdout, "Created task abc-123-def.")
 		return 0, nil
@@ -221,8 +228,36 @@ func TestHandleAdd_Success(t *testing.T) {
 	if code != 0 {
 		t.Fatalf("add code = %d, want 0", code)
 	}
-	if !strings.Contains(stdout.String(), "abc-123-def") {
-		t.Fatalf("output missing UUID: %s", stdout.String())
+	if got := strings.TrimSpace(stdout.String()); got != "1" {
+		t.Fatalf("stdout = %q, want alias 1", stdout.String())
+	}
+	cache := readTaskAliasCacheSnapshot(t)
+	entry := findTaskAliasEntry(t, cache, "abc-123-def")
+	if entry.Alias != "1" {
+		t.Fatalf("created task alias = %q, want 1", entry.Alias)
+	}
+}
+
+func TestHandleAdd_AliasAssignmentFailure(t *testing.T) {
+	oldRoot := taskAliasCacheRoot
+	taskAliasCacheRoot = func() (string, error) { return "", io.ErrUnexpectedEOF }
+	defer func() { taskAliasCacheRoot = oldRoot }()
+
+	d := NewDispatcher(&spyRunner{runFn: func(ctx context.Context, args []string, stdin io.Reader, stdout, stderr io.Writer) (int, error) {
+		io.WriteString(stdout, "Created task abc-123-def.")
+		return 0, nil
+	}})
+
+	var stdout, stderr bytes.Buffer
+	code, _ := d.Dispatch(context.Background(), []string{"add", "New task description"}, nil, &stdout, &stderr)
+	if code != 1 {
+		t.Fatalf("add code = %d, want 1", code)
+	}
+	if stdout.Len() != 0 {
+		t.Fatalf("stdout = %q, want empty output on alias assignment failure", stdout.String())
+	}
+	if !strings.Contains(stderr.String(), "failed to assign task alias") {
+		t.Fatalf("stderr = %q, want alias assignment failure", stderr.String())
 	}
 }
 
diff --git a/internal/askcli/dispatch.go b/internal/askcli/dispatch.go
index 7cb217b..12407e3 100644
--- a/internal/askcli/dispatch.go
+++ b/internal/askcli/dispatch.go
@@ -88,7 +88,7 @@ func (d Dispatcher) Dispatch(ctx context.Context, args []string, stdin io.Reader
 func (d Dispatcher) help(w io.Writer) (int, error) {
 	io.WriteString(w, "ask - task management CLI\n")
 	io.WriteString(w, "\nSubcommands:\n")
-	io.WriteString(w, "  ask add \"description\"          Create a new task\n")
+	io.WriteString(w, "  ask add \"description\"          Create a new task and print its ID\n")
 	io.WriteString(w, "  ask list [filters]           List active tasks (default)\n")
 	io.WriteString(w, "  ask ready                   List READY tasks (not blocked)\n")
 	io.WriteString(w, "  ask all [filters]            List all tasks including completed/deleted\n")