Skip to content

feat: add compare-heapsnapshots command#15

Open
aeroxy wants to merge 9 commits into
mainfrom
dev
Open

feat: add compare-heapsnapshots command#15
aeroxy wants to merge 9 commits into
mainfrom
dev

Conversation

@aeroxy

@aeroxy aeroxy commented Jul 3, 2026

Copy link
Copy Markdown
Owner

Summary by CodeRabbit

  • New Features

    • Added an offline compare-heapsnapshots command to diff two local .heapsnapshot files, with a class-diff summary by default and optional per-class output via --class-index (deterministic ordering).
  • Bug Fixes

    • Improved heapsnapshot temp-file cleanup and snapshot parsing/validation, including consistent CSV-style escaping.
    • Updated kill-daemon to accept --force, prompt interactively, and require --force when non-interactive.
    • Added a bounded timeout for Chrome connection attempts with clearer, actionable error messaging.
  • Documentation

    • Expanded agent guidance for daemon/offline failure handling and correct connection-failure retry behavior.
  • Tests

    • Added unit + end-to-end coverage for offline comparison, class aggregation/diffing, and out-of-range --class-index errors.

@coderabbitai

coderabbitai Bot commented Jul 3, 2026

Copy link
Copy Markdown

Review Change Stack

Note

Reviews paused

It looks like this branch is under active development. To avoid overwhelming you with review comments due to an influx of new commits, CodeRabbit has automatically paused this review. You can configure this behavior by changing the reviews.auto_review.auto_pause_after_reviewed_commits setting.

Use the following commands to manage reviews:

  • @coderabbitai resume to resume automatic reviews.
  • @coderabbitai review to trigger a single review.

Use the checkboxes below for quick actions:

  • ▶️ Resume reviews
  • 🔍 Trigger review

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Pro Plus

Run ID: d3692c79-6357-4b06-96f6-faa74fb18270

📥 Commits

Reviewing files that changed from the base of the PR and between b77678a and 2e24379.

📒 Files selected for processing (10)
  • AGENTS.md
  • skill/chrome-devtools/SKILL.md
  • src/cdp.rs
  • src/commands/evaluate.rs
  • src/commands/executor.rs
  • src/commands/memory.rs
  • src/commands/read_page.rs
  • src/commands/screenshot.rs
  • src/commands/third_party.rs
  • src/lib.rs

📝 Walkthrough

Walkthrough

Adds an offline compare-heapsnapshots subcommand for comparing two local heap snapshots, adds a Chrome connection timeout, changes kill-daemon handling, updates related docs, and reflows existing command and test code.

Changes

Compare Heapsnapshots Feature

Layer / File(s) Summary
Snapshot helpers and class diffing
src/commands/memory.rs
Adds shared snapshot parsing, node lookup error handling, per-class aggregation, class diff computation, text and structured rendering, and the offline compare entry point.
Compare heapsnapshot tests
src/commands/memory.rs
Adds synthetic snapshot fixtures and unit plus end-to-end tests for aggregation, diff ordering, summary/detail index mapping, and out-of-range class selection.
CLI subcommand wiring
src/lib.rs, src/commands/executor.rs
Adds the new subcommand variant, telemetry name mapping, offline execution branch, daemon request handling, and argument allowlisting for compare-heapsnapshots.

Daemon Connection and Kill Handling

Layer / File(s) Summary
Chrome connect timeout
src/cdp.rs
Adds a configurable WebSocket handshake timeout and updates the connect error returned when the timeout expires.
Kill daemon decision flow
src/lib.rs, AGENTS.md, skill/chrome-devtools/SKILL.md
Changes kill-daemon to accept --force, adds a decision helper, wires early run handling for interactive confirmation and non-interactive refusal, and documents the new behavior.

Formatting Reflows

