cdk_payment_processor/proto/

server.rs

use std::net::SocketAddr;
use std::path::PathBuf;
use std::pin::Pin;
use std::str::FromStr;
use std::sync::Arc;
use std::time::Duration;

use cdk_common::grpc::create_version_check_interceptor;
use cdk_common::payment::{IncomingPaymentOptions, MintPayment};
use cdk_common::{CurrencyUnit, PublicKey, QuoteId};
use futures::{Stream, StreamExt};
use lightning::offers::offer::Offer;
use tokio::sync::{mpsc, Notify};
use tokio::task::JoinHandle;
use tokio::time::{sleep, Instant};
use tokio_stream::wrappers::ReceiverStream;
use tonic::transport::{Certificate, Identity, Server, ServerTlsConfig};
use tonic::{async_trait, Request, Response, Status};
use tracing::instrument;

use super::cdk_payment_processor_server::{CdkPaymentProcessor, CdkPaymentProcessorServer};
use crate::error::Error;
use crate::proto::{TryFromProtoAmount, *};

type ResponseStream = Pin<Box<dyn Stream<Item = Result<PaymentEventResponse, Status>> + Send>>;

/// Payment Processor
#[derive(Clone)]
pub struct PaymentProcessorServer {
    inner: Arc<dyn MintPayment<Err = cdk_common::payment::Error> + Send + Sync>,
    socket_addr: SocketAddr,
    shutdown: Arc<Notify>,
    handle: Option<Arc<JoinHandle<anyhow::Result<()>>>>,
}

impl std::fmt::Debug for PaymentProcessorServer {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("PaymentProcessorServer")
            .field("socket_addr", &self.socket_addr)
            .finish_non_exhaustive()
    }
}

impl PaymentProcessorServer {
    /// Create new [`PaymentProcessorServer`]
    pub fn new(
        payment_processor: Arc<dyn MintPayment<Err = cdk_common::payment::Error> + Send + Sync>,
        addr: &str,
        port: u16,
    ) -> anyhow::Result<Self> {
        let socket_addr = SocketAddr::new(addr.parse()?, port);
        Ok(Self {
            inner: payment_processor,
            socket_addr,
            shutdown: Arc::new(Notify::new()),
            handle: None,
        })
    }

    /// Start fake wallet grpc server
    pub async fn start(&mut self, tls_dir: Option<PathBuf>) -> anyhow::Result<()> {
        tracing::info!("Starting RPC server {}", self.socket_addr);

        let server = match tls_dir {
            Some(tls_dir) => {
                tracing::info!("TLS configuration found, starting secure server");

                // Check for server.pem
                let server_pem_path = tls_dir.join("server.pem");
                if !server_pem_path.exists() {
                    let err_msg = format!(
                        "TLS certificate file not found: {}",
                        server_pem_path.display()
                    );
                    tracing::error!("{}", err_msg);
                    return Err(anyhow::anyhow!(err_msg));
                }

                // Check for server.key
                let server_key_path = tls_dir.join("server.key");
                if !server_key_path.exists() {
                    let err_msg = format!("TLS key file not found: {}", server_key_path.display());
                    tracing::error!("{}", err_msg);
                    return Err(anyhow::anyhow!(err_msg));
                }

                // Check for ca.pem
                let ca_pem_path = tls_dir.join("ca.pem");
                if !ca_pem_path.exists() {
                    let err_msg =
                        format!("CA certificate file not found: {}", ca_pem_path.display());
                    tracing::error!("{}", err_msg);
                    return Err(anyhow::anyhow!(err_msg));
                }

                let cert = std::fs::read_to_string(&server_pem_path)?;
                let key = std::fs::read_to_string(&server_key_path)?;
                let client_ca_cert = std::fs::read_to_string(&ca_pem_path)?;

                let client_ca_cert = Certificate::from_pem(client_ca_cert);
                let server_identity = Identity::from_pem(cert, key);
                let tls_config = ServerTlsConfig::new()
                    .identity(server_identity)
                    .client_ca_root(client_ca_cert);

                Server::builder().tls_config(tls_config)?.add_service(
                    CdkPaymentProcessorServer::with_interceptor(
                        self.clone(),
                        create_version_check_interceptor(
                            cdk_common::grpc::VERSION_HEADER,
                            cdk_common::PAYMENT_PROCESSOR_PROTOCOL_VERSION,
                        ),
                    ),
                )
            }
            None => {
                tracing::warn!("No valid TLS configuration found, starting insecure server");
                Server::builder().add_service(CdkPaymentProcessorServer::with_interceptor(
                    self.clone(),
                    create_version_check_interceptor(
                        cdk_common::grpc::VERSION_HEADER,
                        cdk_common::PAYMENT_PROCESSOR_PROTOCOL_VERSION,
                    ),
                ))
            }
        };

        let shutdown = self.shutdown.clone();
        let addr = self.socket_addr;

        self.handle = Some(Arc::new(tokio::spawn(async move {
            let server = server.serve_with_shutdown(addr, async {
                shutdown.notified().await;
            });

            server.await?;
            Ok(())
        })));

        Ok(())
    }

