mostro_client/

cli.rs

pub mod add_invoice;
pub mod adm_send_dm;
pub mod conversation_key;
pub mod dm_to_user;
pub mod get_dm;
pub mod get_dm_user;
pub mod last_trade_index;
pub mod list_disputes;
pub mod list_orders;
pub mod new_order;
pub mod orders_info;
pub mod rate_user;
pub mod restore;
pub mod send_dm;
pub mod send_msg;
pub mod take_dispute;
pub mod take_order;

use crate::cli::add_invoice::execute_add_invoice;
use crate::cli::adm_send_dm::execute_adm_send_dm;
use crate::cli::conversation_key::execute_conversation_key;
use crate::cli::dm_to_user::execute_dm_to_user;
use crate::cli::get_dm::execute_get_dm;
use crate::cli::get_dm_user::execute_get_dm_user;
use crate::cli::last_trade_index::execute_last_trade_index;
use crate::cli::list_disputes::execute_list_disputes;
use crate::cli::list_orders::execute_list_orders;
use crate::cli::new_order::execute_new_order;
use crate::cli::orders_info::execute_orders_info;
use crate::cli::rate_user::execute_rate_user;
use crate::cli::restore::execute_restore;
use crate::cli::send_dm::execute_send_dm;
use crate::cli::take_dispute::execute_take_dispute;
use crate::cli::take_order::execute_take_order;
use crate::db::{connect, User};
use crate::util;

use anyhow::{Error, Result};
use clap::{Parser, Subcommand};
use mostro_core::prelude::*;
use nostr_sdk::prelude::*;
use sqlx::SqlitePool;
use std::{
    env::{set_var, var},
    str::FromStr,
};
use take_dispute::*;
use uuid::Uuid;

#[derive(Debug)]
pub struct Context {
    pub client: Client,
    pub identity_keys: Keys,
    pub trade_keys: Keys,
    pub trade_index: i64,
    pub pool: SqlitePool,
    pub context_keys: Option<Keys>,
    pub mostro_pubkey: PublicKey,
}

#[derive(Parser)]
#[command(
    name = "mostro-cli",
    about = "A simple CLI to use Mostro P2P",
    author,
    help_template = "\
{before-help}{name} 🧌

{about-with-newline}
{author-with-newline}
{usage-heading} {usage}

{all-args}{after-help}
",
    version
)]
#[command(propagate_version = true)]
#[command(arg_required_else_help(true))]
pub struct Cli {
    #[command(subcommand)]
    pub command: Option<Commands>,
    #[arg(short, long)]
    pub verbose: bool,
    #[arg(short, long)]
    pub mostropubkey: Option<String>,
    #[arg(short, long)]
    pub relays: Option<String>,
    #[arg(short, long)]
    pub pow: Option<String>,
    #[arg(short, long)]
    pub secret: bool,
}