Layer / File(s) Summary
Formatting reflows
src/commands/evaluate.rs, src/commands/read_page.rs, src/commands/screenshot.rs, src/commands/third_party.rs, src/lib.rs, src/commands/memory.rs
Reformats existing expressions, assertions, and small helper lists across several commands and tests without changing behavior.

Estimated code review effort: 4 (Complex) | ~45 minutes

Possibly related PRs

Poem

A rabbit peered at heaps so bright,
And sorted classes left and right.
It hopped through diffs with tidy care,
Then twitched its nose at timeout air.
“No endless loops,” it softly sings,
“Just compare and tidy things.” 🐇

🚥 Pre-merge checks | ✅ 5
✅ Passed checks (5 passed)
Check name Status Explanation
Description Check ✅ Passed Check skipped - CodeRabbit’s high-level summary is enabled.
Title check ✅ Passed The title clearly and concisely describes the main change: adding the new compare-heapsnapshots command.
Docstring Coverage ✅ Passed No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Linked Issues check ✅ Passed Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check ✅ Passed Check skipped because no linked issues were found for this pull request.

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

Comment @coderabbitai help to get the list of available commands.

@aeroxy

aeroxy commented Jul 3, 2026

Copy link
Copy Markdown
Owner Author

@coderabbitai full review

@coderabbitai

coderabbitai Bot commented Jul 3, 2026

Copy link
Copy Markdown
✅ Action performed

Full review finished.

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a new offline command, compare-heapsnapshots, which allows users to compare two .heapsnapshot files and view per-class differences in added/deleted nodes and size deltas without requiring a Chrome connection. It implements the necessary parsing, aggregation, diffing, and formatting logic, along with comprehensive unit tests. The feedback highlights a critical issue where sorting the diffs is non-deterministic for classes with identical size deltas due to the randomized iteration order of HashSet. This can cause the index used to query detailed class views to change across CLI invocations. To resolve this, a deterministic tie-breaker sorting by class name is recommended.

Important

The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.

Comment thread src/commands/memory.rs Outdated

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 1

🧹 Nitpick comments (1)
src/lib.rs (1)

251-256: 📐 Maintainability & Code Quality | 🔵 Trivial | 💤 Low value

Prefer auto-derived short for consistency.

Elsewhere in this enum (e.g. InspectHeapSnapshotNode, output fields) short flags are auto-derived via #[arg(long, short)] rather than explicitly pinned. Since 'b'/'c' are exactly the auto-derived values here, using the same convention would be more consistent.

♻️ Proposed consistency tweak
         /// Path to the base (earlier) .heapsnapshot file
-        #[arg(long, short = 'b')]
+        #[arg(long, short)]
         base: String,
         /// Path to the current (later) .heapsnapshot file
-        #[arg(long, short = 'c')]
+        #[arg(long, short)]
         current: String,
🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@src/lib.rs` around lines 251 - 256, The `base` and `current` fields in this
enum use explicit short flags even though they match clap’s auto-derived values.
Update the `#[arg(...)]` attributes on these fields to use the same `#[arg(long,
short)]` convention used by `InspectHeapSnapshotNode` and the `output` fields so
the enum stays consistent.
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@src/commands/memory.rs`:
- Line 399: The `diffs.sort_by` ordering in `memory.rs` is only comparing
`size_delta`, so ties can reorder differently across runs because
`all_names.into_iter()` comes from a randomized `HashSet`. Update the sorting in
the summary path to use a deterministic secondary key from the class identity
(for example the class name or another stable field from the `diff` item) so the
row order is reproducible across separate invocations and `--class-index`
matches the displayed summary index.

---

Nitpick comments:
In `@src/lib.rs`:
- Around line 251-256: The `base` and `current` fields in this enum use explicit
short flags even though they match clap’s auto-derived values. Update the
`#[arg(...)]` attributes on these fields to use the same `#[arg(long, short)]`
convention used by `InspectHeapSnapshotNode` and the `output` fields so the enum
stays consistent.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Pro Plus

