uv-workspace: wip for removing package name

Author: BurntSushi

crates/uv-workspace/src/pyproject.rs

@@ -25,7 +25,8 @@ use uv_normalize::{ExtraName, GroupName, PackageName};
 use uv_pep440::{Version, VersionSpecifiers};
 use uv_pep508::MarkerTree;
 use uv_pypi_types::{
-    ConflictingGroupList, RequirementSource, SupportedEnvironments, VerbatimParsedUrl,
+    ConflictingGroup, ConflictingGroupList, ConflictingGroups, RequirementSource,
+    SupportedEnvironments, VerbatimParsedUrl,
 };
 
 #[derive(Error, Debug)]
@@ -483,6 +484,88 @@ pub struct ToolUv {
     pub conflicting_groups: Option<ConflictingGroupList>,
 }
 
+#[derive(Debug, Default, Clone, Eq, PartialEq, serde::Deserialize, serde::Serialize)]
+#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
+pub struct ToolUvConflictingList(Vec<ToolUvConflictingSet>);
+
+impl ToolUvConflictingList {
+    pub fn to_conflicting(&self, package: &PackageName) -> ConflictingGroupList {
+        let mut conflicting = ConflictingGroupList::empty();
+        for tool_uv_set in &self.0 {
+            let mut set = vec![];
+            for item in &tool_uv_set.0 {
+                set.push(ConflictingGroup::from((
+                    package.clone(),
+                    item.extra.clone(),
+                )));
+            }
+            // OK because we guarantee that `ToolUvConflictingList` is
+            // valid and there aren't any new errors that can occur
+            // here.
+            let set = ConflictingGroups::try_from(set).unwrap();
+            conflicting.push(set);
+        }
+        conflicting
+    }
+}
+
+#[derive(Debug, Default, Clone, Eq, PartialEq, serde::Serialize)]
+#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
+pub struct ToolUvConflictingSet(Vec<ToolUvConflictingItem>);
+
+#[derive(
+    Debug,
+    Default,
+    Clone,
+    Eq,
+    Hash,
+    PartialEq,
+    PartialOrd,
+    Ord,
+    serde::Deserialize,
+    serde::Serialize,
+)]
+#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
+pub struct ToolUvConflictingItem {
+    extra: ExtraName,
+}
+
+impl<'de> serde::Deserialize<'de> for ToolUvConflictingSet {
+    fn deserialize<D>(deserializer: D) -> Result<ToolUvConflictingSet, D::Error>
+    where
+        D: serde::Deserializer<'de>,
+    {
+        let items = Vec::<ToolUvConflictingItem>::deserialize(deserializer)?;
+        Self::try_from(items).map_err(serde::de::Error::custom)
+    }
+}
+
+impl TryFrom<Vec<ToolUvConflictingItem>> for ToolUvConflictingSet {
+    type Error = ToolUvConflictingError;
+
+    fn try_from(
+        items: Vec<ToolUvConflictingItem>,
+    ) -> Result<ToolUvConflictingSet, ToolUvConflictingError> {
+        match items.len() {
+            0 => return Err(ToolUvConflictingError::ZeroGroups),
+            1 => return Err(ToolUvConflictingError::OneGroup),
+            _ => {}
+        }
+        Ok(ToolUvConflictingSet(items))
+    }
+}
+
+/// An error that occurs when the given conflicting groups are invalid somehow.
+#[derive(Debug, thiserror::Error)]
+pub enum ToolUvConflictingError {
+    /// An error for when there are zero conflicting groups.
+    #[error("Each set of conflicts must have at least two entries, but found none")]
+    ZeroGroups,
+    /// An error for when there is one conflicting group.
+    #[error("Each set of conflicts must have at least two entries, but found only one")]
+    OneGroup,
+}
+
 #[derive(Default, Debug, Clone, PartialEq, Eq)]
 #[cfg_attr(test, derive(Serialize))]
 #[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]