From d6c3ff5a6acd60b055a9feb4653aeafd61bea2a6 Mon Sep 17 00:00:00 2001
From: Evan <0xIchigo@protonmail.com>
Date: Tue, 30 Apr 2024 15:28:29 -0400
Subject: [PATCH] Add MintApiAuthority

---
 src/types/enums.rs     | 19 +++++++++++++++++++
 tests/test_mint_api.rs | 40 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 59 insertions(+)

diff --git a/src/types/enums.rs b/src/types/enums.rs
index bc312b4..112e8b9 100644
--- a/src/types/enums.rs
+++ b/src/types/enums.rs
@@ -1,5 +1,7 @@
 use serde::{Deserialize, Serialize};
 
+use super::*;
+
 #[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Default)]
 pub enum Interface {
     #[serde(rename = "V1_NFT")]
@@ -131,3 +133,20 @@ pub enum TokenType {
     RegularNft,
     All,
 }
+
+#[derive(Debug, Clone, Copy, PartialEq)]
+pub enum MintApiAuthority {
+    Mainnet(&'static str),
+    Devnet(&'static str),
+}
+
+impl MintApiAuthority {
+    pub fn from_cluster(cluster: Cluster) -> Result<Self, &'static str> {
+        match cluster {
+            Cluster::Devnet => Ok(MintApiAuthority::Devnet("2LbAtCJSaHqTnP9M5QSjvAMXk79RNLusFspFN5Ew67TC")),
+            Cluster::MainnetBeta => Ok(MintApiAuthority::Mainnet(
+                "HnT5KVAywGgQDhmh6Usk4bxRg4RwKxCK4jmECyaDth5R",
+            )),
+        }
+    }
+}
diff --git a/tests/test_mint_api.rs b/tests/test_mint_api.rs
index 30d111e..0316834 100644
--- a/tests/test_mint_api.rs
+++ b/tests/test_mint_api.rs
@@ -155,3 +155,43 @@ async fn test_get_asset_proof_failure() {
     let result: Result<MintResponse, HeliusError> = helius.mint_compressed_nft(request).await;
     assert!(result.is_err(), "Expected an error but got success");
 }
+
+#[tokio::test]
+async fn test_mint_api_authority_from_cluster_success() {
+    let devnet_cluster: Cluster = Cluster::Devnet;
+    let mainnet_cluster: Cluster = Cluster::MainnetBeta;
+
+    let devnet_authority: Result<MintApiAuthority, &str> = MintApiAuthority::from_cluster(devnet_cluster);
+    let mainnet_authority: Result<MintApiAuthority, &str> = MintApiAuthority::from_cluster(mainnet_cluster);
+
+    assert_eq!(
+        devnet_authority.unwrap(),
+        MintApiAuthority::Devnet("2LbAtCJSaHqTnP9M5QSjvAMXk79RNLusFspFN5Ew67TC"),
+        "Devnet authority did not match expected value"
+    );
+    assert_eq!(
+        mainnet_authority.unwrap(),
+        MintApiAuthority::Mainnet("HnT5KVAywGgQDhmh6Usk4bxRg4RwKxCK4jmECyaDth5R"),
+        "Mainnet authority did not match expected value"
+    );
+}
+
+#[tokio::test]
+async fn test_mint_api_authority_from_cluster_failure() {
+    let devnet_cluster: Cluster = Cluster::Devnet;
+    let mainnet_cluster: Cluster = Cluster::MainnetBeta;
+
+    let devnet_authority: Result<MintApiAuthority, &str> = MintApiAuthority::from_cluster(devnet_cluster);
+    let mainnet_authority: Result<MintApiAuthority, &str> = MintApiAuthority::from_cluster(mainnet_cluster);
+
+    assert_ne!(
+        devnet_authority.unwrap(),
+        MintApiAuthority::Devnet("Blade"),
+        "Devnet authority did not match expected value"
+    );
+    assert_ne!(
+        mainnet_authority.unwrap(),
+        MintApiAuthority::Mainnet("Deacon Frost"),
+        "Mainnet authority did not match expected value"
+    );
+}