Run ID: b6d31b44-cc7b-4a10-a3ce-ddb0812f72d6

📥 Commits

Reviewing files that changed from the base of the PR and between b77678a and 273e4df.

📒 Files selected for processing (3)
  • src/commands/executor.rs
  • src/commands/memory.rs
  • src/lib.rs

Comment thread src/commands/memory.rs Outdated
@aeroxy

aeroxy commented Jul 4, 2026

Copy link
Copy Markdown
Owner Author

/gemini review

@aeroxy

aeroxy commented Jul 4, 2026

Copy link
Copy Markdown
Owner Author

@coderabbitai full review

@coderabbitai

coderabbitai Bot commented Jul 4, 2026

Copy link
Copy Markdown
✅ Action performed

Full review finished.

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a new offline command, compare-heapsnapshots, which compares two .heapsnapshot files to report per-class added and deleted nodes without requiring a Chrome connection. It includes helper functions for parsing, aggregating, diffing, and formatting the snapshot data, along with comprehensive unit tests. The reviewer suggested two performance optimizations: first, avoiding redundant string cloning in build_class_aggregates by only cloning when inserting a new class name; second, optimizing diff_snapshots to eliminate an intermediate HashSet allocation and reduce map lookups by iterating over the maps directly.

Important

The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.

Comment thread src/commands/memory.rs Outdated
Comment thread src/commands/memory.rs

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

🧹 Nitpick comments (2)
src/commands/memory.rs (2)

406-415: 📐 Maintainability & Code Quality | 🔵 Trivial | ⚡ Quick win

Reuse csv_escape in format_node_details to avoid divergent escaping.

This is a verbatim copy of the escaping logic already inlined in format_node_details (Lines 226-230). Having two copies invites future drift. Route the existing call through this helper.

♻️ Proposed fix
-        let escaped_name = if name.contains(',') || name.contains('"') || name.contains('\n') || name.contains('\r') {
-            format!("\"{}\"", name.replace('"', "\"\""))
-        } else {
-            name.to_string()
-        };
+        let escaped_name = csv_escape(name);
🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@src/commands/memory.rs` around lines 406 - 415, `format_node_details`
currently duplicates the CSV escaping logic that `csv_escape` already provides,
which can lead to drift between the two paths. Update `format_node_details` to
call `csv_escape` for class/node name escaping instead of keeping its own inline
implementation, and leave `csv_escape` as the single shared helper for this
behavior.

355-397: 🎯 Functional Correctness | 🔵 Trivial | ⚡ Quick win

Detail node ordering is non-deterministic across runs.

added_ids/deleted_ids are populated by iterating cur.nodes / base_a.nodes, which are HashMaps seeded with a per-process random state. The summary sort is now stable (Line 399), but for a class with multiple added/removed nodes, the --class-index detail view rows (format_class_diff_detail) will appear in a different order on each separate invocation. Add a deterministic key so detail output is reproducible.

🔧 Proposed fix: sort node lists by id
         let added_count = added_ids.len();
         let removed_count = deleted_ids.len();
         if added_count == 0 && removed_count == 0 {
             return None;
         }
+        // Sort by id so the per-class detail view is reproducible across
+        // separate `--class-index` invocations (HashMap order is randomized).
+        let mut added: Vec<(u64, u64)> = added_ids.into_iter().zip(added_self_sizes).collect();
+        added.sort_unstable_by_key(|&(id, _)| id);
+        let (added_ids, added_self_sizes): (Vec<u64>, Vec<u64>) = added.into_iter().unzip();
+        let mut deleted: Vec<(u64, u64)> = deleted_ids.into_iter().zip(deleted_self_sizes).collect();
+        deleted.sort_unstable_by_key(|&(id, _)| id);
+        let (deleted_ids, deleted_self_sizes): (Vec<u64>, Vec<u64>) = deleted.into_iter().unzip();
         let count_delta = added_count as i64 - removed_count as i64;
🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@src/commands/memory.rs` around lines 355 - 397, Make the per-class detail
rows deterministic in `memory.rs` by ensuring `HeapSnapshotClassDiff` stores
`added_ids` and `deleted_ids` in a stable order before
`format_class_diff_detail` renders them. The current loops that populate these
vectors from `cur_agg` and `base_agg` iterate `HashMap` entries, so sort the
collected node ids (and keep `added_self_sizes` / `deleted_self_sizes` aligned
with them) using a consistent key such as the node id or a derived deterministic
tuple. This should be done in the aggregation logic that builds
`HeapSnapshotClassDiff`, not in the formatter, so `--class-index` output is
reproducible across runs.
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Nitpick comments:
In `@src/commands/memory.rs`:
- Around line 406-415: `format_node_details` currently duplicates the CSV
escaping logic that `csv_escape` already provides, which can lead to drift
between the two paths. Update `format_node_details` to call `csv_escape` for
class/node name escaping instead of keeping its own inline implementation, and
leave `csv_escape` as the single shared helper for this behavior.
- Around line 355-397: Make the per-class detail rows deterministic in
`memory.rs` by ensuring `HeapSnapshotClassDiff` stores `added_ids` and
`deleted_ids` in a stable order before `format_class_diff_detail` renders them.
The current loops that populate these vectors from `cur_agg` and `base_agg`
iterate `HashMap` entries, so sort the collected node ids (and keep
`added_self_sizes` / `deleted_self_sizes` aligned with them) using a consistent
key such as the node id or a derived deterministic tuple. This should be done in
the aggregation logic that builds `HeapSnapshotClassDiff`, not in the formatter,
so `--class-index` output is reproducible across runs.

