grass/src/builtin/selector.rs

use super::{Builtin, GlobalFunctionMap};

use crate::{
    args::CallArgs,
    common::{Brackets, ListSeparator, QuoteKind},
    error::SassResult,
    parse::Parser,
    selector::{ComplexSelector, ComplexSelectorComponent, Extender, Selector, SelectorList},
    value::Value,
};

fn is_superselector(mut args: CallArgs, parser: &mut Parser<'_>) -> SassResult<Value> {
    args.max_args(2)?;
    let parent_selector = parser
        .arg(&mut args, 0, "super")?
        .to_selector(parser, "super", false)?;
    let child_selector = parser
        .arg(&mut args, 1, "sub")?
        .to_selector(parser, "sub", false)?;

    Ok(Value::bool(
        parent_selector.is_super_selector(&child_selector),
    ))
}

fn simple_selectors(mut args: CallArgs, parser: &mut Parser<'_>) -> SassResult<Value> {
    args.max_args(1)?;
    // todo: Value::to_compound_selector
    let selector = parser
        .arg(&mut args, 0, "selector")?
        .to_selector(parser, "selector", false)?;

    if selector.0.components.len() != 1 {
        return Err(("$selector: expected selector.", args.span()).into());
    }

    let compound = if let Some(ComplexSelectorComponent::Compound(compound)) =
        selector.0.components[0].components.get(0).cloned()
    {
        compound
    } else {
        todo!()
    };

    Ok(Value::List(
        compound
            .components
            .into_iter()
            .map(|simple| Value::String(simple.to_string(), QuoteKind::None))
            .collect(),
        ListSeparator::Comma,
        Brackets::None,
    ))
}

fn selector_parse(mut args: CallArgs, parser: &mut Parser<'_>) -> SassResult<Value> {
    args.max_args(1)?;
    Ok(parser
        .arg(&mut args, 0, "selector")?
        .to_selector(parser, "selector", false)?
        .into_value())
}

fn selector_nest(args: CallArgs, parser: &mut Parser<'_>) -> SassResult<Value> {
    let span = args.span();
    let selectors = parser.variadic_args(args)?;
    if selectors.is_empty() {
        return Err(("$selectors: At least one selector must be passed.", span).into());
    }

    Ok(selectors
        .into_iter()
        .map(|sel| sel.node.to_selector(parser, "selectors", true))
        .collect::<SassResult<Vec<Selector>>>()?
        .into_iter()
        .try_fold(
            Selector::new(span),
            |parent, child| -> SassResult<Selector> {
                Ok(child.resolve_parent_selectors(&parent, true)?)
            },
        )?
        .into_value())
}

fn selector_append(args: CallArgs, parser: &mut Parser<'_>) -> SassResult<Value> {
    let span = args.span();
    let selectors = parser.variadic_args(args)?;
    if selectors.is_empty() {
        return Err(("$selectors: At least one selector must be passed.", span).into());
    }

    let mut parsed_selectors = selectors
        .into_iter()
        .map(|s| {
            let tmp = s.node.to_selector(parser, "selectors", false)?;
            if tmp.contains_parent_selector() {
                Err(("Parent selectors aren't allowed here.", span).into())
            } else {
                Ok(tmp)
            }
        })
        .collect::<SassResult<Vec<Selector>>>()?;

    let first = parsed_selectors.remove(0);
    Ok(parsed_selectors
        .into_iter()
        .try_fold(first, |parent, child| -> SassResult<Selector> {
            Ok(Selector(SelectorList {
                components: child
                    .0
                    .components
                    .into_iter()
                    .map(|complex| -> SassResult<ComplexSelector> {
                        let compound = complex.components.first();
                        if let Some(ComplexSelectorComponent::Compound(compound)) = compound {
                            let mut components = vec![match compound.clone().prepend_parent() {
                                Some(v) => ComplexSelectorComponent::Compound(v),
                                None => {
                                    return Err((
                                        format!("Can't append {} to {}.", complex, parent),
                                        span,
                                    )
                                        .into())
                                }
                            }];
                            components.extend(complex.components.into_iter().skip(1));
                            Ok(ComplexSelector {
                                components,
                                line_break: false,
                            })
                        } else {
                            Err((format!("Can't append {} to {}.", complex, parent), span).into())
                        }
                    })
                    .collect::<SassResult<Vec<ComplexSelector>>>()?,
                span,
            })
            .resolve_parent_selectors(&parent, false)?)
        })?
        .into_value())
}

fn selector_extend(mut args: CallArgs, parser: &mut Parser<'_>) -> SassResult<Value> {
    args.max_args(3)?;
    let selector = parser
        .arg(&mut args, 0, "selector")?
        .to_selector(parser, "selector", false)?;
    let target = parser
        .arg(&mut args, 1, "extendee")?
        .to_selector(parser, "extendee", false)?;
    let source = parser
        .arg(&mut args, 2, "extender")?
        .to_selector(parser, "extender", false)?;

    Ok(Extender::extend(selector.0, source.0, target.0, args.span())?.to_sass_list())
}

fn selector_replace(mut args: CallArgs, parser: &mut Parser<'_>) -> SassResult<Value> {
    args.max_args(3)?;
    let selector = parser
        .arg(&mut args, 0, "selector")?
        .to_selector(parser, "selector", false)?;
    let target = parser
        .arg(&mut args, 1, "original")?
        .to_selector(parser, "original", false)?;
    let source =
        parser
            .arg(&mut args, 2, "replacement")?
            .to_selector(parser, "replacement", false)?;
    Ok(Extender::replace(selector.0, source.0, target.0, args.span())?.to_sass_list())
}

fn selector_unify(mut args: CallArgs, parser: &mut Parser<'_>) -> SassResult<Value> {
    args.max_args(2)?;
    let selector1 =
        parser
            .arg(&mut args, 0, "selector1")?
            .to_selector(parser, "selector1", false)?;

    if selector1.contains_parent_selector() {
        return Err((
            "$selector1: Parent selectors aren't allowed here.",
            args.span(),
        )
            .into());
    }

    let selector2 =
        parser
            .arg(&mut args, 1, "selector2")?
            .to_selector(parser, "selector2", false)?;

    if selector2.contains_parent_selector() {
        return Err((
            "$selector2: Parent selectors aren't allowed here.",
            args.span(),
        )
            .into());
    }

    Ok(match selector1.unify(&selector2) {
        Some(sel) => sel.into_value(),
        None => Value::Null,
    })
}

pub(crate) fn declare(f: &mut GlobalFunctionMap) {
    f.insert("is-superselector", Builtin::new(is_superselector));
    f.insert("simple-selectors", Builtin::new(simple_selectors));
    f.insert("selector-parse", Builtin::new(selector_parse));
    f.insert("selector-nest", Builtin::new(selector_nest));
    f.insert("selector-append", Builtin::new(selector_append));
    f.insert("selector-extend", Builtin::new(selector_extend));
    f.insert("selector-replace", Builtin::new(selector_replace));
    f.insert("selector-unify", Builtin::new(selector_unify));
}