wiiload 0.4.0

A minimal wiiload implementation in rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

// SPDX-FileCopyrightText: 2026 Manuel Quarneti <mq1@ik.me>
// SPDX-License-Identifier: MIT OR Apache-2.0

#![warn(clippy::all, rust_2018_idioms)]

#[cfg(feature = "cli")]
struct Args {
    file: String,
    wii_ip: std::net::IpAddr,
    wii_port: u16,
    compress: bool,
}

#[cfg(feature = "cli")]
fn parse_args() -> Result<Args, lexopt::Error> {
    use lexopt::prelude::*;

    let mut file = None;
    let mut wii_ip = None;
    let mut compress = false;
    let mut wii_port = wiiload::WIILOAD_PORT;

    let mut parser = lexopt::Parser::from_env();
    while let Some(arg) = parser.next()? {
        match arg {
            Short('i') | Long("wii-ip") => {
                wii_ip = Some(parser.value()?.parse()?);
            }
            Short('p') | Long("wii-port") => {
                wii_port = parser.value()?.parse()?;
            }
            Short('c') | Long("compress") => {
                compress = true;
            }
            Value(val) => {
                file = Some(val.string()?);
            }
            Short('h') | Long("help") => {
                println!("Usage: wiiload [-i|--wii-ip=IP] [-p|--wii-port=PORT] [--compress] FILE");
                std::process::exit(0);
            }
            _ => return Err(arg.unexpected()),
        }
    }

    Ok(Args {
        file: file.ok_or("Please specify a file")?,
        wii_ip: wii_ip.ok_or("Please specify a wii ip")?,
        compress,
        wii_port,
    })
}

#[cfg(feature = "cli")]
fn main() -> Result<(), lexopt::Error> {
    let args = parse_args()?;

    let file_path = std::path::Path::new(&args.file);
    let body = std::fs::read(file_path).unwrap();
    let filename = file_path.file_name().unwrap().to_str().unwrap();

    let mut conn = std::net::TcpStream::connect((args.wii_ip, args.wii_port)).unwrap();

    if args.compress {
        #[cfg(feature = "compression")]
        {
            println!("Compressing and sending file...");
            wiiload::compress_then_send(&mut conn, filename, &body).unwrap();
        }
        #[cfg(not(feature = "compression"))]
        {
            println!("Compression not enabled! Please add the `compression` feature to enable it.");
        }
    } else {
        println!("Sending file...");
        wiiload::send(&mut conn, filename, &body).unwrap();
    }

    Ok(())
}

#[cfg(not(feature = "cli"))]
fn main() {
    println!("Please add the `cli` feature to enable the CLI");
}