ℹ️ Review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Pro Plus

Run ID: b406a0b9-24f9-4db0-ba3b-4a8e6cd7310c

📥 Commits

Reviewing files that changed from the base of the PR and between b77678a and 612a303.

📒 Files selected for processing (3)
  • src/commands/executor.rs
  • src/commands/memory.rs
  • src/lib.rs

- Reformatted argument lists and chained method calls for better clarity in `executor.rs`, `memory.rs`, `read_page.rs`, `screenshot.rs`, `third_party.rs`, and `lib.rs`.
- Enhanced error handling messages in `lib.rs` for clearer debugging.
- Updated tests in `read_page.rs`, `screenshot.rs`, and `lib.rs` for improved readability and consistency in assertions.
- Removed unnecessary line breaks and adjusted formatting in various files to adhere to style guidelines.
@aeroxy

aeroxy commented Jul 5, 2026

Copy link
Copy Markdown
Owner Author

/gemini review

@aeroxy

aeroxy commented Jul 5, 2026

Copy link
Copy Markdown
Owner Author

@coderabbitai full review

@coderabbitai

coderabbitai Bot commented Jul 5, 2026

Copy link
Copy Markdown
✅ Action performed

Full review finished.

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a new offline command, compare-heapsnapshots, which compares two local .heapsnapshot files and reports per-class added or deleted nodes without requiring a Chrome connection. It also includes extensive code formatting and style cleanups across several files. The review feedback highlights two important performance and memory optimizations in src/commands/memory.rs: dropping the raw HeapSnapshot ASTs immediately after aggregation to prevent potential Out-Of-Memory (OOM) errors on large snapshots, and pre-allocating vector capacities during the diffing process to minimize reallocations.

Important

The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.

Comment thread src/commands/memory.rs Outdated
Comment thread src/commands/memory.rs Outdated

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

🧹 Nitpick comments (1)
src/commands/memory.rs (1)

333-339: 📐 Maintainability & Code Quality | 🔵 Trivial | 💤 Low value

Consider the Entry API to simplify the insert-or-update branch.

