Codex mid-turn delivery deletes input before the steer is accepted

internal/store/inbox.go

@@ -274,6 +274,18 @@ func (s *Store) MarkDelivered(ctx context.Context, responseIDs []string) error {
 	})
 }
 
+func (s *Store) MarkUndelivered(ctx context.Context, responseIDs []string) error {
+	return s.write(ctx, func(tx *sql.Tx, emit emitFunc) error {
+		for _, id := range responseIDs {
+			_, err := tx.ExecContext(ctx, `UPDATE inbox_responses SET delivered_at = NULL WHERE id = ?`, id)
+			if err != nil {
+				return err
+			}
+		}
+		return nil
+	})
+}
+
 type querier interface {
 	QueryRowContext(ctx context.Context, query string, args ...any) *sql.Row
 	QueryContext(ctx context.Context, query string, args ...any) (*sql.Rows, error)

internal/store/lead.go

@@ -70,6 +70,23 @@ func (s *Store) LeadInputs(ctx context.Context, owner string) ([]LeadInput, erro
 	return inputs, rows.Err()
 }
 
+func (s *Store) RestoreLeadInputs(ctx context.Context, inputs []LeadInput) error {
+	return s.write(ctx, func(tx *sql.Tx, emit emitFunc) error {
+		for _, in := range inputs {
+			images, err := json.Marshal(append([]string{}, in.Images...))
+			if err != nil {
+				return err
+			}
+			_, err = tx.ExecContext(ctx, `INSERT INTO lead_inputs (id, owner, kind, actor, text, images, created_at) VALUES (?, ?, ?, ?, ?, ?, ?)`,
+				in.ID, in.Owner, string(in.Kind), in.Actor, in.Text, string(images), formatTime(in.CreatedAt))
+			if err != nil {
+				return err
+			}
+		}
+		return nil
+	})
+}
+
 func (s *Store) DeleteLeadInputs(ctx context.Context, ids []int64) error {
 	return s.write(ctx, func(tx *sql.Tx, emit emitFunc) error {
 		for _, id := range ids {

internal/store/taskinputs.go

@@ -57,6 +57,23 @@ func (s *Store) TaskInputs(ctx context.Context, taskID string) ([]TaskInput, err
 	return inputs, rows.Err()
 }
 
+func (s *Store) RestoreTaskInputs(ctx context.Context, inputs []TaskInput) error {
+	return s.write(ctx, func(tx *sql.Tx, emit emitFunc) error {
+		for _, in := range inputs {
+			images, err := json.Marshal(append([]string{}, in.Images...))
+			if err != nil {
+				return err
+			}
+			_, err = tx.ExecContext(ctx, `INSERT INTO task_inputs (id, task_id, actor, text, images, created_at) VALUES (?, ?, ?, ?, ?, ?)`,
+				in.ID, in.TaskID, in.Actor, in.Text, string(images), formatTime(in.CreatedAt))
+			if err != nil {
+				return err
+			}
+		}
+		return nil
+	})
+}
+
 func (s *Store) DeleteTaskInputs(ctx context.Context, ids []int64) error {
 	return s.write(ctx, func(tx *sql.Tx, emit emitFunc) error {
 		for _, id := range ids {