    /// Stop fake wallet grpc server
    pub async fn stop(&self) -> anyhow::Result<()> {
        const SHUTDOWN_TIMEOUT: Duration = Duration::from_secs(5);

        if let Some(handle) = &self.handle {
            tracing::info!("Initiating server shutdown");
            self.shutdown.notify_waiters();

            let start = Instant::now();

            while !handle.is_finished() {
                if start.elapsed() >= SHUTDOWN_TIMEOUT {
                    tracing::error!(
                        "Server shutdown timed out after {} seconds, aborting handle",
                        SHUTDOWN_TIMEOUT.as_secs()
                    );
                    handle.abort();
                    break;
                }
                sleep(Duration::from_millis(100)).await;
            }

            if handle.is_finished() {
                tracing::info!("Server shutdown completed successfully");
            }
        } else {
            tracing::info!("No server handle found, nothing to stop");
        }

        Ok(())
    }
}

impl Drop for PaymentProcessorServer {
    fn drop(&mut self) {
        tracing::debug!("Dropping payment process server");
        self.shutdown.notify_one();
    }
}

#[async_trait]
impl CdkPaymentProcessor for PaymentProcessorServer {
    async fn get_settings(
        &self,
        _request: Request<EmptyRequest>,
    ) -> Result<Response<SettingsResponse>, Status> {
        let settings = self
            .inner
            .get_settings()
            .await
            .map_err(|_| Status::internal("Could not get settings"))?;

        Ok(Response::new(SettingsResponse {
            unit: settings.unit,
            bolt11: settings.bolt11.map(|b| super::Bolt11Settings {
                mpp: b.mpp,
                amountless: b.amountless,
                invoice_description: b.invoice_description,
            }),
            bolt12: settings.bolt12.map(|b| super::Bolt12Settings {
                amountless: b.amountless,
            }),
            onchain: settings.onchain.map(|o| super::OnchainSettings {
                confirmations: o.confirmations,
                min_receive_amount_sat: o.min_receive_amount_sat,
                min_send_amount_sat: o.min_send_amount_sat,
            }),
            custom: settings.custom,
        }))
    }

    async fn create_payment(
        &self,
        request: Request<CreatePaymentRequest>,
    ) -> Result<Response<CreatePaymentResponse>, Status> {
        let CreatePaymentRequest { options, .. } = request.into_inner();

        let options = options.ok_or_else(|| Status::invalid_argument("Missing payment options"))?;

        let proto_options = match options
            .options
            .ok_or_else(|| Status::invalid_argument("Missing options"))?
        {
            incoming_payment_options::Options::Custom(opts) => {
                let amount: Option<cdk_common::Amount<CurrencyUnit>> = match opts.amount {
                    Some(a) => Some(
                        a.try_into()
                            .map_err(|_| Status::invalid_argument("Invalid amount"))?,
                    ),
                    None => None,
                };
                if opts.quote_id.is_empty() {
                    return Err(Status::invalid_argument(
                        "Missing quote_id in Custom options",
                    ));
                }
                let quote_id = parse_quote_id(&opts.quote_id)?;
                let pubkey = opts
                    .pubkey
                    .as_deref()
                    .map(PublicKey::from_str)
                    .transpose()
                    .map_err(|_| Status::invalid_argument("Invalid pubkey in Custom options"))?;
                IncomingPaymentOptions::Custom(Box::new(
                    cdk_common::payment::CustomIncomingPaymentOptions {
                        method: "".to_string(),
                        description: opts.description,
                        amount,
                        unix_expiry: opts.unix_expiry,
                        extra_json: opts.extra_json,
                        quote_id,
                        pubkey,
                    },
                ))
            }
            incoming_payment_options::Options::Bolt11(opts) => {
                let amount = opts
                    .amount
                    .ok_or_else(|| Status::invalid_argument("Missing amount"))?
                    .try_into()
                    .map_err(|_| Status::invalid_argument("Invalid amount"))?;
                IncomingPaymentOptions::Bolt11(cdk_common::payment::Bolt11IncomingPaymentOptions {
                    description: opts.description,
                    amount,
                    unix_expiry: opts.unix_expiry,
                })
            }
            incoming_payment_options::Options::Bolt12(opts) => {
                let amount: Option<cdk_common::Amount<CurrencyUnit>> = match opts.amount {
                    Some(a) => Some(
                        a.try_into()
                            .map_err(|_| Status::invalid_argument("Invalid amount"))?,
                    ),
                    None => None,
                };
                IncomingPaymentOptions::Bolt12(Box::new(
                    cdk_common::payment::Bolt12IncomingPaymentOptions {
                        description: opts.description,
                        amount,
                        unix_expiry: opts.unix_expiry,
                    },
                ))
            }
            incoming_payment_options::Options::Onchain(opts) => IncomingPaymentOptions::Onchain(
                cdk_common::payment::OnchainIncomingPaymentOptions {
                    quote_id: opts.quote_id.parse().map_err(|_| {
                        Status::invalid_argument("Invalid quote_id in Onchain options")
                    })?,
                },
            ),
        };

        let invoice_response = self
            .inner
            .create_incoming_payment_request(proto_options)
            .await
            .map_err(|_| Status::internal("Could not create invoice"))?;

        Ok(Response::new(invoice_response.into()))
    }

