ddnet-map-gen 0.2.4

A DDraceNetwork map generator.
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99

use crate::generators::{fly::FlyGenerator, maze::MazeGenerator, MapGenerator};
use clap::{arg, Parser, Subcommand};
use eyre::Result;
use owo_colors::OwoColorize;
use rand::{distributions::Alphanumeric, thread_rng, Rng};
use rand_chacha::ChaCha8Rng;
use rand_seeder::Seeder;
use std::path::PathBuf;

#[derive(Parser)]
#[command(author, version, about, long_about = None)]
struct Cli {
    /// The width of the map.
    #[arg(long, default_value_t = 1000)]
    width: usize,
    /// The height of the map.
    #[arg(long, default_value_t = 1000)]
    height: usize,
    /// The seed used when generating a map. By default a random one.
    #[arg(short, long)]
    seed: Option<String>,
    /// The mapres directory.
    #[arg(short, long, default_value = "mapres")]
    mapres: PathBuf,
    /// The output map file.
    #[arg(short, long, default_value = "generated.map")]
    output: PathBuf,
    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    /// Generate a map for fly techniques.
    Fly {
        #[arg(long, default_value_t = 12)]
        max_fly_width: u16,
        /// The output map file.
        #[arg(long, default_value_t = 3)]
        min_fly_width: u16,
    },
    /// Generate a maze-like map.
    Maze,
}

impl Commands {
    pub fn print(&self) {
        print!("Selected generator: ");
        let data = match self {
            Self::Fly {
                max_fly_width,
                min_fly_width,
            } => format!(
                "{}, parameters: max_fly_width={}, min_fly_width={}",
                "Fly".purple().bold(),
                max_fly_width.purple(),
                min_fly_width.purple()
            ),
            Self::Maze => format!("{}", "Maze".purple().bold()),
        };
        println!("{data}");
    }
}

pub fn run_cli() -> Result<()> {
    let cli = Cli::parse();

    let seed: String = {
        if let Some(x) = &cli.seed {
            x.clone()
        } else {
            thread_rng()
                .sample_iter(&Alphanumeric)
                .take(8)
                .map(char::from)
                .collect()
        }
    };

    println!("Using seed: {}", seed.green().bold());

    let mut rng: ChaCha8Rng = Seeder::from(seed).make_rng();

    cli.command.print();

    match cli.command {
        Commands::Maze => {
            MazeGenerator.save_file(&mut rng, &cli.mapres, cli.width, cli.height, &cli.output)
        }
        Commands::Fly {
            max_fly_width,
            min_fly_width,
        } => FlyGenerator {
            max_fly_width,
            min_fly_width,
        }
        .save_file(&mut rng, &cli.mapres, cli.width, cli.height, &cli.output),
    }
}