Lines
19.03 %
Functions
10 %
use super::util::named_children;
use crate::EssenceParseError;
use conjure_cp_core::ast::{
DeclarationPtr, Domain, DomainPtr, IntVal, Name, Range, RecordEntry, Reference, SetAttr,
SymbolTablePtr,
};
use core::panic;
use std::str::FromStr;
use tree_sitter::Node;
/// Parse an Essence variable domain into its Conjure AST representation.
pub fn parse_domain(
domain: Node,
source_code: &str,
symbols: Option<SymbolTablePtr>,
) -> Result<DomainPtr, EssenceParseError> {
match domain.kind() {
"domain" => parse_domain(
domain.child(0).expect("No domain found"),
source_code,
symbols,
),
"bool_domain" => Ok(Domain::bool()),
"int_domain" => Ok(parse_int_domain(domain, source_code, &symbols)),
"identifier" => {
let decl = get_declaration_ptr_from_identifier(domain, source_code, &symbols)?;
let dom = Domain::reference(decl).ok_or(EssenceParseError::syntax_error(
format!(
"'{}' is not a valid domain declaration",
&source_code[domain.start_byte()..domain.end_byte()]
Some(domain.range()),
))?;
Ok(dom)
}
"tuple_domain" => parse_tuple_domain(domain, source_code, symbols),
"matrix_domain" => parse_matrix_domain(domain, source_code, symbols),
"record_domain" => parse_record_domain(domain, source_code, symbols),
"set_domain" => parse_set_domain(domain, source_code, symbols),
_ => panic!("{} is not a supported domain type", domain.kind()),
fn get_declaration_ptr_from_identifier(
identifier: Node,
symbols_ptr: &Option<SymbolTablePtr>,
) -> Result<DeclarationPtr, EssenceParseError> {
let name = Name::user(&source_code[identifier.start_byte()..identifier.end_byte()]);
let decl = symbols_ptr
.as_ref()
.ok_or(EssenceParseError::syntax_error(
"context needed to resolve identifier".to_string(),
Some(identifier.range()),
))?
.read()
.lookup(&name)
format!("'{name}' is not defined"),
Ok(decl)
/// Parse an integer domain. Can be a single integer or a range.
fn parse_int_domain(
int_domain: Node,
) -> DomainPtr {
if int_domain.child_count() == 1 {
return Domain::int(vec![Range::Bounded(i32::MIN, i32::MAX)]);
let mut ranges: Vec<Range<i32>> = Vec::new();
let mut ranges_unresolved: Vec<Range<IntVal>> = Vec::new();
let range_list = int_domain
.child_by_field_name("ranges")
.expect("No range list found for int domain");
for domain_component in named_children(&range_list) {
match domain_component.kind() {
"atom" => {
let text = &source_code[domain_component.start_byte()..domain_component.end_byte()];
// Try parsing as a literal integer first
if let Ok(integer) = text.parse::<i32>() {
ranges.push(Range::Single(integer));
continue;
// Otherwise, treat as a reference
let decl =
get_declaration_ptr_from_identifier(domain_component, source_code, symbols_ptr);
if let Ok(decl) = decl {
ranges_unresolved.push(Range::Single(IntVal::Reference(Reference::new(decl))));
} else {
panic!("'{}' is not a valid integer", text);
"int_range" => {
let lower_bound: Option<Result<i32, DeclarationPtr>> =
match domain_component.child_by_field_name("lower") {
Some(lower_node) => {
let text = &source_code[lower_node.start_byte()..lower_node.end_byte()];
Some(Ok(integer))
let decl = get_declaration_ptr_from_identifier(
lower_node,
symbols_ptr,
);
Some(Err(decl))
None => None,
let upper_bound: Option<Result<i32, DeclarationPtr>> =
match domain_component.child_by_field_name("upper") {
Some(upper_node) => {
let text = &source_code[upper_node.start_byte()..upper_node.end_byte()];
upper_node,
match (lower_bound, upper_bound) {
(Some(Ok(lower)), Some(Ok(upper))) => ranges.push(Range::Bounded(lower, upper)),
(Some(Ok(lower)), Some(Err(decl))) => {
ranges_unresolved.push(Range::Bounded(
IntVal::Const(lower),
IntVal::Reference(Reference::new(decl)),
));
(Some(Err(decl)), Some(Ok(upper))) => {
IntVal::Const(upper),
(Some(Err(decl_lower)), Some(Err(decl_upper))) => {
IntVal::Reference(Reference::new(decl_lower)),
IntVal::Reference(Reference::new(decl_upper)),
(Some(Ok(lower)), None) => {
ranges.push(Range::UnboundedR(lower));
(Some(Err(decl)), None) => {
ranges_unresolved
.push(Range::UnboundedR(IntVal::Reference(Reference::new(decl))));
(None, Some(Ok(upper))) => {
ranges.push(Range::UnboundedL(upper));
(None, Some(Err(decl))) => {
.push(Range::UnboundedL(IntVal::Reference(Reference::new(decl))));
(None, None) => {
ranges.push(Range::Unbounded);
_ => panic!("unsupported int range type"),
if !ranges_unresolved.is_empty() {
for range in ranges {
match range {
Range::Single(i) => ranges_unresolved.push(Range::Single(IntVal::Const(i))),
Range::Bounded(l, u) => {
ranges_unresolved.push(Range::Bounded(IntVal::Const(l), IntVal::Const(u)))
Range::UnboundedL(l) => ranges_unresolved.push(Range::UnboundedL(IntVal::Const(l))),
Range::UnboundedR(u) => ranges_unresolved.push(Range::UnboundedR(IntVal::Const(u))),
Range::Unbounded => ranges_unresolved.push(Range::Unbounded),
return Domain::int(ranges_unresolved);
Domain::int(ranges)
fn parse_tuple_domain(
tuple_domain: Node,
let mut domains: Vec<DomainPtr> = Vec::new();
for domain in named_children(&tuple_domain) {
domains.push(parse_domain(domain, source_code, symbols.clone())?);
Ok(Domain::tuple(domains))
fn parse_matrix_domain(
matrix_domain: Node,
let index_domain_list = matrix_domain
.child_by_field_name("index_domain_list")
.expect("No index domains found for matrix domain");
for domain in named_children(&index_domain_list) {
let value_domain = parse_domain(
matrix_domain.child_by_field_name("value_domain").ok_or(
EssenceParseError::syntax_error(
"Expected a value domain".to_string(),
Some(matrix_domain.range()),
)?,
)?;
Ok(Domain::matrix(value_domain, domains))
fn parse_record_domain(
record_domain: Node,
let mut record_entries: Vec<RecordEntry> = Vec::new();
for record_entry in named_children(&record_domain) {
let name_node = record_entry
.child_by_field_name("name")
.expect("No name found for record entry");
let name = Name::user(&source_code[name_node.start_byte()..name_node.end_byte()]);
let domain_node = record_entry
.child_by_field_name("domain")
.expect("No domain found for record entry");
let domain = parse_domain(domain_node, source_code, symbols.clone())?;
record_entries.push(RecordEntry { name, domain });
Ok(Domain::record(record_entries))
pub fn parse_set_domain(
set_domain: Node,
let mut set_attribute: Option<SetAttr> = None;
let mut value_domain: Option<DomainPtr> = None;
for child in named_children(&set_domain) {
match child.kind() {
"set_attributes" => {
// Check if we have both minSize and maxSize (minMax case)
let min_value_node = child.child_by_field_name("min_value");
let max_value_node = child.child_by_field_name("max_value");
let size_value_node = child.child_by_field_name("size_value");
if let (Some(min_node), Some(max_node)) = (min_value_node, max_value_node) {
// MinMax case
let min_str = &source_code[min_node.start_byte()..min_node.end_byte()];
let max_str = &source_code[max_node.start_byte()..max_node.end_byte()];
let min_val = i32::from_str(min_str).map_err(|_| {
format!("Invalid integer value for minSize: {}", min_str),
Some(min_node.range()),
)
})?;
let max_val = i32::from_str(max_str).map_err(|_| {
format!("Invalid integer value for maxSize: {}", max_str),
Some(max_node.range()),
set_attribute = Some(SetAttr::new_min_max_size(min_val, max_val));
} else if let Some(size_node) = size_value_node {
// Size case
let size_str = &source_code[size_node.start_byte()..size_node.end_byte()];
let size_val = i32::from_str(size_str).map_err(|_| {
format!("Invalid integer value for size: {}", size_str),
Some(size_node.range()),
set_attribute = Some(SetAttr::new_size(size_val));
} else if let Some(min_node) = min_value_node {
// MinSize only case
set_attribute = Some(SetAttr::new_min_size(min_val));
} else if let Some(max_node) = max_value_node {
// MaxSize only case
set_attribute = Some(SetAttr::new_max_size(max_val));
"domain" => {
value_domain = Some(parse_domain(child, source_code, symbols.clone())?);
_ => {
return Err(EssenceParseError::syntax_error(
format!("Unrecognized set domain child kind: {}", child.kind()),
Some(child.range()),
if let Some(domain) = value_domain {
Ok(Domain::set(set_attribute.unwrap_or_default(), domain))
Err(EssenceParseError::syntax_error(
"Set domain must have a value domain".to_string(),
Some(set_domain.range()),
))