use super::{CodecTarget, Constant, Handle, Handoff, Routine, Scalar};
use crate::adapter::zig::{field_name, zig_map_entry_name, zig_type_name, zig_value_type};
use crate::encode::EncodeError;
use zerodds_ast::{Ast, CharWidth, MemberNode, TypeKind, TypeNode};
use zerodds_ast_core::Id;
use zerodds_lang_cst::common::{Doc, Ident, Literal};
use zerodds_lang_cst::zig;
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct Zig;
const MAX_TYPE_DEPTH: usize = 64;
const RUNTIME: &str = "zdwire";
const SIGNED_CHAR: &str = "signed char";
const FALLIBLE_EXEMPT_METHODS: &[&str] = &[
"reader_status",
"reader_position",
"reader_remaining",
"reader_save",
"reader_restore",
"reader_refuse",
"reader_clear_discard",
"reader_begin_packed_element",
"reader_finish_packed_element",
"begin_packed_element",
"finish_packed_element",
"finish_pid_read",
"position",
"trim_utf8",
"trim_utf16",
];
const FALLIBLE_STATICS: &[&str] = &["utf16_units"];
#[derive(Debug, Clone)]
pub enum ZigExpr {
Plain(zig::Expr),
Fallible(zig::Expr),
OptionalSlot(zig::Expr),
ExternalSlot(zig::Expr),
}
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct ZigType {
pub tree: zig::Type,
pub enum_like: bool,
pub integer_alias: bool,
}
impl ZigType {
fn plain(tree: zig::Type) -> Self {
Self {
tree,
enum_like: false,
integer_alias: false,
}
}
}
fn inner(expr: ZigExpr) -> zig::Expr {
match expr {
ZigExpr::Plain(e) | ZigExpr::OptionalSlot(e) | ZigExpr::ExternalSlot(e) => e,
ZigExpr::Fallible(e) => zig::Expr::Paren(Box::new(zig::Expr::try_(e))),
}
}
fn inner_bare(expr: ZigExpr) -> zig::Expr {
match expr {
ZigExpr::Plain(e) | ZigExpr::OptionalSlot(e) | ZigExpr::ExternalSlot(e) => e,
ZigExpr::Fallible(e) => zig::Expr::try_(e),
}
}
fn plain(e: zig::Expr) -> ZigExpr {
ZigExpr::Plain(e)
}
fn local_name(text: &str) -> String {
if zig::is_primitive_name(text) {
return format!("zd_{text}");
}
text.to_string()
}
impl Zig {
fn runtime_name(name: &str) -> zig::Expr {
zig::Expr::dot(zig::Expr::Name(Ident::new(RUNTIME)), name)
}
fn camel(method: &str) -> String {
let mut out = String::new();
for (index, segment) in method.split('_').enumerate() {
if index == 0 {
out.push_str(segment);
continue;
}
let mut chars = segment.chars();
if let Some(first) = chars.next() {
out.extend(first.to_uppercase());
out.push_str(chars.as_str());
}
}
out
}
fn text(value: &str) -> ZigExpr {
plain(zig::Expr::Literal(Literal::Str(value.to_string())))
}
fn runtime_enum(type_: &str, value: &str) -> ZigExpr {
plain(zig::Expr::dot(Self::runtime_name(type_), value))
}
fn reader_heap() -> zig::Expr {
zig::Expr::dot(zig::Expr::Name(Ident::new("r")), "heap")
}
fn is_enumeration(ast: &Ast, id: Id<TypeNode>) -> bool {
super::resolves_to_enumeration(ast, id)
}
fn is_bitmask(ast: &Ast, id: Id<TypeNode>) -> bool {
let mut current = id;
for _ in 0..MAX_TYPE_DEPTH {
match ast.type_(current).map(|node| node.kind.clone()) {
Ok(TypeKind::Bitmask { .. }) => return true,
Ok(TypeKind::Alias { aliased }) => match aliased.resolved() {
Some(target) => current = target,
None => return false,
},
_ => return false,
}
}
false
}
fn is_integer_type(tree: &zig::Type) -> bool {
match tree {
zig::Type::Primitive(primitive) => matches!(
primitive,
zig::Primitive::I8
| zig::Primitive::I16
| zig::Primitive::I32
| zig::Primitive::I64
| zig::Primitive::U8
| zig::Primitive::U16
| zig::Primitive::U32
| zig::Primitive::U64
),
zig::Type::Named(name) => name.as_str() == "usize",
_ => false,
}
}
}
impl CodecTarget for Zig {
type Expr = ZigExpr;
type Stmt = zig::Stmt;
type Block = Vec<zig::Stmt>;
type Type = ZigType;
type Decl = zig::Decl;
fn scalar(kind: Scalar) -> Self::Type {
ZigType::plain(match kind {
Scalar::Bool => zig::Type::Primitive(zig::Primitive::Bool),
Scalar::Char | Scalar::UInt8 => zig::Type::Primitive(zig::Primitive::U8),
Scalar::Int8 => zig::Type::Primitive(zig::Primitive::I8),
Scalar::Int16 => zig::Type::Primitive(zig::Primitive::I16),
Scalar::UInt16 => zig::Type::Primitive(zig::Primitive::U16),
Scalar::Int32 => zig::Type::Primitive(zig::Primitive::I32),
Scalar::UInt32 | Scalar::Index => zig::Type::Primitive(zig::Primitive::U32),
Scalar::Int64 => zig::Type::Primitive(zig::Primitive::I64),
Scalar::UInt64 => zig::Type::Primitive(zig::Primitive::U64),
Scalar::Float => zig::Type::Primitive(zig::Primitive::F32),
Scalar::Double => zig::Type::Primitive(zig::Primitive::F64),
Scalar::LongDouble => zig::Type::Primitive(zig::Primitive::F128),
Scalar::Size => zig::Type::named("usize"),
})
}
fn named(name: &str) -> Self::Type {
ZigType::plain(zig::Type::named(name))
}
fn nothing() -> Self::Type {
ZigType::plain(zig::Type::Primitive(zig::Primitive::Void))
}
fn handle(kind: Handle) -> Self::Type {
ZigType::plain(zig::Type::named(match kind {
Handle::DHeader => "zdwire.DHeader",
Handle::EmHeader => "zdwire.EmHeader",
Handle::Pid => "zdwire.Pid",
Handle::TryConstruct => "zdwire.TryConstruct",
Handle::ReaderMark => "zdwire.ReaderMark",
Handle::PidRead => "zdwire.PidRead",
Handle::EmHeaderRead => "zdwire.EmHeaderRead",
}))
}
fn code_point(value: Self::Expr, _narrow: bool) -> Self::Expr {
value
}
fn enumerator_ordinal(ast: &Ast, id: Id<TypeNode>, value: Self::Expr) -> Self::Expr {
if !Self::is_enumeration(ast, id) {
return value;
}
plain(zig::Expr::call("@intFromEnum", vec![inner(value)]))
}
fn bool_less(left: Self::Expr, right: Self::Expr) -> Self::Expr {
plain(zig::Expr::Call {
func: Box::new(Self::runtime_name("boolLess")),
args: vec![inner(left), inner(right)],
})
}
fn constant(which: Constant) -> Self::Expr {
match which {
Constant::WcharWidth2 => plain(zig::Expr::int(2)),
Constant::WcharWidth4 => plain(zig::Expr::int(4)),
Constant::WcharFixed => Self::text("fixed"),
Constant::WcharGiopLengthPrefixed => Self::text("giop-length-prefixed"),
Constant::WstringUnitsStream => Self::text("stream"),
Constant::WstringUnitsBigEndianAlways => Self::text("big-endian-always"),
Constant::WstringUnitsByteOrderMark => Self::text("byte-order-mark"),
Constant::WstringUnitsHostUnswapped => Self::text("host-unswapped"),
Constant::WstringSpecification => Self::text("specification"),
Constant::WstringCharacterCount => Self::text("character-count"),
Constant::WstringCharacterCountTerminated => Self::text("character-count-terminated"),
Constant::WstringSpecificationWideUnits => Self::text("specification-wide-units"),
Constant::WstringRosTypeSupport => Self::text("ros-type-support"),
Constant::WstringOctetsTerminated2 => Self::text("octets/terminated/2"),
Constant::WstringOctetsTerminated4 => Self::text("octets/terminated/4"),
Constant::WstringCharactersTerminated2 => Self::text("characters/terminated/2"),
Constant::FixedDeclared => Self::text("declared"),
Constant::FixedValuePrecision => Self::text("value-precision"),
Constant::FixedValueStringDigits => Self::text("value-string-digits"),
Constant::ParameterMember => Self::runtime_enum("ParameterRole", "member"),
Constant::ParameterOptional => Self::runtime_enum("ParameterRole", "optional"),
Constant::TryConstructDiscard => Self::runtime_enum("TryConstruct", "discard"),
Constant::TryConstructTrim => Self::runtime_enum("TryConstruct", "trim"),
Constant::TryConstructUseDefault => Self::runtime_enum("TryConstruct", "use_default"),
Constant::StatusOk => Self::runtime_enum("Status", "ok"),
Constant::StatusDiscarded => Self::runtime_enum("Status", "discarded"),
Constant::StatusImpossibleLength => Self::runtime_enum("Status", "impossible_length"),
Constant::StatusOutOfRange => Self::runtime_enum("Status", "out_of_range"),
Constant::StatusTruncated => Self::runtime_enum("Status", "truncated"),
}
}
fn unsigned_holder(bits: u16) -> Self::Type {
ZigType::plain(zig::Type::Primitive(crate::adapter::zig::zig_holder(bits)))
}
fn value_type(ast: &Ast, id: Id<TypeNode>) -> Self::Type {
let tree = match zig_value_type(ast, id, "codec") {
Ok(tree) => tree,
Err(_) => zig::Type::Primitive(zig::Primitive::Void),
};
ZigType {
tree,
enum_like: Self::is_enumeration(ast, id),
integer_alias: Self::is_bitmask(ast, id),
}
}
fn type_name(ast: &Ast, id: Id<TypeNode>) -> String {
type_tag(ast, id, 0)
}
fn map_entry_name(ast: &Ast, id: Id<TypeNode>) -> String {
zig_map_entry_name(ast, id).unwrap_or_else(|_| format!("ZdUnresolvedEntry{}", id.slot()))
}
fn module_prefix(_ast: &Ast, _id: Id<TypeNode>) -> String {
String::new()
}
fn escape(name: &str) -> String {
field_name(name)
}
fn inherited_path(depth: usize) -> Option<String> {
Some(format!("{}.", field_name("parent")).repeat(depth))
}
fn name(text: &str) -> Self::Expr {
plain(zig::Expr::Name(Ident::new(&local_name(text))))
}
fn int(value: i128) -> Self::Expr {
plain(zig::Expr::int(value))
}
fn literal(value: Literal) -> Self::Expr {
let Literal::Int(text) = &value else {
return plain(zig::Expr::Literal(value));
};
match text.as_str() {
"true" => plain(zig::Expr::Literal(Literal::Bool(true))),
"false" => plain(zig::Expr::Literal(Literal::Bool(false))),
_ => plain(zig::Expr::Literal(value)),
}
}
fn wide_int(value: i128) -> Self::Expr {
plain(zig::Expr::int(value))
}
fn unsigned_int(value: i128) -> Self::Expr {
plain(zig::Expr::int(value))
}
fn wide_unsigned(value: u64) -> Self::Expr {
plain(zig::Expr::int(i128::from(value)))
}
fn wide_signed(value: i128) -> Self::Expr {
plain(zig::Expr::int(value))
}
fn member(base: &str, field: &str) -> Self::Expr {
plain(zig::Expr::dot(zig::Expr::Name(Ident::new(base)), field))
}
fn whole(base: &str) -> Self::Expr {
plain(zig::Expr::Name(Ident::new(&local_name(base))))
}
fn field_of(value: Self::Expr, field: &str) -> Self::Expr {
plain(zig::Expr::dot(inner(value), field))
}
fn index(value: Self::Expr, at: Self::Expr) -> Self::Expr {
plain(zig::Expr::Index {
value: Box::new(inner(value)),
index: Box::new(inner(at)),
})
}
fn count_of(base: &str, fixed: Option<u64>) -> Self::Expr {
match fixed {
Some(bound) => plain(zig::Expr::int(i128::from(bound))),
None => plain(zig::Expr::as_(
zig::Type::Primitive(zig::Primitive::U32),
zig::Expr::call(
"@intCast",
vec![zig::Expr::dot(zig::Expr::Name(Ident::new(base)), "len")],
),
)),
}
}
fn elements(base: &str) -> Self::Expr {
plain(zig::Expr::Name(Ident::new(base)))
}
fn case_member(base: &str, field: &str) -> Self::Expr {
Self::member(base, field)
}
fn entry_half(entry: Self::Expr, key: bool) -> Self::Expr {
plain(zig::Expr::dot(
inner(entry),
if key { "key" } else { "value" },
))
}
fn absent() -> Self::Expr {
plain(zig::Expr::Literal(Literal::Null))
}
fn materialised(_value: Self::Expr) -> Option<Self::Expr> {
None
}
fn optional_value(_ast: &Ast, _id: Id<TypeNode>, base: &str, field: &str) -> Self::Expr {
plain(zig::Expr::dot(inner(Self::member(base, field)), "?"))
}
fn present(base: &str, field: &str) -> Self::Expr {
plain(zig::Expr::Binary {
op: "!=",
lhs: Box::new(inner(Self::member(base, field))),
rhs: Box::new(zig::Expr::Literal(Literal::Null)),
})
}
fn set_present(base: &str, field: &str, present: bool) -> Vec<Self::Stmt> {
if present {
return Vec::new();
}
vec![zig::Stmt::Assign {
place: inner(Self::member(base, field)),
value: zig::Expr::Literal(Literal::Null),
}]
}
fn optional_destination(base: &str, field: &str) -> (Vec<Self::Stmt>, Self::Expr) {
(
Vec::new(),
ZigExpr::OptionalSlot(inner(Self::member(base, field))),
)
}
fn holds_a_cycle_externally() -> bool {
true
}
fn external_value(
_ast: &Ast,
_id: Id<TypeNode>,
base: &str,
field: &str,
optional: bool,
) -> Self::Expr {
let mut place = inner(Self::member(base, field));
if optional {
place = zig::Expr::dot(place, "?");
}
plain(zig::Expr::dot(place, "*"))
}
fn external_destination(
base: &str,
field: &str,
_optional: bool,
) -> (Vec<Self::Stmt>, Self::Expr) {
(
Vec::new(),
ZigExpr::ExternalSlot(inner(Self::member(base, field))),
)
}
fn external_case_value(_ast: &Ast, _id: Id<TypeNode>, base: &str, field: &str) -> Self::Expr {
plain(zig::Expr::dot(inner(Self::case_member(base, field)), "*"))
}
fn external_case_destination(base: &str, field: &str) -> (Vec<Self::Stmt>, Self::Expr) {
(
Vec::new(),
ZigExpr::ExternalSlot(inner(Self::case_member(base, field))),
)
}
fn external_present(base: &str, field: &str) -> Self::Expr {
Self::present(base, field)
}
fn external_set_present(base: &str, field: &str, present: bool) -> Vec<Self::Stmt> {
Self::set_present(base, field, present)
}
fn binary(op: &'static str, lhs: Self::Expr, rhs: Self::Expr) -> Self::Expr {
let spelled = match op {
"&&" => "and",
"||" => "or",
other => other,
};
plain(zig::Expr::Binary {
op: spelled,
lhs: Box::new(inner(lhs)),
rhs: Box::new(inner(rhs)),
})
}
fn unary(op: &'static str, value: Self::Expr) -> Self::Expr {
let operand = inner(value);
if op == "~" {
if let zig::Expr::Literal(Literal::Int(_)) = &operand {
return plain(zig::Expr::Unary {
op: "~",
value: Box::new(zig::Expr::as_(
zig::Type::Primitive(zig::Primitive::U64),
operand,
)),
});
}
}
plain(zig::Expr::Unary {
op,
value: Box::new(operand),
})
}
fn paren(value: Self::Expr) -> Self::Expr {
plain(zig::Expr::Paren(Box::new(inner(value))))
}
fn cast(type_: Self::Type, value: Self::Expr) -> Self::Expr {
let operand = inner(value);
if type_.tree == zig::Type::named(SIGNED_CHAR) {
return plain(zig::Expr::as_(
zig::Type::Primitive(zig::Primitive::I8),
zig::Expr::call("@bitCast", vec![operand]),
));
}
if type_.tree == zig::Type::Primitive(zig::Primitive::Bool) {
if let zig::Expr::Literal(Literal::Int(text)) = &operand {
return plain(zig::Expr::Literal(Literal::Bool(text != "0")));
}
}
if type_.enum_like {
return plain(zig::Expr::as_(
type_.tree,
zig::Expr::call("@enumFromInt", vec![operand]),
));
}
if type_.tree == zig::Type::Primitive(zig::Primitive::I64) {
return plain(zig::Expr::Call {
func: Box::new(Self::runtime_name("asI64")),
args: vec![operand],
});
}
if type_.tree == zig::Type::Primitive(zig::Primitive::U64) {
return plain(zig::Expr::Call {
func: Box::new(Self::runtime_name("asU64")),
args: vec![operand],
});
}
if Self::is_integer_type(&type_.tree) || type_.integer_alias {
if let zig::Expr::Literal(Literal::Int(_)) = &operand {
return plain(zig::Expr::as_(type_.tree, operand));
}
return plain(zig::Expr::as_(
type_.tree,
zig::Expr::call("@intCast", vec![operand]),
));
}
plain(zig::Expr::as_(type_.tree, operand))
}
fn handoff(value: Self::Expr, _form: Handoff) -> Self::Expr {
value
}
fn call(callee: &str, args: Vec<Self::Expr>) -> Self::Expr {
let arguments: Vec<zig::Expr> = args.into_iter().map(inner).collect();
if callee == "strcmp" {
return plain(zig::Expr::Call {
func: Box::new(Self::runtime_name("strcmp")),
args: arguments,
});
}
ZigExpr::Fallible(zig::Expr::call(callee, arguments))
}
fn runtime(receiver: &str, method: &str, args: Vec<Self::Expr>) -> Self::Expr {
let call = zig::Expr::Call {
func: Box::new(zig::Expr::dot(
zig::Expr::Name(Ident::new(receiver)),
&Self::camel(method),
)),
args: args.into_iter().map(inner).collect(),
};
if FALLIBLE_EXEMPT_METHODS.contains(&method) {
return plain(call);
}
ZigExpr::Fallible(call)
}
fn runtime_static(method: &str, args: Vec<Self::Expr>) -> Self::Expr {
if method == "pid_is_user_member" {
let mut arguments = args.into_iter().map(inner);
if let Some(header) = arguments.next() {
return plain(zig::Expr::Call {
func: Box::new(zig::Expr::dot(header, "isUserMember")),
args: Vec::new(),
});
}
return plain(zig::Expr::Call {
func: Box::new(Self::runtime_name("isUserMember")),
args: Vec::new(),
});
}
let call = zig::Expr::Call {
func: Box::new(Self::runtime_name(&Self::camel(method))),
args: args.into_iter().map(inner).collect(),
};
if FALLIBLE_STATICS.contains(&method) {
return ZigExpr::Fallible(call);
}
plain(call)
}
fn carries_hdds_axes() -> bool {
true
}
fn declare(type_: Self::Type, name: &str, value: Option<Self::Expr>) -> Self::Stmt {
zig::Stmt::Var {
name: Ident::new(&local_name(name)),
type_: Some(type_.tree),
value: value.map(inner_bare),
mutable: true,
}
}
fn perform(value: Self::Expr) -> Self::Stmt {
match value {
ZigExpr::Plain(zig::Expr::Binary { op: "=", lhs, rhs }) => zig::Stmt::Assign {
place: *lhs,
value: *rhs,
},
ZigExpr::Fallible(e) => zig::Stmt::Expr(zig::Expr::try_(e)),
other => zig::Stmt::Expr(inner(other)),
}
}
fn give_back(value: Option<Self::Expr>) -> Self::Stmt {
zig::Stmt::Return(Some(match value {
Some(value) => inner(value),
None => zig::Expr::Name(Ident::new("v")),
}))
}
fn when(
condition: Self::Expr,
then: Self::Block,
otherwise: Option<Self::Block>,
) -> Self::Stmt {
zig::Stmt::If {
condition: inner(condition),
then,
otherwise: otherwise.filter(|body| !body.is_empty()),
}
}
fn count_up(counter: &str, bound: Self::Expr, body: Self::Block) -> Self::Stmt {
zig::Stmt::CountUp {
counter: Ident::new(&local_name(counter)),
type_: Box::new(zig::Type::Primitive(zig::Primitive::U32)),
bound: Box::new(inner(bound)),
body,
}
}
fn until(condition: Self::Expr, body: Self::Block) -> Self::Stmt {
zig::Stmt::While {
condition: inner(condition),
body,
}
}
fn leave() -> Self::Stmt {
zig::Stmt::Break
}
fn note(doc: Doc) -> Self::Stmt {
zig::Stmt::Comment(doc)
}
fn drop_value(value: Self::Expr) -> Self::Stmt {
zig::Stmt::DiscardExpr(inner_bare(value))
}
fn mark_used(name: &str) -> Vec<Self::Stmt> {
vec![zig::Stmt::Discard(Ident::new(name))]
}
fn read_call(
callee: &str,
place: Self::Expr,
_form: Handoff,
behaviour: Self::Expr,
) -> Self::Stmt {
let decoded = zig::Expr::call(
callee,
vec![zig::Expr::Name(Ident::new("r")), inner(behaviour)],
);
match place {
ZigExpr::ExternalSlot(field) => zig::Stmt::Assign {
place: field,
value: zig::Expr::try_(zig::Expr::Call {
func: Box::new(Self::runtime_name("zdPtr")),
args: vec![Self::reader_heap(), zig::Expr::try_(decoded)],
}),
},
ZigExpr::OptionalSlot(field) | ZigExpr::Plain(field) | ZigExpr::Fallible(field) => {
zig::Stmt::Assign {
place: field,
value: zig::Expr::try_(decoded),
}
}
}
}
fn read_forward(callee: &str) -> Self::Stmt {
Self::read_call(
callee,
Self::whole("v"),
Handoff::Value,
Self::name("zd_tc"),
)
}
fn block(statements: Vec<Self::Stmt>) -> Self::Block {
statements
}
fn into_statements(block: Self::Block) -> Vec<Self::Stmt> {
block
}
fn make_room(
base: &str,
element: Self::Type,
count: Self::Expr,
_assigned_whole: bool,
fill: Vec<Self::Stmt>,
) -> Vec<Self::Stmt> {
let mut out = vec![zig::Stmt::Assign {
place: zig::Expr::Name(Ident::new(base)),
value: zig::Expr::try_(zig::Expr::Call {
func: Box::new(zig::Expr::dot(Self::reader_heap(), "alloc")),
args: vec![zig::Expr::type_(element.tree), inner(count)],
}),
}];
out.extend(fill);
out
}
fn keep_first(base: &str, count: Self::Expr) -> Vec<Self::Stmt> {
vec![zig::Stmt::Assign {
place: zig::Expr::Name(Ident::new(base)),
value: zig::Expr::Index {
value: Box::new(zig::Expr::Name(Ident::new(base))),
index: Box::new(zig::Expr::Binary {
op: "..",
lhs: Box::new(zig::Expr::int(0)),
rhs: Box::new(inner(count)),
}),
},
}]
}
fn sequence_element(
base: &str,
index: Self::Expr,
_element: Self::Type,
) -> (Vec<Self::Stmt>, Self::Expr, Vec<Self::Stmt>) {
(
Vec::new(),
Self::index(Self::elements(base), index),
Vec::new(),
)
}
fn routine(spec: Routine<Self>) -> Option<Self::Decl> {
let body = spec.body?;
let mut params = vec![zig::FnParam {
name: Ident::new(if spec.writing { "w" } else { "r" }),
type_: zig::Type::pointer(zig::Type::named(if spec.writing {
"zdwire.Writer"
} else {
"zdwire.Reader"
})),
}];
if spec.writing {
params.push(zig::FnParam {
name: Ident::new("v"),
type_: spec.sample.tree.clone(),
});
}
for (type_, name) in spec.extra {
params.push(zig::FnParam {
name: Ident::new(&name),
type_: type_.tree,
});
}
let error = zig::Type::named("zdwire.WireError");
let (returns, mut body) = if spec.writing {
(
zig::Type::error_union(error, zig::Type::Primitive(zig::Primitive::Void)),
body,
)
} else {
let defaulter = crate::default_zig::defaulter_from_decoder(&spec.name);
let mut statements = vec![zig::Stmt::Var {
name: Ident::new("v"),
type_: Some(spec.sample.tree.clone()),
value: Some(zig::Expr::try_(zig::Expr::call(
&defaulter,
vec![Self::reader_heap()],
))),
mutable: true,
}];
statements.extend(body);
statements.push(zig::Stmt::Return(Some(zig::Expr::Name(Ident::new("v")))));
(zig::Type::error_union(error, spec.sample.tree), statements)
};
let assigned = assigned_roots(&body);
fix_mutability(&mut body, &assigned);
settle_discards(&mut body, ¶ms);
Some(zig::Decl::Func(zig::FuncDecl {
doc: spec.doc,
is_pub: true,
name: Ident::new(&spec.name),
params,
returns,
body,
}))
}
fn default_into(
ast: &Ast,
id: Id<TypeNode>,
place: Self::Expr,
) -> Result<Self::Stmt, EncodeError> {
let defaulted = zig::Expr::call(
&crate::default_zig::defaulter_name(ast, id),
vec![Self::reader_heap()],
);
Ok(match place {
ZigExpr::ExternalSlot(field) => zig::Stmt::Assign {
place: field,
value: zig::Expr::try_(zig::Expr::Call {
func: Box::new(Self::runtime_name("zdPtr")),
args: vec![Self::reader_heap(), zig::Expr::try_(defaulted)],
}),
},
ZigExpr::OptionalSlot(field) | ZigExpr::Plain(field) | ZigExpr::Fallible(field) => {
zig::Stmt::Assign {
place: field,
value: zig::Expr::try_(defaulted),
}
}
})
}
fn member_default(
ast: &Ast,
node: &MemberNode,
type_: Id<TypeNode>,
place: Self::Expr,
) -> Result<Vec<Self::Stmt>, EncodeError> {
crate::default_zig::member_default(ast, node, type_, inner(place), Self::reader_heap())
}
fn enumerator_constant(ast: &Ast, id: Id<TypeNode>, enumerator: &str) -> Self::Expr {
plain(zig::Expr::dot(
zig::Expr::Name(Ident::new(&zig_type_name(ast, id))),
&field_name(enumerator),
))
}
}
fn assigned_roots(body: &[zig::Stmt]) -> Vec<String> {
let mut out = Vec::new();
collect_assigned(body, &mut out);
out
}
fn collect_assigned(body: &[zig::Stmt], out: &mut Vec<String>) {
for statement in body {
match statement {
zig::Stmt::Assign { place, .. } => {
if let Some(root) = root_name(place) {
if !out.contains(&root) {
out.push(root);
}
}
}
zig::Stmt::If {
then, otherwise, ..
} => {
collect_assigned(then, out);
if let Some(otherwise) = otherwise {
collect_assigned(otherwise, out);
}
}
zig::Stmt::CountUp { body, .. } | zig::Stmt::While { body, .. } => {
collect_assigned(body, out);
}
_ => {}
}
}
}
fn root_name(place: &zig::Expr) -> Option<String> {
match place {
zig::Expr::Name(name) => Some(name.as_str().to_string()),
zig::Expr::Dot { value, .. }
| zig::Expr::Index { value, .. }
| zig::Expr::Paren(value)
| zig::Expr::Try(value) => root_name(value),
_ => None,
}
}
fn fix_mutability(body: &mut [zig::Stmt], assigned: &[String]) {
for statement in body {
match statement {
zig::Stmt::Var {
name,
value,
mutable,
..
} => {
let is_assigned = assigned.iter().any(|root| root == name.as_str());
*mutable = is_assigned || value.is_none();
}
zig::Stmt::If {
then, otherwise, ..
} => {
fix_mutability(then, assigned);
if let Some(otherwise) = otherwise {
fix_mutability(otherwise, assigned);
}
}
zig::Stmt::CountUp { body, .. } | zig::Stmt::While { body, .. } => {
fix_mutability(body, assigned);
}
_ => {}
}
}
}
fn settle_discards(body: &mut Vec<zig::Stmt>, params: &[zig::FnParam]) {
let mut mentioned: Vec<String> = Vec::new();
collect_mentions(body, &mut mentioned);
prune_discards(body, &mentioned);
let mut fronts = Vec::new();
for param in params {
let name = param.name.as_str();
if !mentioned.iter().any(|m| m == name) {
fronts.push(zig::Stmt::Discard(Ident::new(name)));
}
}
if !fronts.is_empty() {
fronts.append(body);
*body = fronts;
}
}
fn collect_mentions(body: &[zig::Stmt], out: &mut Vec<String>) {
for statement in body {
match statement {
zig::Stmt::Var { value, .. } => {
if let Some(value) = value {
mentions_in_expr(value, out);
}
}
zig::Stmt::Assign { place, value } => {
mentions_in_expr(place, out);
mentions_in_expr(value, out);
}
zig::Stmt::Expr(value)
| zig::Stmt::DiscardExpr(value)
| zig::Stmt::Return(Some(value)) => {
mentions_in_expr(value, out);
}
zig::Stmt::If {
condition,
then,
otherwise,
} => {
mentions_in_expr(condition, out);
collect_mentions(then, out);
if let Some(otherwise) = otherwise {
collect_mentions(otherwise, out);
}
}
zig::Stmt::CountUp { bound, body, .. } => {
mentions_in_expr(bound, out);
collect_mentions(body, out);
}
zig::Stmt::While { condition, body } => {
mentions_in_expr(condition, out);
collect_mentions(body, out);
}
zig::Stmt::Discard(_)
| zig::Stmt::Return(None)
| zig::Stmt::Break
| zig::Stmt::Comment(_) => {}
}
}
}
fn mentions_in_expr(expr: &zig::Expr, out: &mut Vec<String>) {
match expr {
zig::Expr::Name(name) => {
let text = name.as_str().to_string();
if !out.contains(&text) {
out.push(text);
}
}
zig::Expr::Binary { lhs, rhs, .. } => {
mentions_in_expr(lhs, out);
mentions_in_expr(rhs, out);
}
zig::Expr::Paren(value)
| zig::Expr::Unary { value, .. }
| zig::Expr::Try(value)
| zig::Expr::Dot { value, .. } => mentions_in_expr(value, out),
zig::Expr::Index { value, index } => {
mentions_in_expr(value, out);
mentions_in_expr(index, out);
}
zig::Expr::Call { func, args } => {
mentions_in_expr(func, out);
for arg in args {
mentions_in_expr(arg, out);
}
}
zig::Expr::StructLit(fields) => {
for (_, value) in fields {
mentions_in_expr(value, out);
}
}
zig::Expr::ListLit(elements) => {
for element in elements {
mentions_in_expr(element, out);
}
}
zig::Expr::Literal(_) | zig::Expr::Type(_) => {}
}
}
fn prune_discards(body: &mut Vec<zig::Stmt>, mentioned: &[String]) {
body.retain(|statement| match statement {
zig::Stmt::Discard(name) => !mentioned.iter().any(|m| m == name.as_str()),
_ => true,
});
for statement in body {
match statement {
zig::Stmt::If {
then, otherwise, ..
} => {
prune_discards(then, mentioned);
if let Some(otherwise) = otherwise {
prune_discards(otherwise, mentioned);
}
}
zig::Stmt::CountUp { body, .. } | zig::Stmt::While { body, .. } => {
prune_discards(body, mentioned);
}
_ => {}
}
}
}
fn bound_tag(ast: &Ast, bound: &zerodds_ast::Bound) -> String {
match bound.expr() {
None => String::new(),
Some(expr) => match ast.eval_integral(expr) {
Ok(limit) => format!("_b{limit}"),
Err(_) => "_bx".to_string(),
},
}
}
#[must_use]
pub fn type_tag(ast: &Ast, id: Id<TypeNode>, depth: usize) -> String {
if depth > MAX_TYPE_DEPTH {
return "unresolved".to_string();
}
let node = match ast.type_(id) {
Ok(node) => node,
Err(_) => return "unresolved".to_string(),
};
match &node.kind {
TypeKind::Primitive(kind) => kind.as_str().replace(' ', "_"),
TypeKind::String {
width: CharWidth::Char,
bound,
} => format!("string{}", bound_tag(ast, bound)),
TypeKind::String {
width: CharWidth::WChar,
bound,
} => format!("wstring{}", bound_tag(ast, bound)),
TypeKind::Fixed { digits, scale } => format!("fixed_{digits}_{scale}"),
TypeKind::Sequence { element, bound } => match element.resolved() {
Some(element) => format!(
"seq{}_{}",
bound_tag(ast, bound),
type_tag(ast, element, depth + 1)
),
None => format!("seq_unresolved{}", bound_tag(ast, bound)),
},
TypeKind::Array { element, dims } => {
let mut name = match element.resolved() {
Some(element) => format!("arr_{}", type_tag(ast, element, depth + 1)),
None => "arr_unresolved".to_string(),
};
for dim in dims {
match ast.eval_integral(*dim) {
Ok(length) => name.push_str(&format!("_{length}")),
Err(_) => name.push_str("_x"),
}
}
name
}
TypeKind::Map { key, value, bound } => {
let half = |half: Option<Id<TypeNode>>| match half {
Some(target) => type_tag(ast, target, depth + 1),
None => "unresolved".to_string(),
};
format!(
"map{}_{}_{}",
bound_tag(ast, bound),
half(key.resolved()),
half(value.resolved())
)
}
TypeKind::Alias { aliased } => match aliased.resolved() {
Some(target) => type_tag(ast, target, depth + 1),
None => "unresolved".to_string(),
},
TypeKind::ValueBox { boxed } => match boxed.resolved() {
Some(target) => format!("box_{}", type_tag(ast, target, depth + 1)),
None => "box_unresolved".to_string(),
},
TypeKind::Structure { .. }
| TypeKind::Union { .. }
| TypeKind::Enumeration { .. }
| TypeKind::Bitmask { .. }
| TypeKind::Bitset { .. }
| TypeKind::Valuetype { .. }
| TypeKind::Native => zig_type_name(ast, id),
TypeKind::Any => "any".to_string(),
TypeKind::ValueBase => "valuebase".to_string(),
TypeKind::ObjectRef { .. } => "objectref".to_string(),
TypeKind::AnnotationType { .. } => "annotation".to_string(),
}
}