use std::iter::Peekable;

use crate::common::Scope;
use crate::error::SassResult;
use crate::selector::Selector;
use crate::{eat_expr, Expr, RuleSet, Stmt, Token};

pub(crate) fn eat_stmts<I: Iterator<Item = Token>>(
    toks: &mut Peekable<I>,
    scope: &Scope,
    super_selector: &Selector,
) -> SassResult<Vec<Stmt>> {
    let mut stmts = Vec::new();
    let mut scope = scope.clone();
    while let Some(expr) = eat_expr(toks, &scope, super_selector)? {
        match expr {
            Expr::AtRule(a) => stmts.push(Stmt::AtRule(a)),
            Expr::Style(s) => stmts.push(Stmt::Style(s)),
            Expr::Styles(s) => stmts.extend(s.into_iter().map(Box::new).map(Stmt::Style)),
            Expr::Include(s) => stmts.extend(s),
            Expr::MixinDecl(..) | Expr::FunctionDecl(..) | Expr::Debug(..) | Expr::Warn(..) => {
                todo!()
            }
            Expr::Selector(selector) => {
                let rules = eat_stmts(toks, &scope, &super_selector.zip(&selector))?;
                stmts.push(Stmt::RuleSet(RuleSet {
                    super_selector: super_selector.clone(),
                    selector,
                    rules,
                }));
            }
            Expr::VariableDecl(name, val) => {
                scope.insert_var(&name, *val)?;
            }
            Expr::MultilineComment(s) => stmts.push(Stmt::MultilineComment(s)),
        }
    }
    Ok(stmts)
}