#[derive(Subcommand, Clone)]
#[clap(rename_all = "lower")]
pub enum Commands {
    /// Requests open orders from Mostro pubkey
    ListOrders {
        /// Status of the order
        #[arg(short, long)]
        status: Option<String>,
        /// Currency selected
        #[arg(short, long)]
        currency: Option<String>,
        /// Choose an order kind
        #[arg(short, long)]
        kind: Option<String>,
    },
    /// Create a new buy/sell order on Mostro
    NewOrder {
        /// Choose an order kind
        #[arg(short, long)]
        kind: String,
        /// Sats amount - leave empty for market price
        #[arg(short, long)]
        #[clap(default_value_t = 0)]
        amount: i64,
        /// Currency selected
        #[arg(short = 'c', long)]
        fiat_code: String,
        /// Fiat amount
        #[arg(short, long)]
        #[clap(value_parser=check_fiat_range)]
        fiat_amount: (i64, Option<i64>),
        /// Payment method
        #[arg(short = 'm', long)]
        payment_method: String,
        /// Premium on price
        #[arg(short, long)]
        #[clap(default_value_t = 0)]
        #[clap(allow_hyphen_values = true)]
        premium: i64,
        /// Invoice string
        #[arg(short, long)]
        invoice: Option<String>,
        /// Expiration time of a pending Order, in days
        #[arg(short, long)]
        #[clap(default_value_t = 0)]
        expiration_days: i64,
    },
    /// Take a sell order from a Mostro pubkey
    TakeSell {
        /// Order id
        #[arg(short, long)]
        order_id: Uuid,
        /// Invoice string
        #[arg(short, long)]
        invoice: Option<String>,
        /// Amount of fiat to buy
        #[arg(short, long)]
        amount: Option<u32>,
    },
    /// Take a buy order from a Mostro pubkey
    TakeBuy {
        /// Order id
        #[arg(short, long)]
        order_id: Uuid,
        /// Amount of fiat to sell
        #[arg(short, long)]
        amount: Option<u32>,
    },
    /// Buyer add a new invoice to receive the payment
    AddInvoice {
        /// Order id
        #[arg(short, long)]
        order_id: Uuid,
        /// Invoice string
        #[arg(short, long)]
        invoice: String,
    },
    /// Get the latest direct messages
    GetDm {
        /// Since time of the messages in minutes
        #[arg(short, long)]
        #[clap(default_value_t = 30)]
        since: i64,
        /// If true, get messages from counterparty, otherwise from Mostro
        #[arg(short, long)]
        from_user: bool,
    },
    /// Get direct messages sent to any trade keys
    GetDmUser {
        /// Since time of the messages in minutes
        #[arg(short, long)]
        #[clap(default_value_t = 30)]
        since: i64,
    },
    /// Get the latest direct messages for admin
    GetAdminDm {
        /// Since time of the messages in minutes
        #[arg(short, long)]
        #[clap(default_value_t = 30)]
        since: i64,
        /// If true, get messages from counterparty, otherwise from Mostro
        #[arg(short, long)]
        from_user: bool,
    },
    /// Send direct message to a user
    SendDm {
        /// Pubkey of the counterpart
        #[arg(short, long)]
        pubkey: String,
        /// Order id
        #[arg(short, long)]
        order_id: Uuid,
        /// Message to send
        #[arg(short, long)]
        message: String,
    },
    /// Send gift wrapped direct message to a user
    DmToUser {
        /// Pubkey of the recipient
        #[arg(short, long)]
        pubkey: String,
        /// Order id to get ephemeral keys
        #[arg(short, long)]
        order_id: Uuid,
        /// Message to send
        #[arg(short, long)]
        message: String,
    },
    /// Send fiat sent message to confirm payment to other user
    FiatSent {
        /// Order id
        #[arg(short, long)]
        order_id: Uuid,
    },
    /// Settle the hold invoice and pay to buyer.
    Release {
        /// Order id
        #[arg(short, long)]
        order_id: Uuid,
    },
    /// Cancel a pending order
    Cancel {
        /// Order id
        #[arg(short, long)]
        order_id: Uuid,
    },
    /// Rate counterpart after a successful trade
    Rate {
        /// Order id
        #[arg(short, long)]
        order_id: Uuid,
        /// Rating from 1 to 5
        #[arg(short, long)]
        rating: u8,
    },
    /// Restore session to recover all pending orders and disputes
    Restore {},
    /// Start a dispute
    Dispute {
        /// Order id
        #[arg(short, long)]
        order_id: Uuid,
    },
    /// Cancel an order (only admin)
    AdmCancel {
        /// Order id
        #[arg(short, long)]
        order_id: Uuid,
    },
    /// Settle a seller's hold invoice (only admin)
    AdmSettle {
        /// Order id
        #[arg(short, long)]
        order_id: Uuid,
    },
    /// Requests open disputes from Mostro pubkey
    ListDisputes {},
    /// Add a new dispute's solver (only admin)
    AdmAddSolver {
        /// npubkey
        #[arg(short, long)]
        npubkey: String,
    },
    /// Admin or solver take a Pending dispute (only admin)
    AdmTakeDispute {
        /// Dispute id
        #[arg(short, long)]
        dispute_id: Uuid,
    },
    /// Send gift wrapped direct message to a user (only admin)
    AdmSendDm {
        /// Pubkey of the recipient
        #[arg(short, long)]
        pubkey: String,
        /// Message to send
        #[arg(short, long)]
        message: String,
    },
    /// Get the conversation key for direct messaging with a user
    ConversationKey {
        /// Pubkey of the counterpart
        #[arg(short, long)]
        pubkey: String,
    },
    /// Get last trade index of user
    GetLastTradeIndex {},
    /// Request detailed information for specific orders
    OrdersInfo {
        /// Order IDs to request information for
        #[arg(short, long)]
        order_ids: Vec<Uuid>,
    },
}

