vesti 0.2.9

A preprocessor that compiles into LaTeX
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

#![allow(clippy::enum_variant_names)]
#![allow(clippy::derive_partial_eq_without_eq)]
#![deny(bindings_with_variant_name)]

mod codegen;
mod commands;
mod error;
mod exit_status;
mod initialization;
mod lexer;
mod location;
mod parser;

use std::path::PathBuf;
use std::sync::atomic::AtomicUsize;
use std::sync::{Arc, RwLock};
use std::thread::{self, JoinHandle};

use clap::Parser;

#[cfg(target_os = "windows")]
use signal_hook::consts::signal::{SIGILL, SIGINT, SIGTERM};
#[cfg(not(target_os = "windows"))]
use signal_hook::consts::signal::{SIGINT, SIGTERM};
use signal_hook::flag as signal_flag;

use crate::commands::{compile_vesti, VestiOpt};
use crate::error::pretty_print::pretty_print;
use crate::exit_status::ExitCode;
use crate::initialization::generate_vesti_file;

fn main() -> ExitCode {
    let args = commands::VestiOpt::parse();
    let is_loop_end = Arc::new(RwLock::new(false));

    if let VestiOpt::Init { project_name } = args {
        let project_name = if let Some(project_name) = project_name {
            PathBuf::from(project_name)
        } else {
            const ERR_MESSAGE: &str = "cannot get the current directory";
            let tmp = std::env::current_dir().expect(ERR_MESSAGE);
            PathBuf::from(tmp.file_name().expect(ERR_MESSAGE))
        };
        return match generate_vesti_file(project_name) {
            Ok(()) => ExitCode::Success,
            Err(err) => {
                println!("{}", pretty_print(None, err, None));
                ExitCode::Failure
            }
        };
    } else {
        let is_continuous = args.is_continuous_compile();

        let trap = Arc::new(AtomicUsize::new(0));
        // TODO: I do not test this code in windows actually :)
        #[cfg(target_os = "windows")]
        for signal in [SIGINT, SIGTERM, SIGILL].iter() {
            signal_flag::register_usize(*signal, Arc::clone(&trap), *signal as usize)
                .expect("Undefined behavior happened!");
        }
        #[cfg(not(target_os = "windows"))]
        for signal in [SIGINT, SIGTERM].iter() {
            signal_flag::register_usize(*signal, Arc::clone(&trap), *signal as usize)
                .expect("Undefined behavior happened!");
        }

        let file_lists = match args.take_file_name() {
            Ok(inner) => inner,
            Err(err) => {
                println!("{}", pretty_print(None, err, None));
                return ExitCode::Failure;
            }
        };

        let mut handle_vesti: Vec<JoinHandle<_>> = Vec::new();
        for file_name in file_lists {
            let cloned_trap = Arc::clone(&trap);
            let cloned_bool = Arc::clone(&is_loop_end);
            handle_vesti.push(thread::spawn(move || {
                return compile_vesti(cloned_trap, file_name, is_continuous, cloned_bool);
            }));
        }

        if is_continuous {
            println!("Press Ctrl+C to finish the program.");
        }

        for vesti in handle_vesti.into_iter() {
            if vesti.join().unwrap() == ExitCode::Failure {
                return ExitCode::Failure;
            }
        }

        println!("bye!");
    }

    ExitCode::Success
}