feat(pixi_api): Implement package search (#4910)

Author: haecker-felix

Cargo.lock

@@ -203,9 +203,9 @@ async fn test_search_multiple_versions() {
         .replace("\x1b[1m", "")
         .replace("\x1b[2m", "");
 
-    assert_eq!(result.len(), 1);
-    assert_eq!(result[0].package_record.version.as_str(), "0.2.0");
-    assert_eq!(result[0].package_record.build, "h60d57d3_1");
+    let latest_package = result.last().expect("should have at least one result");
+    assert_eq!(latest_package.package_record.version.as_str(), "0.2.0");
+    assert_eq!(latest_package.package_record.build, "h60d57d3_1");
     let output = output
         .lines()
         // Filter out URL line since temporary directory name is random.

crates/pixi/tests/integration_rust/search_tests.rs

@@ -8,6 +8,11 @@ readme.workspace = true
 repository.workspace = true
 version = "0.1.0"
 
+[features]
+default = []
+native-tls = ["rattler_repodata_gateway/native-tls"]
+rustls-tls = ["rattler_repodata_gateway/rustls-tls"]
+
 [dependencies]
 console = { workspace = true }
 dunce = { workspace = true }
@@ -25,8 +30,15 @@ pixi_pypi_spec = { workspace = true }
 pixi_spec = { workspace = true }
 pixi_utils = { workspace = true }
 rattler_conda_types = { workspace = true }
+rattler_lock = { workspace = true }
+rattler_repodata_gateway = { workspace = true, features = [
+  "sparse",
+  "gateway",
+] }
+regex = { workspace = true }
 same-file = { workspace = true }
 serde = { workspace = true, features = ["derive"] }
+strsim = { workspace = true }
 tempfile = { workspace = true }
 tokio = { workspace = true, features = ["fs"] }
 tracing = { workspace = true }

crates/pixi_api/Cargo.toml

@@ -9,11 +9,49 @@ use pixi_manifest::{
 };
 use pixi_pypi_spec::{PixiPypiSpec, PypiPackageName};
 use pixi_spec::PixiSpec;
-use rattler_conda_types::{PackageName, Platform};
+use rattler_conda_types::{Channel, MatchSpec, PackageName, Platform, RepoDataRecord};
 
 use crate::interface::Interface;
 use crate::workspace::{InitOptions, ReinstallOptions};
 
+pub struct DefaultContext<I: Interface> {
+    interface: I,
+}
+
+impl<I: Interface> DefaultContext<I> {
+    pub fn new(interface: I) -> Self {
+        Self { interface }
+    }
+
+    /// Returns all matching package versions sorted by version
+    pub async fn search_exact(
+        &self,
+        match_spec: MatchSpec,
+        channels: IndexSet<Channel>,
+        platform: Platform,
+    ) -> miette::Result<Option<Vec<RepoDataRecord>>> {
+        crate::workspace::search::search_exact(
+            &self.interface,
+            None,
+            match_spec,
+            channels,
+            platform,
+        )
+        .await
+    }
+
+    /// Returns all matching packages with their latest versions
+    pub async fn search_wildcard(
+        &self,
+        search: &str,
+        channels: IndexSet<Channel>,
+        platform: Platform,
+    ) -> miette::Result<Option<Vec<RepoDataRecord>>> {
+        crate::workspace::search::search_wildcard(&self.interface, None, search, channels, platform)
+            .await
+    }
+}
+
 pub struct WorkspaceContext<I: Interface> {
     interface: I,
     workspace: Workspace,
@@ -172,4 +210,38 @@ impl<I: Interface> WorkspaceContext<I> {
         )
         .await
     }
+
+    /// Returns all matching package versions sorted by version
+    pub async fn search_exact(
+        &self,
+        match_spec: MatchSpec,
+        channels: IndexSet<Channel>,
+        platform: Platform,
+    ) -> miette::Result<Option<Vec<RepoDataRecord>>> {
+        crate::workspace::search::search_exact(
+            &self.interface,
+            Some(&self.workspace),
+            match_spec,
+            channels,
+            platform,
+        )
+        .await
+    }
+
+    /// Returns all matching packages with their latest versions
+    pub async fn search_wildcard(
+        &self,
+        search: &str,
+        channels: IndexSet<Channel>,
+        platform: Platform,
+    ) -> miette::Result<Option<Vec<RepoDataRecord>>> {
+        crate::workspace::search::search_wildcard(
+            &self.interface,
+            Some(&self.workspace),
+            search,
+            channels,
+            platform,
+        )
+        .await
+    }
 }

crates/pixi_api/src/context.rs

@@ -1,7 +1,7 @@
 pub mod workspace;
 
 mod context;
-pub use context::WorkspaceContext;
+pub use context::{DefaultContext, WorkspaceContext};
 
 mod interface;
 pub use interface::Interface;

crates/pixi_api/src/lib.rs

@@ -1,9 +1,13 @@
+//! This module contains everything which is related to a Pixi workspace.
+
 pub(crate) mod init;
 pub use init::{GitAttributes, InitOptions, ManifestFormat};
 
 pub(crate) mod reinstall;
 pub use reinstall::ReinstallOptions;
 
+pub(crate) mod search;
+
 pub(crate) mod task;
 
 #[allow(clippy::module_inception)]