epita/epitls-pi
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

formatted

Browse source
This commit is contained in:
matthieu.jolimaitre 2022-09-25 10:30:30 +02:00
parent 37e5b5e555
commit 7f694cd6c2
9 changed files with 333 additions and 332 deletions
Download patch file Download diff file Expand all files Collapse all files

156
src/tasks.rs
View file

@ -1,109 +1,109 @@
use std::{
fs,
path::PathBuf,
process::{Command, ExitStatus},
thread,
time::Duration,
fs,
path::PathBuf,
process::{Command, ExitStatus},
thread,
time::Duration,
};
use crate::utils::{
log_command_run, log_separator_bottom, log_separator_top, tmp_file_path, Apply,
log_command_run, log_separator_bottom, log_separator_top, tmp_file_path, Apply,
};
pub struct CompileTask {
files: Vec<PathBuf>,
addition: Vec<String>,
flags: Vec<String>,
files: Vec<PathBuf>,
addition: Vec<String>,
flags: Vec<String>,
}
// TODO: split compile & compile raw
impl CompileTask {
pub fn new(files: Vec<PathBuf>) -> Self {
Self {
files,
addition: vec![],
flags: vec![],
}
}
pub fn new(files: Vec<PathBuf>) -> Self {
Self {
files,
addition: vec![],
flags: vec![],
}
}
pub fn with_addition(mut self, code: impl ToString) -> Self {
self.addition.push(code.to_string());
self
}
pub fn with_addition(mut self, code: impl ToString) -> Self {
self.addition.push(code.to_string());
self
}
pub fn with_flag(mut self, flag: impl ToString) -> Self {
self.flags.push(flag.to_string());
self
}
pub fn with_flag(mut self, flag: impl ToString) -> Self {
self.flags.push(flag.to_string());
self
}
pub fn run(self) -> Result<PathBuf, ExitStatus> {
let proc_source = self.gen_source();
let mut sources = self.files.clone();
sources.push(proc_source);
self.compile(sources)
}
pub fn run(self) -> Result<PathBuf, ExitStatus> {
let proc_source = self.gen_source();
let mut sources = self.files.clone();
sources.push(proc_source);
self.compile(sources)
}
pub fn gen_source(&self) -> PathBuf {
let mut output_path = tmp_file_path();
// TODO: make use of supplement
output_path.set_extension("c");
fs::write(&output_path, "").unwrap();
output_path
}
pub fn gen_source(&self) -> PathBuf {
let mut output_path = tmp_file_path();
// TODO: make use of supplement
output_path.set_extension("c");
fs::write(&output_path, "").unwrap();
output_path
}
pub fn compile(&self, sources: Vec<PathBuf>) -> Result<PathBuf, ExitStatus> {
let output_path = tmp_file_path().apply(|o| o.set_extension("b"));
let output_path_ref = output_path.to_str().unwrap();
let mut command = Command::new("gcc");
command
.args(["-o", output_path_ref])
.args(self.flags.clone())
.args(sources.iter().map(|s| s.to_str().unwrap()));
log_command_run(&command);
log_separator_top();
let status = command.status().unwrap();
log_separator_bottom();
thread::sleep(Duration::from_millis(100));
status.success().then_some(output_path).ok_or(status)
}
pub fn compile(&self, sources: Vec<PathBuf>) -> Result<PathBuf, ExitStatus> {
let output_path = tmp_file_path().apply(|o| o.set_extension("b"));
let output_path_ref = output_path.to_str().unwrap();
let mut command = Command::new("gcc");
command
.args(["-o", output_path_ref])
.args(self.flags.clone())
.args(sources.iter().map(|s| s.to_str().unwrap()));
log_command_run(&command);
log_separator_top();
let status = command.status().unwrap();
log_separator_bottom();
thread::sleep(Duration::from_millis(100));
status.success().then_some(output_path).ok_or(status)
}
}
pub struct RunTask {
file: PathBuf,
file: PathBuf,
}
impl RunTask {
pub fn new(file: PathBuf) -> Self {
Self { file }
}
pub fn new(file: PathBuf) -> Self {
Self { file }
}
pub fn run(self) -> Result<(), ExitStatus> {
let mut command = Command::new(self.file);
log_command_run(&command);
log_separator_top();
let status = command.status().unwrap();
log_separator_bottom();
if status.success() {
Ok(())
} else {
Err(status)
}
}
pub fn run(self) -> Result<(), ExitStatus> {
let mut command = Command::new(self.file);
log_command_run(&command);
log_separator_top();
let status = command.status().unwrap();
log_separator_bottom();
if status.success() {
Ok(())
} else {
Err(status)
}
}
}
pub struct GenTask {
content: String,
content: String,
}
impl GenTask {
pub fn new(content: String) -> Self {
Self { content }
}
pub fn run(self) -> PathBuf {
let output_path = tmp_file_path().apply(|o| o.set_extension("c"));
fs::write(&output_path, &self.content).unwrap();
dbg!(fs::read_to_string(&output_path).unwrap());
output_path
}
pub fn new(content: String) -> Self {
Self { content }
}
pub fn run(self) -> PathBuf {
let output_path = tmp_file_path().apply(|o| o.set_extension("c"));
fs::write(&output_path, &self.content).unwrap();
dbg!(fs::read_to_string(&output_path).unwrap());
output_path
}
}