The get_mut/else-insert pattern duplicates the agg.nodes.insert(id, self_size) logic across both branches. entry(...).or_insert_with(...) collapses this to one line without behavior change.

♻️ Proposed refactor
-        if let Some(agg) = aggregates.get_mut(name_ref) {
-            agg.nodes.insert(id, self_size);
-        } else {
-            let mut agg = ClassAggregate::new();
-            agg.nodes.insert(id, self_size);
-            aggregates.insert(name_ref.clone(), agg);
-        }
+        aggregates
+            .entry(name_ref.clone())
+            .or_insert_with(ClassAggregate::new)
+            .nodes
+            .insert(id, self_size);
🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@src/commands/memory.rs` around lines 333 - 339, The insert-or-update logic in
the aggregate handling duplicates the same node insertion in both branches.
Refactor the `aggregates` update in `memory.rs` to use the `Entry` API on
`name_ref` so `ClassAggregate::new()` is only created when missing and
`agg.nodes.insert(id, self_size)` happens in one place. Keep the existing
behavior unchanged while simplifying the branch around the `get_mut`/insert
path.
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Nitpick comments:
In `@src/commands/memory.rs`:
- Around line 333-339: The insert-or-update logic in the aggregate handling
duplicates the same node insertion in both branches. Refactor the `aggregates`
update in `memory.rs` to use the `Entry` API on `name_ref` so
`ClassAggregate::new()` is only created when missing and `agg.nodes.insert(id,
self_size)` happens in one place. Keep the existing behavior unchanged while
simplifying the branch around the `get_mut`/insert path.

ℹ️ Review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Pro Plus

Run ID: 36e8f73a-3cf5-4170-be03-af9aa790539c

📥 Commits

Reviewing files that changed from the base of the PR and between b77678a and 9f5098e.

📒 Files selected for processing (7)
  • src/commands/evaluate.rs
  • src/commands/executor.rs
  • src/commands/memory.rs
  • src/commands/read_page.rs
  • src/commands/screenshot.rs
  • src/commands/third_party.rs
  • src/lib.rs

* preallocate added/deleted vectors using upper-bound capacities to avoid reallocations
* compute base length once to size deletion buffers efficiently
* scope snapshot parsing to drop raw data immediately after building aggregates, lowering peak memory usage during diff
@aeroxy

aeroxy commented Jul 6, 2026

Copy link
Copy Markdown
Owner Author

/gemini review

@aeroxy

aeroxy commented Jul 6, 2026

Copy link
Copy Markdown
Owner Author

@coderabbitai full review

@coderabbitai

coderabbitai Bot commented Jul 6, 2026

Copy link
Copy Markdown
✅ Action performed

Full review finished.

@gemini-code-assist

Copy link
Copy Markdown
Contributor

Warning

Gemini encountered an error creating the review. You can try again by commenting /gemini review.

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

🧹 Nitpick comments (1)
src/commands/memory.rs (1)

316-342: 🚀 Performance & Scalability | 🔵 Trivial | ⚡ Quick win

Use entry() to avoid double hashmap lookups in the hot loop.

get_mut followed by insert performs two lookups per node; entry() does it in one. This loop runs once per node in the entire snapshot, so for large snapshots this adds up.

♻️ Proposed refactor
-        if let Some(agg) = aggregates.get_mut(name_ref) {
-            agg.nodes.insert(id, self_size);
-        } else {
-            let mut agg = ClassAggregate::new();
-            agg.nodes.insert(id, self_size);
-            aggregates.insert(name_ref.clone(), agg);
-        }
+        aggregates
+            .entry(name_ref.clone())
+            .or_insert_with(ClassAggregate::new)
+            .nodes
+            .insert(id, self_size);
🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@src/commands/memory.rs` around lines 316 - 342, The node aggregation loop in
`src/commands/memory.rs` is doing two HashMap lookups per class name via
`aggregates.get_mut()` followed by `insert`; switch this hot path to
`HashMap::entry()` in the `while current_idx + node_size <= nodes.len()` loop so
each `name_ref` is handled with a single lookup. Update the `ClassAggregate`
insertion logic to use the entry API while preserving the current
`ClassAggregate::new()` initialization and `agg.nodes.insert(id, self_size)`
behavior.
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Nitpick comments:
In `@src/commands/memory.rs`:
- Around line 316-342: The node aggregation loop in `src/commands/memory.rs` is
doing two HashMap lookups per class name via `aggregates.get_mut()` followed by
`insert`; switch this hot path to `HashMap::entry()` in the `while current_idx +
node_size <= nodes.len()` loop so each `name_ref` is handled with a single
lookup. Update the `ClassAggregate` insertion logic to use the entry API while
preserving the current `ClassAggregate::new()` initialization and
`agg.nodes.insert(id, self_size)` behavior.

