Fix compressed rollout search path matching

codex-rs/rollout/src/compression_tests.rs

@@ -106,14 +106,13 @@ async fn append_rollout_item_materializes_compressed_rollout() -> anyhow::Result
 }
 
 #[tokio::test]
-async fn search_rollout_matches_returns_compressed_snippet() -> anyhow::Result<()> {
+async fn search_rollout_matches_uses_logical_path_for_compressed_rollout() -> anyhow::Result<()> {
     let home = TempDir::new()?;
     let uuid = Uuid::from_u128(15);
     let thread_id = ThreadId::from_string(&uuid.to_string())?;
     let rollout_path = rollout_path(home.path(), "2025-01-03T12-00-00", uuid);
     write_rollout(&rollout_path, thread_id, "targeted search term")?;
     compress_now(&rollout_path)?;
-    let compressed_path = compressed_rollout_path(&rollout_path);
 
     let matches = search_rollout_matches(
         std::path::Path::new("missing-rg-for-test"),
@@ -124,7 +123,7 @@ async fn search_rollout_matches_returns_compressed_snippet() -> anyhow::Result<(
     .await?;
 
     assert_eq!(
-        matches.get(compressed_path.as_path()),
+        matches.get(rollout_path.as_path()),
         Some(&Some("targeted search term".to_string()))
     );
     Ok(())

codex-rs/rollout/src/search.rs

@@ -21,6 +21,7 @@ use super::compression;
 const MATCH_CONTEXT_BEFORE_CHARS: usize = 48;
 const MATCH_CONTEXT_AFTER_CHARS: usize = 96;
 
+/// Search matches keyed by the canonical `.jsonl` path for each rollout.
 pub type RolloutSearchMatches = HashMap<PathBuf, Option<String>>;
 
 pub async fn search_rollout_paths(
@@ -145,7 +146,10 @@ async fn scan_rollout_matches(
                 if let Some(snippet) =
                     first_rollout_content_match_snippet(rollout_file.path(), search_term).await?
                 {
-                    matches.insert(rollout_file.into_path(), Some(snippet));
+                    matches.insert(
+                        compression::plain_rollout_path(rollout_file.path()),
+                        Some(snippet),
+                    );
                 }
                 continue;
             }
@@ -217,7 +221,10 @@ async fn scan_compressed_rollout_matches(
             if let Some(snippet) =
                 first_rollout_content_match_snippet(rollout_file.path(), search_term).await?
             {
-                matches.insert(rollout_file.into_path(), Some(snippet));
+                matches.insert(
+                    compression::plain_rollout_path(rollout_file.path()),
+                    Some(snippet),
+                );
             }
         }
     }

codex-rs/thread-store/src/local/search_threads.rs

@@ -109,7 +109,8 @@ pub(super) async fn search_threads(
         )
         .await?;
         for item in page.items {
-            let Some(snippet) = (match remaining_rollouts.remove(item.path.as_path()) {
+            let logical_path = codex_rollout::plain_rollout_path(item.path.as_path());
+            let Some(snippet) = (match remaining_rollouts.remove(logical_path.as_path()) {
                 Some(Some(snippet)) => Some(snippet),
                 Some(None) => first_rollout_content_match_snippet(item.path.as_path(), search_term)
                     .await