podcasts/hammond-gtk/src/views/podcasts.rs

use gtk;
use gtk::prelude::*;
use gdk_pixbuf::Pixbuf;
use diesel::associations::Identifiable;

use hammond_data::dbqueries;
use hammond_data::Podcast;

use widgets::podcast::*;
use utils::get_pixbuf_from_path;

#[derive(Debug, Clone)]
pub struct PopulatedView {
    pub container: gtk::Box,
    pub flowbox: gtk::FlowBox,
    viewport: gtk::Viewport,
}

impl PopulatedView {
    pub fn new() -> PopulatedView {
        let builder = gtk::Builder::new_from_resource("/org/gnome/hammond/gtk/podcasts_view.ui");
        let container: gtk::Box = builder.get_object("fb_parent").unwrap();
        let flowbox: gtk::FlowBox = builder.get_object("flowbox").unwrap();
        let viewport: gtk::Viewport = builder.get_object("viewport").unwrap();

        PopulatedView {
            container,
            flowbox,
            viewport,
        }
    }

    pub fn init(&self, content_stack: &gtk::Stack) {
        use gtk::WidgetExt;

        // TODO: handle unwraps.
        let stack = content_stack;
        self.flowbox
            .connect_child_activated(clone!(stack => move |_, child| {
            // This is such an ugly hack...
            // let id = child.get_name().unwrap().parse::<i32>().unwrap();
            let id = WidgetExt::get_name(child).unwrap().parse::<i32>().unwrap();
            let parent = dbqueries::get_podcast_from_id(id).unwrap();
            on_flowbox_child_activate(&stack, &parent);
        }));
        // Populate the flowbox with the Podcasts.
        populate_flowbox(&self.flowbox);
    }
}

fn show_empty_view(stack: &gtk::Stack) {
    stack.set_visible_child_name("empty");

    info!("Empty view.");
}

fn populate_flowbox(flowbox: &gtk::FlowBox) {
    let podcasts = dbqueries::get_podcasts();

    if let Ok(pds) = podcasts {
        pds.iter().for_each(|parent| {
            let f = create_flowbox_child(parent);
            flowbox.add(&f);
        });
        flowbox.show_all();
    }
}

fn create_flowbox_child(pd: &Podcast) -> gtk::FlowBoxChild {
    let builder = gtk::Builder::new_from_resource("/org/gnome/hammond/gtk/podcasts_child.ui");

    // Copy of gnome-music AlbumWidget
    let box_: gtk::Box = builder.get_object("fb_child").unwrap();
    let pd_title: gtk::Label = builder.get_object("pd_title").unwrap();
    let pd_cover: gtk::Image = builder.get_object("pd_cover").unwrap();
    let banner: gtk::Image = builder.get_object("banner").unwrap();
    let banner_title: gtk::Label = builder.get_object("banner_label").unwrap();

    pd_title.set_text(pd.title());

    let cover = get_pixbuf_from_path(pd);
    if let Some(img) = cover {
        pd_cover.set_from_pixbuf(&img);
    };

    configure_banner(pd, &banner, &banner_title);

    let fbc = gtk::FlowBoxChild::new();
    // There's probably a better way to store the id somewhere.
    // fbc.set_name(&pd.id().to_string());
    WidgetExt::set_name(&fbc, &pd.id().to_string());
    fbc.add(&box_);
    fbc
}

fn configure_banner(pd: &Podcast, banner: &gtk::Image, banner_title: &gtk::Label) {
    let bann = Pixbuf::new_from_resource_at_scale("/org/gnome/hammond/banner.png", 256, 256, true);
    if let Ok(b) = bann {
        banner.set_from_pixbuf(&b);

        let new_episodes = dbqueries::get_pd_unplayed_episodes(pd);

        if let Ok(n) = new_episodes {
            if !n.is_empty() {
                banner_title.set_text(&n.len().to_string());
                banner.show();
                banner_title.show();
            }
        }
    }
}

fn on_flowbox_child_activate(stack: &gtk::Stack, parent: &Podcast) {
    let old = stack.get_child_by_name("pdw").unwrap();
    let pdw = PodcastWidget::new();
    pdw.init(stack, parent);

    stack.remove(&old);
    stack.add_named(&pdw.container, "pdw");
    stack.set_visible_child_name("pdw");

    // aggresive memory cleanup
    // probably not needed
    old.destroy();
}

fn setup_podcasts_flowbox(stack: &gtk::Stack) -> gtk::FlowBox {
    let builder = gtk::Builder::new_from_resource("/org/gnome/hammond/gtk/podcasts_view.ui");
    let fb_parent: gtk::Box = builder.get_object("fb_parent").unwrap();
    let flowbox: gtk::FlowBox = builder.get_object("flowbox").unwrap();
    init_flowbox(stack, &flowbox);

    stack.add_named(&fb_parent, "fb_parent");

    if flowbox.get_children().is_empty() {
        show_empty_view(stack);
    } else {
        stack.set_visible_child(&fb_parent);
    };

    flowbox
}

pub fn update_podcasts_view(stack: &gtk::Stack) {
    let vis = stack.get_visible_child_name().unwrap();
    let old = stack.get_child_by_name("fb_parent").unwrap();
    stack.remove(&old);

    let flowbox = setup_podcasts_flowbox(stack);

    if vis == "empty" && !flowbox.get_children().is_empty() {
        stack.set_visible_child_name("fb_parent");
    } else if vis == "fb_parent" && flowbox.get_children().is_empty() {
        stack.set_visible_child_name("empty");
    } else {
        // preserve the visible widget
        stack.set_visible_child_name(&vis);
    };

    // aggresive memory cleanup
    // probably not needed
    old.destroy();
}

fn init_flowbox(stack: &gtk::Stack, flowbox: &gtk::FlowBox) {
    use gtk::WidgetExt;

    // TODO: handle unwraps.
    flowbox.connect_child_activated(clone!(stack => move |_, child| {
        // This is such an ugly hack...
        // let id = child.get_name().unwrap().parse::<i32>().unwrap();
        let id = WidgetExt::get_name(child).unwrap().parse::<i32>().unwrap();
        let parent = dbqueries::get_podcast_from_id(id).unwrap();
        on_flowbox_child_activate(&stack, &parent);
    }));
    // Populate the flowbox with the Podcasts.
    populate_flowbox(flowbox);
}