fennec-modbus 0.92.0

#![cfg(feature = "cli")]

mod cli;

use anyhow::Error;
use clap::{Parser, Subcommand};
use fennec_modbus::tcp::{UnitId, tokio::Client};
use tracing::level_filters::LevelFilter;
use tracing_subscriber::{EnvFilter, Layer, layer::SubscriberExt, util::SubscriberInitExt};

type Result<T = (), E = Error> = core::result::Result<T, E>;

#[tokio::main]
async fn main() -> Result {
    let args = Args::parse();
    let env_filter =
        EnvFilter::builder().with_default_directive(LevelFilter::WARN.into()).from_env()?;
    tracing_subscriber::registry()
        .with(tracing_subscriber::fmt::layer().without_time().compact().with_filter(env_filter))
        .init();

    let client = args.endpoint.client();
    let unit_id = args.unit_id;

    match args.command {
        Command::Read(device) => match device {
            Device::Mq2200 => {
                cli::mq2200::read(client, unit_id).await?;
            }
        },
    }

    Ok(())
}

#[derive(Parser)]
struct Args {
    #[clap(flatten)]
    endpoint: Endpoint,

    /// Unit ID aka «slave ID».
    #[clap(long = "unit-id", alias = "slave-id", env = "UNIT_ID")]
    unit_id: UnitId,

    #[clap(subcommand)]
    command: Command,
}

#[derive(Parser)]
struct Endpoint {
    /// Connection endpoint.
    #[clap(name = "ENDPOINT", env = "ENDPOINT")]
    inner: String,
}

impl Endpoint {
    pub fn client(self) -> Client<String> {
        Client::new(self.inner)
    }
}

#[derive(Copy, Clone, Subcommand)]
enum Command {
    /// Test readings from a live device.
    #[clap(subcommand)]
    Read(Device),
}

#[derive(Copy, Clone, Subcommand)]
enum Device {
    /// Fox ESS MQ2200 (Mini Qube), Solakon ONE, and Avocado 22 Pro.
    #[clap(alias = "solakon-one", alias = "avocado-22-pro")]
    Mq2200,
}