    async fn get_payment_quote(
        &self,
        request: Request<PaymentQuoteRequest>,
    ) -> Result<Response<PaymentQuoteResponse>, Status> {
        let request = request.into_inner();

        let unit = CurrencyUnit::from_str(&request.unit)
            .map_err(|_| Status::invalid_argument("Invalid currency unit"))?;

        let quote_id = parse_quote_id(&request.quote_id)?;

        let options = match request.request_type() {
            OutgoingPaymentRequestType::Bolt11Invoice => {
                let bolt11: cdk_common::Bolt11Invoice =
                    request.request.parse().map_err(Error::Invoice)?;

                cdk_common::payment::OutgoingPaymentOptions::Bolt11(Box::new(
                    cdk_common::payment::Bolt11OutgoingPaymentOptions {
                        bolt11,
                        max_fee_amount: None,
                        timeout_secs: None,
                        melt_options: request.options.map(TryInto::try_into).transpose()?,
                        quote_id,
                    },
                ))
            }
            OutgoingPaymentRequestType::Bolt12Offer => {
                // Parse offer to verify it's valid, but store as string
                let _: Offer = request.request.parse().map_err(|_| Error::Bolt12Parse)?;

                cdk_common::payment::OutgoingPaymentOptions::Bolt12(Box::new(
                    cdk_common::payment::Bolt12OutgoingPaymentOptions {
                        offer: Offer::from_str(&request.request)
                            .expect("Already validated offer above"),
                        max_fee_amount: None,
                        timeout_secs: None,
                        melt_options: request.options.map(TryInto::try_into).transpose()?,
                        quote_id,
                    },
                ))
            }
            OutgoingPaymentRequestType::Custom => {
                let amount = request
                    .amount
                    .try_from_proto()
                    .map_err(|_| Status::invalid_argument("Invalid amount"))?;

                // Custom payment method - pass request as-is with no validation
                cdk_common::payment::OutgoingPaymentOptions::Custom(Box::new(
                    cdk_common::payment::CustomOutgoingPaymentOptions {
                        method: String::new(), // Will be set from variant
                        request: request.request.clone(),
                        amount,
                        max_fee_amount: None,
                        timeout_secs: None,
                        melt_options: request.options.map(TryInto::try_into).transpose()?,
                        extra_json: request.extra_json.clone(),
                        quote_id,
                    },
                ))
            }
            OutgoingPaymentRequestType::Onchain => {
                let opts = request.onchain_options.ok_or_else(|| {
                    Status::invalid_argument("Missing onchain_options for onchain quote")
                })?;
                let amount = opts
                    .amount
                    .ok_or_else(|| Status::invalid_argument("Missing amount in onchain quote"))?
                    .try_into()
                    .map_err(|_| Status::invalid_argument("Invalid amount"))?;
                let max_fee_amount = opts
                    .max_fee_amount
                    .try_from_proto()
                    .map_err(|_| Status::invalid_argument("Invalid max_fee_amount"))?;
                let onchain_quote_id = parse_quote_id(&opts.quote_id)?;
                if onchain_quote_id != quote_id {
                    return Err(Status::invalid_argument(
                        "quote_id does not match onchain_options quote_id",
                    ));
                }

                cdk_common::payment::OutgoingPaymentOptions::Onchain(Box::new(
                    cdk_common::payment::OnchainOutgoingPaymentOptions {
                        address: opts.address,
                        amount,
                        max_fee_amount,
                        quote_id,
                        fee_index: opts.fee_index,
                        metadata: opts.metadata,
                    },
                ))
            }
            OutgoingPaymentRequestType::Unspecified => {
                return Err(Status::invalid_argument("Unspecified payment request type"));
            }
        };

        let payment_quote = self
            .inner
            .get_payment_quote(&unit, options)
            .await
            .map_err(|err| {
                tracing::error!("Could not get payment quote: {}", err);
                Status::internal("Could not get quote")
            })?;

        Ok(Response::new(payment_quote.into()))
    }

