summaryrefslogtreecommitdiff
path: root/src/init.rs
blob: d051fc421251fdacc694c7e7e0ca964d0cc9244f (plain)
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
100
101
102
103
104
105
106
107
108
109
110
use anyhow::{bail, Context, Result};
use std::{
    env::set_current_dir,
    fs::{create_dir, OpenOptions},
    io::{self, ErrorKind, Write},
    path::Path,
};

use crate::{embedded::EMBEDDED_FILES, info_file::ExerciseInfo, trust::trust_current_dir};

fn create_cargo_toml(exercise_infos: &[ExerciseInfo]) -> io::Result<()> {
    let mut cargo_toml = Vec::with_capacity(1 << 13);
    cargo_toml.extend_from_slice(b"bin = [\n");
    for exercise_info in exercise_infos {
        cargo_toml.extend_from_slice(b"  { name = \"");
        cargo_toml.extend_from_slice(exercise_info.name.as_bytes());
        cargo_toml.extend_from_slice(b"\", path = \"exercises/");
        if let Some(dir) = &exercise_info.dir {
            cargo_toml.extend_from_slice(dir.as_bytes());
            cargo_toml.push(b'/');
        }
        cargo_toml.extend_from_slice(exercise_info.name.as_bytes());
        cargo_toml.extend_from_slice(b".rs\" },\n");
    }

    cargo_toml.extend_from_slice(
        br#"]

[package]
name = "rustlings"
edition = "2021"
publish = false
"#,
    );
    OpenOptions::new()
        .create_new(true)
        .write(true)
        .open("Cargo.toml")?
        .write_all(&cargo_toml)
}

fn create_gitignore() -> io::Result<()> {
    OpenOptions::new()
        .create_new(true)
        .write(true)
        .open(".gitignore")?
        .write_all(GITIGNORE)
}

fn create_vscode_dir() -> Result<()> {
    create_dir(".vscode").context("Failed to create the directory `.vscode`")?;
    OpenOptions::new()
        .create_new(true)
        .write(true)
        .open(".vscode/extensions.json")?
        .write_all(VS_CODE_EXTENSIONS_JSON)?;

    Ok(())
}

pub fn init(exercise_infos: &[ExerciseInfo]) -> Result<()> {
    if Path::new("exercises").is_dir() && Path::new("Cargo.toml").is_file() {
        bail!(PROBABLY_IN_RUSTLINGS_DIR_ERR);
    }

    let rustlings_path = Path::new("rustlings");
    if let Err(e) = create_dir(rustlings_path) {
        if e.kind() == ErrorKind::AlreadyExists {
            bail!(RUSTLINGS_DIR_ALREADY_EXISTS_ERR);
        }
        return Err(e.into());
    }

    set_current_dir("rustlings")
        .context("Failed to change the current directory to `rustlings`")?;

    EMBEDDED_FILES
        .init_exercises_dir()
        .context("Failed to initialize the `rustlings/exercises` directory")?;

    create_cargo_toml(exercise_infos)
        .context("Failed to create the file `rustlings/Cargo.toml`")?;

    create_gitignore().context("Failed to create the file `rustlings/.gitignore`")?;

    create_vscode_dir().context("Failed to create the file `rustlings/.vscode/extensions.json`")?;

    trust_current_dir()?;

    Ok(())
}

const GITIGNORE: &[u8] = b"/target
/.rustlings-state.txt
";

const VS_CODE_EXTENSIONS_JSON: &[u8] = br#"{"recommendations":["rust-lang.rust-analyzer"]}"#;

const PROBABLY_IN_RUSTLINGS_DIR_ERR: &str =
    "A directory with the name `exercises` and a file with the name `Cargo.toml` already exist
in the current directory. It looks like Rustlings was already initialized here.
Run `rustlings` for instructions on getting started with the exercises.

If you didn't already initialize Rustlings, please initialize it in another directory.";

const RUSTLINGS_DIR_ALREADY_EXISTS_ERR: &str =
    "A directory with the name `rustlings` already exists in the current directory.
You probably already initialized Rustlings.
Run `cd rustlings`
Then run `rustlings` again";