1use std::fmt;
4use std::fmt::Write;
5use std::sync::Arc;
6use std::sync::LazyLock;
7
8use wdl_ast::AstNode;
9use wdl_ast::AstToken;
10use wdl_ast::Diagnostic;
11use wdl_ast::Ident;
12use wdl_ast::Severity;
13use wdl_ast::Span;
14use wdl_ast::SupportedVersion;
15use wdl_ast::TreeNode;
16use wdl_ast::TreeToken;
17use wdl_ast::v1;
18use wdl_ast::v1::AccessExpr;
19use wdl_ast::v1::CallExpr;
20use wdl_ast::v1::Expr;
21use wdl_ast::v1::IfExpr;
22use wdl_ast::v1::IndexExpr;
23use wdl_ast::v1::LiteralArray;
24use wdl_ast::v1::LiteralExpr;
25use wdl_ast::v1::LiteralHints;
26use wdl_ast::v1::LiteralInput;
27use wdl_ast::v1::LiteralMap;
28use wdl_ast::v1::LiteralMapItem;
29use wdl_ast::v1::LiteralObject;
30use wdl_ast::v1::LiteralOutput;
31use wdl_ast::v1::LiteralPair;
32use wdl_ast::v1::LiteralStruct;
33use wdl_ast::v1::LogicalAndExpr;
34use wdl_ast::v1::LogicalNotExpr;
35use wdl_ast::v1::LogicalOrExpr;
36use wdl_ast::v1::NegationExpr;
37use wdl_ast::v1::Placeholder;
38use wdl_ast::v1::PlaceholderOption;
39use wdl_ast::v1::StringPart;
40use wdl_ast::v1::TASK_FIELD_ATTEMPT;
41use wdl_ast::v1::TASK_FIELD_CONTAINER;
42use wdl_ast::v1::TASK_FIELD_CPU;
43use wdl_ast::v1::TASK_FIELD_DISKS;
44use wdl_ast::v1::TASK_FIELD_END_TIME;
45use wdl_ast::v1::TASK_FIELD_EXT;
46use wdl_ast::v1::TASK_FIELD_FPGA;
47use wdl_ast::v1::TASK_FIELD_GPU;
48use wdl_ast::v1::TASK_FIELD_ID;
49use wdl_ast::v1::TASK_FIELD_MEMORY;
50use wdl_ast::v1::TASK_FIELD_META;
51use wdl_ast::v1::TASK_FIELD_NAME;
52use wdl_ast::v1::TASK_FIELD_PARAMETER_META;
53use wdl_ast::v1::TASK_FIELD_RETURN_CODE;
54use wdl_ast::v1::TASK_HINT_DISKS;
55use wdl_ast::v1::TASK_HINT_FPGA;
56use wdl_ast::v1::TASK_HINT_GPU;
57use wdl_ast::v1::TASK_HINT_INPUTS;
58use wdl_ast::v1::TASK_HINT_LOCALIZATION_OPTIONAL;
59use wdl_ast::v1::TASK_HINT_LOCALIZATION_OPTIONAL_ALIAS;
60use wdl_ast::v1::TASK_HINT_MAX_CPU;
61use wdl_ast::v1::TASK_HINT_MAX_CPU_ALIAS;
62use wdl_ast::v1::TASK_HINT_MAX_MEMORY;
63use wdl_ast::v1::TASK_HINT_MAX_MEMORY_ALIAS;
64use wdl_ast::v1::TASK_HINT_OUTPUTS;
65use wdl_ast::v1::TASK_HINT_SHORT_TASK;
66use wdl_ast::v1::TASK_HINT_SHORT_TASK_ALIAS;
67use wdl_ast::v1::TASK_REQUIREMENT_CONTAINER;
68use wdl_ast::v1::TASK_REQUIREMENT_CONTAINER_ALIAS;
69use wdl_ast::v1::TASK_REQUIREMENT_CPU;
70use wdl_ast::v1::TASK_REQUIREMENT_DISKS;
71use wdl_ast::v1::TASK_REQUIREMENT_FPGA;
72use wdl_ast::v1::TASK_REQUIREMENT_GPU;
73use wdl_ast::v1::TASK_REQUIREMENT_MAX_RETRIES;
74use wdl_ast::v1::TASK_REQUIREMENT_MAX_RETRIES_ALIAS;
75use wdl_ast::v1::TASK_REQUIREMENT_MEMORY;
76use wdl_ast::v1::TASK_REQUIREMENT_RETURN_CODES;
77use wdl_ast::v1::TASK_REQUIREMENT_RETURN_CODES_ALIAS;
78use wdl_ast::version::V1;
79
80use super::ArrayType;
81use super::CompoundType;
82use super::MapType;
83use super::Optional;
84use super::PairType;
85use super::PrimitiveType;
86use super::StructType;
87use super::Type;
88use super::TypeNameResolver;
89use crate::SyntaxNodeExt;
90use crate::UNNECESSARY_FUNCTION_CALL;
91use crate::config::DiagnosticsConfig;
92use crate::diagnostics::Io;
93use crate::diagnostics::ambiguous_argument;
94use crate::diagnostics::argument_type_mismatch;
95use crate::diagnostics::cannot_access;
96use crate::diagnostics::cannot_coerce_to_string;
97use crate::diagnostics::cannot_index;
98use crate::diagnostics::comparison_mismatch;
99use crate::diagnostics::if_conditional_mismatch;
100use crate::diagnostics::index_type_mismatch;
101use crate::diagnostics::invalid_placeholder_option;
102use crate::diagnostics::invalid_regex_pattern;
103use crate::diagnostics::logical_and_mismatch;
104use crate::diagnostics::logical_not_mismatch;
105use crate::diagnostics::logical_or_mismatch;
106use crate::diagnostics::map_key_not_primitive;
107use crate::diagnostics::missing_struct_members;
108use crate::diagnostics::multiple_type_mismatch;
109use crate::diagnostics::negation_mismatch;
110use crate::diagnostics::no_common_type;
111use crate::diagnostics::not_a_pair_accessor;
112use crate::diagnostics::not_a_struct;
113use crate::diagnostics::not_a_struct_member;
114use crate::diagnostics::not_a_task_member;
115use crate::diagnostics::numeric_mismatch;
116use crate::diagnostics::string_concat_mismatch;
117use crate::diagnostics::too_few_arguments;
118use crate::diagnostics::too_many_arguments;
119use crate::diagnostics::type_mismatch;
120use crate::diagnostics::unknown_call_io;
121use crate::diagnostics::unknown_function;
122use crate::diagnostics::unknown_task_io;
123use crate::diagnostics::unnecessary_function_call;
124use crate::diagnostics::unsupported_function;
125use crate::document::Task;
126use crate::stdlib::FunctionBindError;
127use crate::stdlib::MAX_PARAMETERS;
128use crate::stdlib::STDLIB;
129use crate::types::Coercible;
130pub fn task_member_type(name: &str) -> Option<Type> {
136 match name {
137 n if n == TASK_FIELD_NAME || n == TASK_FIELD_ID => Some(PrimitiveType::String.into()),
138 n if n == TASK_FIELD_CONTAINER => Some(Type::from(PrimitiveType::String).optional()),
139 n if n == TASK_FIELD_CPU => Some(PrimitiveType::Float.into()),
140 n if n == TASK_FIELD_MEMORY || n == TASK_FIELD_ATTEMPT => {
141 Some(PrimitiveType::Integer.into())
142 }
143 n if n == TASK_FIELD_GPU || n == TASK_FIELD_FPGA => {
144 Some(STDLIB.array_string_type().clone())
145 }
146 n if n == TASK_FIELD_DISKS => Some(STDLIB.map_string_int_type().clone()),
147 n if n == TASK_FIELD_END_TIME || n == TASK_FIELD_RETURN_CODE => {
148 Some(Type::from(PrimitiveType::Integer).optional())
149 }
150 n if n == TASK_FIELD_META || n == TASK_FIELD_PARAMETER_META || n == TASK_FIELD_EXT => {
151 Some(Type::Object)
152 }
153 _ => None,
154 }
155}
156
157pub fn task_requirement_types(version: SupportedVersion, name: &str) -> Option<&'static [Type]> {
161 static CONTAINER_TYPES: LazyLock<Box<[Type]>> = LazyLock::new(|| {
163 Box::new([
164 PrimitiveType::String.into(),
165 STDLIB.array_string_type().clone(),
166 ])
167 });
168 const CPU_TYPES: &[Type] = &[
170 Type::Primitive(PrimitiveType::Integer, false),
171 Type::Primitive(PrimitiveType::Float, false),
172 ];
173 const MEMORY_TYPES: &[Type] = &[
175 Type::Primitive(PrimitiveType::Integer, false),
176 Type::Primitive(PrimitiveType::String, false),
177 ];
178 const GPU_TYPES: &[Type] = &[Type::Primitive(PrimitiveType::Boolean, false)];
180 const FPGA_TYPES: &[Type] = &[Type::Primitive(PrimitiveType::Boolean, false)];
182 static DISKS_TYPES: LazyLock<Box<[Type]>> = LazyLock::new(|| {
184 Box::new([
185 PrimitiveType::Integer.into(),
186 PrimitiveType::String.into(),
187 STDLIB.array_string_type().clone(),
188 ])
189 });
190 const MAX_RETRIES_TYPES: &[Type] = &[Type::Primitive(PrimitiveType::Integer, false)];
192 static RETURN_CODES_TYPES: LazyLock<Box<[Type]>> = LazyLock::new(|| {
194 Box::new([
195 PrimitiveType::Integer.into(),
196 PrimitiveType::String.into(),
197 STDLIB.array_int_type().clone(),
198 ])
199 });
200
201 match name {
202 n if n == TASK_REQUIREMENT_CONTAINER || n == TASK_REQUIREMENT_CONTAINER_ALIAS => {
203 Some(&CONTAINER_TYPES)
204 }
205 n if n == TASK_REQUIREMENT_CPU => Some(CPU_TYPES),
206 n if n == TASK_REQUIREMENT_DISKS => Some(&DISKS_TYPES),
207 n if n == TASK_REQUIREMENT_GPU => Some(GPU_TYPES),
208 n if version >= SupportedVersion::V1(V1::Two) && n == TASK_REQUIREMENT_FPGA => {
209 Some(FPGA_TYPES)
210 }
211 n if version >= SupportedVersion::V1(V1::Two) && n == TASK_REQUIREMENT_MAX_RETRIES => {
212 Some(MAX_RETRIES_TYPES)
213 }
214 n if n == TASK_REQUIREMENT_MAX_RETRIES_ALIAS => Some(MAX_RETRIES_TYPES),
215 n if n == TASK_REQUIREMENT_MEMORY => Some(MEMORY_TYPES),
216 n if version >= SupportedVersion::V1(V1::Two) && n == TASK_REQUIREMENT_RETURN_CODES => {
217 Some(&RETURN_CODES_TYPES)
218 }
219 n if n == TASK_REQUIREMENT_RETURN_CODES_ALIAS => Some(&RETURN_CODES_TYPES),
220 _ => None,
221 }
222}
223
224pub fn task_hint_types(
228 version: SupportedVersion,
229 name: &str,
230 use_hidden_types: bool,
231) -> Option<&'static [Type]> {
232 static DISKS_TYPES: LazyLock<Box<[Type]>> = LazyLock::new(|| {
234 Box::new([
235 PrimitiveType::String.into(),
236 STDLIB.map_string_string_type().clone(),
237 ])
238 });
239 const FPGA_TYPES: &[Type] = &[
241 Type::Primitive(PrimitiveType::Integer, false),
242 Type::Primitive(PrimitiveType::String, false),
243 ];
244 const GPU_TYPES: &[Type] = &[
246 Type::Primitive(PrimitiveType::Integer, false),
247 Type::Primitive(PrimitiveType::String, false),
248 ];
249 const INPUTS_TYPES: &[Type] = &[Type::Object];
251 const INPUTS_HIDDEN_TYPES: &[Type] = &[Type::Input];
253 const LOCALIZATION_OPTIONAL_TYPES: &[Type] = &[Type::Primitive(PrimitiveType::Boolean, false)];
255 const MAX_CPU_TYPES: &[Type] = &[
257 Type::Primitive(PrimitiveType::Integer, false),
258 Type::Primitive(PrimitiveType::Float, false),
259 ];
260 const MAX_MEMORY_TYPES: &[Type] = &[
262 Type::Primitive(PrimitiveType::Integer, false),
263 Type::Primitive(PrimitiveType::String, false),
264 ];
265 const OUTPUTS_TYPES: &[Type] = &[Type::Object];
267 const OUTPUTS_HIDDEN_TYPES: &[Type] = &[Type::Output];
269 const SHORT_TASK_TYPES: &[Type] = &[Type::Primitive(PrimitiveType::Boolean, false)];
271
272 match name {
273 n if n == TASK_HINT_DISKS => Some(&DISKS_TYPES),
274 n if version >= SupportedVersion::V1(V1::Two) && n == TASK_HINT_FPGA => Some(FPGA_TYPES),
275 n if n == TASK_HINT_GPU => Some(GPU_TYPES),
276 n if use_hidden_types
277 && version >= SupportedVersion::V1(V1::Two)
278 && n == TASK_HINT_INPUTS =>
279 {
280 Some(INPUTS_HIDDEN_TYPES)
281 }
282 n if n == TASK_HINT_INPUTS => Some(INPUTS_TYPES),
283 n if version >= SupportedVersion::V1(V1::Two) && n == TASK_HINT_LOCALIZATION_OPTIONAL => {
284 Some(LOCALIZATION_OPTIONAL_TYPES)
285 }
286 n if n == TASK_HINT_LOCALIZATION_OPTIONAL_ALIAS => Some(LOCALIZATION_OPTIONAL_TYPES),
287 n if version >= SupportedVersion::V1(V1::Two) && n == TASK_HINT_MAX_CPU => {
288 Some(MAX_CPU_TYPES)
289 }
290 n if n == TASK_HINT_MAX_CPU_ALIAS => Some(MAX_CPU_TYPES),
291 n if version >= SupportedVersion::V1(V1::Two) && n == TASK_HINT_MAX_MEMORY => {
292 Some(MAX_MEMORY_TYPES)
293 }
294 n if n == TASK_HINT_MAX_MEMORY_ALIAS => Some(MAX_MEMORY_TYPES),
295 n if use_hidden_types
296 && version >= SupportedVersion::V1(V1::Two)
297 && n == TASK_HINT_OUTPUTS =>
298 {
299 Some(OUTPUTS_HIDDEN_TYPES)
300 }
301 n if n == TASK_HINT_OUTPUTS => Some(OUTPUTS_TYPES),
302 n if version >= SupportedVersion::V1(V1::Two) && n == TASK_HINT_SHORT_TASK => {
303 Some(SHORT_TASK_TYPES)
304 }
305 n if n == TASK_HINT_SHORT_TASK_ALIAS => Some(SHORT_TASK_TYPES),
306 _ => None,
307 }
308}
309
310#[derive(Debug, Clone, Copy, PartialEq, Eq)]
312pub enum ComparisonOperator {
313 Equality,
315 Inequality,
317 Less,
319 LessEqual,
321 Greater,
323 GreaterEqual,
325}
326
327impl fmt::Display for ComparisonOperator {
328 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
329 write!(
330 f,
331 "{}",
332 match self {
333 Self::Equality => "==",
334 Self::Inequality => "!=",
335 Self::Less => "<",
336 Self::LessEqual => "<=",
337 Self::Greater => ">",
338 Self::GreaterEqual => ">=",
339 }
340 )
341 }
342}
343
344#[derive(Debug, Clone, Copy, PartialEq, Eq)]
346pub enum NumericOperator {
347 Addition,
349 Subtraction,
351 Multiplication,
353 Division,
355 Modulo,
357 Exponentiation,
359}
360
361impl fmt::Display for NumericOperator {
362 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
363 write!(
364 f,
365 "{}",
366 match self {
367 Self::Addition => "addition",
368 Self::Subtraction => "subtraction",
369 Self::Multiplication => "multiplication",
370 Self::Division => "division",
371 Self::Modulo => "remainder",
372 Self::Exponentiation => "exponentiation",
373 }
374 )
375 }
376}
377
378#[derive(Debug)]
380pub struct AstTypeConverter<R>(R);
381
382impl<R> AstTypeConverter<R>
383where
384 R: TypeNameResolver,
385{
386 pub fn new(resolver: R) -> Self {
388 Self(resolver)
389 }
390
391 pub fn convert_type<N: TreeNode>(&mut self, ty: &v1::Type<N>) -> Result<Type, Diagnostic> {
396 let optional = ty.is_optional();
397
398 let ty: Type = match ty {
399 v1::Type::Map(ty) => {
400 let ty = self.convert_map_type(ty)?;
401 ty.into()
402 }
403 v1::Type::Array(ty) => {
404 let ty = self.convert_array_type(ty)?;
405 ty.into()
406 }
407 v1::Type::Pair(ty) => {
408 let ty = self.convert_pair_type(ty)?;
409 ty.into()
410 }
411 v1::Type::Object(_) => Type::Object,
412 v1::Type::Ref(r) => {
413 let name = r.name();
414 self.0.resolve(name.text(), name.span())?
415 }
416 v1::Type::Primitive(ty) => Type::Primitive(ty.kind().into(), false),
417 };
418
419 if optional { Ok(ty.optional()) } else { Ok(ty) }
420 }
421
422 pub fn convert_array_type<N: TreeNode>(
427 &mut self,
428 ty: &v1::ArrayType<N>,
429 ) -> Result<ArrayType, Diagnostic> {
430 let element_type = self.convert_type(&ty.element_type())?;
431 if ty.is_non_empty() {
432 Ok(ArrayType::non_empty(element_type))
433 } else {
434 Ok(ArrayType::new(element_type))
435 }
436 }
437
438 pub fn convert_pair_type<N: TreeNode>(
443 &mut self,
444 ty: &v1::PairType<N>,
445 ) -> Result<PairType, Diagnostic> {
446 let (left_type, right_type) = ty.types();
447 Ok(PairType::new(
448 self.convert_type(&left_type)?,
449 self.convert_type(&right_type)?,
450 ))
451 }
452
453 pub fn convert_map_type<N: TreeNode>(
458 &mut self,
459 ty: &v1::MapType<N>,
460 ) -> Result<MapType, Diagnostic> {
461 let (key_type, value_type) = ty.types();
462 let optional = key_type.is_optional();
463 Ok(MapType::new(
464 Type::Primitive(key_type.kind().into(), optional),
465 self.convert_type(&value_type)?,
466 ))
467 }
468
469 pub fn convert_struct_type<N: TreeNode>(
474 &mut self,
475 definition: &v1::StructDefinition<N>,
476 ) -> Result<StructType, Diagnostic> {
477 Ok(StructType {
478 name: Arc::new(definition.name().text().to_string()),
479 members: definition
480 .members()
481 .map(|d| Ok((d.name().text().to_string(), self.convert_type(&d.ty())?)))
482 .collect::<Result<_, _>>()?,
483 })
484 }
485}
486
487impl From<v1::PrimitiveTypeKind> for PrimitiveType {
488 fn from(value: v1::PrimitiveTypeKind) -> Self {
489 match value {
490 v1::PrimitiveTypeKind::Boolean => Self::Boolean,
491 v1::PrimitiveTypeKind::Integer => Self::Integer,
492 v1::PrimitiveTypeKind::Float => Self::Float,
493 v1::PrimitiveTypeKind::String => Self::String,
494 v1::PrimitiveTypeKind::File => Self::File,
495 v1::PrimitiveTypeKind::Directory => Self::Directory,
496 }
497 }
498}
499
500pub trait EvaluationContext {
502 fn version(&self) -> SupportedVersion;
504
505 fn resolve_name(&self, name: &str, span: Span) -> Option<Type>;
507
508 fn resolve_type_name(&mut self, name: &str, span: Span) -> Result<Type, Diagnostic>;
510
511 fn task(&self) -> Option<&Task>;
515
516 fn diagnostics_config(&self) -> DiagnosticsConfig;
518
519 fn add_diagnostic(&mut self, diagnostic: Diagnostic);
521}
522
523#[derive(Debug)]
525pub struct ExprTypeEvaluator<'a, C> {
526 context: &'a mut C,
528 placeholders: usize,
536}
537
538impl<'a, C: EvaluationContext> ExprTypeEvaluator<'a, C> {
539 pub fn new(context: &'a mut C) -> Self {
541 Self {
542 context,
543 placeholders: 0,
544 }
545 }
546
547 pub fn evaluate_expr<N: TreeNode + SyntaxNodeExt>(&mut self, expr: &Expr<N>) -> Option<Type> {
551 match expr {
552 Expr::Literal(expr) => self.evaluate_literal_expr(expr),
553 Expr::NameRef(r) => {
554 let name = r.name();
555 self.context.resolve_name(name.text(), name.span())
556 }
557 Expr::Parenthesized(expr) => self.evaluate_expr(&expr.expr()),
558 Expr::If(expr) => self.evaluate_if_expr(expr),
559 Expr::LogicalNot(expr) => self.evaluate_logical_not_expr(expr),
560 Expr::Negation(expr) => self.evaluate_negation_expr(expr),
561 Expr::LogicalOr(expr) => self.evaluate_logical_or_expr(expr),
562 Expr::LogicalAnd(expr) => self.evaluate_logical_and_expr(expr),
563 Expr::Equality(expr) => {
564 let (lhs, rhs) = expr.operands();
565 self.evaluate_comparison_expr(ComparisonOperator::Equality, &lhs, &rhs, expr.span())
566 }
567 Expr::Inequality(expr) => {
568 let (lhs, rhs) = expr.operands();
569 self.evaluate_comparison_expr(
570 ComparisonOperator::Inequality,
571 &lhs,
572 &rhs,
573 expr.span(),
574 )
575 }
576 Expr::Less(expr) => {
577 let (lhs, rhs) = expr.operands();
578 self.evaluate_comparison_expr(ComparisonOperator::Less, &lhs, &rhs, expr.span())
579 }
580 Expr::LessEqual(expr) => {
581 let (lhs, rhs) = expr.operands();
582 self.evaluate_comparison_expr(
583 ComparisonOperator::LessEqual,
584 &lhs,
585 &rhs,
586 expr.span(),
587 )
588 }
589 Expr::Greater(expr) => {
590 let (lhs, rhs) = expr.operands();
591 self.evaluate_comparison_expr(ComparisonOperator::Greater, &lhs, &rhs, expr.span())
592 }
593 Expr::GreaterEqual(expr) => {
594 let (lhs, rhs) = expr.operands();
595 self.evaluate_comparison_expr(
596 ComparisonOperator::GreaterEqual,
597 &lhs,
598 &rhs,
599 expr.span(),
600 )
601 }
602 Expr::Addition(expr) => {
603 let (lhs, rhs) = expr.operands();
604 self.evaluate_numeric_expr(NumericOperator::Addition, expr.span(), &lhs, &rhs)
605 }
606 Expr::Subtraction(expr) => {
607 let (lhs, rhs) = expr.operands();
608 self.evaluate_numeric_expr(NumericOperator::Subtraction, expr.span(), &lhs, &rhs)
609 }
610 Expr::Multiplication(expr) => {
611 let (lhs, rhs) = expr.operands();
612 self.evaluate_numeric_expr(NumericOperator::Multiplication, expr.span(), &lhs, &rhs)
613 }
614 Expr::Division(expr) => {
615 let (lhs, rhs) = expr.operands();
616 self.evaluate_numeric_expr(NumericOperator::Division, expr.span(), &lhs, &rhs)
617 }
618 Expr::Modulo(expr) => {
619 let (lhs, rhs) = expr.operands();
620 self.evaluate_numeric_expr(NumericOperator::Modulo, expr.span(), &lhs, &rhs)
621 }
622 Expr::Exponentiation(expr) => {
623 let (lhs, rhs) = expr.operands();
624 self.evaluate_numeric_expr(NumericOperator::Exponentiation, expr.span(), &lhs, &rhs)
625 }
626 Expr::Call(expr) => self.evaluate_call_expr(expr),
627 Expr::Index(expr) => self.evaluate_index_expr(expr),
628 Expr::Access(expr) => self.evaluate_access_expr(expr),
629 }
630 }
631
632 fn evaluate_literal_expr<N: TreeNode + SyntaxNodeExt>(
634 &mut self,
635 expr: &LiteralExpr<N>,
636 ) -> Option<Type> {
637 match expr {
638 LiteralExpr::Boolean(_) => Some(PrimitiveType::Boolean.into()),
639 LiteralExpr::Integer(_) => Some(PrimitiveType::Integer.into()),
640 LiteralExpr::Float(_) => Some(PrimitiveType::Float.into()),
641 LiteralExpr::String(s) => {
642 for p in s.parts() {
643 if let StringPart::Placeholder(p) = p {
644 self.check_placeholder(&p);
645 }
646 }
647
648 Some(PrimitiveType::String.into())
649 }
650 LiteralExpr::Array(expr) => Some(self.evaluate_literal_array(expr)),
651 LiteralExpr::Pair(expr) => Some(self.evaluate_literal_pair(expr)),
652 LiteralExpr::Map(expr) => Some(self.evaluate_literal_map(expr)),
653 LiteralExpr::Object(expr) => Some(self.evaluate_literal_object(expr)),
654 LiteralExpr::Struct(expr) => self.evaluate_literal_struct(expr),
655 LiteralExpr::None(_) => Some(Type::None),
656 LiteralExpr::Hints(expr) => self.evaluate_literal_hints(expr),
657 LiteralExpr::Input(expr) => self.evaluate_literal_input(expr),
658 LiteralExpr::Output(expr) => self.evaluate_literal_output(expr),
659 }
660 }
661
662 pub(crate) fn check_placeholder<N: TreeNode + SyntaxNodeExt>(
664 &mut self,
665 placeholder: &Placeholder<N>,
666 ) {
667 self.placeholders += 1;
668
669 let expr = placeholder.expr();
672 if let Some(ty) = self.evaluate_expr(&expr) {
673 if let Some(option) = placeholder.option() {
674 let valid = match option {
675 PlaceholderOption::Sep(_) => {
676 ty == Type::Union
677 || ty == Type::None
678 || matches!(&ty,
679 Type::Compound(CompoundType::Array(array_ty), _)
680 if matches!(array_ty.element_type(), Type::Primitive(_, false) | Type::Union))
681 }
682 PlaceholderOption::Default(_) => {
683 matches!(ty, Type::Primitive(..) | Type::Union | Type::None)
684 }
685 PlaceholderOption::TrueFalse(_) => {
686 matches!(
687 ty,
688 Type::Primitive(PrimitiveType::Boolean, _) | Type::Union | Type::None
689 )
690 }
691 };
692
693 if !valid {
694 self.context.add_diagnostic(invalid_placeholder_option(
695 &ty,
696 expr.span(),
697 &option,
698 ));
699 }
700 } else {
701 match ty {
702 Type::Primitive(..) | Type::Union | Type::None => {}
703 _ => {
704 self.context
705 .add_diagnostic(cannot_coerce_to_string(&ty, expr.span()));
706 }
707 }
708 }
709 }
710
711 self.placeholders -= 1;
712 }
713
714 fn evaluate_literal_array<N: TreeNode + SyntaxNodeExt>(
716 &mut self,
717 expr: &LiteralArray<N>,
718 ) -> Type {
719 let mut elements = expr.elements();
722 match elements
723 .next()
724 .and_then(|e| Some((self.evaluate_expr(&e)?, e.span())))
725 {
726 Some((mut expected, mut expected_span)) => {
727 for expr in elements {
729 if let Some(actual) = self.evaluate_expr(&expr) {
730 match expected.common_type(&actual) {
731 Some(ty) => {
732 expected = ty;
733 expected_span = expr.span();
734 }
735 _ => {
736 self.context.add_diagnostic(no_common_type(
737 &expected,
738 expected_span,
739 &actual,
740 expr.span(),
741 ));
742 }
743 }
744 }
745 }
746
747 ArrayType::new(expected).into()
748 }
749 None => ArrayType::new(Type::Union).into(),
751 }
752 }
753
754 fn evaluate_literal_pair<N: TreeNode + SyntaxNodeExt>(
756 &mut self,
757 expr: &LiteralPair<N>,
758 ) -> Type {
759 let (left, right) = expr.exprs();
760 let left = self.evaluate_expr(&left).unwrap_or(Type::Union);
761 let right = self.evaluate_expr(&right).unwrap_or(Type::Union);
762 PairType::new(left, right).into()
763 }
764
765 fn evaluate_literal_map<N: TreeNode + SyntaxNodeExt>(&mut self, expr: &LiteralMap<N>) -> Type {
767 let map_item_type = |item: LiteralMapItem<N>| {
768 let (key, value) = item.key_value();
769 let expected_key = self.evaluate_expr(&key)?;
770 match expected_key {
771 Type::Primitive(..) | Type::None | Type::Union => {
772 }
774 _ => {
775 self.context
776 .add_diagnostic(map_key_not_primitive(key.span(), &expected_key));
777 return None;
778 }
779 }
780
781 Some((
782 expected_key,
783 key.span(),
784 self.evaluate_expr(&value)?,
785 value.span(),
786 ))
787 };
788
789 let mut items = expr.items();
790 match items.next().and_then(map_item_type) {
791 Some((
792 mut expected_key,
793 mut expected_key_span,
794 mut expected_value,
795 mut expected_value_span,
796 )) => {
797 for item in items {
799 let (key, value) = item.key_value();
800 if let Some(actual_key) = self.evaluate_expr(&key)
801 && let Some(actual_value) = self.evaluate_expr(&value)
802 {
803 match expected_key.common_type(&actual_key) {
804 Some(ty) => {
805 expected_key = ty;
806 expected_key_span = key.span();
807 }
808 _ => {
809 self.context.add_diagnostic(no_common_type(
810 &expected_key,
811 expected_key_span,
812 &actual_key,
813 key.span(),
814 ));
815 }
816 }
817
818 match expected_value.common_type(&actual_value) {
819 Some(ty) => {
820 expected_value = ty;
821 expected_value_span = value.span();
822 }
823 _ => {
824 self.context.add_diagnostic(no_common_type(
825 &expected_value,
826 expected_value_span,
827 &actual_value,
828 value.span(),
829 ));
830 }
831 }
832 }
833 }
834
835 MapType::new(expected_key, expected_value).into()
836 }
837 None => MapType::new(Type::Union, Type::Union).into(),
839 }
840 }
841
842 fn evaluate_literal_object<N: TreeNode + SyntaxNodeExt>(
844 &mut self,
845 expr: &LiteralObject<N>,
846 ) -> Type {
847 for item in expr.items() {
849 let (_, v) = item.name_value();
850 self.evaluate_expr(&v);
851 }
852
853 Type::Object
854 }
855
856 fn evaluate_literal_struct<N: TreeNode + SyntaxNodeExt>(
858 &mut self,
859 expr: &LiteralStruct<N>,
860 ) -> Option<Type> {
861 let name = expr.name();
862 match self.context.resolve_type_name(name.text(), name.span()) {
863 Ok(ty) => {
864 let ty = match ty {
865 Type::Compound(CompoundType::Struct(ty), false) => ty,
866 _ => panic!("type should be a required struct"),
867 };
868
869 let mut present = vec![false; ty.members().len()];
871
872 for item in expr.items() {
874 let (n, v) = item.name_value();
875 match ty.members.get_full(n.text()) {
876 Some((index, _, expected)) => {
877 present[index] = true;
878 if let Some(actual) = self.evaluate_expr(&v)
879 && !actual.is_coercible_to(expected)
880 {
881 self.context.add_diagnostic(type_mismatch(
882 expected,
883 n.span(),
884 &actual,
885 v.span(),
886 ));
887 }
888 }
889 _ => {
890 self.context
892 .add_diagnostic(not_a_struct_member(name.text(), &n));
893 }
894 }
895 }
896
897 let mut unspecified = present
899 .iter()
900 .enumerate()
901 .filter_map(|(i, present)| {
902 if *present {
903 return None;
904 }
905
906 let (name, ty) = &ty.members.get_index(i).unwrap();
907 if ty.is_optional() {
908 return None;
909 }
910
911 Some(name.as_str())
912 })
913 .peekable();
914
915 if unspecified.peek().is_some() {
916 let mut members = String::new();
917 let mut count = 0;
918 while let Some(member) = unspecified.next() {
919 match (unspecified.peek().is_none(), count) {
920 (true, c) if c > 1 => members.push_str(", and "),
921 (true, 1) => members.push_str(" and "),
922 (false, c) if c > 0 => members.push_str(", "),
923 _ => {}
924 }
925
926 write!(&mut members, "`{member}`").ok();
927 count += 1;
928 }
929
930 self.context
931 .add_diagnostic(missing_struct_members(&name, count, &members));
932 }
933
934 Some(Type::Compound(CompoundType::Struct(ty), false))
935 }
936 Err(diagnostic) => {
937 self.context.add_diagnostic(diagnostic);
938 None
939 }
940 }
941 }
942
943 pub(crate) fn evaluate_runtime_item<N: TreeNode + SyntaxNodeExt>(
945 &mut self,
946 name: &Ident<N::Token>,
947 expr: &Expr<N>,
948 ) {
949 let expr_ty = self.evaluate_expr(expr).unwrap_or(Type::Union);
950 if !self.evaluate_requirement(name, expr, &expr_ty) {
951 if let Some(expected) = task_hint_types(self.context.version(), name.text(), false)
954 && !expected
955 .iter()
956 .any(|target| expr_ty.is_coercible_to(target))
957 {
958 self.context.add_diagnostic(multiple_type_mismatch(
959 expected,
960 name.span(),
961 &expr_ty,
962 expr.span(),
963 ));
964 }
965 }
966 }
967
968 pub(crate) fn evaluate_requirements_item<N: TreeNode + SyntaxNodeExt>(
970 &mut self,
971 name: &Ident<N::Token>,
972 expr: &Expr<N>,
973 ) {
974 let expr_ty = self.evaluate_expr(expr).unwrap_or(Type::Union);
975 self.evaluate_requirement(name, expr, &expr_ty);
976 }
977
978 fn evaluate_requirement<N: TreeNode>(
984 &mut self,
985 name: &Ident<N::Token>,
986 expr: &Expr<N>,
987 expr_ty: &Type,
988 ) -> bool {
989 if let Some(expected) = task_requirement_types(self.context.version(), name.text()) {
990 if !expected
991 .iter()
992 .any(|target| expr_ty.is_coercible_to(target))
993 {
994 self.context.add_diagnostic(multiple_type_mismatch(
995 expected,
996 name.span(),
997 expr_ty,
998 expr.span(),
999 ));
1000 }
1001
1002 return true;
1003 }
1004
1005 false
1006 }
1007
1008 fn evaluate_literal_hints<N: TreeNode + SyntaxNodeExt>(
1010 &mut self,
1011 expr: &LiteralHints<N>,
1012 ) -> Option<Type> {
1013 self.context.task()?;
1014
1015 for item in expr.items() {
1016 self.evaluate_hints_item(&item.name(), &item.expr())
1017 }
1018
1019 Some(Type::Hints)
1020 }
1021
1022 pub(crate) fn evaluate_hints_item<N: TreeNode + SyntaxNodeExt>(
1025 &mut self,
1026 name: &Ident<N::Token>,
1027 expr: &Expr<N>,
1028 ) {
1029 let expr_ty = self.evaluate_expr(expr).unwrap_or(Type::Union);
1030 if let Some(expected) = task_hint_types(self.context.version(), name.text(), true)
1031 && !expected
1032 .iter()
1033 .any(|target| expr_ty.is_coercible_to(target))
1034 {
1035 self.context.add_diagnostic(multiple_type_mismatch(
1036 expected,
1037 name.span(),
1038 &expr_ty,
1039 expr.span(),
1040 ));
1041 }
1042 }
1043
1044 fn evaluate_literal_input<N: TreeNode + SyntaxNodeExt>(
1046 &mut self,
1047 expr: &LiteralInput<N>,
1048 ) -> Option<Type> {
1049 self.context.task()?;
1051
1052 for item in expr.items() {
1054 self.evaluate_literal_io_item(item.names(), item.expr(), Io::Input);
1055 }
1056
1057 Some(Type::Input)
1058 }
1059
1060 fn evaluate_literal_output<N: TreeNode + SyntaxNodeExt>(
1062 &mut self,
1063 expr: &LiteralOutput<N>,
1064 ) -> Option<Type> {
1065 self.context.task()?;
1067
1068 for item in expr.items() {
1070 self.evaluate_literal_io_item(item.names(), item.expr(), Io::Output);
1071 }
1072
1073 Some(Type::Output)
1074 }
1075
1076 fn evaluate_literal_io_item<N: TreeNode + SyntaxNodeExt>(
1078 &mut self,
1079 names: impl Iterator<Item = Ident<N::Token>>,
1080 expr: Expr<N>,
1081 io: Io,
1082 ) {
1083 let mut names = names.enumerate().peekable();
1084 let expr_ty = self.evaluate_expr(&expr).unwrap_or(Type::Union);
1085
1086 let mut span = None;
1089 let mut s: Option<&StructType> = None;
1090 while let Some((i, name)) = names.next() {
1091 let ty = if i == 0 {
1093 span = Some(name.span());
1094
1095 match if io == Io::Input {
1096 self.context
1097 .task()
1098 .expect("should have task")
1099 .inputs()
1100 .get(name.text())
1101 .map(|i| i.ty())
1102 } else {
1103 self.context
1104 .task()
1105 .expect("should have task")
1106 .outputs()
1107 .get(name.text())
1108 .map(|o| o.ty())
1109 } {
1110 Some(ty) => ty,
1111 None => {
1112 self.context.add_diagnostic(unknown_task_io(
1113 self.context.task().expect("should have task").name(),
1114 &name,
1115 io,
1116 ));
1117 break;
1118 }
1119 }
1120 } else {
1121 let start = span.unwrap().start();
1123 span = Some(Span::new(start, name.span().end() - start));
1124 let s = s.unwrap();
1125 match s.members.get(name.text()) {
1126 Some(ty) => ty,
1127 None => {
1128 self.context
1129 .add_diagnostic(not_a_struct_member(&s.name, &name));
1130 break;
1131 }
1132 }
1133 };
1134
1135 match ty {
1136 Type::Compound(CompoundType::Struct(ty), _) => s = Some(ty),
1137 _ if names.peek().is_some() => {
1138 self.context.add_diagnostic(not_a_struct(&name, i == 0));
1139 break;
1140 }
1141 _ => {
1142 }
1144 }
1145 }
1146
1147 if let Some((_, last)) = names.last() {
1149 let start = span.unwrap().start();
1150 span = Some(Span::new(start, last.span().end() - start));
1151 }
1152
1153 if !expr_ty.is_coercible_to(&Type::Hints) {
1155 self.context.add_diagnostic(type_mismatch(
1156 &Type::Hints,
1157 span.expect("should have span"),
1158 &expr_ty,
1159 expr.span(),
1160 ));
1161 }
1162 }
1163
1164 fn evaluate_if_expr<N: TreeNode + SyntaxNodeExt>(&mut self, expr: &IfExpr<N>) -> Option<Type> {
1166 let (cond_expr, true_expr, false_expr) = expr.exprs();
1167
1168 let cond_ty = self.evaluate_expr(&cond_expr).unwrap_or(Type::Union);
1170 if !cond_ty.is_coercible_to(&PrimitiveType::Boolean.into()) {
1171 self.context
1172 .add_diagnostic(if_conditional_mismatch(&cond_ty, cond_expr.span()));
1173 }
1174
1175 let true_ty = self.evaluate_expr(&true_expr).unwrap_or(Type::Union);
1177 let false_ty = self.evaluate_expr(&false_expr).unwrap_or(Type::Union);
1178
1179 match (true_ty, false_ty) {
1180 (Type::Union, Type::Union) => None,
1181 (Type::Union, false_ty) => Some(false_ty),
1182 (true_ty, Type::Union) => Some(true_ty),
1183 (true_ty, false_ty) => match true_ty.common_type(&false_ty) {
1184 Some(ty) => Some(ty),
1185 _ => {
1186 self.context.add_diagnostic(type_mismatch(
1187 &true_ty,
1188 true_expr.span(),
1189 &false_ty,
1190 false_expr.span(),
1191 ));
1192
1193 None
1194 }
1195 },
1196 }
1197 }
1198
1199 fn evaluate_logical_not_expr<N: TreeNode + SyntaxNodeExt>(
1201 &mut self,
1202 expr: &LogicalNotExpr<N>,
1203 ) -> Option<Type> {
1204 let operand = expr.operand();
1206 let ty = self.evaluate_expr(&operand).unwrap_or(Type::Union);
1207 if !ty.is_coercible_to(&PrimitiveType::Boolean.into()) {
1208 self.context
1209 .add_diagnostic(logical_not_mismatch(&ty, operand.span()));
1210 }
1211
1212 Some(PrimitiveType::Boolean.into())
1213 }
1214
1215 fn evaluate_negation_expr<N: TreeNode + SyntaxNodeExt>(
1217 &mut self,
1218 expr: &NegationExpr<N>,
1219 ) -> Option<Type> {
1220 let operand = expr.operand();
1222 let ty = self.evaluate_expr(&operand)?;
1223
1224 if ty.eq(&PrimitiveType::Integer.into()) {
1227 return Some(PrimitiveType::Integer.into());
1228 }
1229
1230 if !ty.is_coercible_to(&PrimitiveType::Float.into()) {
1231 self.context
1232 .add_diagnostic(negation_mismatch(&ty, operand.span()));
1233 return None;
1235 }
1236
1237 Some(PrimitiveType::Float.into())
1238 }
1239
1240 fn evaluate_logical_or_expr<N: TreeNode + SyntaxNodeExt>(
1242 &mut self,
1243 expr: &LogicalOrExpr<N>,
1244 ) -> Option<Type> {
1245 let (lhs, rhs) = expr.operands();
1247
1248 let ty = self.evaluate_expr(&lhs).unwrap_or(Type::Union);
1249 if !ty.is_coercible_to(&PrimitiveType::Boolean.into()) {
1250 self.context
1251 .add_diagnostic(logical_or_mismatch(&ty, lhs.span()));
1252 }
1253
1254 let ty = self.evaluate_expr(&rhs).unwrap_or(Type::Union);
1255 if !ty.is_coercible_to(&PrimitiveType::Boolean.into()) {
1256 self.context
1257 .add_diagnostic(logical_or_mismatch(&ty, rhs.span()));
1258 }
1259
1260 Some(PrimitiveType::Boolean.into())
1261 }
1262
1263 fn evaluate_logical_and_expr<N: TreeNode + SyntaxNodeExt>(
1265 &mut self,
1266 expr: &LogicalAndExpr<N>,
1267 ) -> Option<Type> {
1268 let (lhs, rhs) = expr.operands();
1270
1271 let ty = self.evaluate_expr(&lhs).unwrap_or(Type::Union);
1272 if !ty.is_coercible_to(&PrimitiveType::Boolean.into()) {
1273 self.context
1274 .add_diagnostic(logical_and_mismatch(&ty, lhs.span()));
1275 }
1276
1277 let ty = self.evaluate_expr(&rhs).unwrap_or(Type::Union);
1278 if !ty.is_coercible_to(&PrimitiveType::Boolean.into()) {
1279 self.context
1280 .add_diagnostic(logical_and_mismatch(&ty, rhs.span()));
1281 }
1282
1283 Some(PrimitiveType::Boolean.into())
1284 }
1285
1286 fn evaluate_comparison_expr<N: TreeNode + SyntaxNodeExt>(
1288 &mut self,
1289 op: ComparisonOperator,
1290 lhs: &Expr<N>,
1291 rhs: &Expr<N>,
1292 span: Span,
1293 ) -> Option<Type> {
1294 let lhs_ty = self.evaluate_expr(lhs).unwrap_or(Type::Union);
1295 let rhs_ty = self.evaluate_expr(rhs).unwrap_or(Type::Union);
1296
1297 if lhs_ty.is_union() || lhs_ty.is_none() || rhs_ty.is_union() || rhs_ty.is_none() {
1299 return Some(PrimitiveType::Boolean.into());
1300 }
1301
1302 for expected in [
1304 Type::from(PrimitiveType::Boolean),
1305 PrimitiveType::Integer.into(),
1306 PrimitiveType::Float.into(),
1307 PrimitiveType::String.into(),
1308 PrimitiveType::File.into(),
1309 PrimitiveType::Directory.into(),
1310 ] {
1311 if op != ComparisonOperator::Equality
1313 && op != ComparisonOperator::Inequality
1314 && (matches!(
1315 lhs_ty.as_primitive(),
1316 Some(PrimitiveType::File) | Some(PrimitiveType::Directory)
1317 ) || matches!(
1318 rhs_ty.as_primitive(),
1319 Some(PrimitiveType::File) | Some(PrimitiveType::Directory)
1320 ))
1321 {
1322 continue;
1323 }
1324
1325 if lhs_ty.is_coercible_to(&expected) && rhs_ty.is_coercible_to(&expected) {
1326 return Some(PrimitiveType::Boolean.into());
1327 }
1328
1329 let expected = expected.optional();
1330 if lhs_ty.is_coercible_to(&expected) && rhs_ty.is_coercible_to(&expected) {
1331 return Some(PrimitiveType::Boolean.into());
1332 }
1333 }
1334
1335 if op == ComparisonOperator::Equality || op == ComparisonOperator::Inequality {
1337 if (lhs_ty.is_coercible_to(&Type::Object) && rhs_ty.is_coercible_to(&Type::Object))
1339 || (lhs_ty.is_coercible_to(&Type::OptionalObject)
1340 && rhs_ty.is_coercible_to(&Type::OptionalObject))
1341 {
1342 return Some(PrimitiveType::Boolean.into());
1343 }
1344
1345 let equal = match (&lhs_ty, &rhs_ty) {
1347 (
1348 Type::Compound(CompoundType::Array(a), _),
1349 Type::Compound(CompoundType::Array(b), _),
1350 ) => a == b,
1351 (
1352 Type::Compound(CompoundType::Pair(a), _),
1353 Type::Compound(CompoundType::Pair(b), _),
1354 ) => a == b,
1355 (
1356 Type::Compound(CompoundType::Map(a), _),
1357 Type::Compound(CompoundType::Map(b), _),
1358 ) => a == b,
1359 (
1360 Type::Compound(CompoundType::Struct(a), _),
1361 Type::Compound(CompoundType::Struct(b), _),
1362 ) => a == b,
1363 _ => false,
1364 };
1365
1366 if equal {
1367 return Some(PrimitiveType::Boolean.into());
1368 }
1369 }
1370
1371 self.context.add_diagnostic(comparison_mismatch(
1373 op,
1374 span,
1375 &lhs_ty,
1376 lhs.span(),
1377 &rhs_ty,
1378 rhs.span(),
1379 ));
1380 Some(PrimitiveType::Boolean.into())
1381 }
1382
1383 fn evaluate_numeric_expr<N: TreeNode + SyntaxNodeExt>(
1385 &mut self,
1386 op: NumericOperator,
1387 span: Span,
1388 lhs: &Expr<N>,
1389 rhs: &Expr<N>,
1390 ) -> Option<Type> {
1391 let lhs_ty = self.evaluate_expr(lhs).unwrap_or(Type::Union);
1392 let rhs_ty = self.evaluate_expr(rhs).unwrap_or(Type::Union);
1393
1394 if lhs_ty.eq(&PrimitiveType::Integer.into()) && rhs_ty.eq(&PrimitiveType::Integer.into()) {
1396 return Some(PrimitiveType::Integer.into());
1397 }
1398
1399 if !lhs_ty.is_union()
1401 && lhs_ty.is_coercible_to(&PrimitiveType::Float.into())
1402 && !rhs_ty.is_union()
1403 && rhs_ty.is_coercible_to(&PrimitiveType::Float.into())
1404 {
1405 return Some(PrimitiveType::Float.into());
1406 }
1407
1408 if op == NumericOperator::Addition {
1412 let allow_optional = self.placeholders > 0;
1413 let other = if (!lhs_ty.is_optional() || allow_optional)
1414 && lhs_ty
1415 .as_primitive()
1416 .map(|p| p == PrimitiveType::String)
1417 .unwrap_or(false)
1418 {
1419 Some((lhs_ty.is_optional(), &rhs_ty, rhs.span()))
1420 } else if (!rhs_ty.is_optional() || allow_optional)
1421 && rhs_ty
1422 .as_primitive()
1423 .map(|p| p == PrimitiveType::String)
1424 .unwrap_or(false)
1425 {
1426 Some((rhs_ty.is_optional(), &lhs_ty, lhs.span()))
1427 } else {
1428 None
1429 };
1430
1431 if let Some((optional, other, span)) = other {
1432 if (!other.is_optional() || allow_optional)
1433 && other
1434 .as_primitive()
1435 .map(|p| p != PrimitiveType::Boolean)
1436 .unwrap_or(other.is_union() || (allow_optional && other.is_none()))
1437 {
1438 let ty: Type = PrimitiveType::String.into();
1439 if optional || other.is_optional() {
1440 return Some(ty.optional());
1441 }
1442
1443 return Some(ty);
1444 }
1445
1446 self.context
1447 .add_diagnostic(string_concat_mismatch(other, span));
1448 return None;
1449 }
1450 }
1451
1452 if !lhs_ty.is_union() && !rhs_ty.is_union() {
1453 self.context.add_diagnostic(numeric_mismatch(
1454 op,
1455 span,
1456 &lhs_ty,
1457 lhs.span(),
1458 &rhs_ty,
1459 rhs.span(),
1460 ));
1461 }
1462
1463 None
1464 }
1465
1466 fn evaluate_call_expr<N: TreeNode + SyntaxNodeExt>(
1468 &mut self,
1469 expr: &CallExpr<N>,
1470 ) -> Option<Type> {
1471 let target = expr.target();
1472 match STDLIB.function(target.text()) {
1473 Some(f) => {
1474 let mut count = 0;
1476 let mut arguments = [const { Type::Union }; MAX_PARAMETERS];
1477
1478 for arg in expr.arguments() {
1479 if count < MAX_PARAMETERS {
1480 arguments[count] = self.evaluate_expr(&arg).unwrap_or(Type::Union);
1481 }
1482
1483 count += 1;
1484 }
1485
1486 match target.text() {
1487 "find" | "matches" | "sub" => {
1488 if let Some(Expr::Literal(LiteralExpr::String(pattern_literal))) =
1490 expr.arguments().nth(1)
1491 && let Some(value) = pattern_literal.text()
1492 {
1493 let pattern = value.text().to_string();
1494 if let Err(e) = regex::Regex::new(&pattern) {
1495 self.context.add_diagnostic(invalid_regex_pattern(
1496 target.text(),
1497 value.text(),
1498 &e,
1499 pattern_literal.span(),
1500 ));
1501 }
1502 }
1503 }
1504 _ => {}
1505 }
1506
1507 let arguments = &arguments[..count.min(MAX_PARAMETERS)];
1508 if count <= MAX_PARAMETERS {
1509 match f.bind(self.context.version(), arguments) {
1510 Ok(binding) => {
1511 if let Some(severity) =
1512 self.context.diagnostics_config().unnecessary_function_call
1513 && !expr.inner().is_rule_excepted(UNNECESSARY_FUNCTION_CALL)
1514 {
1515 self.check_unnecessary_call(
1516 &target,
1517 arguments,
1518 expr.arguments().map(|e| e.span()),
1519 severity,
1520 );
1521 }
1522 return Some(binding.return_type().clone());
1523 }
1524 Err(FunctionBindError::RequiresVersion(minimum)) => {
1525 self.context.add_diagnostic(unsupported_function(
1526 minimum,
1527 target.text(),
1528 target.span(),
1529 ));
1530 }
1531 Err(FunctionBindError::TooFewArguments(minimum)) => {
1532 self.context.add_diagnostic(too_few_arguments(
1533 target.text(),
1534 target.span(),
1535 minimum,
1536 count,
1537 ));
1538 }
1539 Err(FunctionBindError::TooManyArguments(maximum)) => {
1540 self.context.add_diagnostic(too_many_arguments(
1541 target.text(),
1542 target.span(),
1543 maximum,
1544 count,
1545 expr.arguments().skip(maximum).map(|e| e.span()),
1546 ));
1547 }
1548 Err(FunctionBindError::ArgumentTypeMismatch { index, expected }) => {
1549 self.context.add_diagnostic(argument_type_mismatch(
1550 target.text(),
1551 &expected,
1552 &arguments[index],
1553 expr.arguments()
1554 .nth(index)
1555 .map(|e| e.span())
1556 .expect("should have span"),
1557 ));
1558 }
1559 Err(FunctionBindError::Ambiguous { first, second }) => {
1560 self.context.add_diagnostic(ambiguous_argument(
1561 target.text(),
1562 target.span(),
1563 &first,
1564 &second,
1565 ));
1566 }
1567 }
1568 } else {
1569 match f.param_min_max(self.context.version()) {
1571 Some((_, max)) => {
1572 assert!(max <= MAX_PARAMETERS);
1573 self.context.add_diagnostic(too_many_arguments(
1574 target.text(),
1575 target.span(),
1576 max,
1577 count,
1578 expr.arguments().skip(max).map(|e| e.span()),
1579 ));
1580 }
1581 None => {
1582 self.context.add_diagnostic(unsupported_function(
1583 f.minimum_version(),
1584 target.text(),
1585 target.span(),
1586 ));
1587 }
1588 }
1589 }
1590
1591 Some(f.realize_unconstrained_return_type(arguments))
1592 }
1593 None => {
1594 self.context
1595 .add_diagnostic(unknown_function(target.text(), target.span()));
1596 None
1597 }
1598 }
1599 }
1600
1601 fn evaluate_index_expr<N: TreeNode + SyntaxNodeExt>(
1603 &mut self,
1604 expr: &IndexExpr<N>,
1605 ) -> Option<Type> {
1606 let (target, index) = expr.operands();
1607
1608 let target_ty = self.evaluate_expr(&target)?;
1610 let (expected_index_ty, result_ty) = match &target_ty {
1611 Type::Compound(CompoundType::Array(ty), _) => (
1612 Some(PrimitiveType::Integer.into()),
1613 Some(ty.element_type().clone()),
1614 ),
1615 Type::Compound(CompoundType::Map(ty), _) => {
1616 (Some(ty.key_type().clone()), Some(ty.value_type().clone()))
1617 }
1618 _ => (None, None),
1619 };
1620
1621 if let Some(expected_index_ty) = expected_index_ty {
1623 let index_ty = self.evaluate_expr(&index).unwrap_or(Type::Union);
1624 if !index_ty.is_coercible_to(&expected_index_ty) {
1625 self.context.add_diagnostic(index_type_mismatch(
1626 &expected_index_ty,
1627 &index_ty,
1628 index.span(),
1629 ));
1630 }
1631 }
1632
1633 match result_ty {
1634 Some(ty) => Some(ty),
1635 None => {
1636 self.context
1637 .add_diagnostic(cannot_index(&target_ty, target.span()));
1638 None
1639 }
1640 }
1641 }
1642
1643 fn evaluate_access_expr<N: TreeNode + SyntaxNodeExt>(
1645 &mut self,
1646 expr: &AccessExpr<N>,
1647 ) -> Option<Type> {
1648 let (target, name) = expr.operands();
1649 let ty = self.evaluate_expr(&target)?;
1650
1651 if matches!(ty, Type::Task) {
1652 return match task_member_type(name.text()) {
1653 Some(ty) => Some(ty),
1654 None => {
1655 self.context.add_diagnostic(not_a_task_member(&name));
1656 return None;
1657 }
1658 };
1659 }
1660
1661 match &ty {
1663 Type::Compound(CompoundType::Struct(ty), _) => {
1664 if let Some(ty) = ty.members.get(name.text()) {
1665 return Some(ty.clone());
1666 }
1667
1668 self.context
1669 .add_diagnostic(not_a_struct_member(ty.name(), &name));
1670 return None;
1671 }
1672 Type::Compound(CompoundType::Pair(ty), _) => {
1673 return match name.text() {
1675 "left" => Some(ty.left_type.clone()),
1676 "right" => Some(ty.right_type.clone()),
1677 _ => {
1678 self.context.add_diagnostic(not_a_pair_accessor(&name));
1679 None
1680 }
1681 };
1682 }
1683 Type::Call(ty) => {
1684 if let Some(output) = ty.outputs().get(name.text()) {
1685 return Some(output.ty().clone());
1686 }
1687
1688 self.context
1689 .add_diagnostic(unknown_call_io(ty, &name, Io::Output));
1690 return None;
1691 }
1692 _ => {}
1693 }
1694
1695 if ty.is_coercible_to(&Type::OptionalObject) {
1698 return Some(Type::Union);
1699 }
1700
1701 self.context
1702 .add_diagnostic(cannot_access(&ty, target.span()));
1703 None
1704 }
1705
1706 fn check_unnecessary_call<T: TreeToken>(
1708 &mut self,
1709 target: &Ident<T>,
1710 arguments: &[Type],
1711 mut spans: impl Iterator<Item = Span>,
1712 severity: Severity,
1713 ) {
1714 let (label, span, fix) = match target.text() {
1715 "select_first" => {
1716 if let Some(ty) = arguments[0].as_array().map(|a| a.element_type()) {
1717 if ty.is_optional() || ty.is_union() {
1718 return;
1719 }
1720 (
1721 format!("array element type `{ty}` is not optional"),
1722 spans.next().expect("should have span"),
1723 "replace the function call with the array's first element",
1724 )
1725 } else {
1726 return;
1727 }
1728 }
1729 "select_all" => {
1730 if let Some(ty) = arguments[0].as_array().map(|a| a.element_type()) {
1731 if ty.is_optional() || ty.is_union() {
1732 return;
1733 }
1734 (
1735 format!("array element type `{ty}` is not optional"),
1736 spans.next().expect("should have span"),
1737 "replace the function call with the array itself",
1738 )
1739 } else {
1740 return;
1741 }
1742 }
1743 "defined" => {
1744 if arguments[0].is_optional() || arguments[0].is_union() {
1745 return;
1746 }
1747
1748 (
1749 format!("type `{ty}` is not optional", ty = arguments[0]),
1750 spans.next().expect("should have span"),
1751 "replace the function call with `true`",
1752 )
1753 }
1754 _ => return,
1755 };
1756
1757 self.context.add_diagnostic(
1758 unnecessary_function_call(target.text(), target.span(), &label, span)
1759 .with_severity(severity)
1760 .with_fix(fix),
1761 )
1762 }
1763}