ℹ️ Review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Pro Plus

Run ID: 90584542-39aa-4394-8831-6c213c013471

📥 Commits

Reviewing files that changed from the base of the PR and between b77678a and e91d0ee.

📒 Files selected for processing (7)
  • src/commands/evaluate.rs
  • src/commands/executor.rs
  • src/commands/memory.rs
  • src/commands/read_page.rs
  • src/commands/screenshot.rs
  • src/commands/third_party.rs
  • src/lib.rs

…n termination

* add connection timeout for Chrome WebSocket handshake with actionable error messaging
* document failure handling and agent guidance in AGENTS.md and SKILL.md
* introduce guarded `kill-daemon` command with `--force` flag and TTY-based confirmation/refusal logic
* prevent non-interactive agents from killing daemon without explicit force
* add unit-testable decision logic for daemon termination behavior
* update CLI argument parsing and command handling for new `kill-daemon` behavior
@aeroxy

aeroxy commented Jul 7, 2026

Copy link
Copy Markdown
Owner Author

/gemini review

@aeroxy

aeroxy commented Jul 7, 2026

Copy link
Copy Markdown
Owner Author

@coderabbitai full review

@coderabbitai

coderabbitai Bot commented Jul 7, 2026

Copy link
Copy Markdown
✅ Action performed

Full review finished.

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a new offline command compare-heapsnapshots to diff two .heapsnapshot files and report per-class added/deleted nodes, complete with text/JSON formatting and unit tests. It also implements a connection timeout for the Chrome WebSocket handshake to prevent indefinite hangs, and adds a safety guard to the kill-daemon command to refuse non-interactive execution unless --force is specified. The review feedback suggests a valuable performance optimization in the heap snapshot diffing logic to avoid allocating and sorting detailed node vectors when only the summary view is requested.

Important

The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.

Comment thread src/commands/memory.rs
Comment thread src/commands/memory.rs Outdated

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 3

🧹 Nitpick comments (1)
src/commands/memory.rs (1)

57-59: 📐 Maintainability & Code Quality | 🔵 Trivial | ⚡ Quick win

Remove the redundant “what” comment.

Line 57 only restates the next CDP call; either drop it or replace it with the reason this enable step must precede streaming. As per coding guidelines, src/**/*.rs: Rust code comments should explain "why", not "what".

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@src/commands/memory.rs` around lines 57 - 59, The comment before the
HeapProfiler enable call is redundant and only restates what
`client.send_to_target(session_id, "HeapProfiler.enable", json!({}))` already
shows. Remove that “what” comment or replace it with a brief “why” explanation
for `send_to_target` in `memory.rs`, focusing on why HeapProfiler must be
enabled before starting streaming.

Source: Coding guidelines

🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@src/commands/memory.rs`:
- Around line 652-658: The out-of-range classIndex validation in the memory
command is currently using a generic anyhow error, which will surface as an
unspecified CLI code. Update the invalid index path in the class selection logic
to return a typed CliError with ErrorCode::InvalidInput instead of anyhow!, so
the error boundary preserves the stable user-input error code; use the existing
classIndex/diffs lookup in memory command to locate the change.
- Around line 316-343: The flat node parsing in the memory snapshot aggregation
loop currently stops at the last complete record and silently ignores any
trailing partial data. Update the logic around the aggregates-building loop in
the memory command to validate that nodes.len() is evenly divisible by node_size
before iterating, and return an error for malformed snapshots instead of relying
on the current_idx + node_size <= nodes.len() check. Keep the fix localized to
the class aggregation path that uses ClassAggregate and the node offsets so
truncated arrays are rejected early.

