forked from Anri/cal8tor
Mylloon
64911ba351
- ask for first period day of the current one instead of the first one - holidays are not mandatory anymore - also add an option to force the weeks number
109 lines
2.8 KiB
Rust
109 lines
2.8 KiB
Rust
use clap::Parser;
|
|
use dialoguer::Input;
|
|
use regex::Regex;
|
|
|
|
mod filter;
|
|
mod ics;
|
|
mod info;
|
|
mod timetable;
|
|
mod utils;
|
|
|
|
#[allow(clippy::struct_excessive_bools)]
|
|
#[derive(Parser)]
|
|
#[clap(version, about, long_about = None)]
|
|
struct Args {
|
|
/// The class you want to get the timetable, i.e.: M1
|
|
#[clap(value_parser)]
|
|
class: String,
|
|
|
|
/// The semester you want (1 or 2), default to current semester
|
|
#[clap(short, long, value_parser, value_name = "SEMESTER NUMBER")]
|
|
semester: Option<i8>,
|
|
|
|
/// The year, default to current year
|
|
#[clap(short, long, value_parser, value_name = "YEAR")]
|
|
year: Option<i32>,
|
|
|
|
/// Export to iCalendar format (.ics)
|
|
#[clap(short, long, value_name = "FILE NAME")]
|
|
export: Option<String>,
|
|
|
|
/// Doesn't distinguish TD from TP
|
|
#[clap(short, long)]
|
|
td_are_tp: bool,
|
|
|
|
/// Number of weeks for the period
|
|
#[clap(short, long, value_parser)]
|
|
weeks: Option<i32>,
|
|
|
|
/// First day of your year
|
|
#[clap(long)]
|
|
first_day: Option<String>,
|
|
|
|
/// If TD/TP start a week after courses
|
|
#[clap(long)]
|
|
week_skip: bool,
|
|
|
|
/// If the exported ICS file should not use the timezone
|
|
#[clap(long)]
|
|
no_tz: bool,
|
|
|
|
/// Guess the holidays and apply it to the generated calendar
|
|
#[clap(long)]
|
|
holidays: bool,
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let args = Args::parse();
|
|
|
|
let matches = Regex::new(r"(?i)M(?P<level>[1,2])")
|
|
.unwrap()
|
|
.captures(&args.class)
|
|
.unwrap();
|
|
|
|
let level = matches
|
|
.name("level")
|
|
.unwrap()
|
|
.as_str()
|
|
.parse::<i8>()
|
|
.unwrap();
|
|
|
|
let user_agent = format!("cal7tor/{}", env!("CARGO_PKG_VERSION"));
|
|
|
|
println!("Récupération de l'emploi du temps des M{level}...");
|
|
let mut timetable = timetable::timetable(level, args.semester, args.year, &user_agent).await;
|
|
|
|
timetable = filter::timetable(timetable, args.td_are_tp);
|
|
|
|
let date = match args.first_day {
|
|
None => Input::new()
|
|
.with_prompt("Début des cours de la période")
|
|
.default(info::get_start_date(level, args.semester, args.year, &user_agent).await)
|
|
.interact_text()
|
|
.unwrap(),
|
|
Some(day) => day,
|
|
};
|
|
|
|
println!("Récupération des informations par rapport à l'année...");
|
|
let info = info::info(
|
|
args.semester,
|
|
args.year,
|
|
&date,
|
|
args.week_skip,
|
|
args.holidays,
|
|
args.weeks,
|
|
);
|
|
|
|
if let Some(mut filename) = args.export {
|
|
// Export the calendar
|
|
let builded_timetable = timetable::build(&timetable, &info);
|
|
ics::export(builded_timetable, &mut filename, !args.no_tz);
|
|
|
|
println!("Fichier .ICS construit et exporté => {filename}");
|
|
} else {
|
|
// Show the calendar
|
|
println!("Affichage...");
|
|
timetable::display(&timetable);
|
|
}
|
|
}
|