use crate::adapter::constant;
use crate::adapter::cst::{MaterialiseError, no_spelling_collision};
use crate::adapter::interface;
use crate::apply::Applied;
use crate::verbatim;
use seesaw_tgg::ident::GhostId;
use zerodds_ast::{Ast, CharWidth, PrimitiveKind, TypeKind, TypeNode};
use zerodds_ast_core::{Id, NodeId};
use zerodds_lang_cst::common::{Doc, Ident, Literal};
use zerodds_lang_cst::go;
const LANGUAGE: &str = "Go";
pub const PACKAGE: &str = "zdtypes";
const INVENTED_PREFIX: &str = "ZdMapEntry_";
pub fn go_items(ast: &Ast, applied: &Applied) -> Result<Vec<go::Decl>, MaterialiseError> {
let mut out = Vec::new();
let mut emitted: Vec<(String, String)> = Vec::new();
for (id, _) in ast.types() {
let start = out.len();
declare(ast, applied, id, &mut out, &mut emitted)?;
verbatim::weave(
verbatim::declaration_slots(ast, id, verbatim::aliases::GO),
&go_type_name(ast, id),
start,
&mut out,
);
}
for (_, node) in ast.constants() {
if ghost_created_for(applied, node.node_id, "GoConst").is_none() {
continue;
}
let value = ast
.eval_const(node.expr)
.map_err(|_| MaterialiseError::Unevaluated {
construct: "constant's value".to_string(),
at: node.name.clone(),
})?;
let declared = node.type_.resolved();
let type_ = match declared {
Some(target) => go_value_type(ast, target)?,
None => go::Type::Primitive(go::Primitive::Int32),
};
let name = declared_name(ast, &node.name, node.module);
remember(&mut emitted, &name, &node.name)?;
out.push(go::Decl::Const(go::ConstDecl {
doc: idl_note(&name, &node.name),
name: Ident::new(&name),
type_: Some(type_.clone()),
value: go_constant(&value, &type_, &node.name)?,
}));
}
for (_, node) in ast.exceptions() {
if ghost_created_for(applied, node.node_id, "GoError").is_none() {
continue;
}
let name = declared_name(ast, &node.name, node.module);
remember(&mut emitted, &name, &node.name)?;
let mut pairs = Vec::new();
for member in &node.members {
if let Ok(member_node) = ast.member(*member) {
pairs.push((field_name(&member_node.name), member_node.name.clone()));
}
}
no_spelling_collision(as_pairs(&pairs), LANGUAGE, &name)?;
let mut fields = Vec::new();
for member in &node.members {
let Ok(member_node) = ast.member(*member) else {
continue;
};
let Some(target) = member_node.type_.resolved() else {
return Err(MaterialiseError::Unevaluated {
construct: "exception member type".to_string(),
at: member_node.name.clone(),
});
};
fields.push(go::Field {
doc: Doc::none(),
name: Ident::new(&field_name(&member_node.name)),
type_: go_member_type(ast, target, &member_node.name)?,
tag: Some(go::Tag::one("idl", &member_node.name)),
});
}
out.push(go::Decl::Type(go::TypeDecl {
body_text: zerodds_lang_cst::common::BodyText::none(),
doc: with_idl_note(
Doc::line("An exception, as the structure its members make."),
&name,
&node.name,
),
name: Ident::new(&name),
alias: false,
type_: go::Type::Struct(fields),
}));
}
for (id, node) in ast.interfaces() {
if ghost_created_for(applied, node.node_id, "GoInterface").is_none() {
continue;
}
let name = declared_name(ast, &node.name, node.module);
remember(&mut emitted, &name, &node.name)?;
let mut methods = Vec::new();
let mut signatures = Vec::new();
let operations = interface::all_operations_flattened(ast, id)
.map_err(|error| interface::interface_error_to_materialise(error, &node.name))?;
for op in &operations {
let (exported, idl_spelling) = match op.origin {
interface::Origin::AttributeGetter => {
let attr = op.attribute_name();
(go::exported(&go::escape_name(attr)), op.name.clone())
}
interface::Origin::AttributeSetter => {
let attr = op.attribute_name();
(
format!("Set{}", go::exported(&go::escape_name(attr))),
op.name.clone(),
)
}
interface::Origin::Declared => (field_name(&op.name), op.name.clone()),
};
signatures.push((exported.clone(), idl_spelling));
let mut params = Vec::new();
for param in &op.params {
params.push(go::Param {
name: Some(Ident::new(&go::escape_name(¶m.name))),
type_: go_member_type(ast, param.type_, ¶m.name)?,
});
}
let mut results = Vec::new();
if let Some(returns) = op.returns {
results.push(go_member_type(ast, returns, &op.name)?);
}
results.push(go::Type::named("error"));
let doc = op
.raises_line()
.map_or_else(Doc::none, |line| Doc::line(&line));
methods.push(go::Method {
doc,
name: Ident::new(&exported),
params,
results,
});
}
no_spelling_collision(as_pairs(&signatures), LANGUAGE, &name)?;
out.push(go::Decl::Interface(go::InterfaceDecl {
doc: with_idl_note(
Doc::line(
"An interface, as the method set its operations and attributes make \
— an attribute a `Name()`/`SetName()` pair, `readonly` only the \
first. A derived interface's bases are flattened in rather than \
embedded, so this is the whole method set; the transport is the RPC \
arc's business, and the shape is the IDL's. Every method returns an \
`error` besides its own result; `raises`/`getraises`/`setraises` \
name what it can carry, in the method's own doc.",
),
&name,
&node.name,
),
name: Ident::new(&name),
methods,
}));
}
Ok(verbatim::weave_file(ast, verbatim::aliases::GO, out))
}
impl verbatim::Carrier for go::Decl {
fn verbatim(text: String) -> Self {
Self::Verbatim(text)
}
fn take_into_body(
&mut self,
name: &str,
begin: &mut Vec<String>,
end: &mut Vec<String>,
) -> bool {
let Self::Type(node) = self else {
return false;
};
if node.alias || node.name.as_str() != name || !matches!(node.type_, go::Type::Struct(_)) {
return false;
}
node.body_text.begin.append(begin);
node.body_text.end.append(end);
true
}
}
pub fn go_file(ast: &Ast, applied: &Applied) -> Result<go::File, MaterialiseError> {
Ok(go::File {
doc: Doc::lines(&[
"Generated by ZeroDDS from an IDL source. Do not edit.",
"",
"The IDL spelling of every renamed declaration is in the comment",
"above it; the IDL spelling of every member is in its `idl` struct",
"tag, because a member's ID under @autoid(HASH) is the MD5 of the",
"IDL name (DDS-XTypes 1.3 7.3.1.2.1.1) and the Go spelling does not",
"invert.",
]),
package: Ident::new(PACKAGE),
imports: Vec::new(),
decls: go_items(ast, applied)?,
})
}
#[allow(clippy::too_many_lines)]
fn declare(
ast: &Ast,
applied: &Applied,
id: Id<TypeNode>,
out: &mut Vec<go::Decl>,
emitted: &mut Vec<(String, String)>,
) -> Result<(), MaterialiseError> {
let Ok(node) = ast.type_(id) else {
return Ok(());
};
match &node.kind {
TypeKind::Primitive(_)
| TypeKind::String { .. }
| TypeKind::Fixed { .. }
| TypeKind::Sequence { .. }
| TypeKind::Array { .. } => {}
TypeKind::Map { key, value, .. } => {
let (Some(key), Some(value)) = (key.resolved(), value.resolved()) else {
return Err(MaterialiseError::Unevaluated {
construct: "map key or value".to_string(),
at: go_type_name(ast, id),
});
};
let name = map_entry_name_at(ast, id, 0)?;
if emitted.iter().any(|(spelled, _)| *spelled == name) {
return Ok(());
}
emitted.push((name.clone(), name.clone()));
out.push(go::Decl::Type(go::TypeDecl {
body_text: zerodds_lang_cst::common::BodyText::none(),
doc: Doc::line(
"One entry of a map. DDS-XTypes 1.3 7.4.1.1.4 defines a map as a \
sequence of these, and the rules write the two halves with no \
framing between them.",
),
name: Ident::new(&name),
alias: false,
type_: go::Type::Struct(vec![
go::Field {
doc: Doc::none(),
name: Ident::new("Key"),
type_: go_member_type(ast, key, &name)?,
tag: None,
},
go::Field {
doc: Doc::none(),
name: Ident::new("Value"),
type_: go_member_type(ast, value, &name)?,
tag: None,
},
]),
}));
}
TypeKind::Alias { aliased } => {
if node.name.is_empty() {
return Ok(());
}
if ghost_created_for(applied, node.node_id, "GoAlias").is_none() {
return Ok(());
}
let Some(target) = aliased.resolved() else {
return Err(MaterialiseError::Unevaluated {
construct: "typedef target".to_string(),
at: node.name.clone(),
});
};
let name = go_type_name(ast, id);
remember(emitted, &name, &node.name)?;
out.push(go::Decl::Type(go::TypeDecl {
body_text: zerodds_lang_cst::common::BodyText::none(),
doc: idl_note(&name, &node.name),
name: Ident::new(&name),
alias: true,
type_: go_member_type(ast, target, &node.name)?,
}));
}
TypeKind::Enumeration { enumerators } => {
if ghost_created_for(applied, node.node_id, "GoEnum").is_none() {
return Ok(());
}
let name = go_type_name(ast, id);
remember(emitted, &name, &node.name)?;
out.push(go::Decl::Type(go::TypeDecl {
body_text: zerodds_lang_cst::common::BodyText::none(),
doc: with_idl_note(
Doc::line(
"An enumeration. The holder is signed because DDS-XTypes 1.3 \
7.4.3.5.3 makes it so; the width on the wire is the \
@bit_bound's and is applied by the runtime.",
),
&name,
&node.name,
),
name: Ident::new(&name),
alias: false,
type_: go::Type::Primitive(go::Primitive::Int32),
}));
let prefix = module_prefix(ast, id);
for enumerator in enumerators {
let Ok(enum_node) = ast.enumerator(*enumerator) else {
continue;
};
if ghost_created_for(applied, enum_node.node_id, "GoEnumerator").is_none() {
continue;
}
let value = ast.effective_enumerator_value(*enumerator).ok_or_else(|| {
MaterialiseError::Unevaluated {
construct: "enumerator value".to_string(),
at: enum_node.name.clone(),
}
})?;
let constant = format!("{prefix}{}", field_name(&enum_node.name));
remember(emitted, &constant, &enum_node.name)?;
out.push(go::Decl::Const(go::ConstDecl {
doc: idl_note(&constant, &enum_node.name),
name: Ident::new(&constant),
type_: Some(go::Type::named(&name)),
value: go::Expr::int(value),
}));
}
}
TypeKind::Bitmask { bits } => {
if ghost_created_for(applied, node.node_id, "GoBitmask").is_none() {
return Ok(());
}
let name = go_type_name(ast, id);
remember(emitted, &name, &node.name)?;
let bound = ast.effective_bit_bound(id);
out.push(go::Decl::Type(go::TypeDecl {
body_text: zerodds_lang_cst::common::BodyText::none(),
doc: with_idl_note(
Doc::line(
"A bitmask, as the unsigned integer Table 33 of DDS-XTypes 1.3 \
7.4.1.1.3.2 gives for its bound.",
),
&name,
&node.name,
),
name: Ident::new(&name),
alias: false,
type_: go::Type::Primitive(go_holder(bound)),
}));
let prefix = module_prefix(ast, id);
for bit in bits {
let Ok(bit_node) = ast.bit_value(*bit) else {
continue;
};
if ghost_created_for(applied, bit_node.node_id, "GoBitConst").is_none() {
continue;
}
let position = ast.effective_bit_position(*bit).ok_or_else(|| {
MaterialiseError::Unevaluated {
construct: "bit position".to_string(),
at: bit_node.name.clone(),
}
})?;
let constant = format!("{prefix}{}", field_name(&bit_node.name));
remember(emitted, &constant, &bit_node.name)?;
out.push(go::Decl::Const(go::ConstDecl {
doc: idl_note(&constant, &bit_node.name),
name: Ident::new(&constant),
type_: Some(go::Type::named(&name)),
value: go::Expr::Binary {
op: "<<",
lhs: Box::new(go::Expr::int(1)),
rhs: Box::new(go::Expr::int(i128::from(position))),
},
}));
}
}
TypeKind::Bitset { .. } => {
if ghost_created_for(applied, node.node_id, "GoBitset").is_none() {
return Ok(());
}
let name = go_type_name(ast, id);
remember(emitted, &name, &node.name)?;
let mut pairs = Vec::new();
for field in &ast.all_bitfields_of(id) {
let Ok(field_node) = ast.bitfield(*field) else {
continue;
};
let Some(field_ident) = field_node.name.as_ref() else {
continue;
};
pairs.push((field_name(field_ident), field_ident.clone()));
}
no_spelling_collision(as_pairs(&pairs), LANGUAGE, &name)?;
let mut fields = Vec::new();
let mut index = 0;
for field in &ast.all_bitfields_of(id) {
let Ok(field_node) = ast.bitfield(*field) else {
continue;
};
let Some(field_ident) = field_node.name.as_ref() else {
continue;
};
fields.push(go::Field {
doc: Doc::none(),
name: Ident::new(&pairs[index].0),
type_: go::Type::Primitive(go_holder(field_node.bits)),
tag: Some(go::Tag::one("idl", field_ident)),
});
index += 1;
}
out.push(go::Decl::Type(go::TypeDecl {
body_text: zerodds_lang_cst::common::BodyText::none(),
doc: with_idl_note(
Doc::line(
"A bitset. The fields are separate integers here; the packing \
into the anchor is the codec's, because the order on the wire \
is the declaration's.",
),
&name,
&node.name,
),
name: Ident::new(&name),
alias: false,
type_: go::Type::Struct(fields),
}));
}
TypeKind::Structure { .. } | TypeKind::Valuetype { .. } => {
let created = if matches!(node.kind, TypeKind::Valuetype { .. }) {
"GoValue"
} else {
"GoStruct"
};
if ghost_created_for(applied, node.node_id, created).is_none() {
return Ok(());
}
let name = go_type_name(ast, id);
remember(emitted, &name, &node.name)?;
let members = ast.members_of(id).to_vec();
let base = super::c::struct_base(ast, id);
let mut pairs = Vec::new();
if base.is_some() {
pairs.push((field_name("parent"), "parent".to_string()));
}
for member in &members {
if let Ok(member_node) = ast.member(*member) {
pairs.push((field_name(&member_node.name), member_node.name.clone()));
}
}
no_spelling_collision(as_pairs(&pairs), LANGUAGE, &name)?;
let mut fields = Vec::new();
if let Some(base) = base {
fields.push(go::Field {
doc: Doc::line(
"The base structure, the first member (XTypes 1.3 7.5.1.4.1.3).",
),
name: Ident::new(&field_name("parent")),
type_: go_member_type(ast, base, "parent")?,
tag: Some(go::Tag::one("idl", "parent")),
});
}
for member in &members {
let Ok(member_node) = ast.member(*member) else {
continue;
};
let Some(target) = member_node.type_.resolved() else {
return Err(MaterialiseError::Unevaluated {
construct: "member type".to_string(),
at: member_node.name.clone(),
});
};
let mut type_ = go_member_type(ast, target, &member_node.name)?;
let mut doc = Doc::none();
if member_node.facets.is_optional() {
type_ = go::Type::pointer(type_);
doc = Doc::line(
"Optional. `nil` when the member is absent — DDS-XTypes 1.3 \
7.2.2.4.4.4.7. In a @mutable type an absent member is not \
written at all; outside one its absence is written.",
);
}
fields.push(go::Field {
doc,
name: Ident::new(&field_name(&member_node.name)),
type_,
tag: Some(go::Tag::one("idl", &member_node.name)),
});
}
out.push(go::Decl::Type(go::TypeDecl {
body_text: zerodds_lang_cst::common::BodyText::none(),
doc: idl_note(&name, &node.name),
name: Ident::new(&name),
alias: false,
type_: go::Type::Struct(fields),
}));
}
TypeKind::Union { discriminator, .. } => {
if ghost_created_for(applied, node.node_id, "GoUnion").is_none() {
return Ok(());
}
let name = go_type_name(ast, id);
remember(emitted, &name, &node.name)?;
let Some(disc) = discriminator.resolved() else {
return Err(MaterialiseError::Unevaluated {
construct: "union discriminator".to_string(),
at: name,
});
};
let cases = ast.cases_of(id);
let mut pairs = vec![(DISCRIMINATOR.to_string(), "<discriminator>".to_string())];
for case in cases {
if let Ok(case_node) = ast.case(*case) {
if let Ok(member) = ast.member(case_node.member) {
pairs.push((field_name(&member.name), member.name.clone()));
}
}
}
no_spelling_collision(as_pairs(&pairs), LANGUAGE, &name)?;
let mut fields = vec![go::Field {
doc: Doc::line("Which branch carries the value."),
name: Ident::new(DISCRIMINATOR),
type_: go_member_type(ast, disc, &name)?,
tag: None,
}];
for case in cases {
let Ok(case_node) = ast.case(*case) else {
continue;
};
if ghost_created_for(applied, case_node.node_id, "GoCase").is_none() {
continue;
}
let Ok(member) = ast.member(case_node.member) else {
continue;
};
let Some(target) = member.type_.resolved() else {
return Err(MaterialiseError::Unevaluated {
construct: "union case type".to_string(),
at: member.name.clone(),
});
};
fields.push(go::Field {
doc: Doc::none(),
name: Ident::new(&field_name(&member.name)),
type_: go_member_type(ast, target, &member.name)?,
tag: Some(go::Tag::one("idl", &member.name)),
});
}
out.push(go::Decl::Type(go::TypeDecl {
body_text: zerodds_lang_cst::common::BodyText::none(),
doc: with_idl_note(
Doc::lines(&[
"A union. Go has no overlaid storage, so the branches sit side",
"by side and `D` says which one carries the value; only that",
"branch is written to the wire (DDS-XTypes 1.3 7.4.1.1.5.2).",
]),
&name,
&node.name,
),
name: Ident::new(&name),
alias: false,
type_: go::Type::Struct(fields),
}));
}
TypeKind::Native => {
if ghost_created_for(applied, node.node_id, "GoNative").is_none() {
return Ok(());
}
let name = go_type_name(ast, id);
remember(emitted, &name, &node.name)?;
let mapping = ast
.any_of_type(id)
.ok()
.and_then(|any| ast.facets(any))
.and_then(|facets| facets.corba.as_ref())
.and_then(|corba| corba.native_mapping.clone());
out.push(go::Decl::Type(go::TypeDecl {
body_text: zerodds_lang_cst::common::BodyText::none(),
doc: with_idl_note(
Doc::line("A native type. Opaque unless a facet named its mapping."),
&name,
&node.name,
),
name: Ident::new(&name),
alias: true,
type_: match mapping {
Some(spelling) => go::Type::named(&spelling),
None => go::Type::Primitive(go::Primitive::Any),
},
}));
}
TypeKind::ValueBox { boxed } => {
let Some(target) = boxed.resolved() else {
return Ok(());
};
let name = go_type_name(ast, id);
if emitted.iter().any(|(spelled, _)| *spelled == name) {
return Ok(());
}
remember(emitted, &name, &node.name)?;
out.push(go::Decl::Type(go::TypeDecl {
body_text: zerodds_lang_cst::common::BodyText::none(),
doc: with_idl_note(
Doc::line("A boxed value: a pointer, `nil` when absent."),
&name,
&node.name,
),
name: Ident::new(&name),
alias: true,
type_: go::Type::pointer(go_member_type(ast, target, &name)?),
}));
}
TypeKind::ObjectRef { .. }
| TypeKind::AnnotationType { .. }
| TypeKind::Any
| TypeKind::ValueBase => {}
}
Ok(())
}
pub const DISCRIMINATOR: &str = "D";
pub fn go_member_type(
ast: &Ast,
target: Id<TypeNode>,
at: &str,
) -> Result<go::Type, MaterialiseError> {
let construct = match ast.type_(target).map(|node| &node.kind) {
Ok(TypeKind::Any) => Some("any"),
Ok(TypeKind::ValueBase) => Some("ValueBase"),
Ok(TypeKind::ObjectRef { .. }) => Some("Object"),
_ => None,
};
if let Some(construct) = construct {
return Err(MaterialiseError::NoLanguageForm {
construct: construct.to_string(),
language: LANGUAGE,
at: at.to_string(),
});
}
go_value_type(ast, target)
}
pub fn go_value_type(ast: &Ast, id: Id<TypeNode>) -> Result<go::Type, MaterialiseError> {
value_type_at(ast, id, 0)
}
fn value_type_at(ast: &Ast, id: Id<TypeNode>, depth: usize) -> Result<go::Type, MaterialiseError> {
if depth > MAX_TYPE_DEPTH {
return Err(MaterialiseError::Unevaluated {
construct: format!("type, whose Go spelling nests past {MAX_TYPE_DEPTH}"),
at: go_type_name(ast, id),
});
}
let Ok(node) = ast.type_(id) else {
return Ok(go::Type::Primitive(go::Primitive::Any));
};
Ok(match &node.kind {
TypeKind::Primitive(kind) => go_primitive_type(*kind),
TypeKind::String { .. } | TypeKind::Fixed { .. } => {
go::Type::Primitive(go::Primitive::String)
}
TypeKind::Sequence { element, .. } => {
let Some(element) = element.resolved() else {
return Err(MaterialiseError::Unevaluated {
construct: "sequence element".to_string(),
at: node.name.clone(),
});
};
go::Type::slice(value_type_at(ast, element, depth + 1)?)
}
TypeKind::Array { element, dims } => {
let Some(element) = element.resolved() else {
return Err(MaterialiseError::Unevaluated {
construct: "array element".to_string(),
at: node.name.clone(),
});
};
let mut type_ = value_type_at(ast, element, depth + 1)?;
for dim in dims.iter().rev() {
let length =
ast.eval_integral(*dim)
.map_err(|_| MaterialiseError::Unevaluated {
construct: "array length".to_string(),
at: node.name.clone(),
})?;
type_ = go::Type::array(length, type_);
}
type_
}
TypeKind::Map { .. } => {
go::Type::slice(go::Type::named(&map_entry_name_at(ast, id, depth + 1)?))
}
TypeKind::Alias { aliased } if node.name.is_empty() => match aliased.resolved() {
Some(target) => value_type_at(ast, target, depth + 1)?,
None => go::Type::Primitive(go::Primitive::Any),
},
_ => go::Type::named(&go_type_name(ast, id)),
})
}
#[must_use]
pub fn go_primitive_type(kind: PrimitiveKind) -> go::Type {
let primitive = match kind {
PrimitiveKind::Boolean => go::Primitive::Bool,
PrimitiveKind::Octet | PrimitiveKind::UInt8 | PrimitiveKind::Char => go::Primitive::UInt8,
PrimitiveKind::Int8 => go::Primitive::Int8,
PrimitiveKind::WChar | PrimitiveKind::UInt16 => go::Primitive::UInt16,
PrimitiveKind::Int16 => go::Primitive::Int16,
PrimitiveKind::Int32 => go::Primitive::Int32,
PrimitiveKind::UInt32 => go::Primitive::UInt32,
PrimitiveKind::Int64 => go::Primitive::Int64,
PrimitiveKind::UInt64 => go::Primitive::UInt64,
PrimitiveKind::Float32 => go::Primitive::Float32,
PrimitiveKind::Float64 => go::Primitive::Float64,
PrimitiveKind::Float128 => go::Primitive::String,
};
go::Type::Primitive(primitive)
}
#[must_use]
pub const fn go_holder(bits: u16) -> go::Primitive {
match bits {
0..=8 => go::Primitive::UInt8,
9..=16 => go::Primitive::UInt16,
17..=32 => go::Primitive::UInt32,
_ => go::Primitive::UInt64,
}
}
#[must_use]
pub fn go_type_name(ast: &Ast, id: Id<TypeNode>) -> String {
let Ok(any) = ast.any_of_type(id) else {
return "ZdUnnamed".to_string();
};
match ast.scoped_name_of(any) {
Some(scoped) => flatten(scoped.segments()),
None => "ZdUnnamed".to_string(),
}
}
pub fn go_map_entry_name(ast: &Ast, id: Id<TypeNode>) -> Result<String, MaterialiseError> {
map_entry_name_at(ast, id, 0)
}
fn map_entry_name_at(
ast: &Ast,
id: Id<TypeNode>,
depth: usize,
) -> Result<String, MaterialiseError> {
if depth > MAX_TYPE_DEPTH {
return Err(MaterialiseError::Unevaluated {
construct: format!("map entry name, which nests past {MAX_TYPE_DEPTH}"),
at: go_type_name(ast, id),
});
}
let Ok(node) = ast.type_(id) else {
return Ok(format!("{INVENTED_PREFIX}Unknown"));
};
let TypeKind::Map { key, value, .. } = &node.kind else {
return Ok(format!("{INVENTED_PREFIX}Unknown"));
};
let key = match key.resolved() {
Some(k) => type_tag(ast, k, depth + 1)?,
None => "Unknown".to_string(),
};
let value = match value.resolved() {
Some(v) => type_tag(ast, v, depth + 1)?,
None => "Unknown".to_string(),
};
Ok(format!("{INVENTED_PREFIX}{key}_{value}"))
}
fn type_tag(ast: &Ast, id: Id<TypeNode>, depth: usize) -> Result<String, MaterialiseError> {
if depth > MAX_TYPE_DEPTH {
return Err(MaterialiseError::Unevaluated {
construct: format!("type tag, which nests past {MAX_TYPE_DEPTH}"),
at: go_type_name(ast, id),
});
}
Ok(match ast.type_(id).map(|node| node.kind.clone()) {
Ok(TypeKind::Primitive(kind)) => go::exported(&kind.as_str().replace(' ', "_")),
Ok(TypeKind::String {
width: CharWidth::Char,
..
}) => "String".to_string(),
Ok(TypeKind::String {
width: CharWidth::WChar,
..
}) => "WString".to_string(),
Ok(TypeKind::Sequence { element, .. }) => match element.resolved() {
Some(element) => format!("SeqOf{}", type_tag(ast, element, depth + 1)?),
None => "SeqOfUnknown".to_string(),
},
Ok(TypeKind::Array { element, dims }) => match element.resolved() {
Some(element) => {
let mut name = format!("ArrOf{}", type_tag(ast, element, depth + 1)?);
for dim in &dims {
match ast.eval_integral(*dim) {
Ok(length) => name.push_str(&format!("_{length}")),
Err(_) => name.push_str("_x"),
}
}
name
}
None => "ArrOfUnknown".to_string(),
},
Ok(TypeKind::Map { .. }) => map_entry_name_at(ast, id, depth + 1)?,
_ => go_type_name(ast, id),
})
}
pub fn module_prefix(ast: &Ast, id: Id<TypeNode>) -> String {
let Ok(any) = ast.any_of_type(id) else {
return String::new();
};
let Some(scoped) = ast.scoped_name_of(any) else {
return String::new();
};
let segments = scoped.segments();
if segments.len() <= 1 {
return String::new();
}
let mut prefix = flatten(&segments[..segments.len() - 1]);
prefix.push('_');
prefix
}
fn declared_name(ast: &Ast, name: &str, module: Id<zerodds_ast::ModuleNode>) -> String {
let mut segments: Vec<String> = Vec::new();
let mut current = Some(module);
let mut depth = 0;
while let Some(id) = current {
if depth > MAX_MODULE_DEPTH {
break;
}
depth += 1;
let Ok(node) = ast.module(id) else { break };
if node.name.is_empty() {
break;
}
segments.push(node.name.clone());
current = node.parent;
}
segments.reverse();
segments.push(name.to_string());
let borrowed: Vec<&str> = segments.iter().map(String::as_str).collect();
flatten_str(&borrowed)
}
const MAX_MODULE_DEPTH: usize = 64;
const MAX_TYPE_DEPTH: usize = 64;
fn flatten(segments: &[String]) -> String {
let borrowed: Vec<&str> = segments.iter().map(String::as_str).collect();
flatten_str(&borrowed)
}
fn flatten_str(segments: &[&str]) -> String {
segments
.iter()
.map(|segment| go::exported(&go::escape_name(segment)))
.collect::<Vec<String>>()
.join("_")
}
#[must_use]
pub fn field_name(name: &str) -> String {
go::exported(&go::escape_name(name))
}
fn idl_note(spelled: &str, declared: &str) -> Doc {
if spelled == declared {
return Doc::none();
}
Doc::line(&format!("IDL: `{declared}`."))
}
fn with_idl_note(mut doc: Doc, spelled: &str, declared: &str) -> Doc {
if spelled != declared {
doc.push(&format!("IDL: `{declared}`."));
}
doc
}
fn remember(
emitted: &mut Vec<(String, String)>,
spelled: &str,
declared: &str,
) -> Result<(), MaterialiseError> {
if let Some((_, first)) = emitted.iter().find(|(other, _)| other == spelled) {
return Err(MaterialiseError::SpellingCollision {
first: first.clone(),
second: declared.to_string(),
spelled: spelled.to_string(),
language: LANGUAGE,
at: "the generated package".to_string(),
});
}
emitted.push((spelled.to_string(), declared.to_string()));
Ok(())
}
fn as_pairs(pairs: &[(String, String)]) -> Vec<(&str, &str)> {
pairs
.iter()
.map(|(spelled, declared)| (spelled.as_str(), declared.as_str()))
.collect()
}
fn ghost_created_for(applied: &Applied, node: NodeId, type_name: &str) -> Option<GhostId> {
let anchor = applied
.origin
.iter()
.find(|(_, model)| **model == node)
.map(|(ghost, _)| *ghost)?;
let typ = applied.graph.types.lookup(type_name)?;
applied
.graph
.nodes_of_type(typ)
.map(|created| created.id)
.find(|created| {
applied.graph.connected(&anchor, created)
|| applied
.graph
.parts(&anchor)
.any(|part| applied.graph.connected(&part.other, created))
})
}
fn go_constant(
value: &zerodds_ast::ConstValue,
type_: &go::Type,
at: &str,
) -> Result<go::Expr, MaterialiseError> {
use zerodds_ast::ConstValue;
let refuse = |construct: &str| MaterialiseError::NoLanguageForm {
construct: construct.to_string(),
language: LANGUAGE,
at: at.to_string(),
};
let is_string = matches!(type_, go::Type::Primitive(go::Primitive::String));
Ok(match value {
ConstValue::Boolean(flag) => go::Expr::Literal(Literal::Bool(*flag)),
ConstValue::Integer(number) => go::Expr::Literal(Literal::int(*number)),
ConstValue::Char(character) => {
go::Expr::Literal(Literal::int(i128::from(u32::from(*character))))
}
ConstValue::WChar(point) => go::Expr::Literal(Literal::int(i128::from(*point))),
ConstValue::Float { mantissa, exponent } => {
if is_string {
return Err(refuse("a `long double` constant, which has no Go literal"));
}
go::Expr::Literal(Literal::Float(constant::float_text(
mantissa,
exponent.as_deref(),
)))
}
ConstValue::Fixed { digits, scale } => {
go::Expr::Literal(Literal::Str(constant::fixed_decimal(digits, *scale)))
}
ConstValue::String(text) => go::Expr::Literal(Literal::Str(text.clone())),
ConstValue::WString(points) => go::Expr::Literal(Literal::Str(
constant::wstring_utf8(points)
.ok_or_else(|| refuse("a `wstring` constant holding a lone surrogate"))?,
)),
_ => return Err(refuse("a constant form this binding has no literal for")),
})
}