storm_node 0.9.0

// Storm node providing distributed storage & messaging for lightning network.
//
// Written in 2022 by
//     Dr. Maxim Orlovsky <orlovsky@lnp-bp.org>
//
// Copyright (C) 2022 by LNP/BP Standards Association, Switzerland.
//
// You should have received a copy of the MIT License along with this software.
// If not, see <https://opensource.org/licenses/MIT>.

use std::process::Command;

use microservices::error::BootstrapError;
use microservices::{DaemonHandle, Launcher, LauncherError};

use super::Runtime;
use crate::{chatd, downpourd, stormd, transferd, LaunchError};

/// Daemons that can be launched by lnpd
#[derive(Clone, Eq, PartialEq, Debug, Display)]
pub enum Daemon {
    #[display("transferd")]
    Transferd,

    #[display("chatd")]
    Chatd,

    #[display("downpourd")]
    Downpourd,
}

impl Launcher for Daemon {
    type RunError = BootstrapError<LaunchError>;
    type Config = crate::Config;

    fn bin_name(&self) -> &'static str {
        match self {
            Daemon::Transferd => "transferd",
            Daemon::Chatd => "chatd",
            Daemon::Downpourd => "downpourd",
        }
    }

    fn cmd_args(&self, cmd: &mut Command) -> Result<(), LauncherError<Self>> {
        cmd.args(std::env::args().skip(1).filter(|arg| {
            !["--threaded", "--chat", "--downpour"].iter().any(|pat| arg.starts_with(pat))
        }));

        Ok(())
    }

    fn run_impl(self, config: crate::Config) -> Result<(), Self::RunError> {
        match self {
            Daemon::Transferd => transferd::run(config),
            Daemon::Chatd => chatd::run(config),
            Daemon::Downpourd => downpourd::run(config),
        }
    }
}

impl Runtime {
    pub(super) fn launch_daemon(
        &self,
        daemon: Daemon,
        config: crate::Config<stormd::Config>,
    ) -> Result<DaemonHandle<Daemon>, LauncherError<Daemon>> {
        if self.config.ext.threaded {
            daemon.thread_daemon(config.into())
        } else {
            daemon.exec_daemon()
        }
    }
}