In `@src/lib.rs`:
- Around line 874-876: The match on kill-daemon handling is partially moving cli
by destructuring cli.command, which breaks later uses of cli.command and
build_request(&cli) in the same flow. Update the Commands::KillDaemon branch to
borrow cli.command instead of taking ownership, and dereference the force field
inside the pattern so the existing cli value remains available for the later
matches and request-building logic.

---

Nitpick comments:
In `@src/commands/memory.rs`:
- Around line 57-59: The comment before the HeapProfiler enable call is
redundant and only restates what `client.send_to_target(session_id,
"HeapProfiler.enable", json!({}))` already shows. Remove that “what” comment or
replace it with a brief “why” explanation for `send_to_target` in `memory.rs`,
focusing on why HeapProfiler must be enabled before starting streaming.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Pro Plus

Run ID: e4a715b7-0eb2-443f-bd39-be280522faec

📥 Commits

Reviewing files that changed from the base of the PR and between b77678a and 2b3f4fd.

📒 Files selected for processing (10)
  • AGENTS.md
  • skill/chrome-devtools/SKILL.md
  • src/cdp.rs
  • src/commands/evaluate.rs
  • src/commands/executor.rs
  • src/commands/memory.rs
  • src/commands/read_page.rs
  • src/commands/screenshot.rs
  • src/commands/third_party.rs
  • src/lib.rs

Comment thread src/commands/memory.rs
Comment thread src/commands/memory.rs
Comment thread src/lib.rs
* enforce nodes array length is multiple of node\_size to detect truncated/malformed snapshots
* improve error message for corrupted heap snapshots
* replace anyhow error with typed CliError (InvalidInput) for invalid classIndex
* update tests to cover truncated nodes case and validate error typing
* clarify HeapProfiler.enable requirement before taking heap snapshot
@aeroxy

aeroxy commented Jul 7, 2026

Copy link
Copy Markdown
Owner Author

/gemini review

@aeroxy

aeroxy commented Jul 7, 2026

Copy link
Copy Markdown
Owner Author

@coderabbitai full review

@coderabbitai

coderabbitai Bot commented Jul 7, 2026

Copy link
Copy Markdown
✅ Action performed

Full review finished.

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces an offline compare-heapsnapshots command to diff heap snapshots, implements a connection timeout for the CDP client, and guards the kill-daemon command to prevent non-interactive callers from executing it without --force. It also updates documentation and adds comprehensive unit tests. Feedback on the changes suggests replacing is_multiple_of with the standard modulo operator to prevent MSRV issues, utilizing the more idiomatic entry API to avoid double lookups, and refactoring parallel vectors into a single vector of structs or tuples to simplify data handling.

Important

The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.

Comment thread src/commands/memory.rs
Comment thread src/commands/memory.rs
Comment thread src/commands/memory.rs

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

♻️ Duplicate comments (1)
src/lib.rs (1)

874-876: 🎯 Functional Correctness | 🔴 Critical | ⚡ Quick win

Borrow cli.command before the kill-daemon decision.

Line 874 still moves cli.command, but the function later uses &cli.command and &cli. Match by reference and dereference force.