    async fn make_payment(
        &self,
        request: Request<MakePaymentRequest>,
    ) -> Result<Response<MakePaymentResponse>, Status> {
        let request = request.into_inner();

        let unit = CurrencyUnit::from_str(&request.unit)
            .map_err(|_| Status::invalid_argument("Invalid currency unit"))?;

        let options = request
            .payment_options
            .ok_or_else(|| Status::invalid_argument("Missing payment options"))?;

        let payment_options = match options
            .options
            .ok_or_else(|| Status::invalid_argument("Missing options"))?
        {
            outgoing_payment_variant::Options::Bolt11(opts) => {
                let bolt11: cdk_common::Bolt11Invoice =
                    opts.bolt11.parse().map_err(Error::Invoice)?;

                let max_fee_amount = opts
                    .max_fee_amount
                    .try_from_proto()
                    .map_err(|_| Status::invalid_argument("Invalid max_fee_amount"))?;
                let quote_id = parse_quote_id(&opts.quote_id)?;

                cdk_common::payment::OutgoingPaymentOptions::Bolt11(Box::new(
                    cdk_common::payment::Bolt11OutgoingPaymentOptions {
                        bolt11,
                        max_fee_amount,
                        timeout_secs: opts.timeout_secs,
                        melt_options: opts.melt_options.map(TryInto::try_into).transpose()?,
                        quote_id,
                    },
                ))
            }
            outgoing_payment_variant::Options::Bolt12(opts) => {
                let offer = Offer::from_str(&opts.offer).map_err(|_| Error::Bolt12Parse)?;

                let max_fee_amount = opts
                    .max_fee_amount
                    .try_from_proto()
                    .map_err(|_| Status::invalid_argument("Invalid max_fee_amount"))?;
                let quote_id = parse_quote_id(&opts.quote_id)?;

                cdk_common::payment::OutgoingPaymentOptions::Bolt12(Box::new(
                    cdk_common::payment::Bolt12OutgoingPaymentOptions {
                        offer,
                        max_fee_amount,
                        timeout_secs: opts.timeout_secs,
                        melt_options: opts.melt_options.map(TryInto::try_into).transpose()?,
                        quote_id,
                    },
                ))
            }
            outgoing_payment_variant::Options::Custom(opts) => {
                let max_fee_amount = opts
                    .max_fee_amount
                    .try_from_proto()
                    .map_err(|_| Status::invalid_argument("Invalid max_fee_amount"))?;
                let quote_id = parse_quote_id(&opts.quote_id)?;
                let amount: Option<cdk_common::Amount<CurrencyUnit>> = match opts.amount {
                    Some(a) => Some(
                        a.try_into()
                            .map_err(|_| Status::invalid_argument("Invalid amount"))?,
                    ),
                    None => None,
                };

                cdk_common::payment::OutgoingPaymentOptions::Custom(Box::new(
                    cdk_common::payment::CustomOutgoingPaymentOptions {
                        method: String::new(), // Method will be determined from context
                        request: opts.offer,   // Reusing offer field for custom request string
                        amount,
                        max_fee_amount,
                        timeout_secs: opts.timeout_secs,
                        melt_options: opts.melt_options.map(TryInto::try_into).transpose()?,
                        extra_json: opts.extra_json,
                        quote_id,
                    },
                ))
            }
            outgoing_payment_variant::Options::Onchain(opts) => {
                let amount = opts
                    .amount
                    .ok_or_else(|| Status::invalid_argument("Missing amount"))?
                    .try_into()
                    .map_err(|_| Status::invalid_argument("Invalid amount"))?;

                let max_fee_amount = opts
                    .max_fee_amount
                    .try_from_proto()
                    .map_err(|_| Status::invalid_argument("Invalid max_fee_amount"))?;

                cdk_common::payment::OutgoingPaymentOptions::Onchain(Box::new(
                    cdk_common::payment::OnchainOutgoingPaymentOptions {
                        address: opts.address,
                        amount,
                        max_fee_amount,
                        quote_id: opts.quote_id.parse().map_err(|_| {
                            Status::invalid_argument("Invalid quote_id in Onchain options")
                        })?,
                        fee_index: opts.fee_index,
                        metadata: opts.metadata,
                    },
                ))
            }
        };

        let pay_response = self
            .inner
            .make_payment(&unit, payment_options)
            .await
            .map_err(|err| {
                tracing::error!("Could not make payment: {}", err);

                match err {
                    cdk_common::payment::Error::InvoiceAlreadyPaid => {
                        Status::already_exists("Payment request already paid")
                    }
                    cdk_common::payment::Error::InvoicePaymentPending => {
                        Status::already_exists("Payment request pending")
                    }
                    _ => Status::internal("Could not pay invoice"),
                }
            })?;

        Ok(Response::new(pay_response.into()))
    }

