//! Extension traits.

use crate::layout::{AlignNode, MoveNode, PadNode};
use crate::prelude::*;
use crate::text::{EmphNode, FontFamily, FontList, StrongNode, TextNode, UnderlineNode};

/// Additional methods on content.
pub trait ContentExt {
    /// Make this content strong.
    fn strong(self) -> Self;

    /// Make this content emphasized.
    fn emph(self) -> Self;

    /// Underline this content.
    fn underlined(self) -> Self;

    /// Link the content somewhere.
    fn linked(self, link: Link) -> Self;

    /// Set alignments for this content.
    fn aligned(self, aligns: Axes<Option<GenAlign>>) -> Self;

    /// Pad this content at the sides.
    fn padded(self, padding: Sides<Rel<Length>>) -> Self;

    /// Transform this content's contents without affecting layout.
    fn moved(self, delta: Axes<Rel<Length>>) -> Self;
}

impl ContentExt for Content {
    fn strong(self) -> Self {
        StrongNode::new(self).pack()
    }

    fn emph(self) -> Self {
        EmphNode::new(self).pack()
    }

    fn underlined(self) -> Self {
        UnderlineNode::new(self).pack()
    }

    fn linked(self, link: Link) -> Self {
        self.styled(MetaNode::set_data(vec![Meta::Link(link)]))
    }

    fn aligned(self, aligns: Axes<Option<GenAlign>>) -> Self {
        self.styled(AlignNode::set_alignment(aligns))
    }

    fn padded(self, padding: Sides<Rel<Length>>) -> Self {
        PadNode::new(self)
            .with_left(padding.left)
            .with_top(padding.top)
            .with_right(padding.right)
            .with_bottom(padding.bottom)
            .pack()
    }

    fn moved(self, delta: Axes<Rel<Length>>) -> Self {
        MoveNode::new(self).with_dx(delta.x).with_dy(delta.y).pack()
    }
}

/// Additional methods for style maps.
pub trait StyleMapExt {
    /// Set a font family composed of a preferred family and existing families
    /// from a style chain.
    fn set_family(&mut self, preferred: FontFamily, existing: StyleChain);
}

impl StyleMapExt for StyleMap {
    fn set_family(&mut self, preferred: FontFamily, existing: StyleChain) {
        self.set(TextNode::set_font(FontList(
            std::iter::once(preferred)
                .chain(TextNode::font_in(existing))
                .collect(),
        )));
    }
}
