vesti 0.4.1

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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

// Copyright (c) 2022 Sungbae Jeong
//
// This software is released under the MIT License.
// https://opensource.org/licenses/MIT

use std::fs;
use std::path::{Path, PathBuf};
use std::sync::atomic::{AtomicUsize, Ordering};
use std::sync::{Arc, RwLock};
use std::thread;
use std::time::{Duration, SystemTime};

use crate::codegen::make_latex_format;
use crate::error::{self, pretty_print::pretty_print};
use crate::exit_status::ExitCode;
use crate::lexer::Lexer;
use crate::parser::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};

#[cfg(target_os = "windows")]
pub const SIGNALS: [i32; 3] = [SIGINT, SIGTERM, SIGILL];
#[cfg(not(target_os = "windows"))]
pub const SIGNALS: [i32; 2] = [SIGINT, SIGTERM];

macro_rules! unwrap_err {
    ($name: ident := $to_unwrap: expr, $source: expr, $file_name: expr, $is_loop_end: expr) => {
        let $name = match $to_unwrap {
            Ok(inner) => inner,
            Err(err) => {
                pretty_print($source, err, $file_name).unwrap();
                let mut writer = $is_loop_end.write().unwrap();
                *writer = true;
                return ExitCode::Failure;
            }
        };
    };
    (mut $name: ident := $to_unwrap: expr, $source: expr, $file_name: expr, $is_loop_end: expr) => {
        let mut $name = match $to_unwrap {
            Ok(inner) => inner,
            Err(err) => {
                pretty_print($source, err, $file_name).unwrap();
                let mut writer = $is_loop_end.write().unwrap();
                *writer = true;
                return ExitCode::Failure;
            }
        };
    };
    ($name: ident = $to_unwrap: expr, $source: expr, $file_name: expr, $is_loop_end: expr) => {
        $name = match $to_unwrap {
            Ok(inner) => inner,
            Err(err) => {
                pretty_print($source, err, $file_name).unwrap();
                let mut writer = $is_loop_end.write().unwrap();
                *writer = true;
                return ExitCode::Failure;
            }
        };
    };
}

fn output_file_name(file_name: &Path) -> PathBuf {
    file_name.with_extension("tex")
}

fn take_time(file_name: &Path) -> error::Result<SystemTime> {
    let path = file_name;
    Ok(path.metadata()?.modified()?)
}

pub fn compile_vesti(
    trap: Arc<AtomicUsize>,
    file_name: PathBuf,
    is_continuous: bool,
    is_loop_end: Arc<RwLock<bool>>,
) -> ExitCode {
    let mut init_compile = true;
    let output = output_file_name(&file_name);
    unwrap_err!(mut init_time := take_time(&file_name), None, None, is_loop_end);
    let mut now_time = init_time;

    while !SIGNALS.contains(&(trap.load(Ordering::Relaxed) as i32)) {
        if {
            let reader = is_loop_end.read().unwrap();
            *reader
        } {
            return ExitCode::Failure;
        }
        if init_compile || init_time != now_time {
            let source = fs::read_to_string(&file_name).expect("Opening file error occurred!");
            let mut parser = Parser::new(Lexer::new(&source));
            unwrap_err!(contents := make_latex_format::<false>(&mut parser), Some(source.as_ref()), Some(&file_name), is_loop_end);
            drop(parser);

            fs::write(&output, contents).expect("File write failed.");

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

            init_compile = false;
            init_time = now_time;
        }
        unwrap_err!(now_time = take_time(&file_name), None, None, is_loop_end);
        thread::sleep(Duration::from_millis(500));
    }

    ExitCode::Success
}