fn get_env_var(cli: &Cli) {
    // Init logger
    if cli.verbose {
        set_var("RUST_LOG", "info");
        pretty_env_logger::init();
    }

    if let Some(ref mostro_pubkey) = cli.mostropubkey {
        set_var("MOSTRO_PUBKEY", mostro_pubkey.clone());
    }
    let _pubkey = var("MOSTRO_PUBKEY").expect("$MOSTRO_PUBKEY env var needs to be set");

    if let Some(ref relays) = cli.relays {
        set_var("RELAYS", relays.clone());
    }

    if let Some(ref pow) = cli.pow {
        set_var("POW", pow.clone());
    }

    if cli.secret {
        set_var("SECRET", "true");
    }
}

// Check range with two values value
fn check_fiat_range(s: &str) -> Result<(i64, Option<i64>)> {
    if s.contains('-') {
        // Get values from CLI
        let values: Vec<&str> = s.split('-').collect();

        // Check if more than two values
        if values.len() > 2 {
            return Err(Error::msg("Wrong amount syntax"));
        };

        // Get ranged command
        let min = values[0]
            .parse::<i64>()
            .map_err(|e| anyhow::anyhow!("Invalid min value: {}", e))?;
        let max = values[1]
            .parse::<i64>()
            .map_err(|e| anyhow::anyhow!("Invalid max value: {}", e))?;

        // Check min below max
        if min >= max {
            return Err(Error::msg("Range of values must be 100-200 for example..."));
        };
        Ok((min, Some(max)))
    } else {
        match s.parse::<i64>() {
            Ok(s) => Ok((s, None)),
            Err(e) => Err(e.into()),
        }
    }
}

pub async fn run() -> Result<()> {
    let cli = Cli::parse();

    let ctx = init_context(&cli).await?;

    if let Some(cmd) = &cli.command {
        cmd.run(&ctx).await?;
    }

    Ok(())
}

async fn init_context(cli: &Cli) -> Result<Context> {
    // Get environment variables
    get_env_var(cli);

    // Initialize database pool
    let pool = connect().await?;

    // Get identity keys
    let identity_keys = User::get_identity_keys(&pool)
        .await
        .map_err(|e| anyhow::anyhow!("Failed to get identity keys: {}", e))?;

    // Get trade keys
    let (trade_keys, trade_index) = User::get_next_trade_keys(&pool)
        .await
        .map_err(|e| anyhow::anyhow!("Failed to get trade keys: {}", e))?;

    // Load private key of admin - only required for admin commands
    // For regular user commands, this will be None
    let context_keys = if is_admin_command(&cli.command) {
        Some(
            std::env::var("ADMIN_NSEC")
                .map_err(|e| {
                    anyhow::anyhow!("ADMIN_NSEC not set (required for admin commands): {}", e)
                })?
                .parse::<Keys>()
                .map_err(|e| anyhow::anyhow!("Failed to parse ADMIN_NSEC: {}", e))?,
        )
    } else {
        None
    };

    // Resolve Mostro pubkey from env (required for all flows)
    let mostro_pubkey = PublicKey::from_str(
        &std::env::var("MOSTRO_PUBKEY")
            .map_err(|e| anyhow::anyhow!("Failed to get MOSTRO_PUBKEY: {}", e))?,
    )?;

    // Connect to Nostr relays
    let client = util::connect_nostr().await?;

    Ok(Context {
        client,
        identity_keys,
        trade_keys,
        trade_index,
        pool,
        context_keys,
        mostro_pubkey,
    })
}