Proposed fix
-    if let Commands::KillDaemon { force } = cli.command {
+    if let Commands::KillDaemon { force } = &cli.command {
         use std::io::IsTerminal;
-        match kill_daemon_decision(force, std::io::stdin().is_terminal()) {
+        match kill_daemon_decision(*force, std::io::stdin().is_terminal()) {
🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@src/lib.rs` around lines 874 - 876, The kill-daemon branch in lib.rs is
moving cli.command, which conflicts with later uses of &cli.command and &cli in
the same function. Update the match on Commands::KillDaemon in the
kill_daemon_decision flow to borrow cli.command by reference instead of moving
it, and dereference force when passing it onward so the rest of the code can
still use cli and cli.command safely.
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Duplicate comments:
In `@src/lib.rs`:
- Around line 874-876: The kill-daemon branch in lib.rs is moving cli.command,
which conflicts with later uses of &cli.command and &cli in the same function.
Update the match on Commands::KillDaemon in the kill_daemon_decision flow to
borrow cli.command by reference instead of moving it, and dereference force when
passing it onward so the rest of the code can still use cli and cli.command
safely.

ℹ️ Review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Pro Plus

Run ID: 2603a71f-dd25-4b8f-9581-6a366f5ccecf

📥 Commits

Reviewing files that changed from the base of the PR and between b77678a and 15b1efb.

📒 Files selected for processing (10)
  • AGENTS.md
  • skill/chrome-devtools/SKILL.md
  • src/cdp.rs
  • src/commands/evaluate.rs
  • src/commands/executor.rs
  • src/commands/memory.rs
  • src/commands/read_page.rs
  • src/commands/screenshot.rs
  • src/commands/third_party.rs
  • src/lib.rs

@aeroxy

aeroxy commented Jul 7, 2026

Copy link
Copy Markdown
Owner Author

/gemini review

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces an offline compare-heapsnapshots command to diff heap snapshots, adds a timeout to Chrome WebSocket connections to prevent indefinite hangs, and implements a safety guard on kill-daemon to prevent automated agents from killing active connections without --force. Feedback on these changes includes optimizing the heap snapshot parser by avoiding redundant string allocations during node iteration, replacing is_multiple_of with the standard modulo operator % to maintain backward compatibility with older Rust versions, and caching the connection timeout duration in a local variable to avoid duplicate environment variable reads.

Important

The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.

Comment thread src/commands/memory.rs Outdated
Comment thread src/commands/memory.rs
Comment thread src/cdp.rs Outdated
@aeroxy

aeroxy commented Jul 7, 2026

Copy link
Copy Markdown
Owner Author

/gemini review

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a new offline command compare-heapsnapshots to diff two .heapsnapshot files, along with corresponding documentation and tests. It also adds a connection timeout to the Chrome WebSocket handshake to prevent indefinite hangs, and guards the kill-daemon command to prevent automated agents from executing it non-interactively without the --force flag. The review feedback suggests optimizing the diff_snapshots function by passing the base and current aggregates by value rather than by reference, which avoids unnecessary string cloning and simplifies the lookup logic.

Important

The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.

Comment thread src/commands/memory.rs
Comment thread src/commands/memory.rs Outdated
Comment thread src/commands/memory.rs Outdated
Comment thread src/commands/memory.rs Outdated
Comment thread src/commands/memory.rs Outdated
Comment thread src/commands/memory.rs Outdated
…ment passing and reducing unnecessary lookups
@aeroxy

aeroxy commented Jul 7, 2026

Copy link
Copy Markdown
Owner Author

@coderabbitai full review

@coderabbitai

coderabbitai Bot commented Jul 7, 2026

Copy link
Copy Markdown
✅ Action performed

Full review finished.

@aeroxy

aeroxy commented Jul 7, 2026

Copy link
Copy Markdown
Owner Author

/gemini review

@gemini-code-assist

Copy link
Copy Markdown
Contributor

Warning

Gemini encountered an error creating the review. You can try again by commenting /gemini review.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant