From 1ccfa00ee29c36b364119e80f2377634fba671e6 Mon Sep 17 00:00:00 2001
From: Adam Harvey <adam@adamharvey.name>
Date: Fri, 13 Oct 2023 17:01:52 +0200
Subject: [PATCH] feat: allow direct extraction of the squatted package

---
 src/checks/mod.rs | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/checks/mod.rs b/src/checks/mod.rs
index 11c6802..7be8b39 100644
--- a/src/checks/mod.rs
+++ b/src/checks/mod.rs
@@ -56,6 +56,23 @@ impl Squat {
             package: package.into(),
         }
     }
+
+    /// Retrieves the name of the package that may be squatted.
+    pub fn package(&self) -> &str {
+        match self {
+            Squat::Bitflip(package) => package,
+            Squat::OmittedCharacter(package) => package,
+            Squat::RepeatedCharacter(package) => package,
+            Squat::SwappedCharacters(package) => package,
+            Squat::SwappedWords(package) => package,
+            Squat::Typo(package) => package,
+            Squat::Version(package) => package,
+            Squat::Custom {
+                message: _message,
+                package,
+            } => package,
+        }
+    }
 }
 
 impl Display for Squat {