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

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

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

use utils::get_pixbuf_from_path;
use content::ShowStack;
use headerbar::Header;

use std::rc::Rc;

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

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

impl ShowsPopulated {
    pub fn new() -> ShowsPopulated {
        let builder = gtk::Builder::new_from_resource("/org/gnome/hammond/gtk/shows_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();

        ShowsPopulated {
            container,
            flowbox,
            viewport,
        }
    }

    #[allow(dead_code)]
    pub fn new_initialized(show: Rc<ShowStack>, header: Rc<Header>) -> ShowsPopulated {
        let pop = ShowsPopulated::new();
        pop.init(show, header);
        pop
    }

    pub fn init(&self, show: Rc<ShowStack>, header: Rc<Header>) {
        use gtk::WidgetExt;

        // TODO: handle unwraps.
        // TODO: implement back button.
        self.flowbox
            .connect_child_activated(clone!(show => move |_, child| {
            // This is such an ugly hack...
            let id = WidgetExt::get_name(child).unwrap().parse::<i32>().unwrap();
            let pd = dbqueries::get_podcast_from_id(id).unwrap();

            show.replace_widget(&pd);
            // header.set_show_title(pd.title());
            header.switch_to_back();
            show.switch_widget_animated();
        }));
        // 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 = ShowsChild::new_initialized(parent);
                self.flowbox.add(&flowbox_child.child);
            });
            self.flowbox.show_all();
        }
    }

    pub fn is_empty(&self) -> bool {
        self.flowbox.get_children().is_empty()
    }
}

impl ShowsChild {
    fn new() -> ShowsChild {
        let builder = gtk::Builder::new_from_resource("/org/gnome/hammond/gtk/shows_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);

        ShowsChild {
            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) -> ShowsChild {
        let child = ShowsChild::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();
                }
            }
        }
    }
}