initial commit
This commit is contained in:
commit
1b87769456
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
@ -0,0 +1 @@
|
|||||||
|
/target
|
14
.vscode/tasks.json
vendored
Normal file
14
.vscode/tasks.json
vendored
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
{
|
||||||
|
"version": "2.0.0",
|
||||||
|
"tasks": [
|
||||||
|
{
|
||||||
|
"type": "cargo",
|
||||||
|
"command": "run",
|
||||||
|
"problemMatcher": [
|
||||||
|
"$rustc"
|
||||||
|
],
|
||||||
|
"group": "build",
|
||||||
|
"label": "rust: cargo run"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
2985
Cargo.lock
generated
Normal file
2985
Cargo.lock
generated
Normal file
File diff suppressed because it is too large
Load Diff
19
Cargo.toml
Normal file
19
Cargo.toml
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
[package]
|
||||||
|
name = "ktu-timetable"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2021"
|
||||||
|
|
||||||
|
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
egui = "0.20.1"
|
||||||
|
eframe = "0.20.1"
|
||||||
|
ureq = { version = "2.6.1", features = ["native-tls", "cookies"]}
|
||||||
|
native-tls = "0.2.11"
|
||||||
|
ical = "0.7.*"
|
||||||
|
chrono = "0.4.23"
|
||||||
|
image = "0.24.5"
|
||||||
|
lazy-regex = "2.4.1"
|
||||||
|
directories-next = "2.0.0"
|
||||||
|
toml = "0.5.11"
|
||||||
|
serde = { version = "1.0.152", features = ["derive"]}
|
BIN
assets/break-area.png
Normal file
BIN
assets/break-area.png
Normal file
Binary file not shown.
After Width: | Height: | Size: 152 B |
BIN
assets/icon.ico
Normal file
BIN
assets/icon.ico
Normal file
Binary file not shown.
After Width: | Height: | Size: 4.2 KiB |
BIN
assets/icon.png
Normal file
BIN
assets/icon.png
Normal file
Binary file not shown.
After Width: | Height: | Size: 1.3 KiB |
476
src/app.rs
Normal file
476
src/app.rs
Normal file
@ -0,0 +1,476 @@
|
|||||||
|
use std::ops::Add;
|
||||||
|
|
||||||
|
use eframe::{egui, CreationContext, epaint::text::TextWrapping};
|
||||||
|
use chrono::{Datelike, Timelike, Utc, NaiveDate, Weekday, IsoWeek, Duration, DateTime, NaiveDateTime, Days, Local};
|
||||||
|
use egui::{Color32, ColorImage, TextureHandle, TextureOptions, Rect, text::LayoutJob, Visuals, Stroke};
|
||||||
|
use crate::{timetable::{Timetable, get_timetable, Event, EventCategory}, config::ConfigStorage};
|
||||||
|
|
||||||
|
struct EventsTableStyle {
|
||||||
|
highlight_color: Color32,
|
||||||
|
bg_fill: Color32,
|
||||||
|
fg_stroke: egui::Stroke
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
let highlight_color = style.visuals.selection.bg_fill;
|
||||||
|
let bg_fill = style.visuals.widgets.noninteractive.bg_fill; //Color32::from_rgb(18, 18, 18);
|
||||||
|
let fg_stroke = style.visuals.widgets.active.fg_stroke;
|
||||||
|
|
||||||
|
let light_bg_col = highlight_color; // shift_color(bg_col, 1.5); // Color32::from_rgb(30, 30, 30);
|
||||||
|
let dark_bg_col = shift_color(bg_fill, 0.5); // style.visuals.widgets.noninteractive.bg_stroke.color; Color32::from_rgb(11, 8, 8);
|
||||||
|
//let fg_col = Color32::from_rgb(252, 232, 195);
|
||||||
|
let now_line_col = shift_color(highlight_color, 1.5); // fg_stroke.color; // shift_color(highlight_color, 1.5); // Color32::from_rgb(44, 120, 191);
|
||||||
|
*/
|
||||||
|
impl EventsTableStyle {
|
||||||
|
fn from_visuals(visuals: &Visuals) -> Self {
|
||||||
|
let bg_fill = visuals.widgets.noninteractive.bg_fill;
|
||||||
|
Self {
|
||||||
|
highlight_color: visuals.selection.bg_fill,
|
||||||
|
bg_fill,
|
||||||
|
fg_stroke: visuals.widgets.active.fg_stroke,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn dark_bg_fill(&self) -> Color32 {
|
||||||
|
shift_color(self.bg_fill, 0.5)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn now_line_fill(&self) -> Color32 {
|
||||||
|
shift_color(self.highlight_color, 1.5)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct AppAssets {
|
||||||
|
break_texture: egui::TextureHandle
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct MainApp {
|
||||||
|
pub vidko: Option<String>,
|
||||||
|
pub timetable: Option<Timetable>,
|
||||||
|
shown_week: IsoWeek,
|
||||||
|
shown_events: Vec<Event>,
|
||||||
|
|
||||||
|
assets: Option<AppAssets>,
|
||||||
|
storage: ConfigStorage
|
||||||
|
}
|
||||||
|
|
||||||
|
fn count_minutes(time: &str) -> u32 {
|
||||||
|
let (time_h, time_m) = time.split_once(":").unwrap();
|
||||||
|
return 60*time_h.parse::<u32>().unwrap() + time_m.parse::<u32>().unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_category_bg(category: EventCategory) -> Color32 {
|
||||||
|
match category {
|
||||||
|
EventCategory::Default => Color32::GRAY,
|
||||||
|
EventCategory::Yellow => Color32::from_rgb(251, 184, 41),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn is_bright_color(color: Color32) -> bool {
|
||||||
|
return color.r() + color.g() + color.b() > 128 * 3
|
||||||
|
}
|
||||||
|
|
||||||
|
fn load_image_from_memory(image_data: &[u8]) -> Result<ColorImage, image::ImageError> {
|
||||||
|
let image = image::load_from_memory(image_data)?;
|
||||||
|
let size = [image.width() as _, image.height() as _];
|
||||||
|
let image_buffer = image.to_rgba8();
|
||||||
|
let pixels = image_buffer.as_flat_samples();
|
||||||
|
Ok(ColorImage::from_rgba_unmultiplied(
|
||||||
|
size,
|
||||||
|
pixels.as_slice(),
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn draw_repeating_texture(ui: &mut egui::Ui, texture: &TextureHandle, rect: Rect, tint: Color32) {
|
||||||
|
use egui::*;
|
||||||
|
let painter = ui.painter();
|
||||||
|
|
||||||
|
let texture_size = texture.size_vec2();
|
||||||
|
let horizontal_count = rect.width()/texture_size[0];
|
||||||
|
let vertical_count = rect.height()/texture_size[1];
|
||||||
|
|
||||||
|
let draw_tile = |ix: u32, iy: u32, scale: Vec2| {
|
||||||
|
painter.image(
|
||||||
|
texture.id(),
|
||||||
|
Rect::from_min_size(
|
||||||
|
rect.left_top() + texture_size * vec2(ix as f32, iy as f32),
|
||||||
|
texture_size * scale
|
||||||
|
),
|
||||||
|
Rect::from_min_max(pos2(0.0, 0.0), scale.to_pos2()),
|
||||||
|
tint
|
||||||
|
);
|
||||||
|
};
|
||||||
|
|
||||||
|
{
|
||||||
|
let scale = vec2(1.0, 1.0);
|
||||||
|
for ix in 0..horizontal_count.floor() as u32 {
|
||||||
|
for iy in 0..vertical_count.floor() as u32 {
|
||||||
|
draw_tile(ix, iy, scale);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// right edge
|
||||||
|
{
|
||||||
|
let scale = vec2(horizontal_count % 1.0, 1.0);
|
||||||
|
let ix = horizontal_count.floor() as u32;
|
||||||
|
for iy in 0..vertical_count.floor() as u32 {
|
||||||
|
draw_tile(ix, iy, scale);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// bottom edge
|
||||||
|
{
|
||||||
|
let scale = vec2(1.0, vertical_count % 1.0);
|
||||||
|
let iy = vertical_count.floor() as u32;
|
||||||
|
for ix in 0..horizontal_count.floor() as u32 {
|
||||||
|
draw_tile(ix, iy, scale);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// left bottom corner
|
||||||
|
{
|
||||||
|
let scale = vec2(horizontal_count % 1.0, vertical_count % 1.0);
|
||||||
|
draw_tile(horizontal_count.floor() as u32, vertical_count.floor() as u32, scale);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn show_event_card(ui: &mut egui::Ui, event: &Event, mut rect: Rect) {
|
||||||
|
use egui::*;
|
||||||
|
let margin = 6.0;
|
||||||
|
let border_size = 4.0;
|
||||||
|
let text_size = egui::TextStyle::Body.resolve(ui.style()).size;
|
||||||
|
let text_color = Color32::BLACK;
|
||||||
|
|
||||||
|
rect.set_width(rect.width().max(text_size*6.0));
|
||||||
|
|
||||||
|
let painter = ui.painter();
|
||||||
|
let color = get_category_bg(event.category);
|
||||||
|
let rounding = Rounding::from(5.0);
|
||||||
|
let border_color = color.linear_multiply(1.25);
|
||||||
|
painter.rect_filled(rect, rounding, color);
|
||||||
|
painter.rect_stroke(rect.shrink(border_size/2.0), rounding, (border_size, border_color));
|
||||||
|
|
||||||
|
ui.allocate_ui_at_rect(rect.shrink(margin), |ui| {
|
||||||
|
let font = FontId::proportional(text_size * 0.8);
|
||||||
|
let summary_format = TextFormat {
|
||||||
|
color: text_color,
|
||||||
|
font_id: font.clone(),
|
||||||
|
..TextFormat::default()
|
||||||
|
};
|
||||||
|
let module_name = event.module_name.as_ref().unwrap_or(&event.summary);
|
||||||
|
let mut job = LayoutJob::single_section(module_name.to_string(), summary_format);
|
||||||
|
job.wrap = TextWrapping {
|
||||||
|
max_rows: 2,
|
||||||
|
..Default::default()
|
||||||
|
};
|
||||||
|
ui.label(job);
|
||||||
|
|
||||||
|
ui.add_space(text_size*0.2);
|
||||||
|
let time_label = format!("{} - {}", event.start_time.format("%H:%M"), event.end_time.format("%H:%M"));
|
||||||
|
ui.label(RichText::new(time_label).color(text_color).font(font));
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn is_weekend(time: NaiveDate) -> bool {
|
||||||
|
let day = time.weekday();
|
||||||
|
return day == Weekday::Sat || day == Weekday::Sun;
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn shift_color(color: Color32, amount: f32) -> Color32 {
|
||||||
|
return Color32::from_rgb(
|
||||||
|
(color.r() as f32 * amount) as u8,
|
||||||
|
(color.g() as f32 * amount) as u8,
|
||||||
|
(color.b() as f32 * amount) as u8
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_current_week() -> IsoWeek {
|
||||||
|
let now = Local::now();
|
||||||
|
if is_weekend(now.date_naive()) {
|
||||||
|
now.add(Duration::days(7)).iso_week()
|
||||||
|
} else {
|
||||||
|
now.iso_week()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn show_events_table_header(
|
||||||
|
ui: &mut egui::Ui,
|
||||||
|
style: &EventsTableStyle,
|
||||||
|
rect: Rect,
|
||||||
|
week: IsoWeek
|
||||||
|
) {
|
||||||
|
use egui::*;
|
||||||
|
let painter = ui.painter();
|
||||||
|
painter.rect_filled(rect, Rounding::none(), style.dark_bg_fill());
|
||||||
|
|
||||||
|
let column_width = rect.width()/5.0;
|
||||||
|
let header_size = rect.height();
|
||||||
|
let text_size = egui::TextStyle::Body.resolve(ui.style()).size;
|
||||||
|
|
||||||
|
// Draw day names
|
||||||
|
for (i, name) in ["Pir", "Ant", "Tre", "Ket", "Pen"].iter().enumerate() {
|
||||||
|
let offset = column_width * (i as f32 + 0.5);
|
||||||
|
|
||||||
|
painter.text(
|
||||||
|
rect.left_top() + vec2(offset, header_size/2.5),
|
||||||
|
Align2::CENTER_CENTER,
|
||||||
|
name,
|
||||||
|
FontId::monospace(text_size*1.2),
|
||||||
|
style.fg_stroke.color
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Draw dates
|
||||||
|
let year = week.year();
|
||||||
|
let week = week.week();
|
||||||
|
let mut week_date = NaiveDate::from_isoywd_opt(year, week, Weekday::Mon).expect("Invalid week or year given");
|
||||||
|
for i in 1..=5 {
|
||||||
|
let offset = column_width * i as f32;
|
||||||
|
|
||||||
|
painter.text(
|
||||||
|
rect.left_top() + vec2(offset-3.0, header_size-3.0),
|
||||||
|
Align2::RIGHT_BOTTOM,
|
||||||
|
week_date.format("%m-%d").to_string(),
|
||||||
|
FontId::proportional(text_size*0.85),
|
||||||
|
style.fg_stroke.color
|
||||||
|
);
|
||||||
|
week_date = week_date.add(Duration::days(1));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[inline]
|
||||||
|
fn show_events_table_body(
|
||||||
|
ui: &mut egui::Ui,
|
||||||
|
style: &EventsTableStyle,
|
||||||
|
assets: &AppAssets,
|
||||||
|
rect: Rect,
|
||||||
|
week: IsoWeek,
|
||||||
|
now: NaiveDateTime,
|
||||||
|
events: &[Event]
|
||||||
|
) {
|
||||||
|
use egui::*;
|
||||||
|
|
||||||
|
let painter = ui.painter();
|
||||||
|
let column_width = rect.width()/5.0;
|
||||||
|
let column_gap = 3.0;
|
||||||
|
|
||||||
|
let timestamps = ["9:00", "10:30", "11:00", "12:30", "13:30", "15:00", "15:30", "17:00"];
|
||||||
|
let timestamps_mins = timestamps.map(count_minutes);
|
||||||
|
let total_minutes = timestamps_mins.last().unwrap() - timestamps_mins.first().unwrap();
|
||||||
|
let minute_to_pixel_scale = rect.height()/total_minutes as f32;
|
||||||
|
|
||||||
|
// draw bg
|
||||||
|
painter.rect_filled(
|
||||||
|
rect,
|
||||||
|
Rounding::none(),
|
||||||
|
style.bg_fill
|
||||||
|
);
|
||||||
|
|
||||||
|
// Highlight current day column
|
||||||
|
if now.iso_week() == week && !is_weekend(now.date()) {
|
||||||
|
let days_from_monday = now.weekday().num_days_from_monday() as f32;
|
||||||
|
let rect = Rect::from_min_max(
|
||||||
|
rect.left_top() + vec2(column_width * days_from_monday, 0.0),
|
||||||
|
rect.left_bottom() + vec2(column_width * (1.0+days_from_monday), 0.0)
|
||||||
|
);
|
||||||
|
painter.rect_filled(rect, Rounding::none(), style.highlight_color);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Draw gaps between columns
|
||||||
|
for i in 1..5 {
|
||||||
|
let offset = column_width * i as f32;
|
||||||
|
painter.line_segment([
|
||||||
|
rect.left_top() + vec2(offset, 0.0),
|
||||||
|
rect.left_bottom() + vec2(offset, 0.0)
|
||||||
|
], (column_gap, style.dark_bg_fill()))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Mark break times
|
||||||
|
for i in (1..timestamps_mins.len()-1).step_by(2) {
|
||||||
|
let from = (timestamps_mins[i] - timestamps_mins[0]) as f32 * minute_to_pixel_scale;
|
||||||
|
let to = (timestamps_mins[i+1] - timestamps_mins[0]) as f32 * minute_to_pixel_scale;
|
||||||
|
draw_repeating_texture(
|
||||||
|
ui,
|
||||||
|
&assets.break_texture,
|
||||||
|
Rect::from_min_size(
|
||||||
|
rect.left_top() + vec2(0.0, from),
|
||||||
|
vec2(rect.width(), to - from)
|
||||||
|
),
|
||||||
|
style.dark_bg_fill()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Draw event cards
|
||||||
|
for event in events {
|
||||||
|
let day = event.date.weekday().num_days_from_monday() as usize;
|
||||||
|
let duration = (event.end_time - event.start_time).num_minutes() as f32;
|
||||||
|
let start_time = event.start_time.hour()*60 + event.start_time.minute() - timestamps_mins[0];
|
||||||
|
let event_rect = Rect::from_min_size(
|
||||||
|
rect.left_top() + vec2(column_width*day as f32, start_time as f32*minute_to_pixel_scale),
|
||||||
|
vec2(column_width, duration*minute_to_pixel_scale)
|
||||||
|
).shrink2(vec2(10.0, 0.0));
|
||||||
|
show_event_card(ui, event, event_rect);
|
||||||
|
}
|
||||||
|
|
||||||
|
// now line
|
||||||
|
let painter = ui.painter();
|
||||||
|
let current_time = now.minute() + now.hour() * 60 - timestamps_mins[0];
|
||||||
|
if current_time > 0 && current_time < *timestamps_mins.last().unwrap_or(&0) && !is_weekend(now.date()) {
|
||||||
|
let offset = current_time as f32 * minute_to_pixel_scale;
|
||||||
|
let points = [
|
||||||
|
rect.left_top() + vec2(0.0, offset),
|
||||||
|
rect.right_top() + vec2(0.0, offset)
|
||||||
|
];
|
||||||
|
let thickness = 2.0;
|
||||||
|
let border_size = 2.0;
|
||||||
|
painter.line_segment(points, (thickness + 2.0 * border_size, style.dark_bg_fill()));
|
||||||
|
painter.line_segment(points, (thickness, style.highlight_color));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn show_events_table(
|
||||||
|
assets: &AppAssets,
|
||||||
|
ui: &mut egui::Ui,
|
||||||
|
rect: Rect,
|
||||||
|
shown_week: IsoWeek,
|
||||||
|
now: NaiveDateTime,
|
||||||
|
events: &[Event]
|
||||||
|
) {
|
||||||
|
use egui::*;
|
||||||
|
|
||||||
|
let header_size = 50.0;
|
||||||
|
|
||||||
|
//let now = now.checked_add_days(Days::new(1)).unwrap();
|
||||||
|
let style = EventsTableStyle::from_visuals(&ui.style().visuals);
|
||||||
|
|
||||||
|
show_events_table_header(
|
||||||
|
ui, &style,
|
||||||
|
Rect::from_min_size(
|
||||||
|
rect.left_top(),
|
||||||
|
vec2(rect.width(), header_size)
|
||||||
|
),
|
||||||
|
shown_week
|
||||||
|
);
|
||||||
|
|
||||||
|
show_events_table_body(
|
||||||
|
ui, &style, assets,
|
||||||
|
Rect::from_min_max(
|
||||||
|
rect.left_top() + vec2(0.0, header_size),
|
||||||
|
rect.right_bottom()
|
||||||
|
),
|
||||||
|
shown_week,
|
||||||
|
now,
|
||||||
|
events
|
||||||
|
);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
impl MainApp {
|
||||||
|
pub fn new(storage: ConfigStorage) -> MainApp {
|
||||||
|
MainApp {
|
||||||
|
vidko: None,
|
||||||
|
timetable: None,
|
||||||
|
shown_week: get_current_week(),
|
||||||
|
shown_events: vec![],
|
||||||
|
assets: None,
|
||||||
|
storage
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn on_creation(&mut self, cc: &CreationContext) {
|
||||||
|
self.storage.attempt_load();
|
||||||
|
self.vidko = self.storage.config.vidko_code.clone();
|
||||||
|
|
||||||
|
let break_image = load_image_from_memory(include_bytes!("../assets/break-area.png")).expect("Failed to decode break area texture");
|
||||||
|
let texture_handle = cc.egui_ctx.load_texture("break-area", break_image, TextureOptions::LINEAR);
|
||||||
|
self.assets = Some(AppAssets {
|
||||||
|
break_texture: texture_handle
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn refresh_timetable(&mut self) {
|
||||||
|
if self.vidko.is_none() {
|
||||||
|
self.shown_events = vec![];
|
||||||
|
self.timetable = None;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let timetable = get_timetable(self.vidko.as_ref().unwrap());
|
||||||
|
if timetable.is_err() { return; }
|
||||||
|
let timetable = timetable.unwrap();
|
||||||
|
self.shown_events = timetable.by_week(self.shown_week);
|
||||||
|
self.timetable = Some(timetable);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn shift_shown_week(&mut self, shift: i32) {
|
||||||
|
let year = self.shown_week.year();
|
||||||
|
let week = self.shown_week.week();
|
||||||
|
let week_date = NaiveDate::from_isoywd_opt(year, week, Weekday::Mon).expect("Invalid week or year given");
|
||||||
|
let shifted_week;
|
||||||
|
if shift > 0 {
|
||||||
|
shifted_week = week_date.checked_add_days(Days::new(7 * (shift as u64)));
|
||||||
|
} else {
|
||||||
|
shifted_week = week_date.checked_sub_days(Days::new(7 * ((-shift) as u64)));
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(shifted_week) = shifted_week {
|
||||||
|
self.set_shown_week(shifted_week.iso_week());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn set_shown_week(&mut self, week: IsoWeek) {
|
||||||
|
self.shown_week = week;
|
||||||
|
|
||||||
|
if let Some(timetable) = &self.timetable {
|
||||||
|
self.shown_events = timetable.by_week(self.shown_week);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl eframe::App for MainApp {
|
||||||
|
fn on_exit(&mut self, _gl: Option<&eframe::glow::Context>) {
|
||||||
|
self.storage.attempt_save();
|
||||||
|
}
|
||||||
|
|
||||||
|
fn update(&mut self, ctx: &egui::Context, frame: &mut eframe::Frame) {
|
||||||
|
use egui::*;
|
||||||
|
|
||||||
|
//ctx.set_visuals(egui::Visuals::light());
|
||||||
|
egui::CentralPanel::default()
|
||||||
|
.frame(Frame::none())
|
||||||
|
.show(ctx, |ui| {
|
||||||
|
if ctx.input().key_pressed(Key::D) {
|
||||||
|
self.shift_shown_week(1);
|
||||||
|
}
|
||||||
|
if ctx.input().key_pressed(Key::A) {
|
||||||
|
self.shift_shown_week(-1);
|
||||||
|
}
|
||||||
|
if ctx.input().key_pressed(Key::S) {
|
||||||
|
self.set_shown_week(get_current_week());
|
||||||
|
}
|
||||||
|
if ctx.input().key_pressed(Key::F2) {
|
||||||
|
if ctx.style().visuals.dark_mode {
|
||||||
|
ctx.set_visuals(egui::Visuals::light());
|
||||||
|
} else {
|
||||||
|
ctx.set_visuals(egui::Visuals::dark());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let rect = ui.allocate_rect(ui.min_rect(), Sense::hover()).rect;
|
||||||
|
show_events_table(
|
||||||
|
self.assets.as_ref().unwrap(),
|
||||||
|
ui,
|
||||||
|
rect,
|
||||||
|
self.shown_week,
|
||||||
|
Local::now().naive_local(),
|
||||||
|
&self.shown_events
|
||||||
|
);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
64
src/config.rs
Normal file
64
src/config.rs
Normal file
@ -0,0 +1,64 @@
|
|||||||
|
use std::{path::{Path, PathBuf}, fs};
|
||||||
|
|
||||||
|
use directories_next::ProjectDirs;
|
||||||
|
use eframe::Storage;
|
||||||
|
use egui::util::cache::CacheStorage;
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
|
|
||||||
|
#[derive(Deserialize, Serialize)]
|
||||||
|
pub struct Config {
|
||||||
|
pub vidko_code: Option<String>
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for Config {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self {
|
||||||
|
vidko_code: Some("E0000".into())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct ConfigStorage {
|
||||||
|
pub config: Config,
|
||||||
|
|
||||||
|
config_file: Option<PathBuf>
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for ConfigStorage {
|
||||||
|
fn default() -> Self {
|
||||||
|
let project_dirs = ProjectDirs::from("", "", "KTU Timetable").expect("Failed to determine project directories");
|
||||||
|
let config_dir = project_dirs.config_dir();
|
||||||
|
Self {
|
||||||
|
config: Config::default(),
|
||||||
|
config_file: Some(config_dir.join("config.toml"))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ConfigStorage {
|
||||||
|
pub fn memory() -> Self {
|
||||||
|
let mut config = Self::default();
|
||||||
|
config.config_file = None;
|
||||||
|
config
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn attempt_load(&mut self) {
|
||||||
|
if self.config_file.is_none() { return; }
|
||||||
|
let config_file = self.config_file.as_ref().unwrap();
|
||||||
|
let config_str = fs::read_to_string(config_file);
|
||||||
|
if let Err(_) = config_str {
|
||||||
|
fs::write(config_file, toml::to_string_pretty(&Config::default()).unwrap()).unwrap();
|
||||||
|
}
|
||||||
|
let config_str = config_str.unwrap();
|
||||||
|
self.config = toml::from_str(&config_str).unwrap_or_default();
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn attempt_save(&self) {
|
||||||
|
if self.config_file.is_none() { return; }
|
||||||
|
let config_file = self.config_file.as_ref().unwrap();
|
||||||
|
let config_str = toml::to_string_pretty(&self.config).unwrap();
|
||||||
|
fs::write(config_file, config_str).unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
48
src/main.rs
Normal file
48
src/main.rs
Normal file
@ -0,0 +1,48 @@
|
|||||||
|
#![windows_subsystem = "windows"]
|
||||||
|
|
||||||
|
mod timetable;
|
||||||
|
mod app;
|
||||||
|
mod config;
|
||||||
|
|
||||||
|
use app::MainApp;
|
||||||
|
use config::ConfigStorage;
|
||||||
|
use eframe::egui;
|
||||||
|
|
||||||
|
// TODO: use lazy_static!() to load assets
|
||||||
|
// TODO: convert events_table to egui widget
|
||||||
|
// TODO: show errors when loading config
|
||||||
|
// TODO: Settings menu
|
||||||
|
// TODO: use "confy" for config loading?
|
||||||
|
// TODO: refactor persistence
|
||||||
|
|
||||||
|
fn main() -> Result<(), ureq::Error> {
|
||||||
|
let mut config_storage = ConfigStorage::default();
|
||||||
|
config_storage.config.vidko_code = Some("E1810".into());
|
||||||
|
|
||||||
|
let mut native_options = eframe::NativeOptions::default();
|
||||||
|
native_options.decorated = true;
|
||||||
|
native_options.resizable = true;
|
||||||
|
native_options.min_window_size = Some(egui::vec2(480.0, 320.0));
|
||||||
|
native_options.initial_window_size = Some(egui::vec2(500.0, 320.0));
|
||||||
|
native_options.icon_data = Some(eframe::IconData {
|
||||||
|
rgba: image::load_from_memory(include_bytes!("../assets/icon.png"))
|
||||||
|
.expect("Failed to load icon")
|
||||||
|
.into_rgb8()
|
||||||
|
.into_raw(),
|
||||||
|
width: 32,
|
||||||
|
height: 32,
|
||||||
|
});
|
||||||
|
let mut app = MainApp::new(config_storage);
|
||||||
|
|
||||||
|
eframe::run_native(
|
||||||
|
"KTU timetable",
|
||||||
|
native_options,
|
||||||
|
Box::new(move |cc| {
|
||||||
|
app.on_creation(cc);
|
||||||
|
app.refresh_timetable();
|
||||||
|
Box::new(app)
|
||||||
|
})
|
||||||
|
);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
125
src/timetable.rs
Normal file
125
src/timetable.rs
Normal file
@ -0,0 +1,125 @@
|
|||||||
|
use ical::property::Property;
|
||||||
|
use std::{error::Error, fmt};
|
||||||
|
use std::io::BufReader;
|
||||||
|
use chrono::{NaiveDate, NaiveTime, IsoWeek, Datelike};
|
||||||
|
use lazy_regex::{regex, regex_captures};
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Copy)]
|
||||||
|
pub enum EventCategory {
|
||||||
|
Default,
|
||||||
|
Yellow
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
|
pub struct Event {
|
||||||
|
pub category: EventCategory,
|
||||||
|
pub date: NaiveDate,
|
||||||
|
pub start_time: NaiveTime,
|
||||||
|
pub end_time: NaiveTime,
|
||||||
|
pub description: String,
|
||||||
|
pub summary: String,
|
||||||
|
pub location: String,
|
||||||
|
|
||||||
|
pub module_name: Option<String>
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: Make errors more descriptive
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct Timetable {
|
||||||
|
events: Vec<Event>
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum GetTimetableError {
|
||||||
|
NotFound
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Timetable {
|
||||||
|
pub fn by_week(&self, week: IsoWeek) -> Vec<Event> {
|
||||||
|
return self.events.iter()
|
||||||
|
.filter(|e| e.date.iso_week() == week)
|
||||||
|
.map(|e| e.clone())
|
||||||
|
.collect();
|
||||||
|
}
|
||||||
|
pub fn max_end_time(&self) -> Option<NaiveTime> {
|
||||||
|
return self.events.iter()
|
||||||
|
.map(|e| e.end_time)
|
||||||
|
.max();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Error for GetTimetableError {}
|
||||||
|
|
||||||
|
impl fmt::Display for GetTimetableError {
|
||||||
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||||
|
write!(f, "Failed to get timetable")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn guess_module_name(summary: &str) -> Option<String> {
|
||||||
|
let captures = regex_captures!(r"^P\d{3}B\d{3} (.+)", summary);
|
||||||
|
if let Some((_, module_name)) = captures {
|
||||||
|
return Some(module_name.into());
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_timetable(vidko: &str) -> Result<Timetable, GetTimetableError> {
|
||||||
|
fn find_property<'a>(properties: &'a [Property], name: &str) -> Result<&'a Property, GetTimetableError> {
|
||||||
|
for prop in properties {
|
||||||
|
if prop.name == name {
|
||||||
|
return Ok(prop);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
panic!("Property '{}' not found", name);
|
||||||
|
}
|
||||||
|
|
||||||
|
let resp = ureq::get(&format!("https://uais.cr.ktu.lt/ktuis/tv_rprt2.ical1?p={}&t=basic.ics", vidko))
|
||||||
|
.call()
|
||||||
|
.map_err(|_| GetTimetableError::NotFound)?;
|
||||||
|
|
||||||
|
let mut reader = ical::IcalParser::new(BufReader::new(resp.into_reader()));
|
||||||
|
let cal = reader.next();
|
||||||
|
if cal.is_none() {
|
||||||
|
return Err(GetTimetableError::NotFound)
|
||||||
|
}
|
||||||
|
let cal = cal.unwrap().unwrap();
|
||||||
|
|
||||||
|
let mut timetable = Timetable { events: vec![] };
|
||||||
|
for event in cal.events {
|
||||||
|
let category_prop = find_property(&event.properties, "CATEGORIES")?;
|
||||||
|
let start_prop = find_property(&event.properties, "DTSTART")?;
|
||||||
|
let end_prop = find_property(&event.properties, "DTEND")?;
|
||||||
|
let description_prop = find_property(&event.properties, "DESCRIPTION")?;
|
||||||
|
let summary_prop = find_property(&event.properties, "SUMMARY")?;
|
||||||
|
let location_prop = find_property(&event.properties, "LOCATION")?;
|
||||||
|
|
||||||
|
let mut category = EventCategory::Default;
|
||||||
|
if let Some(category_value) = &category_prop.value {
|
||||||
|
if category_value == "Yellow Category" {
|
||||||
|
category = EventCategory::Yellow;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let start_str = start_prop.value.clone().unwrap();
|
||||||
|
let end_str = end_prop.value.clone().unwrap();
|
||||||
|
let (start_date, start_time) = start_str.split_once('T').unwrap();
|
||||||
|
let (_end_date, end_time) = end_str.split_once('T').unwrap();
|
||||||
|
let summary = summary_prop.value.clone().unwrap();
|
||||||
|
|
||||||
|
timetable.events.push(Event {
|
||||||
|
category,
|
||||||
|
date: NaiveDate::parse_from_str(start_date, "%Y%m%d").unwrap(),
|
||||||
|
start_time: NaiveTime::parse_from_str(start_time, "%H%M%S").unwrap(),
|
||||||
|
end_time: NaiveTime::parse_from_str(end_time, "%H%M%S").unwrap(),
|
||||||
|
description: description_prop.value.clone().unwrap(),
|
||||||
|
module_name: guess_module_name(&summary),
|
||||||
|
summary,
|
||||||
|
location: location_prop.value.clone().unwrap()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
timetable.events.sort_by_key(|event| (event.date, event.start_time));
|
||||||
|
|
||||||
|
Ok(timetable)
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user