bencodex-rs 0.5.1

use bencodex::json::{from_json, to_json_with_options, BinaryEncoding, JsonEncodeOptions};
use bencodex::{Decode, Encode};
use clap::Parser;
use std::io::{Read, Write};
use std::process::ExitCode;

/// A program to encode and decode between Bencodex and JSON.
#[derive(Parser, Debug)]
#[command(version, about, long_about = None)]
struct Args {
    /// Encode Bencodex Binary by base64 string.
    /// If not given, it will encode as hexadecimal string.
    #[arg(short, long)]
    base64: bool,

    /// Decode to Bencodex from JSON.
    #[arg(short, long)]
    decode: bool,
}

fn main() -> ExitCode {
    let args = Args::parse();

    if !args.decode {
        encode(&args)
    } else {
        decode()
    }
}

fn decode() -> ExitCode {
    let mut buf = Vec::new();
    if let Err(err) = std::io::stdin().read_to_end(&mut buf) {
        eprintln!("{:?}", err);
        return ExitCode::FAILURE;
    }

    let json = match serde_json::from_slice(&buf) {
        Ok(x) => x,
        Err(_) => return ExitCode::FAILURE,
    };

    let bencodex_value = match from_json(&json) {
        Ok(x) => x,
        Err(_) => return ExitCode::FAILURE,
    };

    buf = Vec::new();
    if bencodex_value.encode(&mut buf).is_err() {
        return ExitCode::FAILURE;
    }

    let _ = std::io::stdout().write_all(&buf);

    ExitCode::SUCCESS
}

fn encode(args: &Args) -> ExitCode {
    let mut buf = Vec::new();
    if let Err(err) = std::io::stdin().read_to_end(&mut buf) {
        eprintln!("{:?}", err);
        return ExitCode::FAILURE;
    }

    let decoded = match buf.decode() {
        Ok(value) => value,
        Err(err) => {
            eprintln!("Failed to decode. {:?}", err);
            return ExitCode::FAILURE;
        }
    };

    let json_encode_options = JsonEncodeOptions {
        binary_encoding: if args.base64 {
            BinaryEncoding::Base64
        } else {
            BinaryEncoding::Hex
        },
    };

    println!("{}", to_json_with_options(&decoded, json_encode_options));

    ExitCode::SUCCESS
}