Skip to main content

wit_bindgen_cpp/
lib.rs

1use anyhow::bail;
2use heck::{ToPascalCase, ToShoutySnakeCase, ToSnakeCase, ToUpperCamelCase};
3use std::{
4    collections::{HashMap, HashSet},
5    fmt::{self, Display, Write as FmtWrite},
6    io::{Read, Write},
7    path::PathBuf,
8    process::{Command, Stdio},
9    str::FromStr,
10};
11use symbol_name::{make_external_component, make_external_symbol};
12use wit_bindgen_c::to_c_ident;
13use wit_bindgen_core::{
14    Files, InterfaceGenerator, Source, Types, WorldGenerator,
15    abi::{self, AbiVariant, Bindgen, Bitcast, LiftLower, WasmSignature, WasmType},
16    name_package_module, uwrite, uwriteln,
17    wit_parser::{
18        Alignment, ArchitectureSize, Docs, Function, FunctionKind, Handle, Int, InterfaceId, Param,
19        Resolve, SizeAlign, Stability, Type, TypeDef, TypeDefKind, TypeId, TypeOwner, WorldId,
20        WorldKey,
21    },
22};
23
24// mod wamr;
25mod symbol_name;
26
27pub const RESOURCE_IMPORT_BASE_CLASS_NAME: &str = "ResourceImportBase";
28pub const RESOURCE_EXPORT_BASE_CLASS_NAME: &str = "ResourceExportBase";
29pub const RESOURCE_TABLE_NAME: &str = "ResourceTable";
30pub const OWNED_CLASS_NAME: &str = "Owned";
31pub const POINTER_SIZE_EXPRESSION: &str = "sizeof(void*)";
32
33type CppType = String;
34
35#[derive(Clone, Copy, Debug)]
36enum Flavor {
37    Argument(AbiVariant),
38    Result(AbiVariant),
39    InStruct,
40    BorrowedArgument,
41}
42
43#[derive(Default)]
44struct HighlevelSignature {
45    /// this is a constructor or destructor without a written type
46    // implicit_result: bool, -> empty result
47    const_member: bool,
48    static_member: bool,
49    result: CppType,
50    arguments: Vec<(String, CppType)>,
51    name: String,
52    namespace: Vec<String>,
53    implicit_self: bool,
54    post_return: bool,
55}
56
57// follows https://google.github.io/styleguide/cppguide.html
58
59#[derive(Default)]
60struct Includes {
61    needs_vector: bool,
62    needs_expected: bool,
63    needs_string: bool,
64    needs_string_view: bool,
65    needs_optional: bool,
66    needs_cstring: bool,
67    needs_imported_resources: bool,
68    needs_exported_resources: bool,
69    needs_variant: bool,
70    needs_tuple: bool,
71    needs_assert: bool,
72    needs_bit: bool,
73    needs_span: bool,
74    // needs wit types
75    needs_wit: bool,
76    needs_memory: bool,
77    needs_array: bool,
78}
79
80#[derive(Default)]
81struct SourceWithState {
82    src: Source,
83    namespace: Vec<String>,
84}
85
86#[derive(Eq, Hash, PartialEq, Clone, Copy, Debug)]
87enum Direction {
88    Import,
89    Export,
90}
91
92#[derive(Default)]
93struct Cpp {
94    opts: Opts,
95    c_src: SourceWithState,
96    h_src: SourceWithState,
97    c_src_head: Source,
98    extern_c_decls: Source,
99    dependencies: Includes,
100    includes: Vec<String>,
101    world: String,
102    world_id: Option<WorldId>,
103    imported_interfaces: HashSet<InterfaceId>,
104    user_class_files: HashMap<String, String>,
105    defined_types: HashSet<(Vec<String>, String)>,
106    types: Types,
107
108    // needed for symmetric disambiguation
109    interface_prefixes: HashMap<(Direction, WorldKey), String>,
110    import_prefix: Option<String>,
111}
112
113#[cfg(feature = "clap")]
114fn parse_with(s: &str) -> Result<(String, String), String> {
115    let (k, v) = s.split_once('=').ok_or_else(|| {
116        format!("expected string of form `<key>=<value>[,<key>=<value>...]`; got `{s}`")
117    })?;
118    Ok((k.to_string(), v.to_string()))
119}
120
121#[derive(Default, Debug, Clone)]
122#[cfg_attr(feature = "clap", derive(clap::Args))]
123pub struct Opts {
124    /// Call clang-format on the generated code
125    #[cfg_attr(feature = "clap", arg(long, default_value_t = bool::default()))]
126    pub format: bool,
127
128    /// Place each interface in its own file,
129    /// this enables sharing bindings across projects
130    #[cfg_attr(feature = "clap", arg(long, default_value_t = bool::default()))]
131    pub split_interfaces: bool,
132
133    /// Optionally prefix any export names with the specified value.
134    ///
135    /// This is useful to avoid name conflicts when testing.
136    #[cfg_attr(feature = "clap", arg(long))]
137    pub export_prefix: Option<String>,
138
139    /// Wrap all C++ classes inside a custom namespace.
140    ///
141    /// This avoids identical names across components, useful for native
142    #[cfg_attr(feature = "clap", arg(long))]
143    pub internal_prefix: Option<String>,
144
145    /// Set API style to symmetric or asymmetric
146    #[cfg_attr(
147        feature = "clap",
148        arg(
149            long,
150            default_value_t = APIStyle::default(),
151            value_name = "STYLE",
152        ),
153    )]
154    pub api_style: APIStyle,
155
156    /// Whether to generate owning or borrowing type definitions for `record` arguments to imported functions.
157    ///
158    /// Valid values include:
159    ///
160    /// - `owning`: Generated types will be composed entirely of owning fields,
161    ///   regardless of whether they are used as parameters to imports or not.
162    ///
163    /// - `coarse-borrowing`: Generated types used as parameters to imports will be
164    ///   "deeply borrowing", i.e. contain references rather than owned values,
165    ///   so long as they don't contain resources, in which case they will be
166    ///   owning.
167    ///
168    /// - `fine-borrowing": Generated types used as parameters to imports will be
169    ///   "deeply borrowing", i.e. contain references rather than owned values
170    ///   for all fields that are not resources, which will be owning.
171    #[cfg_attr(feature = "clap", arg(long, default_value_t = Ownership::Owning))]
172    pub ownership: Ownership,
173
174    /// Where to place output files
175    #[cfg_attr(feature = "clap", arg(skip))]
176    out_dir: Option<PathBuf>,
177
178    /// Importing wit interface from custom include
179    ///
180    /// Argument must be of the form `k=v` and this option can be passed
181    /// multiple times or one option can be comma separated, for example
182    /// `k1=v1,k2=v2`.
183    #[cfg_attr(feature = "clap", arg(long, value_parser = parse_with, value_delimiter = ','))]
184    pub with: Vec<(String, String)>,
185}
186
187/// Supported API styles for the generated bindings.
188#[derive(Default, Debug, Copy, Clone, PartialEq, Eq, Hash)]
189pub enum APIStyle {
190    /// Imported functions borrow arguments, while exported functions receive owned arguments. Reduces the allocation overhead for the canonical ABI.
191    #[default]
192    Asymmetric,
193    /// Same API for imported and exported functions. Reduces the allocation overhead for symmetric ABI.
194    Symmetric,
195}
196
197impl Display for APIStyle {
198    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
199        match self {
200            APIStyle::Asymmetric => write!(f, "asymmetric"),
201            APIStyle::Symmetric => write!(f, "symmetric"),
202        }
203    }
204}
205
206impl FromStr for APIStyle {
207    type Err = anyhow::Error;
208
209    fn from_str(s: &str) -> Result<Self, Self::Err> {
210        match s {
211            "asymmetric" => Ok(APIStyle::Asymmetric),
212            "symmetric" => Ok(APIStyle::Symmetric),
213            _ => bail!("unrecognized API style: `{s}`; expected `asymmetric` or `symmetric`"),
214        }
215    }
216}
217
218#[derive(Default, Debug, Clone, Copy)]
219pub enum Ownership {
220    /// Generated types will be composed entirely of owning fields, regardless
221    /// of whether they are used as parameters to imports or not.
222    #[default]
223    Owning,
224
225    /// Generated types used as parameters to imports will be "deeply
226    /// borrowing", i.e. contain references rather than owned values when
227    /// applicable.
228    CoarseBorrowing,
229
230    /// Generated types used as parameters to imports will be "deeply
231    /// borrowing", i.e. contain references rather than owned values
232    /// for all fields that are not resources, which will be owning.
233    FineBorrowing,
234}
235
236impl FromStr for Ownership {
237    type Err = String;
238
239    fn from_str(s: &str) -> Result<Self, Self::Err> {
240        match s {
241            "owning" => Ok(Self::Owning),
242            "coarse-borrowing" => Ok(Self::CoarseBorrowing),
243            "fine-borrowing" => Ok(Self::FineBorrowing),
244            _ => Err(format!(
245                "unrecognized ownership: `{s}`; \
246                 expected `owning`, `coarse-borrowing`, or `fine-borrowing`"
247            )),
248        }
249    }
250}
251
252impl fmt::Display for Ownership {
253    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
254        f.write_str(match self {
255            Ownership::Owning => "owning",
256            Ownership::CoarseBorrowing => "coarse-borrowing",
257            Ownership::FineBorrowing => "fine-borrowing",
258        })
259    }
260}
261
262impl Opts {
263    pub fn build(mut self, out_dir: Option<&PathBuf>) -> Box<dyn WorldGenerator> {
264        let mut r = Cpp::new();
265        self.out_dir = out_dir.cloned();
266        r.opts = self;
267        Box::new(r)
268    }
269
270    fn is_only_handle(&self, variant: AbiVariant) -> bool {
271        !matches!(variant, AbiVariant::GuestExport)
272    }
273
274    fn ptr_type(&self) -> &'static str {
275        "uint8_t*"
276    }
277}
278
279impl Cpp {
280    fn new() -> Cpp {
281        Cpp::default()
282    }
283
284    pub fn is_first_definition(&mut self, ns: &Vec<String>, name: &str) -> bool {
285        let owned = (ns.to_owned(), name.to_owned());
286        if !self.defined_types.contains(&owned) {
287            self.defined_types.insert(owned);
288            true
289        } else {
290            false
291        }
292    }
293
294    fn include(&mut self, s: &str) {
295        self.includes.push(s.to_string());
296    }
297
298    /// Returns true if the function is a fallible constructor.
299    ///
300    /// Fallible constructors are constructors that return `result<T, E>` instead of just `T`.
301    /// In the generated C++ code, these become static factory methods named `Create` that
302    /// return `std::expected<T, E>`, rather than regular constructors.
303    fn is_fallible_constructor(&self, resolve: &Resolve, func: &Function) -> bool {
304        matches!(&func.kind, FunctionKind::Constructor(_))
305            && func.result.as_ref().is_some_and(|ty| {
306                if let Type::Id(id) = ty {
307                    matches!(&resolve.types[*id].kind, TypeDefKind::Result(_))
308                } else {
309                    false
310                }
311            })
312    }
313
314    fn interface<'a>(
315        &'a mut self,
316        resolve: &'a Resolve,
317        name: Option<&'a WorldKey>,
318        in_guest_import: bool,
319        wasm_import_module: Option<String>,
320    ) -> CppInterfaceGenerator<'a> {
321        let mut sizes = SizeAlign::default();
322        sizes.fill(resolve);
323
324        CppInterfaceGenerator {
325            _src: Source::default(),
326            r#gen: self,
327            resolve,
328            interface: None,
329            _name: name,
330            sizes,
331            in_guest_import,
332            wasm_import_module,
333        }
334    }
335
336    fn clang_format(code: &mut String) {
337        let mut child = Command::new("clang-format")
338            .stdin(Stdio::piped())
339            .stdout(Stdio::piped())
340            .spawn()
341            .expect("failed to spawn `clang-format`");
342        child
343            .stdin
344            .take()
345            .unwrap()
346            .write_all(code.as_bytes())
347            .unwrap();
348        code.truncate(0);
349        child.stdout.take().unwrap().read_to_string(code).unwrap();
350        let status = child.wait().unwrap();
351        assert!(status.success());
352    }
353
354    fn perform_cast(&mut self, op: &str, cast: &Bitcast) -> String {
355        match cast {
356            Bitcast::I32ToF32 | Bitcast::I64ToF32 => {
357                self.dependencies.needs_bit = true;
358                format!("std::bit_cast<float, int32_t>({op})")
359            }
360            Bitcast::F32ToI32 | Bitcast::F32ToI64 => {
361                self.dependencies.needs_bit = true;
362                format!("std::bit_cast<int32_t, float>({op})")
363            }
364            Bitcast::I64ToF64 => {
365                self.dependencies.needs_bit = true;
366                format!("std::bit_cast<double, int64_t>({op})")
367            }
368            Bitcast::F64ToI64 => {
369                self.dependencies.needs_bit = true;
370                format!("std::bit_cast<int64_t, double>({op})")
371            }
372            Bitcast::I32ToI64 | Bitcast::LToI64 | Bitcast::PToP64 => {
373                format!("(int64_t) {op}")
374            }
375            Bitcast::I64ToI32 | Bitcast::PToI32 | Bitcast::LToI32 => {
376                format!("(int32_t) {op}")
377            }
378            Bitcast::P64ToI64 | Bitcast::None | Bitcast::I64ToP64 => op.to_string(),
379            Bitcast::P64ToP | Bitcast::I32ToP | Bitcast::LToP => {
380                format!("(uint8_t*) {op}")
381            }
382            Bitcast::PToL | Bitcast::I32ToL | Bitcast::I64ToL => {
383                format!("(size_t) {op}")
384            }
385            Bitcast::Sequence(sequence) => {
386                let [first, second] = &**sequence;
387                let inner = self.perform_cast(op, first);
388                self.perform_cast(&inner, second)
389            }
390        }
391    }
392
393    fn finish_includes(&mut self) {
394        self.include("<cstdint>");
395        self.include("<utility>"); // for std::move
396        if self.dependencies.needs_string {
397            self.include("<string>");
398        }
399        if self.dependencies.needs_string_view {
400            self.include("<string_view>");
401        }
402        if self.dependencies.needs_vector {
403            self.include("<vector>");
404        }
405        if self.dependencies.needs_expected {
406            self.include("<expected>");
407        }
408        if self.dependencies.needs_optional {
409            self.include("<optional>");
410        }
411        if self.dependencies.needs_cstring {
412            self.include("<cstring>");
413        }
414        if self.dependencies.needs_imported_resources {
415            self.include("<cassert>");
416        }
417        if self.dependencies.needs_exported_resources {
418            self.include("<map>");
419        }
420        if self.dependencies.needs_variant {
421            self.include("<variant>");
422        }
423        if self.dependencies.needs_tuple {
424            self.include("<tuple>");
425        }
426        if self.dependencies.needs_wit {
427            self.include("\"wit.h\"");
428        }
429        if self.dependencies.needs_memory {
430            self.include("<memory>");
431        }
432        if self.dependencies.needs_array {
433            self.include("<array>");
434        }
435        if self.dependencies.needs_bit {
436            self.include("<bit>");
437        }
438    }
439
440    fn start_new_file(&mut self, condition: Option<bool>) -> FileContext {
441        if condition == Some(true) || self.opts.split_interfaces {
442            FileContext {
443                includes: std::mem::take(&mut self.includes),
444                src: std::mem::take(&mut self.h_src),
445                dependencies: std::mem::take(&mut self.dependencies),
446            }
447        } else {
448            Default::default()
449        }
450    }
451
452    fn finish_file(&mut self, namespace: &[String], store: FileContext) {
453        if !store.src.src.is_empty() {
454            let mut header = String::default();
455            self.finish_includes();
456            self.h_src.change_namespace(&[]);
457            uwriteln!(header, "#pragma once");
458            for include in self.includes.iter() {
459                uwriteln!(header, "#include {include}");
460            }
461            header.push_str(&self.h_src.src);
462            let mut filename = namespace.join("-");
463            filename.push_str(".h");
464            if self.opts.format {
465                Self::clang_format(&mut header);
466            }
467            self.user_class_files.insert(filename.clone(), header);
468
469            let _ = std::mem::replace(&mut self.includes, store.includes);
470            let _ = std::mem::replace(&mut self.h_src, store.src);
471            let _ = std::mem::replace(&mut self.dependencies, store.dependencies);
472            self.includes.push(String::from("\"") + &filename + "\"");
473        }
474    }
475}
476
477#[derive(Default)]
478struct FileContext {
479    includes: Vec<String>,
480    src: SourceWithState,
481    dependencies: Includes,
482}
483
484impl WorldGenerator for Cpp {
485    fn preprocess(&mut self, resolve: &Resolve, world: WorldId) {
486        let name = &resolve.worlds[world].name;
487        self.world = name.to_string();
488        self.types.analyze(resolve);
489        self.world_id = Some(world);
490        uwriteln!(
491            self.c_src_head,
492            r#"#include "{}_cpp.h"
493            #include <cstdlib> // realloc
494
495            extern "C" void *cabi_realloc(void *ptr, size_t old_size, size_t align, size_t new_size);
496
497            __attribute__((__weak__, __export_name__("cabi_realloc")))
498            void *cabi_realloc(void *ptr, size_t old_size, size_t align, size_t new_size) {{
499                (void) old_size;
500                if (new_size == 0) return (void*) align;
501                void *ret = realloc(ptr, new_size);
502                if (!ret) abort();
503                return ret;
504            }}
505
506            "#,
507            self.world.to_snake_case(),
508        );
509    }
510
511    fn import_interface(
512        &mut self,
513        resolve: &Resolve,
514        name: &WorldKey,
515        id: InterfaceId,
516        _files: &mut Files,
517    ) -> anyhow::Result<()> {
518        self.imported_interfaces.insert(id);
519
520        let full_name = resolve.name_world_key(name);
521        match self.opts.with.iter().find(|e| e.0 == full_name) {
522            None => {
523                if let Some(prefix) = self
524                    .interface_prefixes
525                    .get(&(Direction::Import, name.clone()))
526                {
527                    self.import_prefix = Some(prefix.clone());
528                }
529
530                let store = self.start_new_file(None);
531                let wasm_import_module = resolve.name_world_key(name);
532                let binding = Some(name);
533                let mut r#gen = self.interface(resolve, binding, true, Some(wasm_import_module));
534                r#gen.interface = Some(id);
535                r#gen.types(id);
536                let namespace =
537                    namespace(resolve, &TypeOwner::Interface(id), false, &r#gen.r#gen.opts);
538
539                for (_name, func) in resolve.interfaces[id].functions.iter() {
540                    if matches!(func.kind, FunctionKind::Freestanding) {
541                        r#gen.r#gen.h_src.change_namespace(&namespace);
542                        r#gen.generate_function(
543                            func,
544                            &TypeOwner::Interface(id),
545                            AbiVariant::GuestImport,
546                        );
547                    }
548                }
549                self.finish_file(&namespace, store);
550            }
551            Some((_, val)) => {
552                let with_quotes = format!("\"{val}\"");
553                if !self.includes.contains(&with_quotes) {
554                    self.includes.push(with_quotes);
555                }
556            }
557        }
558        let _ = self.import_prefix.take();
559        Ok(())
560    }
561
562    fn export_interface(
563        &mut self,
564        resolve: &Resolve,
565        name: &WorldKey,
566        id: InterfaceId,
567        _files: &mut Files,
568    ) -> anyhow::Result<()> {
569        let old_prefix = self.opts.export_prefix.clone();
570        if let Some(prefix) = self
571            .interface_prefixes
572            .get(&(Direction::Export, name.clone()))
573        {
574            self.opts.export_prefix =
575                Some(prefix.clone() + old_prefix.as_ref().unwrap_or(&String::new()));
576        }
577        let store = self.start_new_file(None);
578        self.h_src
579            .src
580            .push_str(&format!("// export_interface {name:?}\n"));
581        self.imported_interfaces.remove(&id);
582        let wasm_import_module = resolve.name_world_key(name);
583        let binding = Some(name);
584        let mut r#gen = self.interface(resolve, binding, false, Some(wasm_import_module));
585        r#gen.interface = Some(id);
586        r#gen.types(id);
587        let namespace = namespace(resolve, &TypeOwner::Interface(id), true, &r#gen.r#gen.opts);
588
589        for (_name, func) in resolve.interfaces[id].functions.iter() {
590            if matches!(func.kind, FunctionKind::Freestanding) {
591                r#gen.r#gen.h_src.change_namespace(&namespace);
592                r#gen.generate_function(func, &TypeOwner::Interface(id), AbiVariant::GuestExport);
593            }
594        }
595        self.finish_file(&namespace, store);
596        self.opts.export_prefix = old_prefix;
597        Ok(())
598    }
599
600    fn import_funcs(
601        &mut self,
602        resolve: &Resolve,
603        world: WorldId,
604        funcs: &[(&str, &Function)],
605        _files: &mut Files,
606    ) {
607        let name = WorldKey::Name("$root".to_string()); //WorldKey::Name(resolve.worlds[world].name.clone());
608        let wasm_import_module = resolve.name_world_key(&name);
609        let binding = Some(name);
610        let mut r#gen = self.interface(resolve, binding.as_ref(), true, Some(wasm_import_module));
611        let namespace = namespace(resolve, &TypeOwner::World(world), false, &r#gen.r#gen.opts);
612
613        for (_name, func) in funcs.iter() {
614            if matches!(func.kind, FunctionKind::Freestanding) {
615                r#gen.r#gen.h_src.change_namespace(&namespace);
616                r#gen.generate_function(func, &TypeOwner::World(world), AbiVariant::GuestImport);
617            }
618        }
619    }
620
621    fn export_funcs(
622        &mut self,
623        resolve: &Resolve,
624        world: WorldId,
625        funcs: &[(&str, &Function)],
626        _files: &mut Files,
627    ) -> anyhow::Result<()> {
628        let name = WorldKey::Name(resolve.worlds[world].name.clone());
629        let binding = Some(name);
630        let mut r#gen = self.interface(resolve, binding.as_ref(), false, None);
631        let namespace = namespace(resolve, &TypeOwner::World(world), true, &r#gen.r#gen.opts);
632
633        for (_name, func) in funcs.iter() {
634            if matches!(func.kind, FunctionKind::Freestanding) {
635                r#gen.r#gen.h_src.change_namespace(&namespace);
636                r#gen.generate_function(func, &TypeOwner::World(world), AbiVariant::GuestExport);
637            }
638        }
639        Ok(())
640    }
641
642    fn import_types(
643        &mut self,
644        resolve: &Resolve,
645        _world: WorldId,
646        types: &[(&str, TypeId)],
647        _files: &mut Files,
648    ) {
649        let mut r#gen = self.interface(resolve, None, true, Some("$root".to_string()));
650        for (name, id) in types.iter() {
651            r#gen.define_type(name, *id);
652        }
653    }
654
655    fn finish(
656        &mut self,
657        resolve: &Resolve,
658        world_id: WorldId,
659        files: &mut Files,
660    ) -> std::result::Result<(), anyhow::Error> {
661        let world = &resolve.worlds[world_id];
662        let snake = world.name.to_snake_case();
663        let linking_symbol = wit_bindgen_c::component_type_object::linking_symbol(&world.name);
664
665        let mut h_str = SourceWithState::default();
666        let mut c_str = SourceWithState::default();
667
668        let version = env!("CARGO_PKG_VERSION");
669        uwriteln!(
670            h_str.src,
671            "// Generated by `wit-bindgen` {version}. DO NOT EDIT!"
672        );
673
674        uwrite!(
675            h_str.src,
676            "#ifndef __CPP_GUEST_BINDINGS_{0}_H
677            #define __CPP_GUEST_BINDINGS_{0}_H\n",
678            world.name.to_shouty_snake_case(),
679        );
680        self.finish_includes();
681
682        for include in self.includes.iter() {
683            uwriteln!(h_str.src, "#include {include}");
684        }
685
686        uwriteln!(
687            c_str.src,
688            "// Generated by `wit-bindgen` {version}. DO NOT EDIT!"
689        );
690        uwriteln!(
691            c_str.src,
692            "\n// Ensure that the *_component_type.o object is linked in"
693        );
694        uwrite!(
695            c_str.src,
696            "#ifdef __wasm32__
697                   extern \"C\" void {linking_symbol}(void);
698                   __attribute__((used))
699                   void {linking_symbol}_public_use_in_this_compilation_unit(void) {{
700                       {linking_symbol}();
701                   }}
702                   #endif
703               ",
704        );
705        if self.dependencies.needs_assert {
706            uwriteln!(c_str.src, "#include <assert.h>");
707        }
708
709        h_str.change_namespace(&Vec::default());
710
711        self.c_src.change_namespace(&Vec::default());
712        c_str.src.push_str(&self.c_src_head);
713        c_str.src.push_str(&self.extern_c_decls);
714        c_str.src.push_str(&self.c_src.src);
715        self.h_src.change_namespace(&Vec::default());
716        h_str.src.push_str(&self.h_src.src);
717
718        uwriteln!(c_str.src, "\n// Component Adapters");
719
720        uwriteln!(
721            h_str.src,
722            "
723            #endif"
724        );
725
726        if self.opts.format {
727            Self::clang_format(c_str.src.as_mut_string());
728            Self::clang_format(h_str.src.as_mut_string());
729        }
730
731        files.push(&format!("{snake}.cpp"), c_str.src.as_bytes());
732        files.push(&format!("{snake}_cpp.h"), h_str.src.as_bytes());
733        for (name, content) in self.user_class_files.iter() {
734            // if the user class file exists create an updated .template
735            let dst = match &self.opts.out_dir {
736                Some(path) => path.join(name),
737                None => name.into(),
738            };
739            if std::path::Path::exists(&dst) {
740                files.push(&(String::from(name) + ".template"), content.as_bytes());
741            } else {
742                files.push(name, content.as_bytes());
743            }
744        }
745        files.push(
746            &format!("{snake}_component_type.o",),
747            wit_bindgen_c::component_type_object::object(
748                resolve,
749                world_id,
750                &world.name,
751                wit_component::StringEncoding::UTF8,
752                None,
753            )
754            .unwrap()
755            .as_slice(),
756        );
757
758        if self.dependencies.needs_wit {
759            files.push("wit.h", include_bytes!("../helper-types/wit.h"));
760        }
761        Ok(())
762    }
763}
764
765// determine namespace (for the lifted C++ function)
766fn namespace(resolve: &Resolve, owner: &TypeOwner, guest_export: bool, opts: &Opts) -> Vec<String> {
767    let mut result = Vec::default();
768    if let Some(prefix) = &opts.internal_prefix {
769        result.push(prefix.clone());
770    }
771    if guest_export {
772        result.push(String::from("exports"));
773    }
774    match owner {
775        TypeOwner::World(w) => result.push(to_c_ident(&resolve.worlds[*w].name)),
776        TypeOwner::Interface(i) => {
777            let iface = &resolve.interfaces[*i];
778            let pkg_id = iface.package.unwrap();
779            let pkg = &resolve.packages[pkg_id];
780            result.push(to_c_ident(&pkg.name.namespace));
781            // Use name_package_module to get version-specific package names
782            result.push(to_c_ident(&name_package_module(resolve, pkg_id)));
783            if let Some(name) = &iface.name {
784                result.push(to_c_ident(name));
785            }
786        }
787        TypeOwner::None => (),
788    }
789    result
790}
791
792impl SourceWithState {
793    fn change_namespace(&mut self, target: &[String]) {
794        let mut same = 0;
795        // itertools::fold_while?
796        for (a, b) in self.namespace.iter().zip(target.iter()) {
797            if a == b {
798                same += 1;
799            } else {
800                break;
801            }
802        }
803        for _i in same..self.namespace.len() {
804            uwrite!(self.src, "}}\n");
805        }
806        self.namespace.truncate(same);
807        for i in target.iter().skip(same) {
808            uwrite!(self.src, "namespace {} {{\n", i);
809            self.namespace.push(i.clone());
810        }
811    }
812
813    fn qualify(&mut self, target: &[String]) {
814        let mut same = 0;
815        // itertools::fold_while?
816        for (a, b) in self.namespace.iter().zip(target.iter()) {
817            if a == b {
818                same += 1;
819            } else {
820                break;
821            }
822        }
823        if same == 0 && !target.is_empty() {
824            // if the root namespace exists below the current namespace we need to start at root
825            // Also ensure absolute qualification when crossing from exports to imports
826            if self.namespace.contains(target.first().unwrap())
827                || (self.namespace.first().map(|s| s.as_str()) == Some("exports")
828                    && target.first().map(|s| s.as_str()) != Some("exports"))
829            {
830                self.src.push_str("::");
831            }
832        }
833        if same == target.len() && self.namespace.len() != target.len() && same > 0 {
834            // namespace is parent, qualify at least one namespace (and cross fingers)
835            uwrite!(self.src, "{}::", target[same - 1]);
836        } else {
837            for i in target.iter().skip(same) {
838                uwrite!(self.src, "{i}::");
839            }
840        }
841    }
842}
843
844struct CppInterfaceGenerator<'a> {
845    _src: Source,
846    r#gen: &'a mut Cpp,
847    resolve: &'a Resolve,
848    interface: Option<InterfaceId>,
849    _name: Option<&'a WorldKey>,
850    sizes: SizeAlign,
851    in_guest_import: bool,
852    pub wasm_import_module: Option<String>,
853}
854
855impl CppInterfaceGenerator<'_> {
856    fn types(&mut self, iface: InterfaceId) {
857        let iface_data = &self.resolve().interfaces[iface];
858
859        // First pass: emit forward declarations for all resources
860        // This ensures resources can reference each other in method signatures
861        for (name, id) in iface_data.types.iter() {
862            let ty = &self.resolve().types[*id];
863            if matches!(&ty.kind, TypeDefKind::Resource) {
864                let pascal = name.to_upper_camel_case();
865                let guest_import = self.r#gen.imported_interfaces.contains(&iface);
866                let namespc = namespace(self.resolve, &ty.owner, !guest_import, &self.r#gen.opts);
867                self.r#gen.h_src.change_namespace(&namespc);
868                uwriteln!(self.r#gen.h_src.src, "class {pascal};");
869            }
870        }
871
872        // Second pass: emit full type definitions
873        for (name, id) in iface_data.types.iter() {
874            self.define_type(name, *id);
875        }
876    }
877
878    fn define_type(&mut self, name: &str, id: TypeId) {
879        let ty = &self.resolve().types[id];
880        match &ty.kind {
881            TypeDefKind::Record(record) => self.type_record(id, name, record, &ty.docs),
882            TypeDefKind::Resource => self.type_resource(id, name, &ty.docs),
883            TypeDefKind::Flags(flags) => self.type_flags(id, name, flags, &ty.docs),
884            TypeDefKind::Tuple(tuple) => self.type_tuple(id, name, tuple, &ty.docs),
885            TypeDefKind::Enum(enum_) => self.type_enum(id, name, enum_, &ty.docs),
886            TypeDefKind::Variant(variant) => self.type_variant(id, name, variant, &ty.docs),
887            TypeDefKind::Option(t) => self.type_option(id, name, t, &ty.docs),
888            TypeDefKind::Result(r) => self.type_result(id, name, r, &ty.docs),
889            TypeDefKind::List(t) => self.type_list(id, name, t, &ty.docs),
890            TypeDefKind::Type(t) => self.type_alias(id, name, t, &ty.docs),
891            TypeDefKind::Future(_) => todo!("generate for future"),
892            TypeDefKind::Stream(_) => todo!("generate for stream"),
893            TypeDefKind::Handle(_) => todo!("generate for handle"),
894            TypeDefKind::FixedLengthList(_, _) => todo!(),
895            TypeDefKind::Map(_, _) => todo!(),
896            TypeDefKind::Unknown => unreachable!(),
897        }
898    }
899
900    /// This describes the C++ side name
901    fn func_namespace_name(
902        &self,
903        func: &Function,
904        guest_export: bool,
905        cpp_file: bool,
906    ) -> (Vec<String>, String) {
907        let (object, owner) = match &func.kind {
908            FunctionKind::Freestanding => None,
909            FunctionKind::Method(i) => Some(i),
910            FunctionKind::Static(i) => Some(i),
911            FunctionKind::Constructor(i) => Some(i),
912            FunctionKind::AsyncFreestanding => todo!(),
913            FunctionKind::AsyncMethod(_id) => todo!(),
914            FunctionKind::AsyncStatic(_id) => todo!(),
915        }
916        .map(|i| {
917            let ty = &self.resolve.types[*i];
918            (ty.name.as_ref().unwrap().to_pascal_case(), ty.owner)
919        })
920        .unwrap_or((
921            Default::default(),
922            self.interface
923                .map(TypeOwner::Interface)
924                .unwrap_or(TypeOwner::World(self.r#gen.world_id.unwrap())),
925        ));
926        let mut namespace = namespace(self.resolve, &owner, guest_export, &self.r#gen.opts);
927        let is_drop = is_special_method(func);
928        let func_name_h = if !matches!(&func.kind, FunctionKind::Freestanding) {
929            namespace.push(object.clone());
930            if let FunctionKind::Constructor(_i) = &func.kind {
931                // Fallible constructors return result<T, E> and are static factory methods
932                let is_fallible_constructor =
933                    self.r#gen.is_fallible_constructor(self.resolve, func);
934
935                if is_fallible_constructor {
936                    String::from("Create")
937                } else if guest_export && cpp_file {
938                    String::from("New")
939                } else {
940                    object.clone()
941                }
942            } else {
943                match is_drop {
944                    SpecialMethod::ResourceDrop => {
945                        if guest_export {
946                            "ResourceDrop".to_string()
947                        } else {
948                            "~".to_string() + &object
949                        }
950                    }
951                    SpecialMethod::Dtor => "Dtor".to_string(),
952                    SpecialMethod::ResourceNew => "ResourceNew".to_string(),
953                    SpecialMethod::ResourceRep => "ResourceRep".to_string(),
954                    SpecialMethod::Allocate => "New".to_string(),
955                    SpecialMethod::None => func.item_name().to_pascal_case(),
956                }
957            }
958        } else {
959            func.name.to_pascal_case()
960        };
961        (namespace, func_name_h)
962    }
963
964    // print the signature of the guest export (lowered (wasm) function calling into highlevel)
965    fn print_export_signature(&mut self, func: &Function, variant: AbiVariant) -> Vec<String> {
966        let is_drop = is_special_method(func);
967        let id_type = WasmType::I32;
968        let signature = match is_drop {
969            SpecialMethod::ResourceDrop => WasmSignature {
970                params: vec![id_type],
971                results: Vec::new(),
972                indirect_params: false,
973                retptr: false,
974            },
975            SpecialMethod::ResourceRep => WasmSignature {
976                params: vec![id_type],
977                results: vec![WasmType::Pointer],
978                indirect_params: false,
979                retptr: false,
980            },
981            SpecialMethod::Dtor => WasmSignature {
982                params: vec![WasmType::Pointer],
983                results: Vec::new(),
984                indirect_params: false,
985                retptr: false,
986            },
987            SpecialMethod::ResourceNew => WasmSignature {
988                params: vec![WasmType::Pointer],
989                results: vec![id_type],
990                indirect_params: false,
991                retptr: false,
992            },
993            SpecialMethod::None => {
994                // TODO perhaps remember better names for the arguments
995                self.resolve.wasm_signature(variant, func)
996            }
997            SpecialMethod::Allocate => WasmSignature {
998                params: vec![],
999                results: vec![],
1000                indirect_params: false,
1001                retptr: false,
1002            },
1003        };
1004        let mut module_name = self.wasm_import_module.clone();
1005        let symbol_variant = variant;
1006        if matches!(variant, AbiVariant::GuestExport)
1007            && matches!(
1008                is_drop,
1009                SpecialMethod::ResourceNew
1010                    | SpecialMethod::ResourceDrop
1011                    | SpecialMethod::ResourceRep
1012            )
1013        {
1014            module_name = Some(String::from("[export]") + &module_name.unwrap());
1015        }
1016        let func_name = func.name.clone();
1017        let module_prefix = module_name.as_ref().map_or(String::default(), |name| {
1018            let mut res = name.clone();
1019            res.push('#');
1020            res
1021        });
1022        uwriteln!(
1023            self.r#gen.c_src.src,
1024            r#"extern "C" __attribute__((__export_name__("{module_prefix}{func_name}")))"#
1025        );
1026        let return_via_pointer = false;
1027        self.r#gen
1028            .c_src
1029            .src
1030            .push_str(if signature.results.is_empty() || return_via_pointer {
1031                "void"
1032            } else {
1033                wit_bindgen_c::wasm_type(signature.results[0])
1034            });
1035        self.r#gen.c_src.src.push_str(" ");
1036        let export_name = match module_name {
1037            Some(ref module_name) => make_external_symbol(module_name, &func_name, symbol_variant),
1038            None => make_external_component(&func_name),
1039        };
1040        // Add prefix to C ABI export functions to avoid conflicts with C++ namespaces
1041        self.r#gen.c_src.src.push_str("__wasm_export_");
1042        if let Some(prefix) = self.r#gen.opts.export_prefix.as_ref() {
1043            self.r#gen.c_src.src.push_str(prefix);
1044        }
1045        self.r#gen.c_src.src.push_str(&export_name);
1046        self.r#gen.c_src.src.push_str("(");
1047        let mut first_arg = true;
1048        let mut params = Vec::new();
1049        for (n, ty) in signature.params.iter().enumerate() {
1050            let name = format!("arg{n}");
1051            if !first_arg {
1052                self.r#gen.c_src.src.push_str(", ");
1053            } else {
1054                first_arg = false;
1055            }
1056            self.r#gen.c_src.src.push_str(wit_bindgen_c::wasm_type(*ty));
1057            self.r#gen.c_src.src.push_str(" ");
1058            self.r#gen.c_src.src.push_str(&name);
1059            params.push(name);
1060        }
1061        if return_via_pointer {
1062            if !first_arg {
1063                self.r#gen.c_src.src.push_str(", ");
1064            }
1065            self.r#gen.c_src.src.push_str(self.r#gen.opts.ptr_type());
1066            self.r#gen.c_src.src.push_str(" resultptr");
1067            params.push("resultptr".into());
1068        }
1069        self.r#gen.c_src.src.push_str(")\n");
1070        params
1071    }
1072
1073    fn high_level_signature(
1074        &mut self,
1075        func: &Function,
1076        abi_variant: AbiVariant,
1077        outer_namespace: &[String],
1078    ) -> HighlevelSignature {
1079        let mut res = HighlevelSignature::default();
1080
1081        let (namespace, func_name_h) =
1082            self.func_namespace_name(func, matches!(abi_variant, AbiVariant::GuestExport), false);
1083        res.name = func_name_h;
1084        res.namespace = namespace;
1085        let is_drop = is_special_method(func);
1086        // we might want to separate c_sig and h_sig
1087        // let mut sig = String::new();
1088
1089        // Check if this is a fallible constructor (returns result<T, E>)
1090        let is_fallible_constructor = self.r#gen.is_fallible_constructor(self.resolve, func);
1091
1092        // not for ctor nor imported dtor on guest (except fallible constructors)
1093        if (!matches!(&func.kind, FunctionKind::Constructor(_)) || is_fallible_constructor)
1094            && !(matches!(is_drop, SpecialMethod::ResourceDrop)
1095                && matches!(abi_variant, AbiVariant::GuestImport))
1096        {
1097            if matches!(is_drop, SpecialMethod::Allocate) {
1098                res.result.push_str("Owned");
1099            } else if let Some(ty) = &func.result {
1100                res.result.push_str(
1101                    &(self.type_name(ty, outer_namespace, Flavor::Result(abi_variant))
1102                        + if matches!(is_drop, SpecialMethod::ResourceRep) {
1103                            "*"
1104                        } else {
1105                            ""
1106                        }),
1107                );
1108            } else {
1109                res.result = "void".into();
1110            }
1111            if matches!(abi_variant, AbiVariant::GuestExport)
1112                && abi::guest_export_needs_post_return(self.resolve, func)
1113            {
1114                res.post_return = true;
1115            }
1116        }
1117        if (matches!(func.kind, FunctionKind::Static(_)) || is_fallible_constructor)
1118            && !(matches!(&is_drop, SpecialMethod::ResourceDrop)
1119                && matches!(abi_variant, AbiVariant::GuestImport))
1120        {
1121            res.static_member = true;
1122        }
1123        for (
1124            i,
1125            Param {
1126                name, ty: param, ..
1127            },
1128        ) in func.params.iter().enumerate()
1129        {
1130            if i == 0
1131                && name == "self"
1132                && (matches!(&func.kind, FunctionKind::Method(_))
1133                    || (matches!(&is_drop, SpecialMethod::ResourceDrop)
1134                        && matches!(abi_variant, AbiVariant::GuestImport)))
1135            {
1136                res.implicit_self = true;
1137                continue;
1138            }
1139            let is_pointer = if i == 0
1140                && name == "self"
1141                && matches!(&is_drop, SpecialMethod::Dtor | SpecialMethod::ResourceNew)
1142                && matches!(abi_variant, AbiVariant::GuestExport)
1143            {
1144                "*"
1145            } else {
1146                ""
1147            };
1148            res.arguments.push((
1149                to_c_ident(name),
1150                self.type_name(param, &res.namespace, Flavor::Argument(abi_variant)) + is_pointer,
1151            ));
1152        }
1153        // default to non-const when exporting a method
1154        let import = matches!(abi_variant, AbiVariant::GuestImport);
1155        if matches!(func.kind, FunctionKind::Method(_)) && import {
1156            res.const_member = true;
1157        }
1158        res
1159    }
1160
1161    fn print_signature(
1162        &mut self,
1163        func: &Function,
1164        variant: AbiVariant,
1165        import: bool,
1166    ) -> Vec<String> {
1167        let is_special = is_special_method(func);
1168        let from_namespace = self.r#gen.h_src.namespace.clone();
1169        let cpp_sig = self.high_level_signature(func, variant, &from_namespace);
1170        if cpp_sig.static_member {
1171            self.r#gen.h_src.src.push_str("static ");
1172        }
1173        self.r#gen.h_src.src.push_str(&cpp_sig.result);
1174        if !cpp_sig.result.is_empty() {
1175            self.r#gen.h_src.src.push_str(" ");
1176        }
1177        self.r#gen.h_src.src.push_str(&cpp_sig.name);
1178        self.r#gen.h_src.src.push_str("(");
1179        for (num, (arg, typ)) in cpp_sig.arguments.iter().enumerate() {
1180            if num > 0 {
1181                self.r#gen.h_src.src.push_str(", ");
1182            }
1183            self.r#gen.h_src.src.push_str(typ);
1184            self.r#gen.h_src.src.push_str(" ");
1185            self.r#gen.h_src.src.push_str(arg);
1186        }
1187        self.r#gen.h_src.src.push_str(")");
1188        if cpp_sig.const_member {
1189            self.r#gen.h_src.src.push_str(" const");
1190        }
1191        match (&is_special, false, &variant) {
1192            (SpecialMethod::Allocate, _, _) => {
1193                uwriteln!(
1194                    self.r#gen.h_src.src,
1195                    "{{\
1196                        return {OWNED_CLASS_NAME}(new {}({}));\
1197                    }}",
1198                    cpp_sig.namespace.last().unwrap(), //join("::"),
1199                    cpp_sig
1200                        .arguments
1201                        .iter()
1202                        .map(|(arg, _)| format!("std::move({arg})"))
1203                        .collect::<Vec<_>>()
1204                        .join(", ")
1205                );
1206                // body is inside the header
1207                return Vec::default();
1208            }
1209            (SpecialMethod::Dtor, _, _ /*AbiVariant::GuestImport*/)
1210            | (SpecialMethod::ResourceDrop, true, _) => {
1211                uwriteln!(
1212                    self.r#gen.h_src.src,
1213                    "{{\
1214                        delete {};\
1215                    }}",
1216                    cpp_sig.arguments.first().unwrap().0
1217                );
1218            }
1219            _ => self.r#gen.h_src.src.push_str(";\n"),
1220        }
1221
1222        // we want to separate the lowered signature (wasm) and the high level signature
1223        if !import
1224            && !matches!(
1225                &is_special,
1226                SpecialMethod::ResourceDrop
1227                    | SpecialMethod::ResourceNew
1228                    | SpecialMethod::ResourceRep
1229            )
1230        {
1231            self.print_export_signature(func, variant)
1232        } else {
1233            // recalulate with c file namespace
1234            let c_namespace = self.r#gen.c_src.namespace.clone();
1235            let cpp_sig = self.high_level_signature(func, variant, &c_namespace);
1236            let mut params = Vec::new();
1237            self.r#gen.c_src.src.push_str(&cpp_sig.result);
1238            if !cpp_sig.result.is_empty() {
1239                self.r#gen.c_src.src.push_str(" ");
1240            }
1241            self.r#gen.c_src.qualify(&cpp_sig.namespace);
1242            self.r#gen.c_src.src.push_str(&cpp_sig.name);
1243            self.r#gen.c_src.src.push_str("(");
1244            if cpp_sig.implicit_self {
1245                params.push("(*this)".into());
1246            }
1247            for (num, (arg, typ)) in cpp_sig.arguments.iter().enumerate() {
1248                if num > 0 {
1249                    self.r#gen.c_src.src.push_str(", ");
1250                }
1251                self.r#gen.c_src.src.push_str(typ);
1252                self.r#gen.c_src.src.push_str(" ");
1253                self.r#gen.c_src.src.push_str(arg);
1254                params.push(arg.clone());
1255            }
1256            self.r#gen.c_src.src.push_str(")");
1257            if cpp_sig.const_member {
1258                self.r#gen.c_src.src.push_str(" const");
1259            }
1260            self.r#gen.c_src.src.push_str("\n");
1261            params
1262        }
1263    }
1264
1265    fn generate_function(&mut self, func: &Function, owner: &TypeOwner, variant: AbiVariant) {
1266        fn class_namespace(
1267            cifg: &CppInterfaceGenerator,
1268            func: &Function,
1269            variant: AbiVariant,
1270        ) -> Vec<String> {
1271            let owner = &cifg.resolve.types[match &func.kind {
1272                FunctionKind::Static(id) => *id,
1273                _ => panic!("special func should be static"),
1274            }];
1275            let mut namespace = namespace(
1276                cifg.resolve,
1277                &owner.owner,
1278                matches!(variant, AbiVariant::GuestExport),
1279                &cifg.r#gen.opts,
1280            );
1281            namespace.push(owner.name.as_ref().unwrap().to_upper_camel_case());
1282            namespace
1283        }
1284
1285        let export = match variant {
1286            AbiVariant::GuestImport => false,
1287            AbiVariant::GuestExport => true,
1288            AbiVariant::GuestImportAsync => todo!(),
1289            AbiVariant::GuestExportAsync => todo!(),
1290            AbiVariant::GuestExportAsyncStackful => todo!(),
1291        };
1292        let params = self.print_signature(func, variant, !export);
1293        let special = is_special_method(func);
1294        if !matches!(special, SpecialMethod::Allocate) {
1295            self.r#gen.c_src.src.push_str("{\n");
1296            let needs_dealloc = if self.r#gen.opts.api_style == APIStyle::Symmetric
1297                && matches!(variant, AbiVariant::GuestExport)
1298            {
1299                self.r#gen
1300                    .c_src
1301                    .src
1302                    .push_str("std::vector<void*> _deallocate;\n");
1303                self.r#gen.dependencies.needs_vector = true;
1304                true
1305            } else {
1306                false
1307            };
1308            let lift_lower = if export {
1309                LiftLower::LiftArgsLowerResults
1310            } else {
1311                LiftLower::LowerArgsLiftResults
1312            };
1313            match is_special_method(func) {
1314                SpecialMethod::ResourceDrop => match lift_lower {
1315                    LiftLower::LiftArgsLowerResults => {
1316                        let module_name =
1317                            String::from("[export]") + &self.wasm_import_module.clone().unwrap();
1318                        let wasm_sig =
1319                            self.declare_import(&module_name, &func.name, &[WasmType::I32], &[]);
1320                        uwriteln!(
1321                            self.r#gen.c_src.src,
1322                            "{wasm_sig}({});",
1323                            func.params.first().unwrap().name
1324                        );
1325                    }
1326                    LiftLower::LowerArgsLiftResults => {
1327                        let module_name = self.wasm_import_module.clone().unwrap();
1328                        let name =
1329                            self.declare_import(&module_name, &func.name, &[WasmType::I32], &[]);
1330                        uwriteln!(
1331                            self.r#gen.c_src.src,
1332                            "   if (handle>=0) {{
1333                                {name}(handle);
1334                            }}"
1335                        );
1336                    }
1337                },
1338                SpecialMethod::Dtor => {
1339                    let classname = class_namespace(self, func, variant).join("::");
1340                    uwriteln!(self.r#gen.c_src.src, "(({classname}*)arg0)->handle=-1;");
1341                    uwriteln!(self.r#gen.c_src.src, "{0}::Dtor(({0}*)arg0);", classname);
1342                }
1343                SpecialMethod::ResourceNew => {
1344                    let module_name =
1345                        String::from("[export]") + &self.wasm_import_module.clone().unwrap();
1346                    let wasm_sig = self.declare_import(
1347                        &module_name,
1348                        &func.name,
1349                        &[WasmType::Pointer],
1350                        &[WasmType::I32],
1351                    );
1352                    uwriteln!(
1353                        self.r#gen.c_src.src,
1354                        "return {wasm_sig}(({}){});",
1355                        self.r#gen.opts.ptr_type(),
1356                        func.params.first().unwrap().name
1357                    );
1358                }
1359                SpecialMethod::ResourceRep => {
1360                    let module_name =
1361                        String::from("[export]") + &self.wasm_import_module.clone().unwrap();
1362                    let wasm_sig = self.declare_import(
1363                        &module_name,
1364                        &func.name,
1365                        &[WasmType::I32],
1366                        &[WasmType::Pointer],
1367                    );
1368                    let classname = class_namespace(self, func, variant).join("::");
1369                    uwriteln!(
1370                        self.r#gen.c_src.src,
1371                        "return ({}*){wasm_sig}({});",
1372                        classname,
1373                        func.params.first().unwrap().name
1374                    );
1375                }
1376                SpecialMethod::Allocate => unreachable!(),
1377                SpecialMethod::None => {
1378                    // normal methods
1379                    let namespace = if matches!(func.kind, FunctionKind::Freestanding) {
1380                        namespace(
1381                            self.resolve,
1382                            owner,
1383                            matches!(variant, AbiVariant::GuestExport),
1384                            &self.r#gen.opts,
1385                        )
1386                    } else {
1387                        let owner = &self.resolve.types[match &func.kind {
1388                            FunctionKind::Static(id) => *id,
1389                            FunctionKind::Constructor(id) => *id,
1390                            FunctionKind::Method(id) => *id,
1391                            FunctionKind::Freestanding => unreachable!(),
1392                            FunctionKind::AsyncFreestanding => todo!(),
1393                            FunctionKind::AsyncMethod(_id) => todo!(),
1394                            FunctionKind::AsyncStatic(_id) => todo!(),
1395                        }]
1396                        .clone();
1397                        let mut namespace = namespace(
1398                            self.resolve,
1399                            &owner.owner,
1400                            matches!(variant, AbiVariant::GuestExport),
1401                            &self.r#gen.opts,
1402                        );
1403                        namespace.push(owner.name.as_ref().unwrap().to_upper_camel_case());
1404                        namespace
1405                    };
1406                    let mut f = FunctionBindgen::new(self, params);
1407                    if !export {
1408                        f.namespace = namespace.clone();
1409                    }
1410                    f.variant = variant;
1411                    f.needs_dealloc = needs_dealloc;
1412                    f.cabi_post = None;
1413                    abi::call(f.r#gen.resolve, variant, lift_lower, func, &mut f, false);
1414                    let ret_area_decl = f.emit_ret_area_if_needed();
1415                    let code = format!("{}{}", ret_area_decl, String::from(f.src));
1416                    self.r#gen.c_src.src.push_str(&code);
1417                }
1418            }
1419            self.r#gen.c_src.src.push_str("}\n");
1420            // cabi_post
1421            if matches!(variant, AbiVariant::GuestExport)
1422                && abi::guest_export_needs_post_return(self.resolve, func)
1423            {
1424                let sig = self.resolve.wasm_signature(variant, func);
1425                let module_name = self.wasm_import_module.clone();
1426                let export_name = match module_name {
1427                    Some(ref module_name) => {
1428                        format!("{module_name}#{}", func.name)
1429                    }
1430                    None => make_external_component(&func.name),
1431                };
1432                let import_name = match module_name {
1433                    Some(ref module_name) => {
1434                        make_external_symbol(module_name, &func.name, AbiVariant::GuestExport)
1435                    }
1436                    None => make_external_component(&func.name),
1437                };
1438                uwriteln!(
1439                    self.r#gen.c_src.src,
1440                    "extern \"C\" __attribute__((__weak__, __export_name__(\"cabi_post_{export_name}\")))"
1441                );
1442                uwrite!(self.r#gen.c_src.src, "void cabi_post_{import_name}(");
1443
1444                let mut params = Vec::new();
1445                for (i, result) in sig.results.iter().enumerate() {
1446                    let name = format!("arg{i}");
1447                    uwrite!(
1448                        self.r#gen.c_src.src,
1449                        "{} {name}",
1450                        wit_bindgen_c::wasm_type(*result)
1451                    );
1452                    params.push(name);
1453                }
1454                self.r#gen.c_src.src.push_str(") {\n");
1455
1456                let mut f = FunctionBindgen::new(self, params.clone());
1457                f.params = params;
1458                abi::post_return(f.r#gen.resolve, func, &mut f);
1459                let ret_area_decl = f.emit_ret_area_if_needed();
1460                let code = format!("{}{}", ret_area_decl, String::from(f.src));
1461                self.r#gen.c_src.src.push_str(&code);
1462                self.r#gen.c_src.src.push_str("}\n");
1463            }
1464        }
1465    }
1466
1467    // in C this is print_optional_ty
1468    fn optional_type_name(
1469        &mut self,
1470        ty: Option<&Type>,
1471        from_namespace: &[String],
1472        flavor: Flavor,
1473    ) -> String {
1474        match ty {
1475            Some(ty) => self.type_name(ty, from_namespace, flavor),
1476            None => "void".into(),
1477        }
1478    }
1479
1480    fn scoped_record_name(
1481        &self,
1482        id: TypeId,
1483        from_namespace: &[String],
1484        guest_export: bool,
1485        flavor: Flavor,
1486    ) -> String {
1487        let name = self.scoped_type_name(id, from_namespace, guest_export);
1488
1489        if let Flavor::Argument(AbiVariant::GuestImport) = flavor {
1490            match self.r#gen.opts.ownership {
1491                Ownership::Owning => name.to_string(),
1492                Ownership::CoarseBorrowing => {
1493                    if self.r#gen.types.get(id).has_own_handle {
1494                        name.to_string()
1495                    } else {
1496                        format!("{name}Param")
1497                    }
1498                }
1499                Ownership::FineBorrowing => {
1500                    format!("{name}Param")
1501                }
1502            }
1503        } else {
1504            name
1505        }
1506    }
1507
1508    fn scoped_type_name(
1509        &self,
1510        id: TypeId,
1511        from_namespace: &[String],
1512        guest_export: bool,
1513    ) -> String {
1514        let ty = &self.resolve.types[id];
1515        let namespc = namespace(self.resolve, &ty.owner, guest_export, &self.r#gen.opts);
1516        let mut relative = SourceWithState {
1517            namespace: Vec::from(from_namespace),
1518            ..Default::default()
1519        };
1520        relative.qualify(&namespc);
1521        format!(
1522            "{}{}",
1523            &*relative.src,
1524            ty.name.as_ref().unwrap().to_pascal_case()
1525        )
1526    }
1527
1528    fn type_name(&mut self, ty: &Type, from_namespace: &[String], flavor: Flavor) -> String {
1529        match ty {
1530            Type::Bool => "bool".into(),
1531            Type::Char => "uint32_t".into(),
1532            Type::U8 => "uint8_t".into(),
1533            Type::S8 => "int8_t".into(),
1534            Type::U16 => "uint16_t".into(),
1535            Type::S16 => "int16_t".into(),
1536            Type::U32 => "uint32_t".into(),
1537            Type::S32 => "int32_t".into(),
1538            Type::U64 => "uint64_t".into(),
1539            Type::S64 => "int64_t".into(),
1540            Type::F32 => "float".into(),
1541            Type::F64 => "double".into(),
1542            Type::String => match flavor {
1543                Flavor::BorrowedArgument => {
1544                    self.r#gen.dependencies.needs_string_view = true;
1545                    "std::string_view".into()
1546                }
1547                Flavor::Argument(var)
1548                    if matches!(var, AbiVariant::GuestImport)
1549                        || self.r#gen.opts.api_style == APIStyle::Symmetric =>
1550                {
1551                    self.r#gen.dependencies.needs_string_view = true;
1552                    "std::string_view".into()
1553                }
1554                Flavor::Argument(AbiVariant::GuestExport) => {
1555                    self.r#gen.dependencies.needs_wit = true;
1556                    "wit::string".into()
1557                }
1558                _ => {
1559                    self.r#gen.dependencies.needs_wit = true;
1560                    "wit::string".into()
1561                }
1562            },
1563            Type::Id(id) => match &self.resolve.types[*id].kind {
1564                TypeDefKind::Record(_) => {
1565                    let guest_export = self.is_exported_type(&self.resolve.types[*id]);
1566                    self.scoped_record_name(*id, from_namespace, guest_export, flavor)
1567                }
1568                TypeDefKind::Resource => {
1569                    let guest_export = self.is_exported_type(&self.resolve.types[*id]);
1570                    self.scoped_type_name(*id, from_namespace, guest_export)
1571                }
1572                TypeDefKind::Handle(Handle::Own(id)) => {
1573                    let mut typename = self.type_name(&Type::Id(*id), from_namespace, flavor);
1574                    let ty = &self.resolve.types[*id];
1575
1576                    // Follow type aliases to find the actual resource definition
1577                    // When a resource is `use`d in another interface, we have a type alias
1578                    // with the new interface as owner. We need to follow to the original resource.
1579                    let resource_ty = match &ty.kind {
1580                        TypeDefKind::Type(Type::Id(resource_id)) => {
1581                            &self.resolve.types[*resource_id]
1582                        }
1583                        _ => ty,
1584                    };
1585
1586                    let is_exported = self.is_exported_type(resource_ty);
1587                    match (false, flavor) {
1588                        (false, Flavor::Argument(AbiVariant::GuestImport))
1589                        | (true, Flavor::Argument(AbiVariant::GuestExport)) => {
1590                            typename.push_str("&&")
1591                        }
1592                        (false, Flavor::Argument(AbiVariant::GuestExport))
1593                        | (false, Flavor::Result(AbiVariant::GuestExport))
1594                        | (true, Flavor::Argument(AbiVariant::GuestImport))
1595                        | (true, Flavor::Result(AbiVariant::GuestImport)) => {
1596                            // Only exported resources have ::Owned typedef
1597                            if is_exported {
1598                                typename.push_str(&format!("::{OWNED_CLASS_NAME}"))
1599                            } else {
1600                                typename.push_str("&&")
1601                            }
1602                        }
1603                        (false, Flavor::Result(AbiVariant::GuestImport))
1604                        | (true, Flavor::Result(AbiVariant::GuestExport)) => (),
1605                        (_, Flavor::InStruct) => (),
1606                        (false, Flavor::BorrowedArgument) => (),
1607                        (_, _) => todo!(),
1608                    }
1609                    if matches!(flavor, Flavor::InStruct) && is_exported {
1610                        typename.push_str(&format!("::{OWNED_CLASS_NAME}"))
1611                    }
1612                    typename
1613                }
1614                TypeDefKind::Handle(Handle::Borrow(id)) => {
1615                    "std::reference_wrapper<const ".to_string()
1616                        + &self.type_name(&Type::Id(*id), from_namespace, flavor)
1617                        + ">"
1618                }
1619                TypeDefKind::Flags(_f) => {
1620                    let ty = &self.resolve.types[*id];
1621                    let guest_export = self.is_exported_type(ty);
1622                    self.scoped_type_name(*id, from_namespace, guest_export)
1623                }
1624                TypeDefKind::Tuple(t) => {
1625                    let types = t.types.iter().fold(String::new(), |mut a, b| {
1626                        if !a.is_empty() {
1627                            a += ", ";
1628                        }
1629                        a + &self.type_name(b, from_namespace, flavor)
1630                    });
1631                    self.r#gen.dependencies.needs_tuple = true;
1632                    String::from("std::tuple<") + &types + ">"
1633                }
1634                TypeDefKind::Variant(_v) => {
1635                    let ty = &self.resolve.types[*id];
1636                    let guest_export = self.is_exported_type(ty);
1637                    self.scoped_type_name(*id, from_namespace, guest_export)
1638                }
1639                TypeDefKind::Enum(_e) => {
1640                    let ty = &self.resolve.types[*id];
1641                    let guest_export = self.is_exported_type(ty);
1642                    self.scoped_type_name(*id, from_namespace, guest_export)
1643                }
1644                TypeDefKind::Option(o) => {
1645                    // Template parameters need base types without && or other decorations
1646                    // For import arguments, use BorrowedArgument flavor to get string_view
1647                    let template_flavor = match flavor {
1648                        Flavor::Argument(AbiVariant::GuestImport) => Flavor::BorrowedArgument,
1649                        _ => Flavor::InStruct,
1650                    };
1651                    self.r#gen.dependencies.needs_optional = true;
1652                    "std::optional<".to_string()
1653                        + &self.type_name(o, from_namespace, template_flavor)
1654                        + ">"
1655                }
1656                TypeDefKind::Result(r) => {
1657                    // Template parameters need base types without && or other decorations
1658                    let template_flavor = Flavor::InStruct;
1659                    let err_type = r.err.as_ref().map_or(String::from("wit::Void"), |ty| {
1660                        self.type_name(ty, from_namespace, template_flavor)
1661                    });
1662                    self.r#gen.dependencies.needs_expected = true;
1663                    "std::expected<".to_string()
1664                        + &self.optional_type_name(r.ok.as_ref(), from_namespace, template_flavor)
1665                        + ", "
1666                        + &err_type
1667                        + ">"
1668                }
1669                TypeDefKind::List(ty) => {
1670                    // For list elements, use BorrowedArgument flavor for imported functions
1671                    // to get std::string_view instead of wit::string. Otherwise use InStruct
1672                    // flavor to avoid adding && to owned resources (lists contain values, not rvalue references)
1673                    let element_flavor = match flavor {
1674                        Flavor::BorrowedArgument | Flavor::Argument(AbiVariant::GuestImport) => {
1675                            Flavor::BorrowedArgument
1676                        }
1677                        _ => Flavor::InStruct,
1678                    };
1679                    let inner = self.type_name(ty, from_namespace, element_flavor);
1680                    match flavor {
1681                        Flavor::BorrowedArgument => {
1682                            self.r#gen.dependencies.needs_span = true;
1683                            format!("std::span<{inner} const>")
1684                        }
1685                        Flavor::Argument(var)
1686                            if matches!(var, AbiVariant::GuestImport)
1687                                || self.r#gen.opts.api_style == APIStyle::Symmetric =>
1688                        {
1689                            self.r#gen.dependencies.needs_span = true;
1690                            // If the list has an owning handle, it must support moving, so can't be const
1691                            let constness = if self.r#gen.types.get(*id).has_own_handle {
1692                                ""
1693                            } else {
1694                                " const"
1695                            };
1696                            format!("std::span<{inner}{constness}>")
1697                        }
1698                        Flavor::Argument(AbiVariant::GuestExport) => {
1699                            self.r#gen.dependencies.needs_wit = true;
1700                            format!("wit::vector<{inner}>")
1701                        }
1702                        _ => {
1703                            self.r#gen.dependencies.needs_wit = true;
1704                            format!("wit::vector<{inner}>")
1705                        }
1706                    }
1707                }
1708                TypeDefKind::Future(_) => todo!(),
1709                TypeDefKind::Stream(_) => todo!(),
1710                TypeDefKind::Type(ty) => self.type_name(ty, from_namespace, flavor),
1711                TypeDefKind::FixedLengthList(ty, size) => {
1712                    self.r#gen.dependencies.needs_array = true;
1713                    format!(
1714                        "std::array<{}, {size}>",
1715                        self.type_name(ty, from_namespace, flavor)
1716                    )
1717                }
1718                TypeDefKind::Map(_, _) => todo!(),
1719                TypeDefKind::Unknown => todo!(),
1720            },
1721            Type::ErrorContext => todo!(),
1722        }
1723    }
1724
1725    fn declare_import2(
1726        &self,
1727        module_name: &str,
1728        name: &str,
1729        args: &str,
1730        result: &str,
1731        variant: AbiVariant,
1732    ) -> (String, String) {
1733        let mut extern_name = String::from("__wasm_import_");
1734        extern_name.push_str(&make_external_symbol(module_name, name, variant));
1735        let import = format!(
1736            "extern \"C\" __attribute__((import_module(\"{module_name}\")))\n __attribute__((import_name(\"{name}\")))\n {result} {extern_name}({args});\n"
1737        );
1738        (extern_name, import)
1739    }
1740
1741    fn declare_import(
1742        &mut self,
1743        module_name: &str,
1744        name: &str,
1745        params: &[WasmType],
1746        results: &[WasmType],
1747    ) -> String {
1748        let mut args = String::default();
1749        for (n, param) in params.iter().enumerate() {
1750            args.push_str(wit_bindgen_c::wasm_type(*param));
1751            if n + 1 != params.len() {
1752                args.push_str(", ");
1753            }
1754        }
1755        let result = if results.is_empty() {
1756            "void"
1757        } else {
1758            wit_bindgen_c::wasm_type(results[0])
1759        };
1760        let variant = AbiVariant::GuestImport;
1761        let (name, code) = self.declare_import2(module_name, name, &args, result, variant);
1762        self.r#gen.extern_c_decls.push_str(&code);
1763        name
1764    }
1765
1766    fn docs(src: &mut Source, docs: &Docs) {
1767        if let Some(docs) = docs.contents.as_ref() {
1768            for line in docs.trim().lines() {
1769                src.push_str("/// ");
1770                src.push_str(line);
1771                src.push_str("\n");
1772            }
1773        }
1774    }
1775
1776    fn type_record_param(
1777        &mut self,
1778        id: TypeId,
1779        name: &str,
1780        record: &wit_bindgen_core::wit_parser::Record,
1781        namespc: &[String],
1782    ) {
1783        let (flavor, needs_param_type) = {
1784            match self.r#gen.opts.ownership {
1785                Ownership::Owning => (Flavor::InStruct, false),
1786                Ownership::CoarseBorrowing => {
1787                    if self.r#gen.types.get(id).has_own_handle {
1788                        (Flavor::InStruct, false)
1789                    } else {
1790                        (Flavor::BorrowedArgument, true)
1791                    }
1792                }
1793                Ownership::FineBorrowing => (Flavor::BorrowedArgument, true),
1794            }
1795        };
1796
1797        if needs_param_type {
1798            let pascal = format!("{name}-param").to_pascal_case();
1799
1800            uwriteln!(self.r#gen.h_src.src, "struct {pascal} {{");
1801            for field in record.fields.iter() {
1802                let typename = self.type_name(&field.ty, namespc, flavor);
1803                let fname = to_c_ident(&field.name);
1804                uwriteln!(self.r#gen.h_src.src, "{typename} {fname};");
1805            }
1806            uwriteln!(self.r#gen.h_src.src, "}};");
1807        }
1808    }
1809
1810    fn is_exported_type(&self, ty: &TypeDef) -> bool {
1811        match ty.owner {
1812            TypeOwner::Interface(intf) => {
1813                // For resources used in export functions, check if the resource's owner
1814                // interface is in imported_interfaces (which was populated during import())
1815                !self.r#gen.imported_interfaces.contains(&intf)
1816            }
1817            TypeOwner::World(_) => {
1818                // World-level resources are treated as imports, not exports
1819                false
1820            }
1821            TypeOwner::None => true,
1822        }
1823    }
1824}
1825
1826impl<'a> wit_bindgen_core::InterfaceGenerator<'a> for CppInterfaceGenerator<'a> {
1827    fn resolve(&self) -> &'a Resolve {
1828        self.resolve
1829    }
1830
1831    fn type_record(
1832        &mut self,
1833        id: TypeId,
1834        name: &str,
1835        record: &wit_bindgen_core::wit_parser::Record,
1836        docs: &wit_bindgen_core::wit_parser::Docs,
1837    ) {
1838        let ty = &self.resolve.types[id];
1839        let guest_export = self.is_exported_type(ty);
1840        let namespc = namespace(self.resolve, &ty.owner, guest_export, &self.r#gen.opts);
1841
1842        if self.r#gen.is_first_definition(&namespc, name) {
1843            self.r#gen.h_src.change_namespace(&namespc);
1844            Self::docs(&mut self.r#gen.h_src.src, docs);
1845            let pascal = name.to_pascal_case();
1846
1847            uwriteln!(self.r#gen.h_src.src, "struct {pascal} {{");
1848            for field in record.fields.iter() {
1849                Self::docs(&mut self.r#gen.h_src.src, &field.docs);
1850                let typename = self.type_name(&field.ty, &namespc, Flavor::InStruct);
1851                let fname = to_c_ident(&field.name);
1852                uwriteln!(self.r#gen.h_src.src, "{typename} {fname};");
1853            }
1854            uwriteln!(self.r#gen.h_src.src, "}};");
1855            self.type_record_param(id, name, record, namespc.as_slice());
1856        }
1857    }
1858
1859    fn type_resource(
1860        &mut self,
1861        id: TypeId,
1862        name: &str,
1863        _docs: &wit_bindgen_core::wit_parser::Docs,
1864    ) {
1865        let type_ = &self.resolve.types[id];
1866        if let TypeOwner::Interface(intf) = type_.owner {
1867            let guest_import = self.r#gen.imported_interfaces.contains(&intf);
1868            let definition = !(guest_import);
1869            let store = self.r#gen.start_new_file(Some(definition));
1870            let mut world_name = to_c_ident(&self.r#gen.world);
1871            world_name.push_str("::");
1872            let namespc = namespace(self.resolve, &type_.owner, !guest_import, &self.r#gen.opts);
1873            let pascal = name.to_upper_camel_case();
1874            let mut user_filename = namespc.clone();
1875            user_filename.push(pascal.clone());
1876            if definition {
1877                uwriteln!(
1878                    self.r#gen.h_src.src,
1879                    r#"/* User class definition file, autogenerated once, then user modified
1880                    * Updated versions of this file are generated into {pascal}.template.
1881                    */"#
1882                );
1883            }
1884            self.r#gen.h_src.change_namespace(&namespc);
1885
1886            if !definition {
1887                self.r#gen.dependencies.needs_imported_resources = true;
1888            } else {
1889                self.r#gen.dependencies.needs_exported_resources = true;
1890            }
1891            self.r#gen.dependencies.needs_wit = true;
1892
1893            let base_type = match (definition, false) {
1894                (true, false) => format!("wit::{RESOURCE_EXPORT_BASE_CLASS_NAME}<{pascal}>"),
1895                (false, false) => {
1896                    String::from_str("wit::").unwrap() + RESOURCE_IMPORT_BASE_CLASS_NAME
1897                }
1898                (false, true) => {
1899                    String::from_str("wit::").unwrap() + RESOURCE_EXPORT_BASE_CLASS_NAME
1900                }
1901                (true, true) => format!("wit::{RESOURCE_IMPORT_BASE_CLASS_NAME}<{pascal}>"),
1902            };
1903            let derive = format!(" : public {base_type}");
1904            uwriteln!(self.r#gen.h_src.src, "class {pascal}{derive} {{\n");
1905            uwriteln!(self.r#gen.h_src.src, "public:\n");
1906            let variant = if guest_import {
1907                AbiVariant::GuestImport
1908            } else {
1909                AbiVariant::GuestExport
1910            };
1911            {
1912                // destructor
1913                let name = match variant {
1914                    AbiVariant::GuestImport => "[resource-drop]",
1915                    AbiVariant::GuestExport => "[dtor]",
1916                    AbiVariant::GuestImportAsync => todo!(),
1917                    AbiVariant::GuestExportAsync => todo!(),
1918                    AbiVariant::GuestExportAsyncStackful => todo!(),
1919                }
1920                .to_string()
1921                    + name;
1922                let func = Function {
1923                    name,
1924                    kind: FunctionKind::Static(id),
1925                    params: vec![Param {
1926                        name: "self".into(),
1927                        ty: Type::Id(id),
1928                        span: Default::default(),
1929                    }],
1930                    result: None,
1931                    docs: Docs::default(),
1932                    stability: Stability::Unknown,
1933                    span: Default::default(),
1934                };
1935                self.generate_function(&func, &TypeOwner::Interface(intf), variant);
1936            }
1937            let funcs = self.resolve.interfaces[intf].functions.values();
1938            for func in funcs {
1939                if match &func.kind {
1940                    FunctionKind::Freestanding => false,
1941                    FunctionKind::Method(mid) => *mid == id,
1942                    FunctionKind::Static(mid) => *mid == id,
1943                    FunctionKind::Constructor(mid) => *mid == id,
1944                    FunctionKind::AsyncFreestanding => todo!(),
1945                    FunctionKind::AsyncMethod(_id) => todo!(),
1946                    FunctionKind::AsyncStatic(_id) => todo!(),
1947                } {
1948                    self.generate_function(func, &TypeOwner::Interface(intf), variant);
1949                    // For non-fallible constructors on export side, generate a New allocator method
1950                    // For fallible constructors, the user provides their own Create method
1951                    let is_fallible_constructor =
1952                        self.r#gen.is_fallible_constructor(self.resolve, func);
1953
1954                    if matches!(func.kind, FunctionKind::Constructor(_))
1955                        && matches!(variant, AbiVariant::GuestExport)
1956                        && !is_fallible_constructor
1957                    {
1958                        // functional safety requires the option to use a different allocator, so move new into the implementation
1959                        let func2 = Function {
1960                            name: "$alloc".to_string(),
1961                            kind: FunctionKind::Static(id),
1962                            // same params as constructor
1963                            params: func.params.clone(),
1964                            result: Some(Type::Id(id)),
1965                            docs: Docs::default(),
1966                            stability: Stability::Unknown,
1967                            span: Default::default(),
1968                        };
1969                        self.generate_function(&func2, &TypeOwner::Interface(intf), variant);
1970                    }
1971                }
1972            }
1973
1974            if !definition {
1975                // consuming constructor from handle (bindings)
1976                uwriteln!(self.r#gen.h_src.src, "{pascal}({base_type} &&);",);
1977                uwriteln!(self.r#gen.h_src.src, "{pascal}({pascal}&&) = default;");
1978                uwriteln!(
1979                    self.r#gen.h_src.src,
1980                    "{pascal}& operator=({pascal}&&) = default;"
1981                );
1982                self.r#gen.c_src.qualify(&namespc);
1983                uwriteln!(
1984                    self.r#gen.c_src.src,
1985                    "{pascal}::{pascal}({base_type}&&b) : {base_type}(std::move(b)) {{}}"
1986                );
1987            }
1988            if matches!(variant, AbiVariant::GuestExport) {
1989                let id_type = Type::S32;
1990                let func = Function {
1991                    name: "[resource-new]".to_string() + name,
1992                    kind: FunctionKind::Static(id),
1993                    params: vec![Param {
1994                        name: "self".into(),
1995                        ty: Type::Id(id),
1996                        span: Default::default(),
1997                    }],
1998                    result: Some(id_type),
1999                    docs: Docs::default(),
2000                    stability: Stability::Unknown,
2001                    span: Default::default(),
2002                };
2003                self.generate_function(&func, &TypeOwner::Interface(intf), variant);
2004
2005                let func1 = Function {
2006                    name: "[resource-rep]".to_string() + name,
2007                    kind: FunctionKind::Static(id),
2008                    params: vec![Param {
2009                        name: "id".into(),
2010                        ty: id_type,
2011                        span: Default::default(),
2012                    }],
2013                    result: Some(Type::Id(id)),
2014                    docs: Docs::default(),
2015                    stability: Stability::Unknown,
2016                    span: Default::default(),
2017                };
2018                self.generate_function(&func1, &TypeOwner::Interface(intf), variant);
2019
2020                let func2 = Function {
2021                    name: "[resource-drop]".to_string() + name,
2022                    kind: FunctionKind::Static(id),
2023                    params: vec![Param {
2024                        name: "id".into(),
2025                        ty: id_type,
2026                        span: Default::default(),
2027                    }],
2028                    result: None,
2029                    docs: Docs::default(),
2030                    stability: Stability::Unknown,
2031                    span: Default::default(),
2032                };
2033                self.generate_function(&func2, &TypeOwner::Interface(intf), variant);
2034            }
2035            uwriteln!(self.r#gen.h_src.src, "}};\n");
2036            self.r#gen.finish_file(&user_filename, store);
2037        } else if matches!(type_.owner, TypeOwner::World(_)) {
2038            // Handle world-level resources - treat as imported resources
2039            let guest_export = false; // World-level resources are treated as imports
2040            let namespc = namespace(self.resolve, &type_.owner, guest_export, &self.r#gen.opts);
2041            self.r#gen.h_src.change_namespace(&namespc);
2042
2043            let pascal = name.to_upper_camel_case();
2044            self.r#gen.dependencies.needs_imported_resources = true;
2045            self.r#gen.dependencies.needs_wit = true;
2046
2047            let base_type = format!("wit::{RESOURCE_IMPORT_BASE_CLASS_NAME}");
2048            let derive = format!(" : public {base_type}");
2049            uwriteln!(self.r#gen.h_src.src, "class {pascal}{derive}{{\n");
2050            uwriteln!(self.r#gen.h_src.src, "public:\n");
2051
2052            // Add destructor and constructor
2053            uwriteln!(self.r#gen.h_src.src, "~{pascal}();");
2054            uwriteln!(
2055                self.r#gen.h_src.src,
2056                "{pascal}(wit::{RESOURCE_IMPORT_BASE_CLASS_NAME} &&);"
2057            );
2058            uwriteln!(self.r#gen.h_src.src, "{pascal}({pascal}&&) = default;");
2059            uwriteln!(
2060                self.r#gen.h_src.src,
2061                "{pascal}& operator=({pascal}&&) = default;"
2062            );
2063            uwriteln!(self.r#gen.h_src.src, "}};\n");
2064        }
2065    }
2066
2067    fn type_flags(
2068        &mut self,
2069        id: TypeId,
2070        name: &str,
2071        flags: &wit_bindgen_core::wit_parser::Flags,
2072        docs: &wit_bindgen_core::wit_parser::Docs,
2073    ) {
2074        let ty = &self.resolve.types[id];
2075        let guest_export = self.is_exported_type(ty);
2076        let namespc = namespace(self.resolve, &ty.owner, guest_export, &self.r#gen.opts);
2077        if self.r#gen.is_first_definition(&namespc, name) {
2078            self.r#gen.h_src.change_namespace(&namespc);
2079            Self::docs(&mut self.r#gen.h_src.src, docs);
2080            let pascal = name.to_pascal_case();
2081            let int_repr = wit_bindgen_c::int_repr(wit_bindgen_c::flags_repr(flags));
2082            uwriteln!(self.r#gen.h_src.src, "enum class {pascal} : {int_repr} {{");
2083            uwriteln!(self.r#gen.h_src.src, "k_None = 0,");
2084            for (n, field) in flags.flags.iter().enumerate() {
2085                Self::docs(&mut self.r#gen.h_src.src, &field.docs);
2086                let fname = to_c_ident(&field.name).to_pascal_case();
2087                uwriteln!(self.r#gen.h_src.src, "k{fname} = (1ULL<<{n}),");
2088            }
2089            uwriteln!(self.r#gen.h_src.src, "}};");
2090            uwriteln!(
2091                self.r#gen.h_src.src,
2092                r#"static inline {pascal} operator|({pascal} a, {pascal} b) {{ return {pascal}({int_repr}(a)|{int_repr}(b)); }}
2093        static inline {pascal} operator&({pascal} a, {pascal} b) {{ return {pascal}({int_repr}(a)&{int_repr}(b)); }}"#
2094            );
2095        }
2096    }
2097
2098    fn type_tuple(
2099        &mut self,
2100        _id: TypeId,
2101        _name: &str,
2102        _flags: &wit_bindgen_core::wit_parser::Tuple,
2103        _docs: &wit_bindgen_core::wit_parser::Docs,
2104    ) {
2105        // I assume I don't need to do anything ...
2106    }
2107
2108    fn type_variant(
2109        &mut self,
2110        id: TypeId,
2111        name: &str,
2112        variant: &wit_bindgen_core::wit_parser::Variant,
2113        docs: &wit_bindgen_core::wit_parser::Docs,
2114    ) {
2115        let ty = &self.resolve.types[id];
2116        let guest_export = self.is_exported_type(ty);
2117        let namespc = namespace(self.resolve, &ty.owner, guest_export, &self.r#gen.opts);
2118        if self.r#gen.is_first_definition(&namespc, name) {
2119            self.r#gen.h_src.change_namespace(&namespc);
2120            Self::docs(&mut self.r#gen.h_src.src, docs);
2121            let pascal = name.to_pascal_case();
2122            uwriteln!(self.r#gen.h_src.src, "struct {pascal} {{");
2123            let mut inner_namespace = namespc.clone();
2124            inner_namespace.push(pascal.clone());
2125            let mut all_types = String::new();
2126            for case in variant.cases.iter() {
2127                Self::docs(&mut self.r#gen.h_src.src, &case.docs);
2128                let case_pascal = to_c_ident(&case.name).to_pascal_case();
2129                if !all_types.is_empty() {
2130                    all_types += ", ";
2131                }
2132                all_types += &case_pascal;
2133                uwrite!(self.r#gen.h_src.src, "struct {case_pascal} {{");
2134                if let Some(ty) = case.ty.as_ref() {
2135                    let typestr = self.type_name(ty, &inner_namespace, Flavor::InStruct);
2136                    uwrite!(self.r#gen.h_src.src, " {typestr} value; ")
2137                }
2138                uwriteln!(self.r#gen.h_src.src, "}};");
2139            }
2140            uwriteln!(
2141                self.r#gen.h_src.src,
2142                "  std::variant<{all_types}> variants;"
2143            );
2144            uwriteln!(self.r#gen.h_src.src, "}};");
2145            self.r#gen.dependencies.needs_variant = true;
2146        }
2147    }
2148
2149    fn type_option(
2150        &mut self,
2151        _id: TypeId,
2152        _name: &str,
2153        _payload: &wit_bindgen_core::wit_parser::Type,
2154        _docs: &wit_bindgen_core::wit_parser::Docs,
2155    ) {
2156        // nothing to do here
2157    }
2158
2159    fn type_result(
2160        &mut self,
2161        _id: TypeId,
2162        _name: &str,
2163        _result: &wit_bindgen_core::wit_parser::Result_,
2164        _docs: &wit_bindgen_core::wit_parser::Docs,
2165    ) {
2166        // nothing to do here
2167    }
2168
2169    fn type_enum(
2170        &mut self,
2171        id: TypeId,
2172        name: &str,
2173        enum_: &wit_bindgen_core::wit_parser::Enum,
2174        docs: &wit_bindgen_core::wit_parser::Docs,
2175    ) {
2176        let ty = &self.resolve.types[id];
2177        let guest_export = self.is_exported_type(ty);
2178        let namespc = namespace(self.resolve, &ty.owner, guest_export, &self.r#gen.opts);
2179        if self.r#gen.is_first_definition(&namespc, name) {
2180            self.r#gen.h_src.change_namespace(&namespc);
2181            let pascal = name.to_pascal_case();
2182            Self::docs(&mut self.r#gen.h_src.src, docs);
2183            let int_t = wit_bindgen_c::int_repr(enum_.tag());
2184            uwriteln!(self.r#gen.h_src.src, "enum class {pascal} : {int_t} {{");
2185            for (i, case) in enum_.cases.iter().enumerate() {
2186                Self::docs(&mut self.r#gen.h_src.src, &case.docs);
2187                uwriteln!(
2188                    self.r#gen.h_src.src,
2189                    " k{} = {i},",
2190                    to_c_ident(&case.name).to_pascal_case(),
2191                );
2192            }
2193            uwriteln!(self.r#gen.h_src.src, "}};\n");
2194        }
2195    }
2196
2197    fn type_alias(
2198        &mut self,
2199        id: TypeId,
2200        name: &str,
2201        alias_type: &wit_bindgen_core::wit_parser::Type,
2202        docs: &wit_bindgen_core::wit_parser::Docs,
2203    ) {
2204        let ty = &self.resolve.types[id];
2205        let guest_export = self.is_exported_type(ty);
2206        let namespc = namespace(self.resolve, &ty.owner, guest_export, &self.r#gen.opts);
2207        self.r#gen.h_src.change_namespace(&namespc);
2208        let pascal = name.to_pascal_case();
2209        Self::docs(&mut self.r#gen.h_src.src, docs);
2210        let typename = self.type_name(alias_type, &namespc, Flavor::InStruct);
2211        uwriteln!(self.r#gen.h_src.src, "using {pascal} = {typename};");
2212    }
2213
2214    fn type_list(
2215        &mut self,
2216        _id: TypeId,
2217        _name: &str,
2218        _ty: &wit_bindgen_core::wit_parser::Type,
2219        _docs: &wit_bindgen_core::wit_parser::Docs,
2220    ) {
2221        // nothing to do here
2222    }
2223
2224    fn type_builtin(
2225        &mut self,
2226        _id: TypeId,
2227        _name: &str,
2228        _ty: &wit_bindgen_core::wit_parser::Type,
2229        _docs: &wit_bindgen_core::wit_parser::Docs,
2230    ) {
2231        todo!()
2232    }
2233
2234    fn type_future(&mut self, _id: TypeId, _name: &str, _ty: &Option<Type>, _docs: &Docs) {
2235        todo!()
2236    }
2237
2238    fn type_stream(&mut self, _id: TypeId, _name: &str, _ty: &Option<Type>, _docs: &Docs) {
2239        todo!()
2240    }
2241}
2242
2243struct CabiPostInformation {
2244    module: String,
2245    name: String,
2246    ret_type: String,
2247}
2248
2249struct FunctionBindgen<'a, 'b> {
2250    r#gen: &'b mut CppInterfaceGenerator<'a>,
2251    params: Vec<String>,
2252    tmp: usize,
2253    namespace: Vec<String>,
2254    src: Source,
2255    block_storage: Vec<wit_bindgen_core::Source>,
2256    /// intermediate calculations for contained objects
2257    blocks: Vec<(String, Vec<String>)>,
2258    payloads: Vec<String>,
2259    // caching for wasm
2260    variant: AbiVariant,
2261    cabi_post: Option<CabiPostInformation>,
2262    needs_dealloc: bool,
2263    leak_on_insertion: Option<String>,
2264    return_pointer_area_size: ArchitectureSize,
2265    return_pointer_area_align: Alignment,
2266}
2267
2268impl<'a, 'b> FunctionBindgen<'a, 'b> {
2269    fn new(r#gen: &'b mut CppInterfaceGenerator<'a>, params: Vec<String>) -> Self {
2270        Self {
2271            r#gen,
2272            params,
2273            tmp: 0,
2274            namespace: Default::default(),
2275            src: Default::default(),
2276            block_storage: Default::default(),
2277            blocks: Default::default(),
2278            payloads: Default::default(),
2279            variant: AbiVariant::GuestImport,
2280            cabi_post: None,
2281            needs_dealloc: false,
2282            leak_on_insertion: None,
2283            return_pointer_area_size: Default::default(),
2284            return_pointer_area_align: Default::default(),
2285        }
2286    }
2287
2288    fn tmp(&mut self) -> usize {
2289        let ret = self.tmp;
2290        self.tmp += 1;
2291        ret
2292    }
2293
2294    fn tempname(&self, base: &str, idx: usize) -> String {
2295        format!("{base}{idx}")
2296    }
2297
2298    fn push_str(&mut self, s: &str) {
2299        self.src.push_str(s);
2300    }
2301
2302    fn let_results(&mut self, amt: usize, results: &mut Vec<String>) {
2303        if amt > 0 {
2304            let tmp = self.tmp();
2305            let res = format!("result{tmp}");
2306            self.push_str("auto ");
2307            self.push_str(&res);
2308            self.push_str(" = ");
2309            if amt == 1 {
2310                results.push(res);
2311            } else {
2312                for i in 0..amt {
2313                    results.push(format!("std::get<{i}>({res})"));
2314                }
2315            }
2316        }
2317    }
2318
2319    fn load(
2320        &mut self,
2321        ty: &str,
2322        offset: ArchitectureSize,
2323        operands: &[String],
2324        results: &mut Vec<String>,
2325    ) {
2326        results.push(format!(
2327            "*(({}*) ({} + {}))",
2328            ty,
2329            operands[0],
2330            offset.format(POINTER_SIZE_EXPRESSION)
2331        ));
2332    }
2333
2334    fn load_ext(
2335        &mut self,
2336        ty: &str,
2337        offset: ArchitectureSize,
2338        operands: &[String],
2339        results: &mut Vec<String>,
2340    ) {
2341        self.load(ty, offset, operands, results);
2342        let result = results.pop().unwrap();
2343        results.push(format!("(int32_t) ({result})"));
2344    }
2345
2346    fn store(&mut self, ty: &str, offset: ArchitectureSize, operands: &[String]) {
2347        uwriteln!(
2348            self.src,
2349            "*(({}*)({} + {})) = {};",
2350            ty,
2351            operands[1],
2352            offset.format(POINTER_SIZE_EXPRESSION),
2353            operands[0]
2354        );
2355    }
2356
2357    /// Emits a shared return area declaration if needed by this function.
2358    ///
2359    /// During code generation, `return_pointer()` may be called multiple times for:
2360    /// - Indirect parameter storage (when too many/large params)
2361    /// - Return value storage (when return type is too large)
2362    ///
2363    /// **Safety:** This is safe because return pointers are used sequentially:
2364    /// 1. Parameter marshaling (before call)
2365    /// 2. Function execution
2366    /// 3. Return value unmarshaling (after call)
2367    ///
2368    /// The scratch space is reused but never accessed simultaneously.
2369    fn emit_ret_area_if_needed(&self) -> String {
2370        if !self.return_pointer_area_size.is_empty() {
2371            let size_string = self
2372                .return_pointer_area_size
2373                .format(POINTER_SIZE_EXPRESSION);
2374            let tp = match self.return_pointer_area_align {
2375                Alignment::Bytes(bytes) => match bytes.get() {
2376                    1 => "uint8_t",
2377                    2 => "uint16_t",
2378                    4 => "uint32_t",
2379                    8 => "uint64_t",
2380                    // Fallback to uint8_t for unusual alignments (e.g., 16-byte SIMD).
2381                    // This is safe: the size calculation ensures correct buffer size,
2382                    // and uint8_t arrays can store any data regardless of alignment.
2383                    _ => "uint8_t",
2384                },
2385                Alignment::Pointer => "uintptr_t",
2386            };
2387            let static_var = if self.r#gen.in_guest_import {
2388                ""
2389            } else {
2390                "static "
2391            };
2392            format!("{static_var}{tp} ret_area[({size_string}+sizeof({tp})-1)/sizeof({tp})];\n")
2393        } else {
2394            String::new()
2395        }
2396    }
2397}
2398
2399fn move_if_necessary(arg: &str) -> String {
2400    // if it is a name of a variable move it
2401    if !arg.is_empty() && arg.chars().all(char::is_alphanumeric) {
2402        format!("std::move({arg})")
2403    } else {
2404        arg.into()
2405    }
2406}
2407
2408impl<'a, 'b> Bindgen for FunctionBindgen<'a, 'b> {
2409    type Operand = String;
2410
2411    fn emit(
2412        &mut self,
2413        _resolve: &Resolve,
2414        inst: &wit_bindgen_core::abi::Instruction<'_>,
2415        operands: &mut Vec<Self::Operand>,
2416        results: &mut Vec<Self::Operand>,
2417    ) {
2418        let mut top_as = |cvt: &str| {
2419            results.push(format!("({cvt}({}))", operands.pop().unwrap()));
2420        };
2421
2422        match inst {
2423            abi::Instruction::GetArg { nth } => {
2424                if *nth == 0 && self.params[0].as_str() == "self" {
2425                    if self.r#gen.in_guest_import {
2426                        results.push("(*this)".to_string());
2427                    } else {
2428                        results.push("(*lookup_resource(self))".to_string());
2429                    }
2430                } else {
2431                    results.push(self.params[*nth].clone());
2432                }
2433            }
2434            abi::Instruction::I32Const { val } => results.push(format!("(int32_t({val}))")),
2435            abi::Instruction::Bitcasts { casts } => {
2436                for (cast, op) in casts.iter().zip(operands) {
2437                    // let op = op;
2438                    results.push(self.r#gen.r#gen.perform_cast(op, cast));
2439                }
2440            }
2441            abi::Instruction::ConstZero { tys } => {
2442                for ty in tys.iter() {
2443                    match ty {
2444                        WasmType::I32 => results.push("int32_t(0)".to_string()),
2445                        WasmType::I64 => results.push("int64_t(0)".to_string()),
2446                        WasmType::F32 => results.push("0.0f".to_string()),
2447                        WasmType::F64 => results.push("0.0".to_string()),
2448                        WasmType::Length => results.push("size_t(0)".to_string()),
2449                        WasmType::Pointer => results.push("nullptr".to_string()),
2450                        WasmType::PointerOrI64 => results.push("int64_t(0)".to_string()),
2451                    }
2452                }
2453            }
2454            abi::Instruction::I32Load { offset } => {
2455                let tmp = self.tmp();
2456                uwriteln!(
2457                    self.src,
2458                    "int32_t l{tmp} = *((int32_t const*)({} + {offset}));",
2459                    operands[0],
2460                    offset = offset.format(POINTER_SIZE_EXPRESSION)
2461                );
2462                results.push(format!("l{tmp}"));
2463            }
2464            abi::Instruction::I32Load8U { offset } => {
2465                self.load_ext("uint8_t", *offset, operands, results)
2466            }
2467            abi::Instruction::I32Load8S { offset } => {
2468                self.load_ext("int8_t", *offset, operands, results)
2469            }
2470            abi::Instruction::I32Load16U { offset } => {
2471                self.load_ext("uint16_t", *offset, operands, results)
2472            }
2473            abi::Instruction::I32Load16S { offset } => {
2474                self.load_ext("int16_t", *offset, operands, results)
2475            }
2476            abi::Instruction::I64Load { offset } => {
2477                self.load("int64_t", *offset, operands, results)
2478            }
2479            abi::Instruction::F32Load { offset } => self.load("float", *offset, operands, results),
2480            abi::Instruction::F64Load { offset } => self.load("double", *offset, operands, results),
2481            abi::Instruction::I32Store { offset } => self.store("int32_t", *offset, operands),
2482            abi::Instruction::I32Store8 { offset } => self.store("int8_t", *offset, operands),
2483            abi::Instruction::I32Store16 { offset } => self.store("int16_t", *offset, operands),
2484            abi::Instruction::I64Store { offset } => self.store("int64_t", *offset, operands),
2485            abi::Instruction::F32Store { offset } => self.store("float", *offset, operands),
2486            abi::Instruction::F64Store { offset } => self.store("double", *offset, operands),
2487            abi::Instruction::I32FromChar
2488            | abi::Instruction::I32FromBool
2489            | abi::Instruction::I32FromU8
2490            | abi::Instruction::I32FromS8
2491            | abi::Instruction::I32FromU16
2492            | abi::Instruction::I32FromS16
2493            | abi::Instruction::I32FromU32
2494            | abi::Instruction::I32FromS32 => top_as("int32_t"),
2495            abi::Instruction::I64FromU64 | abi::Instruction::I64FromS64 => top_as("int64_t"),
2496            abi::Instruction::F32FromCoreF32 => top_as("float"),
2497            abi::Instruction::F64FromCoreF64 => top_as("double"),
2498            abi::Instruction::S8FromI32 => top_as("int8_t"),
2499            abi::Instruction::U8FromI32 => top_as("uint8_t"),
2500            abi::Instruction::S16FromI32 => top_as("int16_t"),
2501            abi::Instruction::U16FromI32 => top_as("uint16_t"),
2502            abi::Instruction::S32FromI32 => top_as("int32_t"),
2503            abi::Instruction::U32FromI32 => top_as("uint32_t"),
2504            abi::Instruction::S64FromI64 => top_as("int64_t"),
2505            abi::Instruction::U64FromI64 => top_as("uint64_t"),
2506            abi::Instruction::CharFromI32 => top_as("uint32_t"),
2507            abi::Instruction::CoreF32FromF32 => top_as("float"),
2508            abi::Instruction::CoreF64FromF64 => top_as("double"),
2509            abi::Instruction::BoolFromI32 => top_as("bool"),
2510            abi::Instruction::ListCanonLower { realloc, .. } => {
2511                let tmp = self.tmp();
2512                let val = format!("vec{tmp}");
2513                let ptr = format!("ptr{tmp}");
2514                let len = format!("len{tmp}");
2515                self.push_str(&format!("auto&& {} = {};\n", val, operands[0]));
2516                self.push_str(&format!(
2517                    "auto {} = ({})({}.data());\n",
2518                    ptr,
2519                    self.r#gen.r#gen.opts.ptr_type(),
2520                    val
2521                ));
2522                self.push_str(&format!("auto {len} = (size_t)({val}.size());\n"));
2523                if realloc.is_none() {
2524                    results.push(ptr);
2525                } else {
2526                    uwriteln!(self.src, "{}.leak();\n", operands[0]);
2527                    results.push(ptr);
2528                }
2529                results.push(len);
2530            }
2531            abi::Instruction::StringLower { realloc } => {
2532                let tmp = self.tmp();
2533                let val = format!("vec{tmp}");
2534                let ptr = format!("ptr{tmp}");
2535                let len = format!("len{tmp}");
2536                self.push_str(&format!("auto&& {} = {};\n", val, operands[0]));
2537                self.push_str(&format!(
2538                    "auto {} = ({})({}.data());\n",
2539                    ptr,
2540                    self.r#gen.r#gen.opts.ptr_type(),
2541                    val
2542                ));
2543                self.push_str(&format!("auto {len} = (size_t)({val}.size());\n"));
2544                if realloc.is_none() {
2545                    results.push(ptr);
2546                } else {
2547                    uwriteln!(self.src, "{}.leak();\n", operands[0]);
2548                    results.push(ptr);
2549                }
2550                results.push(len);
2551            }
2552            abi::Instruction::ListLower { element, realloc } => {
2553                let tmp = self.tmp();
2554                let body = self.blocks.pop().unwrap();
2555                let val = format!("vec{tmp}");
2556                let ptr = format!("ptr{tmp}");
2557                let len = format!("len{tmp}");
2558                let size = self.r#gen.sizes.size(element);
2559                self.push_str(&format!("auto&& {} = {};\n", val, operands[0]));
2560                self.push_str(&format!(
2561                    "auto {} = ({})({}.data());\n",
2562                    ptr,
2563                    self.r#gen.r#gen.opts.ptr_type(),
2564                    val
2565                ));
2566                self.push_str(&format!("auto {len} = (size_t)({val}.size());\n"));
2567                self.push_str(&format!("for (size_t i = 0; i < {len}; ++i) {{\n"));
2568                self.push_str(&format!(
2569                    "auto base = {ptr} + i * {size};\n",
2570                    size = size.format(POINTER_SIZE_EXPRESSION)
2571                ));
2572                self.push_str(&format!("auto&& iter_elem = {val}[i];\n"));
2573                self.push_str(&format!("{}\n", body.0));
2574                self.push_str("}\n");
2575                if realloc.is_none() {
2576                    results.push(ptr);
2577                } else {
2578                    uwriteln!(self.src, "{}.leak();\n", operands[0]);
2579                    results.push(ptr);
2580                }
2581                results.push(len);
2582            }
2583            abi::Instruction::ListCanonLift { element, .. } => {
2584                let tmp = self.tmp();
2585                let len = format!("len{tmp}");
2586                let inner = self
2587                    .r#gen
2588                    .type_name(element, &self.namespace, Flavor::InStruct);
2589                self.push_str(&format!("auto {} = {};\n", len, operands[1]));
2590                let result = if self.r#gen.r#gen.opts.api_style == APIStyle::Symmetric
2591                    && matches!(self.variant, AbiVariant::GuestExport)
2592                {
2593                    format!(
2594                        "wit::vector<{inner} const>(({inner}*)({}), {len}).get_view()",
2595                        operands[0]
2596                    )
2597                } else {
2598                    format!("wit::vector<{inner}>(({inner}*)({}), {len})", operands[0])
2599                };
2600                results.push(result);
2601            }
2602            abi::Instruction::StringLift => {
2603                let tmp = self.tmp();
2604                let len = format!("len{tmp}");
2605                uwriteln!(self.src, "auto {} = {};\n", len, operands[1]);
2606                let result = if self.r#gen.r#gen.opts.api_style == APIStyle::Symmetric
2607                    && matches!(self.variant, AbiVariant::GuestExport)
2608                {
2609                    assert!(self.needs_dealloc);
2610                    uwriteln!(
2611                        self.src,
2612                        "if ({len}>0) _deallocate.push_back({});\n",
2613                        operands[0]
2614                    );
2615                    format!("std::string_view((char const*)({}), {len})", operands[0])
2616                } else {
2617                    format!("wit::string((char const*)({}), {len})", operands[0])
2618                };
2619                results.push(result);
2620            }
2621            abi::Instruction::ListLift { element, .. } => {
2622                let body = self.blocks.pop().unwrap();
2623                let tmp = self.tmp();
2624                let size = self.r#gen.sizes.size(element);
2625                let _align = self.r#gen.sizes.align(element);
2626                let flavor = if self.r#gen.r#gen.opts.api_style == APIStyle::Symmetric
2627                    && matches!(self.variant, AbiVariant::GuestExport)
2628                {
2629                    Flavor::BorrowedArgument
2630                } else {
2631                    Flavor::InStruct
2632                };
2633                let vtype = self.r#gen.type_name(element, &self.namespace, flavor);
2634                let len = format!("len{tmp}");
2635                let base = format!("base{tmp}");
2636                let result = format!("result{tmp}");
2637                self.push_str(&format!(
2638                    "auto {base} = {operand0};\n",
2639                    operand0 = operands[0]
2640                ));
2641                self.push_str(&format!(
2642                    "auto {len} = {operand1};\n",
2643                    operand1 = operands[1]
2644                ));
2645                self.push_str(&format!(
2646                    r#"auto {result} = wit::vector<{vtype}>::allocate({len});
2647                    "#,
2648                ));
2649
2650                if self.r#gen.r#gen.opts.api_style == APIStyle::Symmetric
2651                    && matches!(self.variant, AbiVariant::GuestExport)
2652                {
2653                    assert!(self.needs_dealloc);
2654                    self.push_str(&format!("if ({len}>0) _deallocate.push_back({base});\n"));
2655                }
2656
2657                uwriteln!(self.src, "for (unsigned i=0; i<{len}; ++i) {{");
2658                uwriteln!(
2659                    self.src,
2660                    "auto base = {base} + i * {size};",
2661                    size = size.format(POINTER_SIZE_EXPRESSION)
2662                );
2663                uwrite!(self.src, "{}", body.0);
2664                uwriteln!(self.src, "auto e{tmp} = {};", move_if_necessary(&body.1[0]));
2665                if let Some(code) = self.leak_on_insertion.take() {
2666                    assert!(self.needs_dealloc);
2667                    uwriteln!(self.src, "{code}");
2668                }
2669                // inplace construct
2670                uwriteln!(self.src, "{result}.initialize(i, std::move(e{tmp}));");
2671                uwriteln!(self.src, "}}");
2672
2673                if self.r#gen.r#gen.opts.api_style == APIStyle::Symmetric
2674                    && matches!(self.variant, AbiVariant::GuestExport)
2675                {
2676                    results.push(format!("{result}.get_const_view()"));
2677                    if self.r#gen.r#gen.opts.api_style == APIStyle::Symmetric
2678                        && matches!(self.variant, AbiVariant::GuestExport)
2679                    {
2680                        self.leak_on_insertion.replace(format!(
2681                            "if ({len}>0) _deallocate.push_back((void*){result}.leak());\n"
2682                        ));
2683                    }
2684                } else {
2685                    results.push(move_if_necessary(&result));
2686                }
2687            }
2688            abi::Instruction::FixedLengthListLift {
2689                element,
2690                size,
2691                id: _,
2692            } => {
2693                let tmp = self.tmp();
2694                let result = format!("result{tmp}");
2695                let typename = self
2696                    .r#gen
2697                    .type_name(element, &self.namespace, Flavor::InStruct);
2698                self.push_str(&format!("std::array<{typename}, {size}> {result} = {{",));
2699                for a in operands.drain(0..(*size as usize)) {
2700                    self.push_str(&a);
2701                    self.push_str(", ");
2702                }
2703                self.push_str("};\n");
2704                results.push(result);
2705            }
2706            abi::Instruction::FixedLengthListLiftFromMemory {
2707                element,
2708                size: elemsize,
2709                id: _,
2710            } => {
2711                let body = self.blocks.pop().unwrap();
2712                let tmp = self.tmp();
2713                let vec = format!("array{tmp}");
2714                let source = operands[0].clone();
2715                let size = self.r#gen.sizes.size(element);
2716                let size_str = size.format(POINTER_SIZE_EXPRESSION);
2717                let typename = self
2718                    .r#gen
2719                    .type_name(element, &self.namespace, Flavor::InStruct);
2720                let ptr_type = self.r#gen.r#gen.opts.ptr_type();
2721                self.push_str(&format!("std::array<{typename}, {elemsize}> {vec};\n"));
2722                self.push_str(&format!(
2723                    "{{
2724                    {ptr_type} outer_base = {source};\n"
2725                ));
2726                let source: String = "outer_base".into();
2727                // let vec: String = "outer_vec".into();
2728                self.push_str(&format!("for (unsigned i = 0; i<{elemsize}; ++i) {{\n",));
2729                self.push_str(&format!("{ptr_type} base = {source} + i * {size_str};\n"));
2730                self.push_str(&body.0);
2731                self.push_str(&format!("{vec}[i] = {};", body.1[0]));
2732                self.push_str("\n}\n}\n");
2733                results.push(vec);
2734            }
2735            abi::Instruction::FixedLengthListLower {
2736                element: _,
2737                size,
2738                id: _,
2739            } => {
2740                for i in 0..(*size as usize) {
2741                    results.push(format!("{}[{i}]", operands[0]));
2742                }
2743            }
2744            abi::Instruction::FixedLengthListLowerToMemory {
2745                element,
2746                size: elemsize,
2747                id: _,
2748            } => {
2749                let body = self.blocks.pop().unwrap();
2750                let vec = operands[0].clone();
2751                let target = operands[1].clone();
2752                let size = self.r#gen.sizes.size(element);
2753                let size_str = size.format(POINTER_SIZE_EXPRESSION);
2754                let typename = self
2755                    .r#gen
2756                    .type_name(element, &self.namespace, Flavor::InStruct);
2757                let ptr_type = self.r#gen.r#gen.opts.ptr_type();
2758                self.push_str(&format!(
2759                    "{{
2760                    {ptr_type} outer_base = {target};\n"
2761                ));
2762                let target: String = "outer_base".into();
2763                self.push_str(&format!(
2764                    "std::array<{typename}, {elemsize}>& outer_vec = {vec};\n"
2765                ));
2766                let vec: String = "outer_vec".into();
2767                self.push_str(&format!("for (unsigned i = 0; i<{vec}.size(); ++i) {{\n",));
2768                self.push_str(&format!(
2769                    "{ptr_type} base = {target} + i * {size_str};
2770                     {typename}& iter_elem = {vec}[i];\n"
2771                ));
2772                self.push_str(&body.0);
2773                self.push_str("\n}\n}\n");
2774            }
2775            abi::Instruction::IterElem { .. } => results.push("iter_elem".to_string()),
2776            abi::Instruction::IterBasePointer => results.push("base".to_string()),
2777            abi::Instruction::RecordLower { record, .. } => {
2778                let op = &operands[0];
2779                for f in record.fields.iter() {
2780                    results.push(format!("({}).{}", op, to_c_ident(&f.name)));
2781                }
2782            }
2783            abi::Instruction::RecordLift { record, ty, .. } => {
2784                let mut result =
2785                    self.r#gen
2786                        .type_name(&Type::Id(*ty), &self.namespace, Flavor::InStruct);
2787                result.push('{');
2788                for (_field, val) in record.fields.iter().zip(operands) {
2789                    result.push_str(&(move_if_necessary(val) + ", "));
2790                }
2791                result.push('}');
2792                results.push(result);
2793            }
2794            abi::Instruction::HandleLower {
2795                handle: Handle::Own(ty),
2796                ..
2797            } => {
2798                let op = &operands[0];
2799
2800                // Check if this is an imported or exported resource
2801                let resource_ty = &self.r#gen.resolve.types[*ty];
2802                let resource_ty = match &resource_ty.kind {
2803                    TypeDefKind::Type(Type::Id(id)) => &self.r#gen.resolve.types[*id],
2804                    _ => resource_ty,
2805                };
2806                let is_exported = self.r#gen.is_exported_type(resource_ty);
2807
2808                if is_exported {
2809                    // Exported resources use .release()->handle
2810                    results.push(format!("{op}.release()->handle"));
2811                } else {
2812                    // Imported resources use .into_handle()
2813                    results.push(format!("{op}.into_handle()"));
2814                }
2815            }
2816            abi::Instruction::HandleLower {
2817                handle: Handle::Borrow(_),
2818                ..
2819            } => {
2820                let op = &operands[0];
2821                if op == "(*this)" {
2822                    // TODO is there a better way to decide?
2823                    results.push(format!("{op}.get_handle()"));
2824                } else {
2825                    results.push(format!("{op}.get().get_handle()"));
2826                }
2827            }
2828            abi::Instruction::HandleLift { handle, .. } => {
2829                let op = &operands[0];
2830                match (handle, false) {
2831                    (Handle::Own(ty), true) => match self.variant {
2832                        AbiVariant::GuestExport => {
2833                            results.push(format!("wit::{RESOURCE_EXPORT_BASE_CLASS_NAME}{{{op}}}"))
2834                        }
2835                        AbiVariant::GuestImport => {
2836                            let tmp = self.tmp();
2837                            let var = self.tempname("obj", tmp);
2838                            let tname = self.r#gen.type_name(
2839                                &Type::Id(*ty),
2840                                &self.namespace,
2841                                Flavor::Argument(self.variant),
2842                            );
2843                            uwriteln!(
2844                                self.src,
2845                                "auto {var} = {tname}::remove_resource({op});
2846                                assert({var}.has_value());"
2847                            );
2848                            results.push(format!("{tname}::Owned(*{var})"));
2849                        }
2850                        AbiVariant::GuestImportAsync => todo!(),
2851                        AbiVariant::GuestExportAsync => todo!(),
2852                        AbiVariant::GuestExportAsyncStackful => todo!(),
2853                    },
2854                    (Handle::Own(ty), false) => match self.variant {
2855                        AbiVariant::GuestImport => {
2856                            results.push(format!("wit::{RESOURCE_IMPORT_BASE_CLASS_NAME}{{{op}}}"))
2857                        }
2858                        AbiVariant::GuestExport => {
2859                            let tmp = self.tmp();
2860                            let var = self.tempname("obj", tmp);
2861                            let tname = self.r#gen.type_name(
2862                                &Type::Id(*ty),
2863                                &self.namespace,
2864                                Flavor::Argument(self.variant),
2865                            );
2866
2867                            // Check if this is an imported or exported resource
2868                            let resource_ty = &self.r#gen.resolve.types[*ty];
2869                            let resource_ty = match &resource_ty.kind {
2870                                TypeDefKind::Type(Type::Id(id)) => &self.r#gen.resolve.types[*id],
2871                                _ => resource_ty,
2872                            };
2873                            let is_exported = self.r#gen.is_exported_type(resource_ty);
2874
2875                            if is_exported {
2876                                // Exported resources use ::Owned typedef
2877                                uwriteln!(
2878                                    self.src,
2879                                    "auto {var} = {tname}::Owned({tname}::ResourceRep({op}));"
2880                                );
2881                            } else {
2882                                // Imported resources construct from ResourceImportBase
2883                                uwriteln!(
2884                                    self.src,
2885                                    "auto {var} = {tname}(wit::{RESOURCE_IMPORT_BASE_CLASS_NAME}{{{op}}});"
2886                                );
2887                            }
2888
2889                            results.push(format!("std::move({var})"))
2890                        }
2891                        AbiVariant::GuestImportAsync => todo!(),
2892                        AbiVariant::GuestExportAsync => todo!(),
2893                        AbiVariant::GuestExportAsyncStackful => todo!(),
2894                    },
2895                    (Handle::Borrow(ty), true) => {
2896                        let tname = self.r#gen.type_name(
2897                            &Type::Id(*ty),
2898                            &self.namespace,
2899                            Flavor::Argument(self.variant),
2900                        );
2901                        results.push(format!("**{tname}::lookup_resource({op})"));
2902                    }
2903                    (Handle::Borrow(ty), false) => match self.variant {
2904                        AbiVariant::GuestImport => results.push(op.clone()),
2905                        AbiVariant::GuestExport => {
2906                            let tname = self.r#gen.type_name(
2907                                &Type::Id(*ty),
2908                                &self.namespace,
2909                                Flavor::Argument(self.variant),
2910                            );
2911                            results.push(format!("std::ref(*({tname} *){op})"));
2912                        }
2913                        AbiVariant::GuestImportAsync => todo!(),
2914                        AbiVariant::GuestExportAsync => todo!(),
2915                        AbiVariant::GuestExportAsyncStackful => todo!(),
2916                    },
2917                }
2918            }
2919            abi::Instruction::TupleLower { tuple, .. } => {
2920                let op = &operands[0];
2921                for n in 0..tuple.types.len() {
2922                    results.push(format!("std::get<{n}>({op})"));
2923                }
2924            }
2925            abi::Instruction::TupleLift { tuple, .. } => {
2926                let name = format!("tuple{}", self.tmp());
2927                uwrite!(self.src, "auto {name} = std::tuple<");
2928                self.src.push_str(
2929                    &(tuple
2930                        .types
2931                        .iter()
2932                        .map(|t| self.r#gen.type_name(t, &self.namespace, Flavor::InStruct)))
2933                    .collect::<Vec<_>>()
2934                    .join(", "),
2935                );
2936                self.src.push_str(">(");
2937                self.src.push_str(
2938                    &operands
2939                        .iter()
2940                        .map(|op| move_if_necessary(op))
2941                        .collect::<Vec<_>>()
2942                        .join(", "),
2943                );
2944                self.src.push_str(");\n");
2945                results.push(format!("std::move({name})"));
2946            }
2947            abi::Instruction::FlagsLower { flags, ty, .. } => {
2948                match wit_bindgen_c::flags_repr(flags) {
2949                    Int::U8 | Int::U16 | Int::U32 => {
2950                        results.push(format!("((int32_t){})", operands.pop().unwrap()));
2951                    }
2952                    Int::U64 => {
2953                        let name =
2954                            self.r#gen
2955                                .type_name(&Type::Id(*ty), &self.namespace, Flavor::InStruct);
2956                        let tmp = self.tmp();
2957                        let tempname = self.tempname("flags", tmp);
2958                        uwriteln!(self.src, "{name} {tempname} = {};", operands[0]);
2959                        results.push(format!("(int32_t)(((uint64_t){tempname}) & 0xffffffff)"));
2960                        results.push(format!(
2961                            "(int32_t)((((uint64_t){tempname}) >> 32) & 0xffffffff)"
2962                        ));
2963                    }
2964                }
2965            }
2966            abi::Instruction::FlagsLift { flags, ty, .. } => {
2967                let typename =
2968                    self.r#gen
2969                        .type_name(&Type::Id(*ty), &self.namespace, Flavor::InStruct);
2970                match wit_bindgen_c::flags_repr(flags) {
2971                    Int::U8 | Int::U16 | Int::U32 => {
2972                        results.push(format!("(({typename}){})", operands.pop().unwrap()));
2973                    }
2974                    Int::U64 => {
2975                        let op0 = &operands[0];
2976                        let op1 = &operands[1];
2977                        results.push(format!(
2978                            "(({typename})(({op0}) | (((uint64_t)({op1})) << 32)))"
2979                        ));
2980                    }
2981                }
2982            }
2983            abi::Instruction::VariantPayloadName => {
2984                let name = format!("payload{}", self.tmp());
2985                results.push(name.clone());
2986                self.payloads.push(name);
2987            }
2988            abi::Instruction::VariantLower {
2989                variant,
2990                results: result_types,
2991                ty: var_ty,
2992                name: _var_name,
2993                ..
2994            } => {
2995                let blocks = self
2996                    .blocks
2997                    .drain(self.blocks.len() - variant.cases.len()..)
2998                    .collect::<Vec<_>>();
2999                let payloads = self
3000                    .payloads
3001                    .drain(self.payloads.len() - variant.cases.len()..)
3002                    .collect::<Vec<_>>();
3003
3004                let mut variant_results = Vec::with_capacity(result_types.len());
3005                for ty in result_types.iter() {
3006                    let name = format!("variant{}", self.tmp());
3007                    results.push(name.clone());
3008                    self.src.push_str(wit_bindgen_c::wasm_type(*ty));
3009                    self.src.push_str(" ");
3010                    self.src.push_str(&name);
3011                    self.src.push_str(";\n");
3012                    variant_results.push(name);
3013                }
3014
3015                let expr_to_match = format!("({}).variants.index()", operands[0]);
3016                let elem_ns =
3017                    self.r#gen
3018                        .type_name(&Type::Id(*var_ty), &self.namespace, Flavor::InStruct);
3019
3020                uwriteln!(self.src, "switch ((int32_t) {}) {{", expr_to_match);
3021                for (i, ((case, (block, block_results)), payload)) in
3022                    variant.cases.iter().zip(blocks).zip(payloads).enumerate()
3023                {
3024                    uwriteln!(self.src, "case {}: {{", i);
3025                    if case.ty.is_some() {
3026                        let case =
3027                            format!("{elem_ns}::{}", to_c_ident(&case.name).to_pascal_case());
3028                        uwriteln!(
3029                            self.src,
3030                            "auto& {} = std::get<{case}>({}.variants).value;",
3031                            payload,
3032                            operands[0],
3033                        );
3034                    }
3035
3036                    self.src.push_str(&block);
3037
3038                    for (name, result) in variant_results.iter().zip(&block_results) {
3039                        uwriteln!(self.src, "{} = {};", name, result);
3040                    }
3041                    self.src.push_str("break;\n}\n");
3042                }
3043                self.src.push_str("}\n");
3044            }
3045            abi::Instruction::VariantLift { variant, ty, .. } => {
3046                let blocks = self
3047                    .blocks
3048                    .drain(self.blocks.len() - variant.cases.len()..)
3049                    .collect::<Vec<_>>();
3050
3051                let ty = self
3052                    .r#gen
3053                    .type_name(&Type::Id(*ty), &self.namespace, Flavor::InStruct);
3054                let resultno = self.tmp();
3055                let result = format!("variant{resultno}");
3056
3057                let op0 = &operands[0];
3058
3059                // Use std::optional to avoid default constructor issues
3060                self.r#gen.r#gen.dependencies.needs_optional = true;
3061                uwriteln!(self.src, "std::optional<{ty}> {result}_opt;");
3062                uwriteln!(self.src, "switch ({op0}) {{");
3063                for (i, (case, (block, block_results))) in
3064                    variant.cases.iter().zip(blocks).enumerate()
3065                {
3066                    let tp = to_c_ident(&case.name).to_pascal_case();
3067                    uwriteln!(self.src, "case {i}: {{ {block}");
3068                    uwriteln!(
3069                        self.src,
3070                        "{result}_opt = {ty}{{{{{ty}::{tp}{{{}}}}}}};",
3071                        move_if_necessary(&block_results.first().cloned().unwrap_or_default())
3072                    );
3073                    uwriteln!(self.src, "}} break;");
3074                }
3075                uwriteln!(self.src, "}}");
3076                uwriteln!(self.src, "{ty} {result} = std::move(*{result}_opt);");
3077
3078                results.push(result);
3079            }
3080            abi::Instruction::EnumLower { .. } => results.push(format!("int32_t({})", operands[0])),
3081            abi::Instruction::EnumLift { ty, .. } => {
3082                let typename =
3083                    self.r#gen
3084                        .type_name(&Type::Id(*ty), &self.namespace, Flavor::InStruct);
3085                results.push(format!("({typename}){}", &operands[0]));
3086            }
3087            abi::Instruction::OptionLower {
3088                payload,
3089                results: result_types,
3090                ..
3091            } => {
3092                let (mut some, some_results) = self.blocks.pop().unwrap();
3093                let (mut none, none_results) = self.blocks.pop().unwrap();
3094                let some_payload = self.payloads.pop().unwrap();
3095                let _none_payload = self.payloads.pop().unwrap();
3096
3097                for (i, ty) in result_types.iter().enumerate() {
3098                    let tmp = self.tmp();
3099                    let name = self.tempname("option", tmp);
3100                    results.push(name.clone());
3101                    self.src.push_str(wit_bindgen_c::wasm_type(*ty));
3102                    self.src.push_str(" ");
3103                    self.src.push_str(&name);
3104                    self.src.push_str(";\n");
3105                    let some_result = &some_results[i];
3106                    uwriteln!(some, "{name} = {some_result};");
3107                    let none_result = &none_results[i];
3108                    uwriteln!(none, "{name} = {none_result};");
3109                }
3110
3111                let op0 = &operands[0];
3112                let flavor = if matches!(self.variant, AbiVariant::GuestImport) {
3113                    Flavor::BorrowedArgument
3114                } else {
3115                    Flavor::InStruct
3116                };
3117                let ty = self.r#gen.type_name(payload, &self.namespace, flavor);
3118                let is_function_param = self.params.iter().any(|p| p == op0);
3119                let value_extract = if matches!(payload, Type::String)
3120                    && matches!(self.variant, AbiVariant::GuestImport)
3121                    && !is_function_param
3122                {
3123                    // Import from struct/variant field: optional<wit::string> needs .get_view()
3124                    format!("(std::move({op0})).value().get_view()")
3125                } else {
3126                    // Direct parameter, export, or non-string: just .value()
3127                    format!("(std::move({op0})).value()")
3128                };
3129                let bind_some = format!("{ty} {some_payload} = {value_extract};");
3130
3131                uwrite!(
3132                    self.src,
3133                    "\
3134                    if (({op0}).has_value()) {{
3135                        {bind_some}
3136                        {some}}} else {{
3137                        {none}}}
3138                    "
3139                );
3140            }
3141            abi::Instruction::OptionLift { payload, .. } => {
3142                let (some, some_results) = self.blocks.pop().unwrap();
3143                let (_none, none_results) = self.blocks.pop().unwrap();
3144                assert!(none_results.is_empty());
3145                assert!(some_results.len() == 1);
3146                let flavor = if self.r#gen.r#gen.opts.api_style == APIStyle::Symmetric
3147                    && matches!(self.variant, AbiVariant::GuestExport)
3148                {
3149                    Flavor::BorrowedArgument
3150                } else {
3151                    Flavor::InStruct
3152                };
3153                let type_name = self.r#gen.type_name(payload, &self.namespace, flavor);
3154                let full_type = format!("std::optional<{type_name}>");
3155                let op0 = &operands[0];
3156
3157                let tmp = self.tmp();
3158                let resultname = self.tempname("option", tmp);
3159                let some_value = move_if_necessary(&some_results[0]);
3160                uwriteln!(
3161                    self.src,
3162                    "{full_type} {resultname};
3163                    if ({op0}) {{
3164                        {some}
3165                        {resultname}.emplace({some_value});
3166                    }}"
3167                );
3168                results.push(format!("std::move({resultname})"));
3169            }
3170            abi::Instruction::ResultLower {
3171                results: result_types,
3172                result,
3173                ..
3174            } => {
3175                let (mut err, err_results) = self.blocks.pop().unwrap();
3176                let (mut ok, ok_results) = self.blocks.pop().unwrap();
3177                let err_payload = self.payloads.pop().unwrap();
3178                let ok_payload = self.payloads.pop().unwrap();
3179
3180                for (i, ty) in result_types.iter().enumerate() {
3181                    let tmp = self.tmp();
3182                    let name = self.tempname("result", tmp);
3183                    results.push(name.clone());
3184                    self.src.push_str(wit_bindgen_c::wasm_type(*ty));
3185                    self.src.push_str(" ");
3186                    self.src.push_str(&name);
3187                    self.src.push_str(";\n");
3188                    let ok_result = &ok_results[i];
3189                    uwriteln!(ok, "{name} = {ok_result};");
3190                    let err_result = &err_results[i];
3191                    uwriteln!(err, "{name} = {err_result};");
3192                }
3193
3194                let op0 = &operands[0];
3195                let ok_ty = self.r#gen.optional_type_name(
3196                    result.ok.as_ref(),
3197                    &self.namespace,
3198                    Flavor::InStruct,
3199                );
3200                let err_ty = self.r#gen.optional_type_name(
3201                    result.err.as_ref(),
3202                    &self.namespace,
3203                    Flavor::InStruct,
3204                );
3205                let bind_ok = if let Some(_ok) = result.ok.as_ref() {
3206                    format!("{ok_ty} {ok_payload} = std::move({op0}).value();")
3207                } else {
3208                    String::new()
3209                };
3210                let bind_err = if let Some(_err) = result.err.as_ref() {
3211                    format!("{err_ty} {err_payload} = std::move({op0}).error();")
3212                } else {
3213                    String::new()
3214                };
3215
3216                uwrite!(
3217                    self.src,
3218                    "\
3219                    if (({op0}).has_value()) {{
3220                        {bind_ok}
3221                        {ok}}} else {{
3222                        {bind_err}
3223                        {err}}}
3224                    "
3225                );
3226            }
3227            abi::Instruction::ResultLift { result, .. } => {
3228                let (mut err, err_results) = self.blocks.pop().unwrap();
3229                let (mut ok, ok_results) = self.blocks.pop().unwrap();
3230                let mut ok_result = String::new();
3231                let err_result;
3232                if result.ok.is_none() {
3233                    ok.clear();
3234                } else {
3235                    ok_result = move_if_necessary(&ok_results[0]);
3236                }
3237                if result.err.is_none() {
3238                    err.clear();
3239                    self.r#gen.r#gen.dependencies.needs_wit = true;
3240                    err_result = String::from("wit::Void{}");
3241                } else {
3242                    err_result = move_if_necessary(&err_results[0]);
3243                }
3244                let ok_type = self.r#gen.optional_type_name(
3245                    result.ok.as_ref(),
3246                    &self.namespace,
3247                    Flavor::InStruct,
3248                );
3249                let err_type = result.err.as_ref().map_or(String::from("wit::Void"), |ty| {
3250                    self.r#gen.type_name(ty, &self.namespace, Flavor::InStruct)
3251                });
3252                let full_type = format!("std::expected<{ok_type}, {err_type}>",);
3253                let err_type = "std::unexpected";
3254                let operand = &operands[0];
3255
3256                let tmp = self.tmp();
3257                let resultname = self.tempname("result", tmp);
3258                // Use std::optional to avoid default constructor issues with std::expected
3259                self.r#gen.r#gen.dependencies.needs_optional = true;
3260                let ok_assign = if result.ok.is_some() {
3261                    format!("{resultname}_opt.emplace({full_type}({ok_result}));")
3262                } else {
3263                    format!("{resultname}_opt.emplace({full_type}());")
3264                };
3265                uwriteln!(
3266                    self.src,
3267                    "std::optional<{full_type}> {resultname}_opt;
3268                    if ({operand}==0) {{
3269                        {ok}
3270                        {ok_assign}
3271                    }} else {{
3272                        {err}
3273                        {resultname}_opt.emplace({err_type}{{{err_result}}});
3274                    }}
3275                    {full_type} {resultname} = std::move(*{resultname}_opt);"
3276                );
3277                results.push(resultname);
3278            }
3279            abi::Instruction::CallWasm { name, sig } => {
3280                let module_name = self
3281                    .r#gen
3282                    .wasm_import_module
3283                    .as_ref()
3284                    .map(|e| {
3285                        self.r#gen
3286                            .r#gen
3287                            .import_prefix
3288                            .as_ref()
3289                            .cloned()
3290                            .unwrap_or_default()
3291                            + e
3292                    })
3293                    .unwrap();
3294
3295                let func = self
3296                    .r#gen
3297                    .declare_import(&module_name, name, &sig.params, &sig.results);
3298
3299                // ... then call the function with all our operands
3300                if !sig.results.is_empty() {
3301                    self.src.push_str("auto ret = ");
3302                    results.push("ret".to_string());
3303                }
3304                self.src.push_str(&func);
3305                self.src.push_str("(");
3306                self.src.push_str(
3307                    &operands
3308                        .iter()
3309                        .map(|op| move_if_necessary(op))
3310                        .collect::<Vec<_>>()
3311                        .join(", "),
3312                );
3313                self.src.push_str(");\n");
3314            }
3315            abi::Instruction::CallInterface { func, .. } => {
3316                // dbg!(func);
3317                self.let_results(if func.result.is_some() { 1 } else { 0 }, results);
3318                let (namespace, func_name_h) = self.r#gen.func_namespace_name(func, true, true);
3319                if matches!(func.kind, FunctionKind::Method(_)) {
3320                    let this = operands.remove(0);
3321                    uwrite!(self.src, "({this}).get().");
3322                } else {
3323                    let mut relative = SourceWithState::default();
3324                    relative.qualify(&namespace);
3325                    self.push_str(&relative.src);
3326                }
3327                self.src.push_str(&func_name_h);
3328                self.push_str("(");
3329                self.push_str(
3330                    &operands
3331                        .iter()
3332                        .map(|op| move_if_necessary(op))
3333                        .collect::<Vec<_>>()
3334                        .join(", "),
3335                );
3336                self.push_str(");\n");
3337                if self.needs_dealloc {
3338                    uwriteln!(
3339                        self.src,
3340                        "for (auto i: _deallocate) {{ free(i); }}\n
3341                        _deallocate.clear();"
3342                    );
3343                }
3344            }
3345            abi::Instruction::Return { amt, func } => {
3346                match amt {
3347                    0 => {}
3348                    _ => {
3349                        assert!(*amt == operands.len());
3350                        // Fallible constructors return expected, not void
3351                        let is_fallible_constructor = self
3352                            .r#gen
3353                            .r#gen
3354                            .is_fallible_constructor(self.r#gen.resolve, func);
3355
3356                        match &func.kind {
3357                            FunctionKind::Constructor(_)
3358                                if self.r#gen.r#gen.opts.is_only_handle(self.variant)
3359                                    && !is_fallible_constructor =>
3360                            {
3361                                // strange but works
3362                                if matches!(self.variant, AbiVariant::GuestExport) {
3363                                    self.src.push_str("this->index = ");
3364                                } else {
3365                                    self.src.push_str("this->handle = ");
3366                                }
3367                            }
3368                            _ => self.src.push_str("return "),
3369                        }
3370                        if let Some(CabiPostInformation {
3371                            module: _,
3372                            name: _cabi_post_name,
3373                            ret_type: cabi_post_type,
3374                        }) = self.cabi_post.as_ref()
3375                        {
3376                            self.src.push_str("wit::guest_owned<");
3377                            self.src.push_str(cabi_post_type);
3378                            self.src.push_str(">(");
3379                        }
3380                        if *amt == 1 {
3381                            if operands[0].starts_with("std::move(") && !operands[0].contains('.') {
3382                                // remove the std::move due to return value optimization (and complex rules about when std::move harms)
3383                                self.src.push_str(&operands[0][9..]);
3384                            } else {
3385                                self.src.push_str(&operands[0]);
3386                            }
3387                        } else {
3388                            todo!();
3389                        }
3390                        if let Some(CabiPostInformation {
3391                            module: func_module,
3392                            name: func_name,
3393                            ret_type: _cabi_post_type,
3394                        }) = self.cabi_post.as_ref()
3395                        {
3396                            let cabi_post_name = self.r#gen.declare_import(
3397                                &format!("cabi_post_{func_module}"),
3398                                func_name,
3399                                &[WasmType::Pointer],
3400                                &[],
3401                            );
3402                            self.src.push_str(&format!(", ret, {cabi_post_name})"));
3403                        }
3404                        if matches!(func.kind, FunctionKind::Constructor(_))
3405                            && self.r#gen.r#gen.opts.is_only_handle(self.variant)
3406                            && !is_fallible_constructor
3407                        {
3408                            // we wrapped the handle in an object, so unpack it
3409
3410                            self.src.push_str(".into_handle()");
3411                        }
3412                        self.src.push_str(";\n");
3413                    }
3414                }
3415            }
3416            abi::Instruction::Malloc { .. } => todo!(),
3417            abi::Instruction::GuestDeallocate { .. } => {
3418                uwriteln!(self.src, "free((void*) ({}));", operands[0]);
3419            }
3420            abi::Instruction::GuestDeallocateString => {
3421                uwriteln!(self.src, "if (({}) > 0) {{", operands[1]);
3422                uwriteln!(
3423                    self.src,
3424                    "wit::string::drop_raw((void*) ({}));",
3425                    operands[0]
3426                );
3427                uwriteln!(self.src, "}}");
3428            }
3429            abi::Instruction::GuestDeallocateList { element } => {
3430                let (body, results) = self.blocks.pop().unwrap();
3431                assert!(results.is_empty());
3432                let tmp = self.tmp();
3433                let ptr = self.tempname("ptr", tmp);
3434                let len = self.tempname("len", tmp);
3435                uwriteln!(self.src, "uint8_t* {ptr} = {};", operands[0]);
3436                uwriteln!(self.src, "size_t {len} = {};", operands[1]);
3437                let i = self.tempname("i", tmp);
3438                uwriteln!(self.src, "for (size_t {i} = 0; {i} < {len}; {i}++) {{");
3439                let size = self.r#gen.sizes.size(element);
3440                uwriteln!(
3441                    self.src,
3442                    "uint8_t* base = {ptr} + {i} * {size};",
3443                    size = size.format(POINTER_SIZE_EXPRESSION)
3444                );
3445                uwriteln!(self.src, "(void) base;");
3446                uwrite!(self.src, "{body}");
3447                uwriteln!(self.src, "}}");
3448                uwriteln!(self.src, "if ({len} > 0) {{");
3449                uwriteln!(self.src, "free((void*) ({ptr}));");
3450                uwriteln!(self.src, "}}");
3451            }
3452            abi::Instruction::GuestDeallocateVariant { blocks } => {
3453                let blocks = self
3454                    .blocks
3455                    .drain(self.blocks.len() - blocks..)
3456                    .collect::<Vec<_>>();
3457
3458                uwriteln!(self.src, "switch ((int32_t) {}) {{", operands[0]);
3459                for (i, (block, results)) in blocks.into_iter().enumerate() {
3460                    assert!(results.is_empty());
3461                    uwriteln!(self.src, "case {}: {{", i);
3462                    self.src.push_str(&block);
3463                    self.src.push_str("break;\n}\n");
3464                }
3465                self.src.push_str("}\n");
3466            }
3467            abi::Instruction::PointerLoad { offset } => {
3468                let ptr_type = self.r#gen.r#gen.opts.ptr_type();
3469                self.load(ptr_type, *offset, operands, results)
3470            }
3471            abi::Instruction::LengthLoad { offset } => {
3472                self.load("size_t", *offset, operands, results)
3473            }
3474            abi::Instruction::PointerStore { offset } => {
3475                let ptr_type = self.r#gen.r#gen.opts.ptr_type();
3476                self.store(ptr_type, *offset, operands)
3477            }
3478            abi::Instruction::LengthStore { offset } => self.store("size_t", *offset, operands),
3479            abi::Instruction::FutureLower { .. } => todo!(),
3480            abi::Instruction::FutureLift { .. } => todo!(),
3481            abi::Instruction::StreamLower { .. } => todo!(),
3482            abi::Instruction::StreamLift { .. } => todo!(),
3483            abi::Instruction::ErrorContextLower { .. } => todo!(),
3484            abi::Instruction::ErrorContextLift { .. } => todo!(),
3485            abi::Instruction::Flush { amt } => {
3486                for i in operands.iter().take(*amt) {
3487                    let tmp = self.tmp();
3488                    let result = format!("result{tmp}");
3489                    uwriteln!(self.src, "auto {result} = {};", move_if_necessary(i));
3490                    results.push(result);
3491                }
3492            }
3493            abi::Instruction::AsyncTaskReturn { .. } => todo!(),
3494            abi::Instruction::DropHandle { .. } => todo!(),
3495        }
3496    }
3497
3498    fn return_pointer(&mut self, size: ArchitectureSize, align: Alignment) -> Self::Operand {
3499        // Track maximum return area requirements
3500        self.return_pointer_area_size = self.return_pointer_area_size.max(size);
3501        self.return_pointer_area_align = self.return_pointer_area_align.max(align);
3502
3503        // Generate pointer to shared ret_area
3504        let tmp = self.tmp();
3505        uwriteln!(
3506            self.src,
3507            "{} ptr{tmp} = ({0})(&ret_area);",
3508            self.r#gen.r#gen.opts.ptr_type(),
3509        );
3510
3511        format!("ptr{tmp}")
3512    }
3513
3514    fn push_block(&mut self) {
3515        let prev = core::mem::take(&mut self.src);
3516        self.block_storage.push(prev);
3517    }
3518
3519    fn finish_block(&mut self, operands: &mut Vec<Self::Operand>) {
3520        let to_restore = self.block_storage.pop().unwrap();
3521        let src = core::mem::replace(&mut self.src, to_restore);
3522        self.blocks.push((src.into(), core::mem::take(operands)));
3523    }
3524
3525    fn sizes(&self) -> &wit_bindgen_core::wit_parser::SizeAlign {
3526        &self.r#gen.sizes
3527    }
3528
3529    fn is_list_canonical(
3530        &self,
3531        resolve: &Resolve,
3532        ty: &wit_bindgen_core::wit_parser::Type,
3533    ) -> bool {
3534        if !resolve.all_bits_valid(ty) {
3535            return false;
3536        }
3537        match ty {
3538            Type::Id(id) => !self.r#gen.r#gen.types.get(*id).has_resource,
3539            _ => true,
3540        }
3541    }
3542}
3543
3544/// This describes the common ABI function referenced or implemented, the C++ side might correspond to a different type
3545enum SpecialMethod {
3546    None,
3547    ResourceDrop, // ([export]) [resource-drop]
3548    ResourceNew,  // [export][resource-new]
3549    ResourceRep,  // [export][resource-rep]
3550    Dtor,         // [dtor] (guest export only)
3551    Allocate,     // internal: allocate new object (called from generated code)
3552}
3553
3554fn is_special_method(func: &Function) -> SpecialMethod {
3555    if matches!(func.kind, FunctionKind::Static(_)) {
3556        if func.name.starts_with("[resource-drop]") {
3557            SpecialMethod::ResourceDrop
3558        } else if func.name.starts_with("[resource-new]") {
3559            SpecialMethod::ResourceNew
3560        } else if func.name.starts_with("[resource-rep]") {
3561            SpecialMethod::ResourceRep
3562        } else if func.name.starts_with("[dtor]") {
3563            SpecialMethod::Dtor
3564        } else if func.name == "$alloc" {
3565            SpecialMethod::Allocate
3566        } else {
3567            SpecialMethod::None
3568        }
3569    } else {
3570        SpecialMethod::None
3571    }
3572}