Add method returning rare sat supply

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 => 2099999990760000,
+      Self::Uncommon => 6926535,
+      Self::Rare => 3432,
+      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());
+  }
 }