#![deny(missing_docs)]
#![allow(unused_doc_comments)]

//! This library contains some widgets for relm.
//!
//! These widgets got extracted from OMMUI-related projects so they can
//! be used in several of them.

#[macro_use]
extern crate log;
#[macro_use]
extern crate relm_derive;
#[macro_use]
extern crate relm;
#[macro_use]
extern crate snafu;

extern crate dbus;
extern crate dbus_udisks2;
extern crate gdk_pixbuf;
extern crate gio;
extern crate glib;
extern crate gtk;
extern crate humansize;
extern crate indexmap;
extern crate ommui_data;
extern crate pango;
extern crate uuid;

pub mod boolean_parameter_entry;
pub mod description_item;
pub mod description_listbox;
pub mod enumeration_parameter_entry;
pub mod error;
pub mod menu;
pub mod menu_page;
pub mod message;
pub mod numeric_parameter_entry;
pub mod parameter_editor;
pub mod parameter_query;
pub mod profile_list;
pub mod progress;
pub mod removable_drives_selector;
pub mod text_entry;

pub use error::Error;

use indexmap::IndexMap;
use ommui_data::samplic;
use snafu::ResultExt;
use std::path::Path;

/// The result returned by functions in this crate.
pub type Result<T, E = Error> = std::result::Result<T, E>;

/// A translation trait.
pub trait XTr: std::fmt::Debug {
    /// Translate a string.
    fn xtr(&self, s: &str) -> String;
}

impl XTr for indexmap::IndexMap<String, String> {
    fn xtr(&self, s: &str) -> String {
        if let Some(t) = self.get(&s.to_string()) {
            t.to_string()
        } else {
            s.to_string()
        }
    }
}

impl<T: XTr> XTr for std::sync::RwLock<T> {
    fn xtr(&self, s: &str) -> String {
        self.read().unwrap().xtr(s)
    }
}

impl<T: XTr> XTr for std::sync::Arc<T> {
    fn xtr(&self, s: &str) -> String {
        (**self).xtr(s)
    }
}

fn tr<S: AsRef<str>>(s: S) -> String {
    s.as_ref().to_string()
}

/// A trait for providing pixbufs.
pub trait Pixbufs {
    /// Get a pixbuf.
    fn get_pixbuf(&self, k: &str) -> Option<gdk_pixbuf::Pixbuf>;

    /// Get a pixbuf, or if not present, a default pixbuf.
    fn get_pixbuf_or_default(&self, k: &str) -> gdk_pixbuf::Pixbuf;
}

impl Pixbufs for indexmap::IndexMap<String, gdk_pixbuf::Pixbuf> {
    fn get_pixbuf(&self, k: &str) -> Option<gdk_pixbuf::Pixbuf> {
        self.get(&k.to_string()).cloned()
    }

    fn get_pixbuf_or_default(&self, k: &str) -> gdk_pixbuf::Pixbuf {
        self.get_pixbuf(k).unwrap_or_else(|| {
            gdk_pixbuf::Pixbuf::load_from_bytes_with_size(
                include_bytes!("../assets/default.svg").as_ref(),
                64u16,
            )
            .unwrap()
        })
    }
}

/// A trait for loading pixbufs from the icons of a source.
pub trait LoadPixbufs {
    /// The pixbufs type which gets loaded.
    type Loaded: Pixbufs;

    /// Load the pixbufs.
    fn load_pixbufs(&self) -> Result<Self::Loaded>;
}

impl<T: samplic::LoadableDevice> LoadPixbufs for T {
    type Loaded = IndexMap<String, gdk_pixbuf::Pixbuf>;

    fn load_pixbufs(&self) -> Result<Self::Loaded> {
        self.load_icon_index()
            .context(error::OmmuiData)?
            .into_iter()
            .map(|k| {
                let raw = self
                    .load_icon(k.as_str())
                    .context(error::OmmuiData)?;
                Ok((
                    k,
                    gdk_pixbuf::Pixbuf::load_from_bytes_with_size(
                        raw.as_bytes(),
                        64u16,
                    )
                    .unwrap(),
                ))
            })
            .collect::<Result<_>>()
    }
}

/// A trait for items loadable from in-memory byte arrays.
pub trait LoadFromBytes: Sized {
    /// Load the item with a maximum width and height.
    fn load_from_bytes_at_scale<B: Into<glib::Bytes>>(
        data: B,
        max_width: u16,
        max_height: u16,
    ) -> Result<Self>;

    /// Load the item with a defined pixel size.
    fn load_from_bytes_with_size<B: Into<glib::Bytes>>(
        data: B,
        pixelsize: u16,
    ) -> Result<Self> {
        Self::load_from_bytes_at_scale(data, pixelsize, pixelsize)
    }
}

impl LoadFromBytes for gdk_pixbuf::Pixbuf {
    fn load_from_bytes_at_scale<B: Into<glib::Bytes>>(
        data: B,
        max_width: u16,
        max_height: u16,
    ) -> Result<Self> {
        let b: glib::Bytes = data.into();
        let stream = gio::MemoryInputStream::new_from_bytes(&b);
        Ok(gdk_pixbuf::Pixbuf::new_from_stream_at_scale(
            &stream,
            i32::from(max_width),
            i32::from(max_height),
            true,
            None::<&gio::Cancellable>,
        )
        .context(error::GLib)?)
    }
}

/// A trait for items loadable from file.
pub trait LoadFromFile: Sized {
    /// Load the item with a maximum width and height.
    fn load_from_file_at_scale<P: AsRef<Path>>(
        filename: P,
        max_width: u16,
        max_height: u16,
    ) -> Result<Self>;

    /// Load the item with a defined pixel size.
    fn load_from_file_with_size<P: AsRef<Path>>(
        filename: P,
        pixelsize: u16,
    ) -> Result<Self> {
        Self::load_from_file_at_scale(filename, pixelsize, pixelsize)
    }
}

impl LoadFromFile for gdk_pixbuf::Pixbuf {
    fn load_from_file_at_scale<P: AsRef<Path>>(
        filename: P,
        max_width: u16,
        max_height: u16,
    ) -> Result<Self> {
        let file = gio::File::new_for_path(filename);
        use gio::FileExt;
        let stream =
            file.read(None::<&gio::Cancellable>).context(error::GLib)?;
        Ok(gdk_pixbuf::Pixbuf::new_from_stream_at_scale(
            &stream,
            i32::from(max_width),
            i32::from(max_height),
            true,
            None::<&gio::Cancellable>,
        )
        .context(error::GLib)?)
    }
}