From 44263347be427e16ae6f5e11f06f084a56945f25 Mon Sep 17 00:00:00 2001
From: Neil Mitchell <ndmitchell@meta.com>
Date: Tue, 28 May 2024 07:05:14 -0700
Subject: [PATCH] Add Changes::filter_by_project_path

Summary: Useful for future work.

Reviewed By: aniketmathur

Differential Revision: D57542273

fbshipit-source-id: 658a4e03846a2bdf3e4a419d40daa725d7b0cd72
---
 btd/src/changes.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/btd/src/changes.rs b/btd/src/changes.rs
index 0a9b3b6..04de66f 100644
--- a/btd/src/changes.rs
+++ b/btd/src/changes.rs
@@ -75,13 +75,17 @@ impl Changes {
         self.contains_cell_path(&package.as_cell_path())
     }
 
-    pub fn filter_by_extension(&self, f: impl Fn(Option<&str>) -> bool) -> Changes {
+    pub fn filter_by_cell_path(&self, f: impl Fn(&CellPath) -> bool) -> Changes {
         let paths = self
             .paths
             .iter()
-            .filter(|x| f(x.get().0.extension()))
+            .filter(|x| f(&x.get().0))
             .cloned()
             .collect();
         Self::from_paths(paths)
     }
+
+    pub fn filter_by_extension(&self, f: impl Fn(Option<&str>) -> bool) -> Changes {
+        self.filter_by_cell_path(|x| f(x.extension()))
+    }
 }