fn is_admin_command(command: &Option<Commands>) -> bool {
    matches!(
        command,
        Some(Commands::AdmCancel { .. })
            | Some(Commands::AdmSettle { .. })
            | Some(Commands::AdmAddSolver { .. })
            | Some(Commands::AdmTakeDispute { .. })
            | Some(Commands::AdmSendDm { .. })
            | Some(Commands::GetAdminDm { .. })
    )
}

impl Commands {
    pub async fn run(&self, ctx: &Context) -> Result<()> {
        match self {
            // Simple order message commands
            Commands::FiatSent { order_id }
            | Commands::Release { order_id }
            | Commands::Dispute { order_id }
            | Commands::Cancel { order_id } => {
                crate::util::run_simple_order_msg(self.clone(), Some(*order_id), ctx).await
            }
            // Last trade index commands
            Commands::GetLastTradeIndex {} => {
                execute_last_trade_index(&ctx.identity_keys, ctx.mostro_pubkey, ctx).await
            }
            // DM commands with pubkey parsing
            Commands::SendDm {
                pubkey,
                order_id,
                message,
            } => execute_send_dm(PublicKey::from_str(pubkey)?, ctx, order_id, message).await,
            Commands::DmToUser {
                pubkey,
                order_id,
                message,
            } => {
                execute_dm_to_user(
                    PublicKey::from_str(pubkey)?,
                    &ctx.client,
                    order_id,
                    message,
                    &ctx.pool,
                )
                .await
            }
            Commands::AdmSendDm { pubkey, message } => {
                execute_adm_send_dm(PublicKey::from_str(pubkey)?, ctx, message).await
            }
            Commands::ConversationKey { pubkey } => {
                execute_conversation_key(&ctx.trade_keys, PublicKey::from_str(pubkey)?).await
            }

            // Order management commands
            Commands::ListOrders {
                status,
                currency,
                kind,
            } => execute_list_orders(kind, currency, status, ctx).await,
            Commands::NewOrder {
                kind,
                fiat_code,
                amount,
                fiat_amount,
                payment_method,
                premium,
                invoice,
                expiration_days,
            } => {
                execute_new_order(
                    kind,
                    fiat_code,
                    fiat_amount,
                    amount,
                    payment_method,
                    premium,
                    invoice,
                    ctx,
                    expiration_days,
                )
                .await
            }
            Commands::TakeSell {
                order_id,
                invoice,
                amount,
            } => execute_take_order(order_id, Action::TakeSell, invoice, *amount, ctx).await,
            Commands::TakeBuy { order_id, amount } => {
                execute_take_order(order_id, Action::TakeBuy, &None, *amount, ctx).await
            }
            Commands::AddInvoice { order_id, invoice } => {
                execute_add_invoice(order_id, invoice, ctx).await
            }
            Commands::Rate { order_id, rating } => execute_rate_user(order_id, rating, ctx).await,

            // DM retrieval commands
            Commands::GetDm { since, from_user } => {
                execute_get_dm(since, false, from_user, ctx).await
            }
            Commands::GetDmUser { since } => execute_get_dm_user(since, ctx).await,
            Commands::GetAdminDm { since, from_user } => {
                execute_get_dm(since, true, from_user, ctx).await
            }

            // Admin commands
            Commands::ListDisputes {} => execute_list_disputes(ctx).await,
            Commands::AdmAddSolver { npubkey } => execute_admin_add_solver(npubkey, ctx).await,
            Commands::AdmSettle { order_id } => execute_admin_settle_dispute(order_id, ctx).await,
            Commands::AdmCancel { order_id } => execute_admin_cancel_dispute(order_id, ctx).await,
            Commands::AdmTakeDispute { dispute_id } => execute_take_dispute(dispute_id, ctx).await,

            // Simple commands
            Commands::Restore {} => {
                execute_restore(&ctx.identity_keys, ctx.mostro_pubkey, ctx).await
            }
            Commands::OrdersInfo { order_ids } => execute_orders_info(order_ids, ctx).await,
        }
    }
}