//! Extension traits.

use crate::prelude::*;

/// 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 to a destination.
    fn linked(self, dest: Destination) -> 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 {
        crate::text::StrongNode::new(self).pack()
    }

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

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

    fn linked(self, dest: Destination) -> Self {
        self.styled(MetaNode::DATA, vec![Meta::Link(dest.clone())])
    }

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

    fn padded(self, padding: Sides<Rel<Length>>) -> Self {
        crate::layout::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 {
        crate::layout::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: crate::text::FontFamily, existing: StyleChain);
}

impl StyleMapExt for StyleMap {
    fn set_family(&mut self, preferred: crate::text::FontFamily, existing: StyleChain) {
        self.set(
            crate::text::TextNode::FONT,
            crate::text::FontList(
                std::iter::once(preferred)
                    .chain(existing.get(crate::text::TextNode::FONT).0.iter().cloned())
                    .collect(),
            ),
        );
    }
}