wldash/src/widgets/clock.rs

use crate::color::Color;
use crate::draw::{Font, ROBOTO_REGULAR};
use crate::widget::{DrawContext, DrawReport, KeyState, ModifiersState, WaitContext, Widget};

use chrono::{Duration, NaiveDateTime, Timelike};

pub struct Clock {
    cur_time: NaiveDateTime,
    clock_cache: Font,
    size: f32,
}

impl Clock {
    pub fn new(time: NaiveDateTime, size: f32) -> Box<Clock> {
        let mut clock_cache = Font::new(&ROBOTO_REGULAR, size);
        clock_cache.add_str_to_cache("0123456789:");

        Box::new(Clock {
            cur_time: time,
            clock_cache,
            size,
        })
    }
}

impl Widget for Clock {
    fn wait(&mut self, ctx: &mut WaitContext) {
        let target = (self.cur_time + Duration::seconds(60))
            .with_second(0)
            .unwrap()
            .with_nanosecond(0)
            .unwrap();
        ctx.set_time(target);
    }
    fn enter(&mut self) {}
    fn leave(&mut self) {}
    fn size(&self) -> (u32, u32) {
        let digit = (self.size * 0.45).ceil() as u32;
        let colon = (self.size * 0.20).ceil() as u32;
        (digit * 4 + colon, self.size.ceil() as u32)
    }

    fn draw(
        &mut self,
        ctx: &mut DrawContext,
        pos: (u32, u32),
    ) -> Result<DrawReport, ::std::io::Error> {
        let (width, height) = self.size();
        if !(ctx.time.date() != self.cur_time.date()
            || ctx.time.hour() != self.cur_time.hour()
            || ctx.time.minute() != self.cur_time.minute()
            || ctx.force)
        {
            return Ok(DrawReport::empty(width, height));
        }

        self.cur_time = ctx.time;

        let buf = &mut ctx.buf.subdimensions((pos.0, pos.1, width, height))?;
        buf.memset(ctx.bg);

        let digit = (self.size * 0.45).ceil() as u32;
        let colon = (self.size * 0.20).ceil() as u32;
        self.clock_cache.draw_text_fixed_width(
            buf,
            ctx.bg,
            &Color::new(1.0, 1.0, 1.0, 1.0),
            &[digit, digit, colon, digit, digit],
            &format!("{:02}:{:02}", ctx.time.hour(), ctx.time.minute()),
        )?;

        Ok(DrawReport {
            width,
            height,
            damage: vec![(pos.0 as i32, pos.1 as i32, width as i32, height as i32)],
            full_damage: false,
        })
    }

    fn keyboard_input(&mut self, _: u32, _: ModifiersState, _: KeyState, _: Option<String>) {}
    fn mouse_click(&mut self, _: u32, _: (u32, u32)) {}
    fn mouse_scroll(&mut self, _: (f64, f64), _: (u32, u32)) {}
}