impl conversion methods for Atom and Cp

1 files changed, 28 insertions, 0 deletions

diff --git a/src/atom/mod.rs b/src/atom/mod.rs
index 6f53f0e..2ecf655 100644
--- a/src/atom/mod.rs
+++ b/src/atom/mod.rs
@@ -133,6 +133,15 @@ pub struct Atom {
     usedeps: Vec<UseDep>,
 }
 
+impl Cpv {
+    pub fn into_cp(self) -> Cp {
+        Cp {
+            name: self.name,
+            category: self.category,
+        }
+    }
+}
+
 impl Atom {
     pub fn version_operator(&self) -> Option<VersionOperator> {
         match self.version {
@@ -140,6 +149,25 @@ impl Atom {
             None => None,
         }
     }
+
+    pub fn into_cp(self) -> Cp {
+        Cp {
+            category: self.category,
+            name: self.name,
+        }
+    }
+
+    pub fn into_cpv(self) -> Option<Cpv> {
+        match self.version {
+            Some((_, version)) => Some(Cpv {
+                category: self.category,
+                name: self.name,
+                version,
+                slot: self.slot,
+            }),
+            None => None,
+        }
+    }
 }
 
 impl PartialEq for VersionSuffix {