    async fn check_incoming_payment(
        &self,
        request: Request<CheckIncomingPaymentRequest>,
    ) -> Result<Response<CheckIncomingPaymentResponse>, Status> {
        let request = request.into_inner();

        let payment_identifier = request
            .request_identifier
            .ok_or_else(|| Status::invalid_argument("Missing request identifier"))?
            .try_into()
            .map_err(|_| Status::invalid_argument("Invalid request identifier"))?;

        let check_responses = self
            .inner
            .check_incoming_payment_status(&payment_identifier)
            .await
            .map_err(|_| Status::internal("Could not check incoming payment status"))?;

        Ok(Response::new(CheckIncomingPaymentResponse {
            payments: check_responses.into_iter().map(|r| r.into()).collect(),
        }))
    }

    async fn check_outgoing_payment(
        &self,
        request: Request<CheckOutgoingPaymentRequest>,
    ) -> Result<Response<MakePaymentResponse>, Status> {
        let request = request.into_inner();

        let payment_identifier = request
            .request_identifier
            .ok_or_else(|| Status::invalid_argument("Missing request identifier"))?
            .try_into()
            .map_err(|_| Status::invalid_argument("Invalid request identifier"))?;

        let check_response = self
            .inner
            .check_outgoing_payment(&payment_identifier)
            .await
            .map_err(|_| Status::internal("Could not check outgoing payment status"))?;

        Ok(Response::new(check_response.into()))
    }

    type WaitPaymentEventStream = ResponseStream;

    #[allow(clippy::incompatible_msrv)]
    #[instrument(skip_all)]
    async fn wait_payment_event(
        &self,
        _request: Request<EmptyRequest>,
    ) -> Result<Response<Self::WaitPaymentEventStream>, Status> {
        tracing::debug!("Server waiting for payment stream");
        let (tx, rx) = mpsc::channel(128);

        let shutdown_clone = self.shutdown.clone();
        let ln = self.inner.clone();
        tokio::spawn(async move {
            loop {
                tokio::select! {
                    _ = shutdown_clone.notified() => {
                        tracing::info!("Shutdown signal received, stopping task");
                        ln.cancel_payment_event_stream();
                        break;
                    }
                    result = ln.wait_payment_event() => {
                        match result {
                            Ok(mut stream) => {
                                while let Some(event) = stream.next().await {
                                    match tx.send(Result::<_, Status>::Ok(event.into())).await {
                                        Ok(_) => {
                                            // Response was queued to be sent to client
                                        }
                                        Err(item) => {
                                            tracing::error!("Error adding payment event to stream: {}", item);
                                            break;
                                        }
                                    }
                                }
                            }
                            Err(err) => {
                                tracing::warn!("Could not get invoice stream: {}", err);
                                tokio::time::sleep(std::time::Duration::from_secs(5)).await;
                            }
                        }
                    }
                }
            }
        });

        let output_stream = ReceiverStream::new(rx);
        Ok(Response::new(
            Box::pin(output_stream) as Self::WaitPaymentEventStream
        ))
    }
}

fn parse_quote_id(s: &str) -> Result<QuoteId, Status> {
    s.parse()
        .map_err(|err| Status::invalid_argument(format!("Invalid quote_id: {err}")))
}