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

//! Example reading text from a file, generating a chain, and then printing `n` amount of tokens.
//!
//! Usage: generate_from_file <FILE_PATH> <N>
//! Using `cargo run`: `cargo run --example generate_from_file -- <FILE_PATH> <N>`

use markovish::Chain;

use rand::thread_rng;
use std::{path::PathBuf, process::exit};

const USAGE: &str = "Usage: generate_from_file <FILE_PATH> <N>";

fn main() {
    let args: Vec<String> = std::env::args().collect();

    if args.len() != 3 {
        println!("{USAGE}");
        exit(1);
    }

    let text =
        std::fs::read_to_string(PathBuf::from(args[1].clone())).expect("could not read file");
    let chain = Chain::from_text(&text).unwrap();
    let gen_text = chain
        .generate_str(
            &mut thread_rng(),
            args[2]
                .parse()
                .expect("did not provide a valid token number"),
        )
        .expect("failed to generate text");
    println!("{}", gen_text.join(""));
}