1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
use std::path::PathBuf;
use comemo::Track;
use ecow::EcoString;
use crate::doc::Frame;
use crate::eval::{eval, Module, Route, Tracer, Value};
use crate::model::{Introspector, Label};
use crate::syntax::{ast, LinkedNode, Source, SyntaxKind};
use crate::util::PathExt;
use crate::World;
/// Try to determine a set of possible values for an expression.
pub fn analyze_expr(world: &(dyn World + 'static), node: &LinkedNode) -> Vec<Value> {
match node.cast::<ast::Expr>() {
Some(ast::Expr::None(_)) => vec![Value::None],
Some(ast::Expr::Auto(_)) => vec![Value::Auto],
Some(ast::Expr::Bool(v)) => vec![Value::Bool(v.get())],
Some(ast::Expr::Int(v)) => vec![Value::Int(v.get())],
Some(ast::Expr::Float(v)) => vec![Value::Float(v.get())],
Some(ast::Expr::Numeric(v)) => vec![Value::numeric(v.get())],
Some(ast::Expr::Str(v)) => vec![Value::Str(v.get().into())],
Some(ast::Expr::FieldAccess(access)) => {
let Some(child) = node.children().next() else { return vec![] };
analyze_expr(world, &child)
.into_iter()
.filter_map(|target| target.field(&access.field()).ok())
.collect()
}
Some(_) => {
if let Some(parent) = node.parent() {
if parent.kind() == SyntaxKind::FieldAccess && node.index() > 0 {
return analyze_expr(world, parent);
}
}
let route = Route::default();
let mut tracer = Tracer::new(Some(node.span()));
typst::eval::eval(
world.track(),
route.track(),
tracer.track_mut(),
world.main(),
)
.and_then(|module| {
typst::model::typeset(
world.track(),
tracer.track_mut(),
&module.content(),
)
})
.ok();
tracer.finish()
}
_ => vec![],
}
}
/// Try to load a module from the current source file.
pub fn analyze_import(
world: &(dyn World + 'static),
source: &Source,
path: &str,
) -> Option<Module> {
let full: PathBuf = if let Some(path) = path.strip_prefix('/') {
world.root().join(path).normalize()
} else if let Some(dir) = source.path().parent() {
dir.join(path).normalize()
} else {
path.into()
};
let route = Route::default();
let mut tracer = Tracer::default();
let id = world.resolve(&full).ok()?;
let source = world.source(id);
eval(world.track(), route.track(), tracer.track_mut(), source).ok()
}
/// Find all labels and details for them.
///
/// Returns:
/// - All labels and descriptions for them, if available
/// - A split offset: All labels before this offset belong to nodes, all after
/// belong to a bibliography.
pub fn analyze_labels(
world: &(dyn World + 'static),
frames: &[Frame],
) -> (Vec<(Label, Option<EcoString>)>, usize) {
let mut output = vec![];
let introspector = Introspector::new(frames);
let items = &world.library().items;
// Labels in the document.
for elem in introspector.all() {
let Some(label) = elem.label().cloned() else { continue };
let details = elem
.field("caption")
.and_then(|field| match field {
Value::Content(content) => Some(content),
_ => None,
})
.as_ref()
.unwrap_or(elem)
.plain_text();
output.push((label, Some(details)));
}
let split = output.len();
// Bibliography keys.
for (key, detail) in (items.bibliography_keys)(world.track(), introspector.track()) {
output.push((Label(key), detail));
}
(output, split)
}
|
