typst/src/eval/walk.rs

use std::rc::Rc;

use super::{Eval, EvalContext, Template, Value};
use crate::diag::TypResult;
use crate::geom::Spec;
use crate::layout::Layout;
use crate::library::{GridNode, ParChild, ParNode, TrackSizing};
use crate::syntax::ast::*;
use crate::util::{BoolExt, EcoString};

/// Walk markup, filling the currently built template.
pub trait Walk {
    /// Walk the node.
    fn walk(&self, ctx: &mut EvalContext) -> TypResult<()>;
}

impl Walk for Markup {
    fn walk(&self, ctx: &mut EvalContext) -> TypResult<()> {
        for node in self.nodes() {
            node.walk(ctx)?;
        }
        Ok(())
    }
}

impl Walk for MarkupNode {
    fn walk(&self, ctx: &mut EvalContext) -> TypResult<()> {
        match self {
            Self::Space => ctx.template.space(),
            Self::Linebreak => ctx.template.linebreak(),
            Self::Parbreak => ctx.template.parbreak(),
            Self::Strong => ctx.template.modify(|s| s.text_mut().strong.flip()),
            Self::Emph => ctx.template.modify(|s| s.text_mut().emph.flip()),
            Self::Text(text) => ctx.template.text(text),
            Self::Raw(raw) => raw.walk(ctx)?,
            Self::Math(math) => math.walk(ctx)?,
            Self::Heading(heading) => heading.walk(ctx)?,
            Self::List(list) => list.walk(ctx)?,
            Self::Enum(enum_) => enum_.walk(ctx)?,
            Self::Expr(expr) => match expr.eval(ctx)? {
                Value::None => {}
                Value::Int(v) => ctx.template.text(format_eco!("{}", v)),
                Value::Float(v) => ctx.template.text(format_eco!("{}", v)),
                Value::Str(v) => ctx.template.text(v),
                Value::Template(v) => ctx.template += v,
                // For values which can't be shown "naturally", we print the
                // representation in monospace.
                other => ctx.template.monospace(other.repr()),
            },
        }
        Ok(())
    }
}

impl Walk for RawNode {
    fn walk(&self, ctx: &mut EvalContext) -> TypResult<()> {
        if self.block {
            ctx.template.parbreak();
        }

        ctx.template.monospace(&self.text);

        if self.block {
            ctx.template.parbreak();
        }

        Ok(())
    }
}

impl Walk for MathNode {
    fn walk(&self, ctx: &mut EvalContext) -> TypResult<()> {
        if self.display {
            ctx.template.parbreak();
        }

        ctx.template.monospace(self.formula.trim());

        if self.display {
            ctx.template.parbreak();
        }

        Ok(())
    }
}

impl Walk for HeadingNode {
    fn walk(&self, ctx: &mut EvalContext) -> TypResult<()> {
        let level = self.level();
        let body = self.body().eval(ctx)?;

        ctx.template.parbreak();
        ctx.template.save();
        ctx.template.modify(move |style| {
            let text = style.text_mut();
            let upscale = (1.6 - 0.1 * level as f64).max(0.75);
            text.size *= upscale;
            text.strong = true;
        });
        ctx.template += body;
        ctx.template.restore();
        ctx.template.parbreak();

        Ok(())
    }
}

impl Walk for ListNode {
    fn walk(&self, ctx: &mut EvalContext) -> TypResult<()> {
        let body = self.body().eval(ctx)?;
        walk_item(ctx, EcoString::from('•'), body);
        Ok(())
    }
}

impl Walk for EnumNode {
    fn walk(&self, ctx: &mut EvalContext) -> TypResult<()> {
        let body = self.body().eval(ctx)?;
        let label = format_eco!("{}.", self.number().unwrap_or(1));
        walk_item(ctx, label, body);
        Ok(())
    }
}

fn walk_item(ctx: &mut EvalContext, label: EcoString, body: Template) {
    ctx.template += Template::from_block(move |style| {
        let label = Layout::pack(ParNode {
            dir: style.par.dir,
            align: style.par.align,
            leading: style.leading(),
            children: vec![ParChild::Text(label.clone(), Rc::clone(&style.text))],
        });

        let spacing = style.text.size / 2.0;
        GridNode {
            tracks: Spec::new(vec![TrackSizing::Auto; 2], vec![]),
            gutter: Spec::new(vec![TrackSizing::Linear(spacing.into())], vec![]),
            children: vec![label, body.pack(style)],
        }
    });
}