monorepo/fitnesstrax/app/src/views/historical_view.rs

/*
Copyright 2023, Savanni D'Gerinel <savanni@luminescent-dreams.com>

This file is part of FitnessTrax.

FitnessTrax is free software: you can redistribute it and/or modify it under the terms of the GNU
General Public License as published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.

FitnessTrax is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
General Public License for more details.

You should have received a copy of the GNU General Public License along with FitnessTrax. If not, see <https://www.gnu.org/licenses/>.
*/

use crate::components::DaySummary;
use emseries::{Recordable, Timestamp};
use ft_core::TraxRecord;
use glib::Object;
use gtk::{prelude::*, subclass::prelude::*};
use std::{cell::RefCell, collections::HashMap};

/// The historical view will show a window into the main database. It will show some version of
/// daily summaries, daily details, and will provide all functions the user may need for editing
/// records.
pub struct HistoricalViewPrivate {}

#[glib::object_subclass]
impl ObjectSubclass for HistoricalViewPrivate {
    const NAME: &'static str = "HistoricalView";
    type Type = HistoricalView;
    type ParentType = gtk::Box;

    fn new() -> Self {
        Self {}
    }
}

impl ObjectImpl for HistoricalViewPrivate {}
impl WidgetImpl for HistoricalViewPrivate {}
impl BoxImpl for HistoricalViewPrivate {}

glib::wrapper! {
    pub struct HistoricalView(ObjectSubclass<HistoricalViewPrivate>) @extends gtk::Box, gtk::Widget, @implements gtk::Orientable;
}

impl HistoricalView {
    pub fn new(records: Vec<TraxRecord>) -> Self {
        let s: Self = Object::builder().build();
        s.set_orientation(gtk::Orientation::Vertical);

        let day_records: GroupedRecords = GroupedRecords::from(records);

        let model = gio::ListStore::new::<DayRecords>();
        model.extend_from_slice(&day_records.0);

        let factory = gtk::SignalListItemFactory::new();
        factory.connect_setup(move |_, list_item| {
            list_item
                .downcast_ref::<gtk::ListItem>()
                .expect("should be a ListItem")
                .set_child(Some(&DaySummary::new()));
        });

        factory.connect_bind(move |_, list_item| {
            let records = list_item
                .downcast_ref::<gtk::ListItem>()
                .expect("should be a ListItem")
                .item()
                .and_downcast::<DayRecords>()
                .expect("should be a DaySummary");

            let summary = list_item
                .downcast_ref::<gtk::ListItem>()
                .expect("should be a ListItem")
                .child()
                .and_downcast::<DaySummary>()
                .expect("should be a DaySummary");

            summary.set_data(records.date(), records.records());
        });

        let lst = gtk::ListView::builder()
            .model(&gtk::NoSelection::new(Some(model)))
            .factory(&factory)
            .single_click_activate(true)
            .build();
        lst.connect_activate(|s, idx| {
            // This gets triggered whenever the user clicks on an item on the list. What we
            // actually want to do here is to open a modal dialog that shows all of the details of
            // the day and which allows the user to edit items within that dialog.
            let item = s.model().unwrap().item(idx).unwrap();
            let records = item.downcast_ref::<DayRecords>().unwrap();
            println!("list item activated: [{:?}] {:?}", idx, records.date());
        });

        s.append(&lst);

        s
    }
}

#[derive(Default)]
pub struct DayRecordsPrivate {
    date: RefCell<chrono::NaiveDate>,
    records: RefCell<Vec<TraxRecord>>,
}

#[glib::object_subclass]
impl ObjectSubclass for DayRecordsPrivate {
    const NAME: &'static str = "DayRecords";
    type Type = DayRecords;
}

impl ObjectImpl for DayRecordsPrivate {}

glib::wrapper! {
    pub struct DayRecords(ObjectSubclass<DayRecordsPrivate>);
}

impl DayRecords {
    pub fn new(date: chrono::NaiveDate, records: Vec<TraxRecord>) -> Self {
        let s: Self = Object::builder().build();

        *s.imp().date.borrow_mut() = date;
        *s.imp().records.borrow_mut() = records;

        s
    }

    pub fn date(&self) -> chrono::NaiveDate {
        self.imp().date.borrow().clone()
    }

    pub fn records(&self) -> Vec<TraxRecord> {
        self.imp().records.borrow().clone()
    }

    pub fn add_record(&self, record: TraxRecord) {
        self.imp().records.borrow_mut().push(record);
    }
}

struct GroupedRecords(Vec<DayRecords>);

impl From<Vec<TraxRecord>> for GroupedRecords {
    fn from(records: Vec<TraxRecord>) -> GroupedRecords {
        GroupedRecords(
            records
                .into_iter()
                .fold(HashMap::new(), |mut acc, rec| {
                    let date = match rec.timestamp() {
                        Timestamp::DateTime(dtz) => dtz.0.date_naive(),
                        Timestamp::Date(date) => date,
                    };
                    acc.entry(date)
                        .and_modify(|entry: &mut DayRecords| (*entry).add_record(rec.clone()))
                        .or_insert(DayRecords::new(date, vec![rec]));
                    acc
                })
                .values()
                .cloned()
                .collect::<Vec<DayRecords>>(),
        )
    }
}

#[cfg(test)]
mod test {
    use super::GroupedRecords;
    use chrono::{NaiveDate, TimeZone};
    use chrono_tz::America::Anchorage;
    use dimensioned::si::{KG, M, S};
    use emseries::DateTimeTz;
    use ft_core::{Steps, TimeDistance, TraxRecord, Weight};

    #[test]
    fn groups_records() {
        let records = vec![
            TraxRecord::Steps(Steps {
                date: NaiveDate::from_ymd_opt(2023, 10, 13).unwrap(),
                count: 1500,
            }),
            TraxRecord::Weight(Weight {
                date: NaiveDate::from_ymd_opt(2023, 10, 13).unwrap(),
                weight: 85. * KG,
            }),
            TraxRecord::Weight(Weight {
                date: NaiveDate::from_ymd_opt(2023, 10, 14).unwrap(),
                weight: 86. * KG,
            }),
            TraxRecord::BikeRide(TimeDistance {
                datetime: DateTimeTz(Anchorage.with_ymd_and_hms(2019, 6, 15, 12, 0, 0).unwrap()),
                distance: Some(1000. * M),
                duration: Some(150. * S),
                comments: Some("Test Comments".to_owned()),
            }),
            TraxRecord::BikeRide(TimeDistance {
                datetime: DateTimeTz(Anchorage.with_ymd_and_hms(2019, 6, 15, 23, 0, 0).unwrap()),
                distance: Some(1000. * M),
                duration: Some(150. * S),
                comments: Some("Test Comments".to_owned()),
            }),
        ];

        let groups = GroupedRecords::from(records).0;
        assert_eq!(groups.len(), 3);
    }
}