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,
}

#[derive(Debug)]
struct PodcastChild {
    container: gtk::Box,
    title: gtk::Label,
    cover: gtk::Image,
    banner: gtk::Image,
    number: gtk::Label,
    child: gtk::FlowBoxChild,
}

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, stack: &gtk::Stack) {
        use gtk::WidgetExt;

        // TODO: handle unwraps.
        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.
        self.populate_flowbox();
    }

    fn populate_flowbox(&self) {
        let podcasts = dbqueries::get_podcasts();

        if let Ok(pds) = podcasts {
            pds.iter().for_each(|parent| {
                let flowbox_child = PodcastChild::new_initialized(parent);
                self.flowbox.add(&flowbox_child.child);
            });
            self.flowbox.show_all();
        }
    }
}

impl PodcastChild {
    fn new() -> PodcastChild {
        let builder = gtk::Builder::new_from_resource("/org/gnome/hammond/gtk/podcasts_child.ui");

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

        let child = gtk::FlowBoxChild::new();
        child.add(&container);

        PodcastChild {
            container,
            title,
            cover,
            banner,
            number,
            child,
        }
    }

    fn init(&self, pd: &Podcast) {
        self.title.set_text(pd.title());

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

        WidgetExt::set_name(&self.child, &pd.id().to_string());
        self.configure_banner(pd);
    }

    pub fn new_initialized(pd: &Podcast) -> PodcastChild {
        let child = PodcastChild::new();
        child.init(pd);

        child
    }

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

            let new_episodes = dbqueries::get_pd_unplayed_episodes(pd);

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

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

    stack.remove(&old);
    stack.add_named(&pdw.container, "widget");
    stack.set_visible_child_full("widget", gtk::StackTransitionType::SlideLeft);

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

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

    let pdw = PopulatedView::new();
    pdw.init(stack);
    stack.add_named(&pdw.container, "podcasts");
    let flowbox = &pdw.flowbox;

    if vis == "empty" && !flowbox.get_children().is_empty() {
        stack.set_visible_child_name("podcasts");
    } else if vis == "podcasts" && 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();
}