Skip to content
This repository has been archived by the owner on Jun 3, 2024. It is now read-only.

add test for modules in modules #36

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

add test for modules in modules #36

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
5297339
add test for modules in modules
yaahc Feb 22, 2024
38a7c78
Add support for calls with more than two path segments
yaahc Feb 22, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Empty file added docs/language_overview.md
View file
Open in desktop
Empty file.
188 changes: 107 additions & 81 deletions src/codegen.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
use tracing::debug;

use crate::{
compiler::{CallTarget, Compiler},
lifetime_checker::AllocationLifetime,
Expand Down Expand Up @@ -494,7 +496,11 @@ impl Codegen {
output.extend_from_slice(format!("allocation_id + {} ", level).as_bytes())
}
AllocationLifetime::Unknown => {
// panic!("found 'unknown' lifetime during codegen")
debug!(
?node_id,
name = ?String::from_utf8_lossy(self.compiler.get_source(node_id)),
"found 'unknown' lifetime during codegen",
);
output.extend_from_slice(b"/* UNKNOWN, */ ")
}
}
Expand Down Expand Up @@ -822,109 +828,129 @@ impl Codegen {
panic!("internal error: expected allocation call during allocation")
}
}
AstNode::NamespacedLookup { item, .. } => match self.compiler.get_node(*item) {
AstNode::Call { head, args } => {
let call_target = self
.compiler
.call_resolution
.get(head)
.expect("internal error: missing call resolution in codegen");

match call_target {
CallTarget::Function(fun_id) => {
output.extend_from_slice(b"/* ");
output.extend_from_slice(
self.compiler
.get_source(self.compiler.functions[fun_id.0].name),
);
output.extend_from_slice(b" */ ");
AstNode::NamespacedLookup { item, .. } => {
let mut codegen_call = |call_target, args: &[NodeId]| match call_target {
CallTarget::Function(fun_id) => {
output.extend_from_slice(b"/* ");
output.extend_from_slice(
self.compiler
.get_source(self.compiler.functions[fun_id.0].name),
);
output.extend_from_slice(b" */ ");

output.extend_from_slice(b"function_");
output.extend_from_slice(fun_id.0.to_string().as_bytes());
output.push(b'(');
output.extend_from_slice(b"function_");
output.extend_from_slice(fun_id.0.to_string().as_bytes());
output.push(b'(');

self.codegen_annotation(node_id, output);
self.codegen_annotation(node_id, output);

for arg in args {
output.extend_from_slice(b", ");
for arg in args {
output.extend_from_slice(b", ");

self.codegen_node(*arg, local_inferences, output)
}
output.push(b')');
self.codegen_node(*arg, local_inferences, output)
}
CallTarget::EnumConstructor(target, offset) => {
output.extend_from_slice(b"enum_case_");
output.extend_from_slice(target.0.to_string().as_bytes());
output.push(b'_');
output.extend_from_slice(offset.0.to_string().as_bytes());
output.push(b'(');
output.push(b')');
}
CallTarget::EnumConstructor(target, offset) => {
output.extend_from_slice(b"enum_case_");
output.extend_from_slice(target.0.to_string().as_bytes());
output.push(b'_');
output.extend_from_slice(offset.0.to_string().as_bytes());
output.push(b'(');

self.codegen_annotation(node_id, output);
self.codegen_annotation(node_id, output);

for arg in args {
output.extend_from_slice(b", ");
for arg in args {
output.extend_from_slice(b", ");

self.codegen_node(*arg, local_inferences, output)
}
output.push(b')');
}
_ => {
unimplemented!("namedspaced lookup of non-namedspaced value")
self.codegen_node(*arg, local_inferences, output)
}
output.push(b')');
}
}
AstNode::Name => {
let call_target = self
.compiler
.call_resolution
.get(item)
.expect("internal error: missing call resolution in codegen");

match call_target {
CallTarget::Function(fun_id) => {
output.extend_from_slice(b"/* ");
output.extend_from_slice(
self.compiler
.get_source(self.compiler.functions[fun_id.0].name),
);
output.extend_from_slice(b" */ ");
_ => {
unimplemented!("namedspaced lookup of non-namedspaced value")
}
};

output.extend_from_slice(b"function_");
output.extend_from_slice(fun_id.0.to_string().as_bytes());
output.push(b'(');
self.compiler.debug_node(node_id);

self.codegen_annotation(node_id, output);
if let Some(call_target) = self.compiler.call_resolution.get(&node_id) {
let mut curr_item = *item;

output.push(b')');
let args = loop {
match self.compiler.get_node(curr_item) {
AstNode::NamespacedLookup { item, .. } => curr_item = *item,
AstNode::Call { args, .. } => break args.clone(),
_ => unreachable!("having a valid call target should guarantee that this namespaced lookup ends in a call"),
}
CallTarget::EnumConstructor(target, offset) => {
output.extend_from_slice(b"enum_case_");
output.extend_from_slice(target.0.to_string().as_bytes());
output.push(b'_');
output.extend_from_slice(offset.0.to_string().as_bytes());
output.push(b'(');
};

self.codegen_annotation(node_id, output);
codegen_call(*call_target, &args);
} else {
match self.compiler.get_node(*item) {
AstNode::Call { head, args } => {
let call_target = self
.compiler
.call_resolution
.get(head)
.expect("internal error: missing call resolution in codegen");

output.push(b')');
codegen_call(*call_target, args)
}
CallTarget::NodeId(target) => {
output.push(b'(');
self.codegen_node(*target, local_inferences, output);
output.push(b')');
AstNode::Name => {
let call_target = self
.compiler
.call_resolution
.get(item)
.expect("internal error: missing call resolution in codegen");

output.push(b'(');
match call_target {
CallTarget::Function(fun_id) => {
output.extend_from_slice(b"/* ");
output.extend_from_slice(
self.compiler
.get_source(self.compiler.functions[fun_id.0].name),
);
output.extend_from_slice(b" */ ");

self.codegen_annotation(node_id, output);
output.extend_from_slice(b"function_");
output.extend_from_slice(fun_id.0.to_string().as_bytes());
output.push(b'(');

self.codegen_annotation(node_id, output);

output.push(b')');
}
CallTarget::EnumConstructor(target, offset) => {
output.extend_from_slice(b"enum_case_");
output.extend_from_slice(target.0.to_string().as_bytes());
output.push(b'_');
output.extend_from_slice(offset.0.to_string().as_bytes());
output.push(b'(');

self.codegen_annotation(node_id, output);

output.push(b')');
}
CallTarget::NodeId(target) => {
output.push(b'(');
self.codegen_node(*target, local_inferences, output);
output.push(b')');

output.push(b')');
output.push(b'(');

self.codegen_annotation(node_id, output);

output.push(b')');
}
}
}
_ => {
panic!("unsupported namespace lookup")
}
}
}
_ => {
panic!("unsupported namespace lookup")
}
},
}
AstNode::NamedValue { value, .. } => {
// FIXME: this should probably be handled cleanly via typecheck+codegen
// rather than ignoring the name
Expand Down
5 changes: 5 additions & 0 deletions src/compiler.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -776,4 +776,9 @@ impl Compiler {
self.module_resolution.insert(module_block, module_id);
module_id
}

pub(crate) fn debug_node(&self, node_id: NodeId) {
let node = String::from_utf8_lossy(self.get_source(node_id));
tracing::debug!(?node);
}
}
Loading