Add price command to current feed price

helium-lib/src/token.rs

@@ -20,7 +20,11 @@ lazy_static::lazy_static! {
     static ref HNT_PRICE_KEY: Pubkey = Pubkey::from_str("4DdmDswskDxXGpwHrXUfn2CNUm9rt21ac79GHNTN3J33").unwrap();
 
     static ref MOBILE_MINT: Pubkey = Pubkey::from_str("mb1eu7TzEc71KxDpsmsKoucSSuuoGLv1drys1oP2jh6").unwrap();
+    static ref MOBILE_PRICE_KEY: Pubkey = Pubkey::from_str("DQ4C1tzvu28cwo1roN1Wm6TW35sfJEjLh517k3ZeWevx").unwrap();
+
     static ref IOT_MINT: Pubkey = Pubkey::from_str("iotEVVZLEywoTn1QdwNPddxPWszn3zFhEot3MfL9fns").unwrap();
+    static ref IOT_PRICE_KEY: Pubkey = Pubkey::from_str("8UYEn5Weq7toHwgcmctvcAxaNJo3SJxXEayM57rpoXr9").unwrap();
+
     static ref DC_MINT: Pubkey = Pubkey::from_str("dcuc8Amr83Wz27ZkQ2K9NS6r8zRpf1J6cvArEBDZDmm").unwrap();
     static ref SOL_MINT: Pubkey = solana_sdk::system_program::ID;
 }
@@ -222,15 +226,22 @@ impl Token {
         vec![Self::Hnt, Self::Iot, Self::Mobile, Self::Dc, Self::Sol]
     }
 
-    pub fn transferrable_value_parser(s: &str) -> StdResult<Self, TokenError> {
-        let transferrable = [Self::Iot, Self::Mobile, Self::Hnt, Self::Sol];
+    fn from_allowed(s: &str, allowed: &[Self]) -> StdResult<Self, TokenError> {
         let result = Self::from_str(s)?;
-        if !transferrable.contains(&result) {
+        if !allowed.contains(&result) {
             return Err(TokenError::InvalidToken(s.to_string()));
         }
         Ok(result)
     }
 
+    pub fn transferrable_value_parser(s: &str) -> StdResult<Self, TokenError> {
+        Self::from_allowed(s, &[Self::Iot, Self::Mobile, Self::Hnt, Self::Sol])
+    }
+
+    pub fn pricekey_value_parser(s: &str) -> StdResult<Self, TokenError> {
+        Self::from_allowed(s, &[Self::Iot, Self::Mobile, Self::Hnt])
+    }
+
     pub fn associated_token_adress(&self, address: &Pubkey) -> Pubkey {
         match self {
             Self::Sol => *address,
@@ -362,6 +373,8 @@ impl Token {
     pub fn price_key(&self) -> Option<&Pubkey> {
         match self {
             Self::Hnt => Some(&HNT_PRICE_KEY),
+            Self::Iot => Some(&IOT_PRICE_KEY),
+            Self::Mobile => Some(&MOBILE_PRICE_KEY),
             _ => None,
         }
     }

helium-wallet/src/cmd/mod.rs

@@ -24,6 +24,7 @@ pub mod dc;
 pub mod export;
 pub mod hotspots;
 pub mod info;
+pub mod price;
 pub mod router;
 pub mod sign;
 pub mod transfer;

helium-wallet/src/cmd/price.rs

@@ -0,0 +1,19 @@
+use crate::cmd::*;
+use helium_lib::token;
+
+#[derive(Clone, Debug, clap::Args)]
+/// Get the current price from the pyth price feed for the given token
+pub struct Cmd {
+    /// Token to look up
+    #[arg(value_parser = token::Token::pricekey_value_parser)]
+    token: token::Token,
+}
+
+impl Cmd {
+    pub async fn run(&self, opts: Opts) -> Result {
+        let settings = opts.try_into()?;
+        let price = token::price::get(&settings, self.token).await?;
+
+        print_json(&price)
+    }
+}

helium-wallet/src/main.rs

@@ -1,6 +1,8 @@
 use clap::Parser;
 use helium_wallet::{
-    cmd::{balance, create, dc, export, hotspots, info, router, sign, transfer, upgrade, Opts},
+    cmd::{
+        balance, create, dc, export, hotspots, info, price, router, sign, transfer, upgrade, Opts,
+    },
     result::Result,
 };
 
@@ -30,6 +32,7 @@ pub enum Cmd {
     Create(create::Cmd),
     Hotspots(Box<hotspots::Cmd>),
     Dc(dc::Cmd),
+    Price(price::Cmd),
     Transfer(transfer::Cmd),
     Export(export::Cmd),
     Sign(sign::Cmd),
@@ -53,6 +56,7 @@ async fn run(cli: Cli) -> Result {
         Cmd::Create(cmd) => cmd.run(cli.opts).await,
         Cmd::Hotspots(cmd) => cmd.run(cli.opts).await,
         Cmd::Dc(cmd) => cmd.run(cli.opts).await,
+        Cmd::Price(cmd) => cmd.run(cli.opts).await,
         Cmd::Transfer(cmd) => cmd.run(cli.opts).await,
         Cmd::Export(cmd) => cmd.run(cli.opts).await,
         Cmd::Sign(cmd) => cmd.run(cli.opts).await,