rustradio 0.16.6

#![allow(unknown_lints)]
#![allow(clippy::collapsible_if)]
#![allow(mismatched_lifetime_syntaxes)]
use std::io::Write;

use anyhow::Result;
use clap::Parser;

use rustradio::block::{Block, BlockRet};
use rustradio::sigmf::{Capture, SigMF};
use rustradio::stream::NCReadStream;
use rustradio::{Error, blockchain};

#[derive(clap::Parser)]
struct Opt {
    #[command(subcommand)]
    command: Commands,
}

#[derive(clap::Args)]
struct CreateOpts {
    /// Sample rate.
    #[arg(long, value_parser = parse_float_with_underscores)]
    sample_rate: f64,

    /// Data type.
    #[arg(long, default_value = "cf32_le")]
    datatype: String,

    /// Capture start time in ISO8601 format.
    ///
    /// YYYY-MM-DDTHH:MM:SS.SSSZ
    #[arg(long, value_parser = validate_iso8601)]
    datetime: Option<String>,

    /// Frequency of capture.
    #[arg(long, value_parser = parse_float_with_underscores)]
    frequency: Option<f64>,

    /// Author.
    #[arg(long)]
    author: Option<String>,

    /// HW.
    #[arg(long)]
    hw: Option<String>,

    /// URL to license.
    #[arg(long)]
    license: Option<String>,

    /// Recorder software.
    #[arg(long)]
    recorder: Option<String>,

    /// Description.
    #[arg(long)]
    description: Option<String>,

    /// SHA512 of the data. If empty, it'll be calculated.
    // TODO: verify format.
    #[arg(long)]
    sha512: Option<String>,

    /// Rename base, excluding `.sigmf-{data,meta}`
    #[arg(long)]
    out: String,

    /// Only print metadata. Don't create or rename files.
    #[arg(long)]
    print: bool,

    raw: std::path::PathBuf,
}

#[allow(clippy::large_enum_variant)]
#[derive(clap::Subcommand)]
enum Commands {
    /// Create a metadata file for a raw data file, making a Recording.
    Create(CreateOpts),
    /// Parse a SigMF Archive/Recording, and check any checksum.
    Check(CheckOpts),
}

#[derive(clap::Args)]
struct CheckOpts {
    /// Archive or base name for Recording.
    archive: std::path::PathBuf,
}

fn validate_iso8601(s: &str) -> Result<String, String> {
    match chrono::DateTime::parse_from_rfc3339(s) {
        Ok(_) => Ok(s.to_string()),
        Err(e) => Err(format!("Invalid ISO8601 datetime: {e}")),
    }
}

fn parse_float_with_underscores(s: &str) -> Result<f64, String> {
    use std::str::FromStr;
    let cleaned = s.replace('_', "");
    f64::from_str(&cleaned).map_err(|e| format!("Invalid float: {e}"))
}

fn main() -> Result<()> {
    let opt = Opt::parse();
    match opt.command {
        Commands::Create(opt) => cmd_create(opt),
        Commands::Check(opt) => cmd_check(opt),
    }
}

use rustradio::block;
use rustradio::rustradio_macros;
#[derive(rustradio_macros::Block)]
#[rustradio(crate, new)]
pub struct CheckHash {
    #[rustradio(in)]
    src: NCReadStream<Vec<u8>>,

    correct: String,
}

impl Block for CheckHash {
    fn work(&mut self) -> Result<BlockRet, Error> {
        let (v, _tags) = match self.src.pop() {
            None => return Ok(BlockRet::WaitForStream(&self.src, 1)),
            Some(x) => x,
        };
        assert_eq!(
            v.iter().map(|v| format!("{v:02x}")).collect::<String>(),
            self.correct
        );
        println!("Hash is correct!");
        Ok(BlockRet::EOF)
    }
}

fn cmd_check(opt: CheckOpts) -> Result<()> {
    use rustradio::blocks::*;
    use rustradio::graph::GraphRunner;
    let mut g = rustradio::mtgraph::MTGraph::new();
    let src = SigMFSource::<u8>::builder(opt.archive)
        .ignore_type_error()
        .build()?;
    let Some(ref in_meta) = src.0.meta().global.core_sha512 else {
        eprintln!("Metadata file doesn't have sha512. Nothing to check");
        return Ok(());
    };
    let in_meta = in_meta.to_owned();
    let prev = blockchain![g, prev, src, sha512(prev)];
    g.add(Box::new(CheckHash::new(prev, in_meta)));
    g.run().map_err(Into::into)
}

fn cmd_create(opt: CreateOpts) -> Result<()> {
    let mut sigmf = SigMF::new(opt.datatype.clone());
    sigmf.global.core_sample_rate = Some(opt.sample_rate);
    sigmf.global.core_author = opt.author;
    sigmf.global.core_hw = opt.hw;
    sigmf.global.core_license = opt.license;
    sigmf.global.core_recorder = opt.recorder;
    sigmf.global.core_description = opt.description;
    let hash = match opt.sha512 {
        Some(ref x) => {
            if x.len() != 128 && !x.is_empty() {
                return Err(Error::msg(
                    "SHA512 string needs to be empty or 128 hex characters (64 bytes)",
                )
                .into());
            }
            if !x.chars().all(|ch| ch.is_ascii_hexdigit()) {
                return Err(Error::msg("SHA512 string needs to be hex bytes").into());
            }
            x.to_string().to_lowercase()
        }
        None => {
            use sha2::Digest;
            use std::io::Read;
            let file = std::fs::File::open(&opt.raw).map_err(|e| Error::file_io(e, &opt.raw))?;
            let mut reader = std::io::BufReader::new(file);
            let mut hasher = sha2::Sha512::new();
            let mut buffer = [0u8; 8192];
            loop {
                let count = reader.read(&mut buffer)?;
                if count == 0 {
                    break;
                }
                hasher.update(&buffer[..count]);
            }
            hasher
                .finalize()
                .iter()
                .map(|v| format!("{v:02x}"))
                .collect()
        }
    };
    if !hash.is_empty() {
        sigmf.global.core_sha512 = Some(hash);
    }
    let mut cap = Capture::new(0);
    cap.core_frequency = opt.frequency;
    cap.core_datetime = opt.datetime;
    sigmf.captures.push(cap);
    let ser = serde_json::to_string(&sigmf)?;

    let dataname = opt.out.clone() + ".sigmf-data";
    let metaname = opt.out.clone() + ".sigmf-meta";

    if std::path::Path::new(&dataname).exists() {
        return Err(anyhow::Error::msg(format!(
            "Data file '{dataname}' already exists"
        )));
    }

    if opt.print {
        println!("{ser}");
        return Ok(());
    }
    {
        let mut meta = std::fs::OpenOptions::new()
            .write(true)
            .create_new(true)
            .open(&metaname)
            .map_err(|e| Error::msg(format!("Failed to create {metaname}: {e}")))?;
        meta.write_all(ser.as_bytes())?;
        meta.flush()?;
    }

    if let Err(e) = std::fs::rename(&opt.raw, &dataname) {
        std::fs::remove_file(&metaname).map_err(|e2|
            anyhow::Error::msg(format!("Failed to delete meta file '{metaname}': {e2} in the error path for renaming '{:?}' to '{dataname}': {e}", opt.raw)))?;
        return Err(e.into());
    }

    Ok(())
}