From c647df46b26c7855c5a23a945e209ebf113a8d6b Mon Sep 17 00:00:00 2001
From: Casey Rodarmor <casey@rodarmor.com>
Date: Thu, 26 Dec 2024 13:02:33 -0800
Subject: [PATCH] Add rare sat supply (#4153)

---
 crates/ordinals/src/rarity.rs | 68 ++++++++++++++++++++++++++++++-----
 1 file changed, 59 insertions(+), 9 deletions(-)

diff --git a/crates/ordinals/src/rarity.rs b/crates/ordinals/src/rarity.rs
index a92fce72a9..e080ba9d4a 100644
--- a/crates/ordinals/src/rarity.rs
+++ b/crates/ordinals/src/rarity.rs
@@ -1,6 +1,8 @@
 use super::*;
 
-#[derive(Debug, PartialEq, PartialOrd, Copy, Clone, DeserializeFromStr, SerializeDisplay)]
+#[derive(
+  Clone, Copy, Debug, DeserializeFromStr, Eq, Hash, PartialEq, PartialOrd, SerializeDisplay,
+)]
 pub enum Rarity {
   Common,
   Uncommon,
@@ -10,6 +12,28 @@ pub enum Rarity {
   Mythic,
 }
 
+impl Rarity {
+  pub const ALL: [Rarity; 6] = [
+    Rarity::Common,
+    Rarity::Uncommon,
+    Rarity::Rare,
+    Rarity::Epic,
+    Rarity::Legendary,
+    Rarity::Mythic,
+  ];
+
+  pub fn supply(self) -> u64 {
+    match self {
+      Self::Common => 2_099_999_990_760_000,
+      Self::Uncommon => 6_926_535,
+      Self::Rare => 3_432,
+      Self::Epic => 27,
+      Self::Legendary => 5,
+      Self::Mythic => 1,
+    }
+  }
+}
+
 impl From<Rarity> for u8 {
   fn from(rarity: Rarity) -> Self {
     rarity as u8
@@ -156,14 +180,7 @@ mod tests {
 
   #[test]
   fn conversions_with_u8() {
-    for &expected in &[
-      Rarity::Common,
-      Rarity::Uncommon,
-      Rarity::Rare,
-      Rarity::Epic,
-      Rarity::Legendary,
-      Rarity::Mythic,
-    ] {
+    for expected in Rarity::ALL {
       let n: u8 = expected.into();
       let actual = Rarity::try_from(n).unwrap();
       assert_eq!(actual, expected);
@@ -176,4 +193,37 @@ mod tests {
   fn error() {
     assert_eq!("foo".parse::<Rarity>().unwrap_err(), "invalid rarity `foo`");
   }
+
+  #[test]
+  fn supply() {
+    let mut i = 0;
+
+    let mut supply = HashMap::<Rarity, u64>::new();
+
+    for height in 0.. {
+      let subsidy = Height(height).subsidy();
+
+      if subsidy == 0 {
+        break;
+      }
+
+      *supply.entry(Sat(i).rarity()).or_default() += 1;
+
+      *supply.entry(Rarity::Common).or_default() += subsidy.saturating_sub(1);
+
+      i += subsidy;
+    }
+
+    for (rarity, supply) in &supply {
+      assert_eq!(
+        rarity.supply(),
+        *supply,
+        "invalid supply for rarity {rarity}"
+      );
+    }
+
+    assert_eq!(supply.values().sum::<u64>(), Sat::SUPPLY);
+
+    assert_eq!(supply.len(), Rarity::ALL.len());
+  }
 }