use ril::prelude::*;
use std::path::Path;

pub struct Generate<T: AsRef<str>, V: AsRef<Path>> {
    elements: Vec<Element<T, V>>,
    base: Image<Rgba>,
}

impl<T: AsRef<str>, V: AsRef<Path>> Generate<T, V> {
    pub fn new(base: Vec<u8>, elements: Vec<Element<T, V>>) -> Result<Self, CowError> {
        Ok(Self {
            base: Image::<Rgba>::from_bytes_inferred(&base)?,
            elements,
        })
    }

    pub fn produce(self) -> Result<Vec<u8>, CowError> {
        let size = self.base.dimensions();
        let mut base_image = self.base.with_overlay_mode(OverlayMode::Merge);
        for element in self.elements {
            let mut pos = element.position;
            let mut img: Image = element.into_image()?;
            if pos.0 < 0 {
                img.crop(-pos.0 as u32, 0, size.0, size.1);
                pos.0 = 0;
            }
            if pos.1 < 0 {
                img.crop(0, -pos.1 as u32, size.0, size.1);
                pos.1 = 0;
            }

            base_image.draw(&Paste::new(&img.convert()).with_position(pos.0 as u32, pos.1 as u32));
        }
        let mut buf = Vec::<u8>::new();
        base_image.encode(ImageFormat::Png, &mut buf).unwrap();

        Ok(buf)
    }
}

pub struct Element<T: AsRef<str>, V: AsRef<Path>> {
    position: (i32, i32),
    dimensions: (u32, u32),
    media: Media<T, V>,
}

impl<T: AsRef<str>, V: AsRef<Path>> Element<T, V> {
    pub fn new(media: Media<T, V>, position: (i32, i32), dimensions: (u32, u32)) -> Self {
        Self {
            position,
            dimensions,
            media,
        }
    }

    fn into_image(self) -> Result<Image, CowError> {
        match self.media {
            Media::Image(image) => Ok(Image::from_bytes_inferred(image)?.resized(
                self.dimensions.0,
                self.dimensions.1,
                ResizeAlgorithm::Bicubic,
            )),
            Media::Text(text) => Ok(Image::new(
                self.dimensions.0,
                self.dimensions.1,
                Dynamic::Rgba(Rgba::transparent()),
            )
            .with(
                &TextSegment::new(
                    &Font::open(text.font, 12.0).unwrap(),
                    text.text,
                    Dynamic::Rgba(Rgba::new(
                        text.fill.0,
                        text.fill.1,
                        text.fill.2,
                        text.fill.3,
                    )),
                )
                .with_size(text.size),
            )),
        }
    }
}

pub enum Media<T: AsRef<str>, V: AsRef<Path>> {
    Text(Text<T, V>),
    Image(Vec<u8>),
}

pub struct Text<T: AsRef<str>, V: AsRef<Path>> {
    text: T,
    font: V,
    size: f32,
    fill: (u8, u8, u8, u8),
}

impl<T: AsRef<str>, V: AsRef<Path>> Text<T, V> {
    pub fn new(text: T, font: V, size: f32, fill: (u8, u8, u8, u8)) -> Self {
        Self {
            text,
            font,
            size,
            fill,
        }
    }
}

#[derive(Clone, Debug)]
pub enum CowError {
    Magick(String),
    Other(String),
}

impl From<ril::Error> for CowError {
    fn from(r: ril::Error) -> Self {
        CowError::Other(r.to_string())
    }
}