1use std::cmp::Ordering;
2use std::collections::hash_map;
3use std::collections::{BTreeMap, HashMap, HashSet};
4use std::fmt;
5use std::mem;
6use std::path::{Path, PathBuf};
7
8use anyhow::{anyhow, bail, Context, Result};
9use id_arena::{Arena, Id};
10use indexmap::{IndexMap, IndexSet};
11use semver::Version;
12#[cfg(feature = "serde")]
13use serde_derive::Serialize;
14
15use crate::ast::lex::Span;
16use crate::ast::{parse_use_path, ParsedUsePath};
17#[cfg(feature = "serde")]
18use crate::serde_::{serialize_arena, serialize_id_map};
19use crate::{
20 AstItem, Docs, Error, Function, FunctionKind, Handle, IncludeName, Interface, InterfaceId,
21 InterfaceSpan, LiftLowerAbi, ManglingAndAbi, PackageName, PackageNotFoundError, SourceMap,
22 Stability, Type, TypeDef, TypeDefKind, TypeId, TypeIdVisitor, TypeOwner, UnresolvedPackage,
23 UnresolvedPackageGroup, World, WorldId, WorldItem, WorldKey, WorldSpan,
24};
25
26mod clone;
27
28#[derive(Default, Clone, Debug)]
42#[cfg_attr(feature = "serde", derive(Serialize))]
43pub struct Resolve {
44 #[cfg_attr(feature = "serde", serde(serialize_with = "serialize_arena"))]
49 pub worlds: Arena<World>,
50
51 #[cfg_attr(feature = "serde", serde(serialize_with = "serialize_arena"))]
56 pub interfaces: Arena<Interface>,
57
58 #[cfg_attr(feature = "serde", serde(serialize_with = "serialize_arena"))]
64 pub types: Arena<TypeDef>,
65
66 #[cfg_attr(feature = "serde", serde(serialize_with = "serialize_arena"))]
71 pub packages: Arena<Package>,
72
73 #[cfg_attr(feature = "serde", serde(skip))]
75 pub package_names: IndexMap<PackageName, PackageId>,
76
77 #[cfg_attr(feature = "serde", serde(skip))]
84 pub features: IndexSet<String>,
85
86 #[cfg_attr(feature = "serde", serde(skip))]
88 pub all_features: bool,
89}
90
91#[derive(Clone, Debug)]
97#[cfg_attr(feature = "serde", derive(Serialize))]
98pub struct Package {
99 pub name: PackageName,
101
102 #[cfg_attr(feature = "serde", serde(skip_serializing_if = "Docs::is_empty"))]
104 pub docs: Docs,
105
106 #[cfg_attr(feature = "serde", serde(serialize_with = "serialize_id_map"))]
109 pub interfaces: IndexMap<String, InterfaceId>,
110
111 #[cfg_attr(feature = "serde", serde(serialize_with = "serialize_id_map"))]
113 pub worlds: IndexMap<String, WorldId>,
114}
115
116pub type PackageId = Id<Package>;
117
118#[derive(Clone, Debug)]
120pub struct PackageSourceMap {
121 sources: Vec<Vec<PathBuf>>,
122 package_id_to_source_map_idx: BTreeMap<PackageId, usize>,
123}
124
125impl PackageSourceMap {
126 fn from_single_source(package_id: PackageId, source: &Path) -> Self {
127 Self {
128 sources: vec![vec![source.to_path_buf()]],
129 package_id_to_source_map_idx: BTreeMap::from([(package_id, 0)]),
130 }
131 }
132
133 fn from_source_maps(
134 source_maps: Vec<SourceMap>,
135 package_id_to_source_map_idx: BTreeMap<PackageId, usize>,
136 ) -> PackageSourceMap {
137 for (package_id, idx) in &package_id_to_source_map_idx {
138 if *idx >= source_maps.len() {
139 panic!(
140 "Invalid source map index: {}, package id: {:?}, source maps size: {}",
141 idx,
142 package_id,
143 source_maps.len()
144 )
145 }
146 }
147
148 Self {
149 sources: source_maps
150 .into_iter()
151 .map(|source_map| {
152 source_map
153 .source_files()
154 .map(|path| path.to_path_buf())
155 .collect()
156 })
157 .collect(),
158 package_id_to_source_map_idx,
159 }
160 }
161
162 pub fn paths(&self) -> impl Iterator<Item = &Path> {
164 self.sources
168 .iter()
169 .flatten()
170 .map(|path_buf| path_buf.as_ref())
171 .collect::<HashSet<&Path>>()
172 .into_iter()
173 }
174
175 pub fn package_paths(&self, id: PackageId) -> Option<impl Iterator<Item = &Path>> {
177 self.package_id_to_source_map_idx
178 .get(&id)
179 .map(|&idx| self.sources[idx].iter().map(|path_buf| path_buf.as_ref()))
180 }
181}
182
183enum ParsedFile {
184 #[cfg(feature = "decoding")]
185 Package(PackageId),
186 Unresolved(UnresolvedPackageGroup),
187}
188
189fn visit<'a>(
191 pkg: &'a UnresolvedPackage,
192 pkg_details_map: &'a BTreeMap<PackageName, (UnresolvedPackage, usize)>,
193 order: &mut IndexSet<PackageName>,
194 visiting: &mut HashSet<&'a PackageName>,
195 source_maps: &[SourceMap],
196) -> Result<()> {
197 if order.contains(&pkg.name) {
198 return Ok(());
199 }
200
201 match pkg_details_map.get(&pkg.name) {
202 Some(pkg_details) => {
203 let (_, source_maps_index) = pkg_details;
204 source_maps[*source_maps_index].rewrite_error(|| {
205 for (i, (dep, _)) in pkg.foreign_deps.iter().enumerate() {
206 let span = pkg.foreign_dep_spans[i];
207 if !visiting.insert(dep) {
208 bail!(Error::new(span, "package depends on itself"));
209 }
210 if let Some(dep) = pkg_details_map.get(dep) {
211 let (dep_pkg, _) = dep;
212 visit(dep_pkg, pkg_details_map, order, visiting, source_maps)?;
213 }
214 assert!(visiting.remove(dep));
215 }
216 assert!(order.insert(pkg.name.clone()));
217 Ok(())
218 })
219 }
220 None => panic!("No pkg_details found for package when doing topological sort"),
221 }
222}
223
224impl Resolve {
225 pub fn new() -> Resolve {
227 Resolve::default()
228 }
229
230 pub fn push_path(&mut self, path: impl AsRef<Path>) -> Result<(PackageId, PackageSourceMap)> {
258 self._push_path(path.as_ref())
259 }
260
261 fn _push_path(&mut self, path: &Path) -> Result<(PackageId, PackageSourceMap)> {
262 if path.is_dir() {
263 self.push_dir(path).with_context(|| {
264 format!(
265 "failed to resolve directory while parsing WIT for path [{}]",
266 path.display()
267 )
268 })
269 } else {
270 let id = self.push_file(path)?;
271 Ok((id, PackageSourceMap::from_single_source(id, path)))
272 }
273 }
274
275 fn sort_unresolved_packages(
276 &mut self,
277 main: UnresolvedPackageGroup,
278 deps: Vec<UnresolvedPackageGroup>,
279 ) -> Result<(PackageId, PackageSourceMap)> {
280 let mut pkg_details_map = BTreeMap::new();
281 let mut source_maps = Vec::new();
282
283 let mut insert = |group: UnresolvedPackageGroup| {
284 let UnresolvedPackageGroup {
285 main,
286 nested,
287 source_map,
288 } = group;
289 let i = source_maps.len();
290 source_maps.push(source_map);
291
292 for pkg in nested.into_iter().chain([main]) {
293 let name = pkg.name.clone();
294 let my_span = pkg.package_name_span;
295 let (prev_pkg, prev_i) = match pkg_details_map.insert(name.clone(), (pkg, i)) {
296 Some(pair) => pair,
297 None => continue,
298 };
299 let loc1 = source_maps[i].render_location(my_span);
300 let loc2 = source_maps[prev_i].render_location(prev_pkg.package_name_span);
301 bail!(
302 "\
303package {name} is defined in two different locations:\n\
304 * {loc1}\n\
305 * {loc2}\n\
306 "
307 )
308 }
309 Ok(())
310 };
311
312 let main_name = main.main.name.clone();
313 insert(main)?;
314 for dep in deps {
315 insert(dep)?;
316 }
317
318 let mut order = IndexSet::new();
322 let mut visiting = HashSet::new();
323 for pkg_details in pkg_details_map.values() {
324 let (pkg, _) = pkg_details;
325 visit(
326 pkg,
327 &pkg_details_map,
328 &mut order,
329 &mut visiting,
330 &source_maps,
331 )?;
332 }
333
334 let mut package_id_to_source_map_idx = BTreeMap::new();
338 let mut main_pkg_id = None;
339 for name in order {
340 let (pkg, source_map_index) = pkg_details_map.remove(&name).unwrap();
341 let source_map = &source_maps[source_map_index];
342 let is_main = pkg.name == main_name;
343 let id = self.push(pkg, source_map)?;
344 if is_main {
345 assert!(main_pkg_id.is_none());
346 main_pkg_id = Some(id);
347 }
348 package_id_to_source_map_idx.insert(id, source_map_index);
349 }
350
351 Ok((
352 main_pkg_id.unwrap(),
353 PackageSourceMap::from_source_maps(source_maps, package_id_to_source_map_idx),
354 ))
355 }
356
357 pub fn push_dir(&mut self, path: impl AsRef<Path>) -> Result<(PackageId, PackageSourceMap)> {
391 self._push_dir(path.as_ref())
392 }
393
394 fn _push_dir(&mut self, path: &Path) -> Result<(PackageId, PackageSourceMap)> {
395 let top_pkg = UnresolvedPackageGroup::parse_dir(path)
396 .with_context(|| format!("failed to parse package: {}", path.display()))?;
397 let deps = path.join("deps");
398 let deps = self
399 .parse_deps_dir(&deps)
400 .with_context(|| format!("failed to parse dependency directory: {}", deps.display()))?;
401
402 self.sort_unresolved_packages(top_pkg, deps)
403 }
404
405 fn parse_deps_dir(&mut self, path: &Path) -> Result<Vec<UnresolvedPackageGroup>> {
406 let mut ret = Vec::new();
407 if !path.exists() {
408 return Ok(ret);
409 }
410 let mut entries = path
411 .read_dir()
412 .and_then(|i| i.collect::<std::io::Result<Vec<_>>>())
413 .context("failed to read directory")?;
414 entries.sort_by_key(|e| e.file_name());
415 for dep in entries {
416 let path = dep.path();
417 let pkg = if dep.file_type()?.is_dir() || path.metadata()?.is_dir() {
418 UnresolvedPackageGroup::parse_dir(&path)
423 .with_context(|| format!("failed to parse package: {}", path.display()))?
424 } else {
425 let filename = dep.file_name();
429 match Path::new(&filename).extension().and_then(|s| s.to_str()) {
430 Some("wit") | Some("wat") | Some("wasm") => match self._push_file(&path)? {
431 #[cfg(feature = "decoding")]
432 ParsedFile::Package(_) => continue,
433 ParsedFile::Unresolved(pkg) => pkg,
434 },
435
436 _ => continue,
440 }
441 };
442 ret.push(pkg);
443 }
444 Ok(ret)
445 }
446
447 pub fn push_file(&mut self, path: impl AsRef<Path>) -> Result<PackageId> {
462 match self._push_file(path.as_ref())? {
463 #[cfg(feature = "decoding")]
464 ParsedFile::Package(id) => Ok(id),
465 ParsedFile::Unresolved(pkg) => self.push_group(pkg),
466 }
467 }
468
469 fn _push_file(&mut self, path: &Path) -> Result<ParsedFile> {
470 let contents = std::fs::read(path)
471 .with_context(|| format!("failed to read path for WIT [{}]", path.display()))?;
472
473 #[cfg(feature = "decoding")]
476 {
477 use crate::decoding::{decode, DecodedWasm};
478
479 #[cfg(feature = "wat")]
480 let is_wasm = wat::Detect::from_bytes(&contents).is_wasm();
481 #[cfg(not(feature = "wat"))]
482 let is_wasm = wasmparser::Parser::is_component(&contents);
483
484 if is_wasm {
485 #[cfg(feature = "wat")]
486 let contents = wat::parse_bytes(&contents).map_err(|mut e| {
487 e.set_path(path);
488 e
489 })?;
490
491 match decode(&contents)? {
492 DecodedWasm::Component(..) => {
493 bail!("found an actual component instead of an encoded WIT package in wasm")
494 }
495 DecodedWasm::WitPackage(resolve, pkg) => {
496 let remap = self.merge(resolve)?;
497 return Ok(ParsedFile::Package(remap.packages[pkg.index()]));
498 }
499 }
500 }
501 }
502
503 let text = match std::str::from_utf8(&contents) {
505 Ok(s) => s,
506 Err(_) => bail!("input file is not valid utf-8 [{}]", path.display()),
507 };
508 let pkgs = UnresolvedPackageGroup::parse(path, text)?;
509 Ok(ParsedFile::Unresolved(pkgs))
510 }
511
512 pub fn push(
523 &mut self,
524 unresolved: UnresolvedPackage,
525 source_map: &SourceMap,
526 ) -> Result<PackageId> {
527 let ret = source_map.rewrite_error(|| Remap::default().append(self, unresolved));
528 if ret.is_ok() {
529 #[cfg(debug_assertions)]
530 self.assert_valid();
531 }
532 ret
533 }
534
535 pub fn push_group(&mut self, unresolved_group: UnresolvedPackageGroup) -> Result<PackageId> {
544 let (pkg_id, _) = self.sort_unresolved_packages(unresolved_group, Vec::new())?;
545 Ok(pkg_id)
546 }
547
548 pub fn push_str(&mut self, path: impl AsRef<Path>, contents: &str) -> Result<PackageId> {
555 self.push_group(UnresolvedPackageGroup::parse(path.as_ref(), contents)?)
556 }
557
558 pub fn all_bits_valid(&self, ty: &Type) -> bool {
559 match ty {
560 Type::U8
561 | Type::S8
562 | Type::U16
563 | Type::S16
564 | Type::U32
565 | Type::S32
566 | Type::U64
567 | Type::S64
568 | Type::F32
569 | Type::F64 => true,
570
571 Type::Bool | Type::Char | Type::String | Type::ErrorContext => false,
572
573 Type::Id(id) => match &self.types[*id].kind {
574 TypeDefKind::List(_)
575 | TypeDefKind::Variant(_)
576 | TypeDefKind::Enum(_)
577 | TypeDefKind::Option(_)
578 | TypeDefKind::Result(_)
579 | TypeDefKind::Future(_)
580 | TypeDefKind::Stream(_) => false,
581 TypeDefKind::Type(t) | TypeDefKind::FixedSizeList(t, ..) => self.all_bits_valid(t),
582
583 TypeDefKind::Handle(h) => match h {
584 crate::Handle::Own(_) => true,
585 crate::Handle::Borrow(_) => true,
586 },
587
588 TypeDefKind::Resource => false,
589 TypeDefKind::Record(r) => r.fields.iter().all(|f| self.all_bits_valid(&f.ty)),
590 TypeDefKind::Tuple(t) => t.types.iter().all(|t| self.all_bits_valid(t)),
591
592 TypeDefKind::Flags(_) => false,
596
597 TypeDefKind::Unknown => unreachable!(),
598 },
599 }
600 }
601
602 pub fn merge(&mut self, resolve: Resolve) -> Result<Remap> {
614 log::trace!(
615 "merging {} packages into {} packages",
616 resolve.packages.len(),
617 self.packages.len()
618 );
619
620 let mut map = MergeMap::new(&resolve, &self);
621 map.build()?;
622 let MergeMap {
623 package_map,
624 interface_map,
625 type_map,
626 world_map,
627 interfaces_to_add,
628 worlds_to_add,
629 ..
630 } = map;
631
632 let mut remap = Remap::default();
651 let Resolve {
652 types,
653 worlds,
654 interfaces,
655 packages,
656 package_names,
657 features: _,
658 ..
659 } = resolve;
660
661 let mut moved_types = Vec::new();
662 for (id, mut ty) in types {
663 let new_id = match type_map.get(&id).copied() {
664 Some(id) => {
665 update_stability(&ty.stability, &mut self.types[id].stability)?;
666 id
667 }
668 None => {
669 log::debug!("moving type {:?}", ty.name);
670 moved_types.push(id);
671 remap.update_typedef(self, &mut ty, None)?;
672 self.types.alloc(ty)
673 }
674 };
675 assert_eq!(remap.types.len(), id.index());
676 remap.types.push(Some(new_id));
677 }
678
679 let mut moved_interfaces = Vec::new();
680 for (id, mut iface) in interfaces {
681 let new_id = match interface_map.get(&id).copied() {
682 Some(id) => {
683 update_stability(&iface.stability, &mut self.interfaces[id].stability)?;
684 id
685 }
686 None => {
687 log::debug!("moving interface {:?}", iface.name);
688 moved_interfaces.push(id);
689 remap.update_interface(self, &mut iface, None)?;
690 self.interfaces.alloc(iface)
691 }
692 };
693 assert_eq!(remap.interfaces.len(), id.index());
694 remap.interfaces.push(Some(new_id));
695 }
696
697 let mut moved_worlds = Vec::new();
698 for (id, mut world) in worlds {
699 let new_id = match world_map.get(&id).copied() {
700 Some(world_id) => {
701 update_stability(&world.stability, &mut self.worlds[world_id].stability)?;
702 for from_import in world.imports.iter() {
703 Resolve::update_world_imports_stability(
704 from_import,
705 &mut self.worlds[world_id].imports,
706 &interface_map,
707 )?;
708 }
709 for from_export in world.exports.iter() {
710 Resolve::update_world_imports_stability(
711 from_export,
712 &mut self.worlds[world_id].exports,
713 &interface_map,
714 )?;
715 }
716 world_id
717 }
718 None => {
719 log::debug!("moving world {}", world.name);
720 moved_worlds.push(id);
721 let mut update = |map: &mut IndexMap<WorldKey, WorldItem>| -> Result<_> {
722 for (mut name, mut item) in mem::take(map) {
723 remap.update_world_key(&mut name, None)?;
724 match &mut item {
725 WorldItem::Function(f) => remap.update_function(self, f, None)?,
726 WorldItem::Interface { id, .. } => {
727 *id = remap.map_interface(*id, None)?
728 }
729 WorldItem::Type(i) => *i = remap.map_type(*i, None)?,
730 }
731 map.insert(name, item);
732 }
733 Ok(())
734 };
735 update(&mut world.imports)?;
736 update(&mut world.exports)?;
737 self.worlds.alloc(world)
738 }
739 };
740 assert_eq!(remap.worlds.len(), id.index());
741 remap.worlds.push(Some(new_id));
742 }
743
744 for (id, mut pkg) in packages {
745 let new_id = match package_map.get(&id).copied() {
746 Some(id) => id,
747 None => {
748 for (_, id) in pkg.interfaces.iter_mut() {
749 *id = remap.map_interface(*id, None)?;
750 }
751 for (_, id) in pkg.worlds.iter_mut() {
752 *id = remap.map_world(*id, None)?;
753 }
754 self.packages.alloc(pkg)
755 }
756 };
757 assert_eq!(remap.packages.len(), id.index());
758 remap.packages.push(new_id);
759 }
760
761 for (name, id) in package_names {
762 let id = remap.packages[id.index()];
763 if let Some(prev) = self.package_names.insert(name, id) {
764 assert_eq!(prev, id);
765 }
766 }
767
768 for id in moved_worlds {
776 let id = remap.map_world(id, None)?;
777 if let Some(pkg) = self.worlds[id].package.as_mut() {
778 *pkg = remap.packages[pkg.index()];
779 }
780 }
781 for id in moved_interfaces {
782 let id = remap.map_interface(id, None)?;
783 if let Some(pkg) = self.interfaces[id].package.as_mut() {
784 *pkg = remap.packages[pkg.index()];
785 }
786 }
787 for id in moved_types {
788 let id = remap.map_type(id, None)?;
789 match &mut self.types[id].owner {
790 TypeOwner::Interface(id) => *id = remap.map_interface(*id, None)?,
791 TypeOwner::World(id) => *id = remap.map_world(*id, None)?,
792 TypeOwner::None => {}
793 }
794 }
795
796 for (name, pkg, iface) in interfaces_to_add {
801 let prev = self.packages[pkg]
802 .interfaces
803 .insert(name, remap.map_interface(iface, None)?);
804 assert!(prev.is_none());
805 }
806 for (name, pkg, world) in worlds_to_add {
807 let prev = self.packages[pkg]
808 .worlds
809 .insert(name, remap.map_world(world, None)?);
810 assert!(prev.is_none());
811 }
812
813 log::trace!("now have {} packages", self.packages.len());
814
815 #[cfg(debug_assertions)]
816 self.assert_valid();
817 Ok(remap)
818 }
819
820 fn update_world_imports_stability(
821 from_item: (&WorldKey, &WorldItem),
822 into_items: &mut IndexMap<WorldKey, WorldItem>,
823 interface_map: &HashMap<Id<Interface>, Id<Interface>>,
824 ) -> Result<()> {
825 match from_item.0 {
826 WorldKey::Name(_) => {
827 Ok(())
829 }
830 key @ WorldKey::Interface(_) => {
831 let new_key = MergeMap::map_name(key, interface_map);
832 if let Some(into) = into_items.get_mut(&new_key) {
833 match (from_item.1, into) {
834 (
835 WorldItem::Interface {
836 id: aid,
837 stability: astability,
838 },
839 WorldItem::Interface {
840 id: bid,
841 stability: bstability,
842 },
843 ) => {
844 let aid = interface_map.get(aid).copied().unwrap_or(*aid);
845 assert_eq!(aid, *bid);
846 update_stability(astability, bstability)?;
847 Ok(())
848 }
849 _ => unreachable!(),
850 }
851 } else {
852 unreachable!()
855 }
856 }
857 }
858 }
859
860 pub fn merge_worlds(&mut self, from: WorldId, into: WorldId) -> Result<()> {
871 let mut new_imports = Vec::new();
872 let mut new_exports = Vec::new();
873
874 let from_world = &self.worlds[from];
875 let into_world = &self.worlds[into];
876
877 log::trace!("merging {} into {}", from_world.name, into_world.name);
878
879 for (name, from_import) in from_world.imports.iter() {
887 let name_str = self.name_world_key(name);
888 match into_world.imports.get(name) {
889 Some(into_import) => {
890 log::trace!("info/from shared import on `{name_str}`");
891 self.merge_world_item(from_import, into_import)
892 .with_context(|| format!("failed to merge world import {name_str}"))?;
893 }
894 None => {
895 log::trace!("new import: `{name_str}`");
896 new_imports.push((name.clone(), from_import.clone()));
897 }
898 }
899 }
900
901 let mut must_be_imported = HashMap::new();
908 for (key, export) in into_world.exports.iter() {
909 for dep in self.world_item_direct_deps(export) {
910 if into_world.exports.contains_key(&WorldKey::Interface(dep)) {
911 continue;
912 }
913 self.foreach_interface_dep(dep, &mut |id| {
914 must_be_imported.insert(id, key.clone());
915 });
916 }
917 }
918
919 for (name, from_export) in from_world.exports.iter() {
922 let name_str = self.name_world_key(name);
923 match into_world.exports.get(name) {
924 Some(into_export) => {
925 log::trace!("info/from shared export on `{name_str}`");
926 self.merge_world_item(from_export, into_export)
927 .with_context(|| format!("failed to merge world export {name_str}"))?;
928 }
929 None => {
930 log::trace!("new export `{name_str}`");
931 self.ensure_can_add_world_export(
934 into_world,
935 name,
936 from_export,
937 &must_be_imported,
938 )
939 .with_context(|| {
940 format!("failed to add export `{}`", self.name_world_key(name))
941 })?;
942 new_exports.push((name.clone(), from_export.clone()));
943 }
944 }
945 }
946
947 let mut cloner = clone::Cloner::new(self, TypeOwner::World(from), TypeOwner::World(into));
961 cloner.register_world_type_overlap(from, into);
962 for (name, item) in new_imports.iter_mut().chain(&mut new_exports) {
963 cloner.world_item(name, item);
964 }
965
966 let into_world = &mut self.worlds[into];
968 for (name, import) in new_imports {
969 let prev = into_world.imports.insert(name, import);
970 assert!(prev.is_none());
971 }
972 for (name, export) in new_exports {
973 let prev = into_world.exports.insert(name, export);
974 assert!(prev.is_none());
975 }
976
977 #[cfg(debug_assertions)]
978 self.assert_valid();
979 Ok(())
980 }
981
982 fn merge_world_item(&self, from: &WorldItem, into: &WorldItem) -> Result<()> {
983 let mut map = MergeMap::new(self, self);
984 match (from, into) {
985 (WorldItem::Interface { id: from, .. }, WorldItem::Interface { id: into, .. }) => {
986 if from == into {
991 return Ok(());
992 }
993
994 map.build_interface(*from, *into)
1004 .context("failed to merge interfaces")?;
1005 }
1006
1007 (WorldItem::Function(from), WorldItem::Function(into)) => {
1010 map.build_function(from, into)
1011 .context("failed to merge functions")?;
1012 }
1013 (WorldItem::Type(from), WorldItem::Type(into)) => {
1014 map.build_type_id(*from, *into)
1015 .context("failed to merge types")?;
1016 }
1017
1018 (WorldItem::Interface { .. }, _)
1020 | (WorldItem::Function { .. }, _)
1021 | (WorldItem::Type { .. }, _) => {
1022 bail!("different kinds of items");
1023 }
1024 }
1025 assert!(map.interfaces_to_add.is_empty());
1026 assert!(map.worlds_to_add.is_empty());
1027 Ok(())
1028 }
1029
1030 fn ensure_can_add_world_export(
1042 &self,
1043 into: &World,
1044 name: &WorldKey,
1045 item: &WorldItem,
1046 must_be_imported: &HashMap<InterfaceId, WorldKey>,
1047 ) -> Result<()> {
1048 assert!(!into.exports.contains_key(name));
1049 let name = self.name_world_key(name);
1050
1051 for dep in self.world_item_direct_deps(item) {
1055 if into.exports.contains_key(&WorldKey::Interface(dep)) {
1056 continue;
1057 }
1058 self.ensure_not_exported(into, dep)
1059 .with_context(|| format!("failed validating export of `{name}`"))?;
1060 }
1061
1062 if let WorldItem::Interface { id, .. } = item {
1066 if let Some(export) = must_be_imported.get(&id) {
1067 let export_name = self.name_world_key(export);
1068 bail!(
1069 "export `{export_name}` depends on `{name}` \
1070 previously as an import which will change meaning \
1071 if `{name}` is added as an export"
1072 );
1073 }
1074 }
1075
1076 Ok(())
1077 }
1078
1079 fn ensure_not_exported(&self, world: &World, id: InterfaceId) -> Result<()> {
1080 let key = WorldKey::Interface(id);
1081 let name = self.name_world_key(&key);
1082 if world.exports.contains_key(&key) {
1083 bail!(
1084 "world exports `{name}` but it's also transitively used by an \
1085 import \
1086 which means that this is not valid"
1087 )
1088 }
1089 for dep in self.interface_direct_deps(id) {
1090 self.ensure_not_exported(world, dep)
1091 .with_context(|| format!("failed validating transitive import dep `{name}`"))?;
1092 }
1093 Ok(())
1094 }
1095
1096 fn world_item_direct_deps(&self, item: &WorldItem) -> impl Iterator<Item = InterfaceId> + '_ {
1102 let mut interface = None;
1103 let mut ty = None;
1104 match item {
1105 WorldItem::Function(_) => {}
1106 WorldItem::Type(id) => ty = Some(*id),
1107 WorldItem::Interface { id, .. } => interface = Some(*id),
1108 }
1109
1110 interface
1111 .into_iter()
1112 .flat_map(move |id| self.interface_direct_deps(id))
1113 .chain(ty.and_then(|t| self.type_interface_dep(t)))
1114 }
1115
1116 fn foreach_interface_dep(&self, id: InterfaceId, f: &mut dyn FnMut(InterfaceId)) {
1120 self._foreach_interface_dep(id, f, &mut HashSet::new())
1121 }
1122
1123 fn _foreach_interface_dep(
1127 &self,
1128 id: InterfaceId,
1129 f: &mut dyn FnMut(InterfaceId),
1130 visited: &mut HashSet<InterfaceId>,
1131 ) {
1132 if !visited.insert(id) {
1133 return;
1134 }
1135 f(id);
1136 for dep in self.interface_direct_deps(id) {
1137 self._foreach_interface_dep(dep, f, visited);
1138 }
1139 }
1140
1141 pub fn id_of(&self, interface: InterfaceId) -> Option<String> {
1145 let interface = &self.interfaces[interface];
1146 Some(self.id_of_name(interface.package.unwrap(), interface.name.as_ref()?))
1147 }
1148
1149 pub fn canonicalized_id_of(&self, interface: InterfaceId) -> Option<String> {
1154 let interface = &self.interfaces[interface];
1155 Some(self.canonicalized_id_of_name(interface.package.unwrap(), interface.name.as_ref()?))
1156 }
1157
1158 pub fn importize(&mut self, world_id: WorldId, out_world_name: Option<String>) -> Result<()> {
1174 let world = &mut self.worlds[world_id];
1179 let pkg = &mut self.packages[world.package.unwrap()];
1180 pkg.worlds.shift_remove(&world.name);
1181 if let Some(name) = out_world_name {
1182 world.name = name.clone();
1183 pkg.worlds.insert(name, world_id);
1184 } else {
1185 world.name.push_str("-importized");
1186 pkg.worlds.insert(world.name.clone(), world_id);
1187 }
1188
1189 world.imports.retain(|_, item| match item {
1192 WorldItem::Type(_) => true,
1193 _ => false,
1194 });
1195
1196 for (name, export) in mem::take(&mut world.exports) {
1197 match (name.clone(), world.imports.insert(name, export)) {
1198 (_, None) => {}
1200
1201 (WorldKey::Name(name), Some(_)) => {
1203 bail!("world export `{name}` conflicts with import of same name");
1204 }
1205
1206 (WorldKey::Interface(_), _) => unreachable!(),
1209 }
1210 }
1211
1212 self.elaborate_world(world_id)?;
1215
1216 #[cfg(debug_assertions)]
1217 self.assert_valid();
1218 Ok(())
1219 }
1220
1221 pub fn id_of_name(&self, pkg: PackageId, name: &str) -> String {
1223 let package = &self.packages[pkg];
1224 let mut base = String::new();
1225 base.push_str(&package.name.namespace);
1226 base.push_str(":");
1227 base.push_str(&package.name.name);
1228 base.push_str("/");
1229 base.push_str(name);
1230 if let Some(version) = &package.name.version {
1231 base.push_str(&format!("@{version}"));
1232 }
1233 base
1234 }
1235
1236 pub fn canonicalized_id_of_name(&self, pkg: PackageId, name: &str) -> String {
1242 let package = &self.packages[pkg];
1243 let mut base = String::new();
1244 base.push_str(&package.name.namespace);
1245 base.push_str(":");
1246 base.push_str(&package.name.name);
1247 base.push_str("/");
1248 base.push_str(name);
1249 if let Some(version) = &package.name.version {
1250 base.push_str("@");
1251 let string = PackageName::version_compat_track_string(version);
1252 base.push_str(&string);
1253 }
1254 base
1255 }
1256
1257 pub fn select_world(&self, package: PackageId, world: Option<&str>) -> Result<WorldId> {
1368 let world_path = match world {
1369 Some(world) => Some(
1370 parse_use_path(world)
1371 .with_context(|| format!("failed to parse world specifier `{world}`"))?,
1372 ),
1373 None => None,
1374 };
1375
1376 let (pkg, world_name) = match world_path {
1377 Some(ParsedUsePath::Name(name)) => (package, name),
1378 Some(ParsedUsePath::Package(pkg, interface)) => {
1379 let pkg = match self.package_names.get(&pkg) {
1380 Some(pkg) => *pkg,
1381 None => {
1382 let mut candidates = self.package_names.iter().filter(|(name, _)| {
1383 pkg.version.is_none()
1384 && pkg.name == name.name
1385 && pkg.namespace == name.namespace
1386 && name.version.is_some()
1387 });
1388 let candidate = candidates.next();
1389 if let Some((c2, _)) = candidates.next() {
1390 let (c1, _) = candidate.unwrap();
1391 bail!(
1392 "package name `{pkg}` is available at both \
1393 versions {} and {} but which is not specified",
1394 c1.version.as_ref().unwrap(),
1395 c2.version.as_ref().unwrap(),
1396 );
1397 }
1398 match candidate {
1399 Some((_, id)) => *id,
1400 None => bail!("unknown package `{pkg}`"),
1401 }
1402 }
1403 };
1404 (pkg, interface.to_string())
1405 }
1406 None => {
1407 let pkg = &self.packages[package];
1408 let worlds = pkg
1409 .worlds
1410 .values()
1411 .map(|world| (package, *world))
1412 .collect::<Vec<_>>();
1413
1414 match &worlds[..] {
1415 [] => bail!("The main package `{}` contains no worlds", pkg.name),
1416 [(_, world)] => return Ok(*world),
1417 _ => bail!(
1418 "multiple worlds found; one must be explicitly chosen:{}",
1419 worlds
1420 .iter()
1421 .map(|(pkg, world)| format!(
1422 "\n {}/{}",
1423 self.packages[*pkg].name, self.worlds[*world].name
1424 ))
1425 .collect::<String>()
1426 ),
1427 }
1428 }
1429 };
1430 let pkg = &self.packages[pkg];
1431 pkg.worlds
1432 .get(&world_name)
1433 .copied()
1434 .ok_or_else(|| anyhow!("no world named `{world_name}` in package"))
1435 }
1436
1437 pub fn name_world_key(&self, key: &WorldKey) -> String {
1439 match key {
1440 WorldKey::Name(s) => s.to_string(),
1441 WorldKey::Interface(i) => self.id_of(*i).expect("unexpected anonymous interface"),
1442 }
1443 }
1444
1445 pub fn name_canonicalized_world_key(&self, key: &WorldKey) -> String {
1448 match key {
1449 WorldKey::Name(s) => s.to_string(),
1450 WorldKey::Interface(i) => self
1451 .canonicalized_id_of(*i)
1452 .expect("unexpected anonymous interface"),
1453 }
1454 }
1455
1456 pub fn type_interface_dep(&self, id: TypeId) -> Option<InterfaceId> {
1462 let ty = &self.types[id];
1463 let dep = match ty.kind {
1464 TypeDefKind::Type(Type::Id(id)) => id,
1465 _ => return None,
1466 };
1467 let other = &self.types[dep];
1468 if ty.owner == other.owner {
1469 None
1470 } else {
1471 match other.owner {
1472 TypeOwner::Interface(id) => Some(id),
1473 _ => unreachable!(),
1474 }
1475 }
1476 }
1477
1478 pub fn interface_direct_deps(&self, id: InterfaceId) -> impl Iterator<Item = InterfaceId> + '_ {
1488 self.interfaces[id]
1489 .types
1490 .iter()
1491 .filter_map(move |(_name, ty)| self.type_interface_dep(*ty))
1492 }
1493
1494 pub fn package_direct_deps(&self, id: PackageId) -> impl Iterator<Item = PackageId> + '_ {
1504 let pkg = &self.packages[id];
1505
1506 pkg.interfaces
1507 .iter()
1508 .flat_map(move |(_name, id)| self.interface_direct_deps(*id))
1509 .chain(pkg.worlds.iter().flat_map(move |(_name, id)| {
1510 let world = &self.worlds[*id];
1511 world
1512 .imports
1513 .iter()
1514 .chain(world.exports.iter())
1515 .filter_map(move |(_name, item)| match item {
1516 WorldItem::Interface { id, .. } => Some(*id),
1517 WorldItem::Function(_) => None,
1518 WorldItem::Type(t) => self.type_interface_dep(*t),
1519 })
1520 }))
1521 .filter_map(move |iface_id| {
1522 let pkg = self.interfaces[iface_id].package?;
1523 if pkg == id {
1524 None
1525 } else {
1526 Some(pkg)
1527 }
1528 })
1529 }
1530
1531 pub fn topological_packages(&self) -> Vec<PackageId> {
1537 let mut pushed = vec![false; self.packages.len()];
1538 let mut order = Vec::new();
1539 for (id, _) in self.packages.iter() {
1540 self.build_topological_package_ordering(id, &mut pushed, &mut order);
1541 }
1542 order
1543 }
1544
1545 fn build_topological_package_ordering(
1546 &self,
1547 id: PackageId,
1548 pushed: &mut Vec<bool>,
1549 order: &mut Vec<PackageId>,
1550 ) {
1551 if pushed[id.index()] {
1552 return;
1553 }
1554 for dep in self.package_direct_deps(id) {
1555 self.build_topological_package_ordering(dep, pushed, order);
1556 }
1557 order.push(id);
1558 pushed[id.index()] = true;
1559 }
1560
1561 #[doc(hidden)]
1562 pub fn assert_valid(&self) {
1563 let mut package_interfaces = Vec::new();
1564 let mut package_worlds = Vec::new();
1565 for (id, pkg) in self.packages.iter() {
1566 let mut interfaces = HashSet::new();
1567 for (name, iface) in pkg.interfaces.iter() {
1568 assert!(interfaces.insert(*iface));
1569 let iface = &self.interfaces[*iface];
1570 assert_eq!(name, iface.name.as_ref().unwrap());
1571 assert_eq!(iface.package.unwrap(), id);
1572 }
1573 package_interfaces.push(pkg.interfaces.values().copied().collect::<HashSet<_>>());
1574 let mut worlds = HashSet::new();
1575 for (name, world) in pkg.worlds.iter() {
1576 assert!(worlds.insert(*world));
1577 assert_eq!(
1578 pkg.worlds.get_key_value(name),
1579 Some((name, world)),
1580 "`MutableKeys` impl may have been used to change a key's hash or equality"
1581 );
1582 let world = &self.worlds[*world];
1583 assert_eq!(*name, world.name);
1584 assert_eq!(world.package.unwrap(), id);
1585 }
1586 package_worlds.push(pkg.worlds.values().copied().collect::<HashSet<_>>());
1587 }
1588
1589 let mut interface_types = Vec::new();
1590 for (id, iface) in self.interfaces.iter() {
1591 assert!(self.packages.get(iface.package.unwrap()).is_some());
1592 if iface.name.is_some() {
1593 assert!(package_interfaces[iface.package.unwrap().index()].contains(&id));
1594 }
1595
1596 for (name, ty) in iface.types.iter() {
1597 let ty = &self.types[*ty];
1598 assert_eq!(ty.name.as_ref(), Some(name));
1599 assert_eq!(ty.owner, TypeOwner::Interface(id));
1600 }
1601 interface_types.push(iface.types.values().copied().collect::<HashSet<_>>());
1602 for (name, f) in iface.functions.iter() {
1603 assert_eq!(*name, f.name);
1604 }
1605 }
1606
1607 let mut world_types = Vec::new();
1608 for (id, world) in self.worlds.iter() {
1609 log::debug!("validating world {}", &world.name);
1610 if let Some(package) = world.package {
1611 assert!(self.packages.get(package).is_some());
1612 assert!(package_worlds[package.index()].contains(&id));
1613 }
1614 assert!(world.includes.is_empty());
1615
1616 let mut types = HashSet::new();
1617 for (name, item) in world.imports.iter().chain(world.exports.iter()) {
1618 log::debug!("validating world item: {}", self.name_world_key(name));
1619 match item {
1620 WorldItem::Interface { id, .. } => {
1621 if matches!(name, WorldKey::Name(_)) {
1624 assert_eq!(self.interfaces[*id].package, world.package);
1625 }
1626 }
1627 WorldItem::Function(f) => {
1628 assert!(!matches!(name, WorldKey::Interface(_)));
1629 assert_eq!(f.name, name.clone().unwrap_name());
1630 }
1631 WorldItem::Type(ty) => {
1632 assert!(!matches!(name, WorldKey::Interface(_)));
1633 assert!(types.insert(*ty));
1634 let ty = &self.types[*ty];
1635 assert_eq!(ty.name, Some(name.clone().unwrap_name()));
1636 assert_eq!(ty.owner, TypeOwner::World(id));
1637 }
1638 }
1639 }
1640 self.assert_world_elaborated(world);
1641 world_types.push(types);
1642 }
1643
1644 for (ty_id, ty) in self.types.iter() {
1645 match ty.owner {
1646 TypeOwner::Interface(id) => {
1647 assert!(self.interfaces.get(id).is_some());
1648 assert!(interface_types[id.index()].contains(&ty_id));
1649 }
1650 TypeOwner::World(id) => {
1651 assert!(self.worlds.get(id).is_some());
1652 assert!(world_types[id.index()].contains(&ty_id));
1653 }
1654 TypeOwner::None => {}
1655 }
1656 }
1657
1658 self.assert_topologically_sorted();
1659 }
1660
1661 fn assert_topologically_sorted(&self) {
1662 let mut positions = IndexMap::new();
1663 for id in self.topological_packages() {
1664 let pkg = &self.packages[id];
1665 log::debug!("pkg {}", pkg.name);
1666 let prev = positions.insert(Some(id), IndexSet::new());
1667 assert!(prev.is_none());
1668 }
1669 positions.insert(None, IndexSet::new());
1670
1671 for (id, iface) in self.interfaces.iter() {
1672 log::debug!("iface {:?}", iface.name);
1673 let ok = positions.get_mut(&iface.package).unwrap().insert(id);
1674 assert!(ok);
1675 }
1676
1677 for (_, world) in self.worlds.iter() {
1678 log::debug!("world {:?}", world.name);
1679
1680 let my_package = world.package;
1681 let my_package_pos = positions.get_index_of(&my_package).unwrap();
1682
1683 for (_, item) in world.imports.iter().chain(&world.exports) {
1684 let id = match item {
1685 WorldItem::Interface { id, .. } => *id,
1686 _ => continue,
1687 };
1688 let other_package = self.interfaces[id].package;
1689 let other_package_pos = positions.get_index_of(&other_package).unwrap();
1690
1691 assert!(other_package_pos <= my_package_pos);
1692 }
1693 }
1694
1695 for (_id, ty) in self.types.iter() {
1696 log::debug!("type {:?} {:?}", ty.name, ty.owner);
1697 let other_id = match ty.kind {
1698 TypeDefKind::Type(Type::Id(ty)) => ty,
1699 _ => continue,
1700 };
1701 let other = &self.types[other_id];
1702 if ty.kind == other.kind {
1703 continue;
1704 }
1705 let my_interface = match ty.owner {
1706 TypeOwner::Interface(id) => id,
1707 _ => continue,
1708 };
1709 let other_interface = match other.owner {
1710 TypeOwner::Interface(id) => id,
1711 _ => continue,
1712 };
1713
1714 let my_package = self.interfaces[my_interface].package;
1715 let other_package = self.interfaces[other_interface].package;
1716 let my_package_pos = positions.get_index_of(&my_package).unwrap();
1717 let other_package_pos = positions.get_index_of(&other_package).unwrap();
1718
1719 if my_package_pos == other_package_pos {
1720 let interfaces = &positions[&my_package];
1721 let my_interface_pos = interfaces.get_index_of(&my_interface).unwrap();
1722 let other_interface_pos = interfaces.get_index_of(&other_interface).unwrap();
1723 assert!(other_interface_pos <= my_interface_pos);
1724 } else {
1725 assert!(other_package_pos < my_package_pos);
1726 }
1727 }
1728 }
1729
1730 fn assert_world_elaborated(&self, world: &World) {
1731 for (key, item) in world.imports.iter() {
1732 log::debug!(
1733 "asserting elaborated world import {}",
1734 self.name_world_key(key)
1735 );
1736 match item {
1737 WorldItem::Type(t) => self.assert_world_imports_type_deps(world, key, *t),
1738
1739 WorldItem::Function(f) => self.assert_world_function_imports_types(world, key, f),
1741
1742 WorldItem::Interface { id, .. } => {
1744 for dep in self.interface_direct_deps(*id) {
1745 assert!(
1746 world.imports.contains_key(&WorldKey::Interface(dep)),
1747 "world import of {} is missing transitive dep of {}",
1748 self.name_world_key(key),
1749 self.id_of(dep).unwrap(),
1750 );
1751 }
1752 }
1753 }
1754 }
1755 for (key, item) in world.exports.iter() {
1756 log::debug!(
1757 "asserting elaborated world export {}",
1758 self.name_world_key(key)
1759 );
1760 match item {
1761 WorldItem::Function(f) => self.assert_world_function_imports_types(world, key, f),
1763
1764 WorldItem::Interface { id, .. } => {
1768 for dep in self.interface_direct_deps(*id) {
1769 let dep_key = WorldKey::Interface(dep);
1770 if world.exports.contains_key(&dep_key) {
1771 continue;
1772 }
1773 self.foreach_interface_dep(dep, &mut |dep| {
1774 let dep_key = WorldKey::Interface(dep);
1775 assert!(
1776 world.imports.contains_key(&dep_key),
1777 "world should import {} (required by {})",
1778 self.name_world_key(&dep_key),
1779 self.name_world_key(key),
1780 );
1781 assert!(
1782 !world.exports.contains_key(&dep_key),
1783 "world should not export {} (required by {})",
1784 self.name_world_key(&dep_key),
1785 self.name_world_key(key),
1786 );
1787 });
1788 }
1789 }
1790
1791 WorldItem::Type(_) => unreachable!(),
1793 }
1794 }
1795 }
1796
1797 fn assert_world_imports_type_deps(&self, world: &World, key: &WorldKey, ty: TypeId) {
1798 let ty = &self.types[ty];
1801 if let TypeDefKind::Type(Type::Id(other)) = ty.kind {
1802 if let TypeOwner::Interface(id) = self.types[other].owner {
1803 let key = WorldKey::Interface(id);
1804 assert!(world.imports.contains_key(&key));
1805 return;
1806 }
1807 }
1808
1809 let mut visitor = MyVisit(self, Vec::new());
1813 visitor.visit_type_def(self, ty);
1814 for ty in visitor.1 {
1815 let ty = &self.types[ty];
1816 let Some(name) = ty.name.clone() else {
1817 continue;
1818 };
1819 let dep_key = WorldKey::Name(name);
1820 assert!(
1821 world.imports.contains_key(&dep_key),
1822 "world import `{}` should also force an import of `{}`",
1823 self.name_world_key(key),
1824 self.name_world_key(&dep_key),
1825 );
1826 }
1827
1828 struct MyVisit<'a>(&'a Resolve, Vec<TypeId>);
1829
1830 impl TypeIdVisitor for MyVisit<'_> {
1831 fn before_visit_type_id(&mut self, id: TypeId) -> bool {
1832 self.1.push(id);
1833 self.0.types[id].name.is_none()
1835 }
1836 }
1837 }
1838
1839 fn assert_world_function_imports_types(&self, world: &World, key: &WorldKey, func: &Function) {
1843 for ty in func
1844 .parameter_and_result_types()
1845 .chain(func.kind.resource().map(Type::Id))
1846 {
1847 let Type::Id(id) = ty else {
1848 continue;
1849 };
1850 self.assert_world_imports_type_deps(world, key, id);
1851 }
1852 }
1853
1854 fn include_stability(
1870 &self,
1871 stability: &Stability,
1872 pkg_id: &PackageId,
1873 span: Option<Span>,
1874 ) -> Result<bool> {
1875 let err = |msg: String| match span {
1876 Some(span) => Error::new(span, msg).into(),
1877 None => anyhow::Error::msg(msg),
1878 };
1879 Ok(match stability {
1880 Stability::Unknown => true,
1881 Stability::Stable { since, .. } => {
1884 let Some(p) = self.packages.get(*pkg_id) else {
1885 return Ok(true);
1893 };
1894
1895 let package_version = p.name.version.as_ref().ok_or_else(|| {
1898 err(format!(
1899 "package [{}] contains a feature gate with a version \
1900 specifier, so it must have a version",
1901 p.name
1902 ))
1903 })?;
1904
1905 if since > package_version {
1909 return Err(err(format!(
1910 "feature gate cannot reference unreleased version \
1911 {since} of package [{}] (current version {package_version})",
1912 p.name
1913 )));
1914 }
1915
1916 true
1917 }
1918 Stability::Unstable { feature, .. } => {
1919 self.features.contains(feature) || self.all_features
1920 }
1921 })
1922 }
1923
1924 fn include_type(&self, ty: &TypeDef, pkgid: PackageId, span: Span) -> Result<bool> {
1927 self.include_stability(&ty.stability, &pkgid, Some(span))
1928 .with_context(|| {
1929 format!(
1930 "failed to process feature gate for type [{}] in package [{}]",
1931 ty.name.as_ref().map(String::as_str).unwrap_or("<unknown>"),
1932 self.packages[pkgid].name,
1933 )
1934 })
1935 }
1936
1937 fn elaborate_world(&mut self, world_id: WorldId) -> Result<()> {
1949 let mut new_imports = IndexMap::new();
1953 let world = &self.worlds[world_id];
1954
1955 let sort_key = |resolve: &Resolve, item: &WorldItem| match item {
1978 WorldItem::Interface { .. } => 0,
1979 WorldItem::Type(ty) => {
1980 let ty = &resolve.types[*ty];
1981 match ty.kind {
1982 TypeDefKind::Type(Type::Id(t)) if resolve.types[t].owner != ty.owner => 1,
1983 _ => 2,
1984 }
1985 }
1986 WorldItem::Function(f) => {
1987 if f.kind.resource().is_none() {
1988 3
1989 } else {
1990 4
1991 }
1992 }
1993 };
1994
1995 let mut world_imports = world.imports.iter().collect::<Vec<_>>();
1998 world_imports.sort_by_key(|(_name, import)| sort_key(self, import));
1999 for (name, item) in world_imports {
2000 match item {
2001 WorldItem::Interface { id, stability } => {
2004 self.elaborate_world_import(&mut new_imports, name.clone(), *id, &stability);
2005 }
2006
2007 WorldItem::Function(_) => {
2010 let prev = new_imports.insert(name.clone(), item.clone());
2011 assert!(prev.is_none());
2012 }
2013
2014 WorldItem::Type(id) => {
2018 if let Some(dep) = self.type_interface_dep(*id) {
2019 self.elaborate_world_import(
2020 &mut new_imports,
2021 WorldKey::Interface(dep),
2022 dep,
2023 &self.types[*id].stability,
2024 );
2025 }
2026 let prev = new_imports.insert(name.clone(), item.clone());
2027 assert!(prev.is_none());
2028 }
2029 }
2030 }
2031
2032 let mut new_exports = IndexMap::new();
2038 let mut export_interfaces = IndexMap::new();
2039 for (name, item) in world.exports.iter() {
2040 match item {
2041 WorldItem::Interface { id, stability } => {
2042 let prev = export_interfaces.insert(*id, (name.clone(), stability));
2043 assert!(prev.is_none());
2044 }
2045 WorldItem::Function(_) => {
2046 let prev = new_exports.insert(name.clone(), item.clone());
2047 assert!(prev.is_none());
2048 }
2049 WorldItem::Type(_) => unreachable!(),
2050 }
2051 }
2052
2053 self.elaborate_world_exports(&export_interfaces, &mut new_imports, &mut new_exports)?;
2054
2055 new_imports.sort_by_cached_key(|_name, import| sort_key(self, import));
2059
2060 log::trace!("imports = {:?}", new_imports);
2063 log::trace!("exports = {:?}", new_exports);
2064 let world = &mut self.worlds[world_id];
2065 world.imports = new_imports;
2066 world.exports = new_exports;
2067
2068 Ok(())
2069 }
2070
2071 fn elaborate_world_import(
2072 &self,
2073 imports: &mut IndexMap<WorldKey, WorldItem>,
2074 key: WorldKey,
2075 id: InterfaceId,
2076 stability: &Stability,
2077 ) {
2078 if imports.contains_key(&key) {
2079 return;
2080 }
2081 for dep in self.interface_direct_deps(id) {
2082 self.elaborate_world_import(imports, WorldKey::Interface(dep), dep, stability);
2083 }
2084 let prev = imports.insert(
2085 key,
2086 WorldItem::Interface {
2087 id,
2088 stability: stability.clone(),
2089 },
2090 );
2091 assert!(prev.is_none());
2092 }
2093
2094 fn elaborate_world_exports(
2141 &self,
2142 export_interfaces: &IndexMap<InterfaceId, (WorldKey, &Stability)>,
2143 imports: &mut IndexMap<WorldKey, WorldItem>,
2144 exports: &mut IndexMap<WorldKey, WorldItem>,
2145 ) -> Result<()> {
2146 let mut required_imports = HashSet::new();
2147 for (id, (key, stability)) in export_interfaces.iter() {
2148 let name = self.name_world_key(&key);
2149 let ok = add_world_export(
2150 self,
2151 imports,
2152 exports,
2153 export_interfaces,
2154 &mut required_imports,
2155 *id,
2156 key,
2157 true,
2158 stability,
2159 );
2160 if !ok {
2161 bail!(
2162 InvalidTransitiveDependency(name),
2180 );
2181 }
2182 }
2183 return Ok(());
2184
2185 fn add_world_export(
2186 resolve: &Resolve,
2187 imports: &mut IndexMap<WorldKey, WorldItem>,
2188 exports: &mut IndexMap<WorldKey, WorldItem>,
2189 export_interfaces: &IndexMap<InterfaceId, (WorldKey, &Stability)>,
2190 required_imports: &mut HashSet<InterfaceId>,
2191 id: InterfaceId,
2192 key: &WorldKey,
2193 add_export: bool,
2194 stability: &Stability,
2195 ) -> bool {
2196 if exports.contains_key(key) {
2197 if add_export {
2198 return true;
2199 } else {
2200 return false;
2201 }
2202 }
2203 if !add_export && required_imports.contains(&id) {
2206 return true;
2207 }
2208 let ok = resolve.interface_direct_deps(id).all(|dep| {
2209 let key = WorldKey::Interface(dep);
2210 let add_export = add_export && export_interfaces.contains_key(&dep);
2211 add_world_export(
2212 resolve,
2213 imports,
2214 exports,
2215 export_interfaces,
2216 required_imports,
2217 dep,
2218 &key,
2219 add_export,
2220 stability,
2221 )
2222 });
2223 if !ok {
2224 return false;
2225 }
2226 let item = WorldItem::Interface {
2227 id,
2228 stability: stability.clone(),
2229 };
2230 if add_export {
2231 if required_imports.contains(&id) {
2232 return false;
2233 }
2234 exports.insert(key.clone(), item);
2235 } else {
2236 required_imports.insert(id);
2237 imports.insert(key.clone(), item);
2238 }
2239 true
2240 }
2241 }
2242
2243 pub fn merge_world_imports_based_on_semver(&mut self, world_id: WorldId) -> Result<()> {
2255 let world = &self.worlds[world_id];
2256
2257 let mut semver_tracks = HashMap::new();
2266 let mut to_remove = HashSet::new();
2267 for (key, _) in world.imports.iter() {
2268 let iface_id = match key {
2269 WorldKey::Interface(id) => *id,
2270 WorldKey::Name(_) => continue,
2271 };
2272 let (track, version) = match self.semver_track(iface_id) {
2273 Some(track) => track,
2274 None => continue,
2275 };
2276 log::debug!(
2277 "{} is on track {}/{}",
2278 self.id_of(iface_id).unwrap(),
2279 track.0,
2280 track.1,
2281 );
2282 match semver_tracks.entry(track.clone()) {
2283 hash_map::Entry::Vacant(e) => {
2284 e.insert((version, iface_id));
2285 }
2286 hash_map::Entry::Occupied(mut e) => match version.cmp(&e.get().0) {
2287 Ordering::Greater => {
2288 to_remove.insert(e.get().1);
2289 e.insert((version, iface_id));
2290 }
2291 Ordering::Equal => {}
2292 Ordering::Less => {
2293 to_remove.insert(iface_id);
2294 }
2295 },
2296 }
2297 }
2298
2299 let mut replacements = HashMap::new();
2302 for id in to_remove {
2303 let (track, _) = self.semver_track(id).unwrap();
2304 let (_, latest) = semver_tracks[&track];
2305 let prev = replacements.insert(id, latest);
2306 assert!(prev.is_none());
2307 }
2308
2309 for (to_replace, replace_with) in replacements.iter() {
2314 self.merge_world_item(
2315 &WorldItem::Interface {
2316 id: *to_replace,
2317 stability: Default::default(),
2318 },
2319 &WorldItem::Interface {
2320 id: *replace_with,
2321 stability: Default::default(),
2322 },
2323 )
2324 .with_context(|| {
2325 let old_name = self.id_of(*to_replace).unwrap();
2326 let new_name = self.id_of(*replace_with).unwrap();
2327 format!(
2328 "failed to upgrade `{old_name}` to `{new_name}`, was \
2329 this semver-compatible update not semver compatible?"
2330 )
2331 })?;
2332 }
2333
2334 for (to_replace, replace_with) in replacements.iter() {
2335 log::debug!(
2336 "REPLACE {} => {}",
2337 self.id_of(*to_replace).unwrap(),
2338 self.id_of(*replace_with).unwrap(),
2339 );
2340 }
2341
2342 for (key, item) in mem::take(&mut self.worlds[world_id].imports) {
2351 if let WorldItem::Interface { id, .. } = item {
2352 if replacements.contains_key(&id) {
2353 continue;
2354 }
2355 }
2356
2357 self.update_interface_deps_of_world_item(&item, &replacements);
2358
2359 let prev = self.worlds[world_id].imports.insert(key, item);
2360 assert!(prev.is_none());
2361 }
2362 for (key, item) in mem::take(&mut self.worlds[world_id].exports) {
2363 self.update_interface_deps_of_world_item(&item, &replacements);
2364 let prev = self.worlds[world_id].exports.insert(key, item);
2365 assert!(prev.is_none());
2366 }
2367
2368 let ids = self.worlds.iter().map(|(id, _)| id).collect::<Vec<_>>();
2374 for world_id in ids {
2375 self.elaborate_world(world_id).with_context(|| {
2376 let name = &self.worlds[world_id].name;
2377 format!(
2378 "failed to elaborate world `{name}` after deduplicating imports \
2379 based on semver"
2380 )
2381 })?;
2382 }
2383
2384 #[cfg(debug_assertions)]
2385 self.assert_valid();
2386
2387 Ok(())
2388 }
2389
2390 fn update_interface_deps_of_world_item(
2391 &mut self,
2392 item: &WorldItem,
2393 replacements: &HashMap<InterfaceId, InterfaceId>,
2394 ) {
2395 match *item {
2396 WorldItem::Type(t) => self.update_interface_dep_of_type(t, &replacements),
2397 WorldItem::Interface { id, .. } => {
2398 let types = self.interfaces[id]
2399 .types
2400 .values()
2401 .copied()
2402 .collect::<Vec<_>>();
2403 for ty in types {
2404 self.update_interface_dep_of_type(ty, &replacements);
2405 }
2406 }
2407 WorldItem::Function(_) => {}
2408 }
2409 }
2410
2411 fn semver_track(&self, id: InterfaceId) -> Option<((PackageName, String), &Version)> {
2422 let iface = &self.interfaces[id];
2423 let pkg = &self.packages[iface.package?];
2424 let version = pkg.name.version.as_ref()?;
2425 let mut name = pkg.name.clone();
2426 name.version = Some(PackageName::version_compat_track(version));
2427 Some(((name, iface.name.clone()?), version))
2428 }
2429
2430 fn update_interface_dep_of_type(
2434 &mut self,
2435 ty: TypeId,
2436 replacements: &HashMap<InterfaceId, InterfaceId>,
2437 ) {
2438 let to_replace = match self.type_interface_dep(ty) {
2439 Some(id) => id,
2440 None => return,
2441 };
2442 let replace_with = match replacements.get(&to_replace) {
2443 Some(id) => id,
2444 None => return,
2445 };
2446 let dep = match self.types[ty].kind {
2447 TypeDefKind::Type(Type::Id(id)) => id,
2448 _ => return,
2449 };
2450 let name = self.types[dep].name.as_ref().unwrap();
2451 let replacement_id = self.interfaces[*replace_with].types[name];
2454 self.types[ty].kind = TypeDefKind::Type(Type::Id(replacement_id));
2455 }
2456
2457 pub fn wasm_import_name(
2464 &self,
2465 mangling: ManglingAndAbi,
2466 import: WasmImport<'_>,
2467 ) -> (String, String) {
2468 match mangling {
2469 ManglingAndAbi::Standard32 => match import {
2470 WasmImport::Func { interface, func } => {
2471 let module = match interface {
2472 Some(key) => format!("cm32p2|{}", self.name_canonicalized_world_key(key)),
2473 None => format!("cm32p2"),
2474 };
2475 (module, func.name.clone())
2476 }
2477 WasmImport::ResourceIntrinsic {
2478 interface,
2479 resource,
2480 intrinsic,
2481 } => {
2482 let name = self.types[resource].name.as_ref().unwrap();
2483 let (prefix, name) = match intrinsic {
2484 ResourceIntrinsic::ImportedDrop => ("", format!("{name}_drop")),
2485 ResourceIntrinsic::ExportedDrop => ("_ex_", format!("{name}_drop")),
2486 ResourceIntrinsic::ExportedNew => ("_ex_", format!("{name}_new")),
2487 ResourceIntrinsic::ExportedRep => ("_ex_", format!("{name}_rep")),
2488 };
2489 let module = match interface {
2490 Some(key) => {
2491 format!("cm32p2|{prefix}{}", self.name_canonicalized_world_key(key))
2492 }
2493 None => {
2494 assert_eq!(prefix, "");
2495 format!("cm32p2")
2496 }
2497 };
2498 (module, name)
2499 }
2500 },
2501 ManglingAndAbi::Legacy(abi) => match import {
2502 WasmImport::Func { interface, func } => {
2503 let module = match interface {
2504 Some(key) => self.name_world_key(key),
2505 None => format!("$root"),
2506 };
2507 (module, format!("{}{}", abi.import_prefix(), func.name))
2508 }
2509 WasmImport::ResourceIntrinsic {
2510 interface,
2511 resource,
2512 intrinsic,
2513 } => {
2514 let name = self.types[resource].name.as_ref().unwrap();
2515 let (prefix, name) = match intrinsic {
2516 ResourceIntrinsic::ImportedDrop => ("", format!("[resource-drop]{name}")),
2517 ResourceIntrinsic::ExportedDrop => {
2518 ("[export]", format!("[resource-drop]{name}"))
2519 }
2520 ResourceIntrinsic::ExportedNew => {
2521 ("[export]", format!("[resource-new]{name}"))
2522 }
2523 ResourceIntrinsic::ExportedRep => {
2524 ("[export]", format!("[resource-rep]{name}"))
2525 }
2526 };
2527 let module = match interface {
2528 Some(key) => format!("{prefix}{}", self.name_world_key(key)),
2529 None => {
2530 assert_eq!(prefix, "");
2531 format!("$root")
2532 }
2533 };
2534 (module, format!("{}{name}", abi.import_prefix()))
2535 }
2536 },
2537 }
2538 }
2539
2540 pub fn wasm_export_name(&self, mangling: ManglingAndAbi, export: WasmExport<'_>) -> String {
2544 match mangling {
2545 ManglingAndAbi::Standard32 => match export {
2546 WasmExport::Func {
2547 interface,
2548 func,
2549 kind,
2550 } => {
2551 let mut name = String::from("cm32p2|");
2552 if let Some(interface) = interface {
2553 let s = self.name_canonicalized_world_key(interface);
2554 name.push_str(&s);
2555 }
2556 name.push_str("|");
2557 name.push_str(&func.name);
2558 match kind {
2559 WasmExportKind::Normal => {}
2560 WasmExportKind::PostReturn => name.push_str("_post"),
2561 WasmExportKind::Callback => todo!(
2562 "not yet supported: \
2563 async callback functions using standard name mangling"
2564 ),
2565 }
2566 name
2567 }
2568 WasmExport::ResourceDtor {
2569 interface,
2570 resource,
2571 } => {
2572 let name = self.types[resource].name.as_ref().unwrap();
2573 let interface = self.name_canonicalized_world_key(interface);
2574 format!("cm32p2|{interface}|{name}_dtor")
2575 }
2576 WasmExport::Memory => "cm32p2_memory".to_string(),
2577 WasmExport::Initialize => "cm32p2_initialize".to_string(),
2578 WasmExport::Realloc => "cm32p2_realloc".to_string(),
2579 },
2580 ManglingAndAbi::Legacy(abi) => match export {
2581 WasmExport::Func {
2582 interface,
2583 func,
2584 kind,
2585 } => {
2586 let mut name = abi.export_prefix().to_string();
2587 match kind {
2588 WasmExportKind::Normal => {}
2589 WasmExportKind::PostReturn => name.push_str("cabi_post_"),
2590 WasmExportKind::Callback => {
2591 assert!(matches!(abi, LiftLowerAbi::AsyncCallback));
2592 name = format!("[callback]{name}")
2593 }
2594 }
2595 if let Some(interface) = interface {
2596 let s = self.name_world_key(interface);
2597 name.push_str(&s);
2598 name.push_str("#");
2599 }
2600 name.push_str(&func.name);
2601 name
2602 }
2603 WasmExport::ResourceDtor {
2604 interface,
2605 resource,
2606 } => {
2607 let name = self.types[resource].name.as_ref().unwrap();
2608 let interface = self.name_world_key(interface);
2609 format!("{}{interface}#[dtor]{name}", abi.export_prefix())
2610 }
2611 WasmExport::Memory => "memory".to_string(),
2612 WasmExport::Initialize => "_initialize".to_string(),
2613 WasmExport::Realloc => "cabi_realloc".to_string(),
2614 },
2615 }
2616 }
2617}
2618
2619#[derive(Debug)]
2621pub enum WasmImport<'a> {
2622 Func {
2624 interface: Option<&'a WorldKey>,
2629
2630 func: &'a Function,
2632 },
2633
2634 ResourceIntrinsic {
2636 interface: Option<&'a WorldKey>,
2638
2639 resource: TypeId,
2641
2642 intrinsic: ResourceIntrinsic,
2644 },
2645}
2646
2647#[derive(Debug)]
2650pub enum ResourceIntrinsic {
2651 ImportedDrop,
2652 ExportedDrop,
2653 ExportedNew,
2654 ExportedRep,
2655}
2656
2657#[derive(Debug)]
2660pub enum WasmExportKind {
2661 Normal,
2663
2664 PostReturn,
2666
2667 Callback,
2669}
2670
2671#[derive(Debug)]
2674pub enum WasmExport<'a> {
2675 Func {
2677 interface: Option<&'a WorldKey>,
2680
2681 func: &'a Function,
2683
2684 kind: WasmExportKind,
2686 },
2687
2688 ResourceDtor {
2690 interface: &'a WorldKey,
2692 resource: TypeId,
2694 },
2695
2696 Memory,
2698
2699 Initialize,
2701
2702 Realloc,
2704}
2705
2706#[derive(Default)]
2709pub struct Remap {
2710 pub types: Vec<Option<TypeId>>,
2711 pub interfaces: Vec<Option<InterfaceId>>,
2712 pub worlds: Vec<Option<WorldId>>,
2713 pub packages: Vec<PackageId>,
2714
2715 own_handles: HashMap<TypeId, TypeId>,
2725
2726 type_has_borrow: Vec<Option<bool>>,
2727}
2728
2729fn apply_map<T>(map: &[Option<Id<T>>], id: Id<T>, desc: &str, span: Option<Span>) -> Result<Id<T>> {
2730 match map.get(id.index()) {
2731 Some(Some(id)) => Ok(*id),
2732 Some(None) => {
2733 let msg = format!(
2734 "found a reference to a {desc} which is excluded \
2735 due to its feature not being activated"
2736 );
2737 match span {
2738 Some(span) => Err(Error::new(span, msg).into()),
2739 None => bail!("{msg}"),
2740 }
2741 }
2742 None => panic!("request to remap a {desc} that has not yet been registered"),
2743 }
2744}
2745
2746impl Remap {
2747 pub fn map_type(&self, id: TypeId, span: Option<Span>) -> Result<TypeId> {
2748 apply_map(&self.types, id, "type", span)
2749 }
2750
2751 pub fn map_interface(&self, id: InterfaceId, span: Option<Span>) -> Result<InterfaceId> {
2752 apply_map(&self.interfaces, id, "interface", span)
2753 }
2754
2755 pub fn map_world(&self, id: WorldId, span: Option<Span>) -> Result<WorldId> {
2756 apply_map(&self.worlds, id, "world", span)
2757 }
2758
2759 fn append(
2760 &mut self,
2761 resolve: &mut Resolve,
2762 unresolved: UnresolvedPackage,
2763 ) -> Result<PackageId> {
2764 let pkgid = resolve.packages.alloc(Package {
2765 name: unresolved.name.clone(),
2766 docs: unresolved.docs.clone(),
2767 interfaces: Default::default(),
2768 worlds: Default::default(),
2769 });
2770 let prev = resolve.package_names.insert(unresolved.name.clone(), pkgid);
2771 if let Some(prev) = prev {
2772 resolve.package_names.insert(unresolved.name.clone(), prev);
2773 bail!(
2774 "attempting to re-add package `{}` when it's already present in this `Resolve`",
2775 unresolved.name,
2776 );
2777 }
2778
2779 self.process_foreign_deps(resolve, pkgid, &unresolved)?;
2780
2781 let foreign_types = self.types.len();
2782 let foreign_interfaces = self.interfaces.len();
2783 let foreign_worlds = self.worlds.len();
2784
2785 assert_eq!(unresolved.types.len(), unresolved.type_spans.len());
2791 for ((id, mut ty), span) in unresolved
2792 .types
2793 .into_iter()
2794 .zip(&unresolved.type_spans)
2795 .skip(foreign_types)
2796 {
2797 if !resolve.include_type(&ty, pkgid, *span)? {
2798 self.types.push(None);
2799 continue;
2800 }
2801
2802 self.update_typedef(resolve, &mut ty, Some(*span))?;
2803 let new_id = resolve.types.alloc(ty);
2804 assert_eq!(self.types.len(), id.index());
2805
2806 let new_id = match resolve.types[new_id] {
2807 TypeDef {
2812 name: None,
2813 owner: TypeOwner::None,
2814 kind: TypeDefKind::Handle(Handle::Own(id)),
2815 docs: _,
2816 stability: _,
2817 } => *self.own_handles.entry(id).or_insert(new_id),
2818
2819 _ => new_id,
2822 };
2823 self.types.push(Some(new_id));
2824 }
2825
2826 assert_eq!(
2829 unresolved.interfaces.len(),
2830 unresolved.interface_spans.len()
2831 );
2832 for ((id, mut iface), span) in unresolved
2833 .interfaces
2834 .into_iter()
2835 .zip(&unresolved.interface_spans)
2836 .skip(foreign_interfaces)
2837 {
2838 if !resolve
2839 .include_stability(&iface.stability, &pkgid, Some(span.span))
2840 .with_context(|| {
2841 format!(
2842 "failed to process feature gate for interface [{}] in package [{}]",
2843 iface
2844 .name
2845 .as_ref()
2846 .map(String::as_str)
2847 .unwrap_or("<unknown>"),
2848 resolve.packages[pkgid].name,
2849 )
2850 })?
2851 {
2852 self.interfaces.push(None);
2853 continue;
2854 }
2855 assert!(iface.package.is_none());
2856 iface.package = Some(pkgid);
2857 self.update_interface(resolve, &mut iface, Some(span))?;
2858 let new_id = resolve.interfaces.alloc(iface);
2859 assert_eq!(self.interfaces.len(), id.index());
2860 self.interfaces.push(Some(new_id));
2861 }
2862
2863 for (i, id) in self.types.iter().enumerate().skip(foreign_types) {
2866 let id = match id {
2867 Some(id) => *id,
2868 None => continue,
2869 };
2870 match &mut resolve.types[id].owner {
2871 TypeOwner::Interface(id) => {
2872 let span = unresolved.type_spans[i];
2873 *id = self.map_interface(*id, Some(span))
2874 .with_context(|| {
2875 "this type is not gated by a feature but its interface is gated by a feature"
2876 })?;
2877 }
2878 TypeOwner::World(_) | TypeOwner::None => {}
2879 }
2880 }
2881
2882 assert_eq!(unresolved.worlds.len(), unresolved.world_spans.len());
2890 for ((id, mut world), span) in unresolved
2891 .worlds
2892 .into_iter()
2893 .zip(&unresolved.world_spans)
2894 .skip(foreign_worlds)
2895 {
2896 if !resolve
2897 .include_stability(&world.stability, &pkgid, Some(span.span))
2898 .with_context(|| {
2899 format!(
2900 "failed to process feature gate for world [{}] in package [{}]",
2901 world.name, resolve.packages[pkgid].name,
2902 )
2903 })?
2904 {
2905 self.worlds.push(None);
2906 continue;
2907 }
2908 self.update_world(&mut world, resolve, &pkgid, &span)?;
2909
2910 let new_id = resolve.worlds.alloc(world);
2911 assert_eq!(self.worlds.len(), id.index());
2912 self.worlds.push(Some(new_id));
2913 }
2914
2915 for (i, id) in self.types.iter().enumerate().skip(foreign_types) {
2917 let id = match id {
2918 Some(id) => *id,
2919 None => continue,
2920 };
2921 match &mut resolve.types[id].owner {
2922 TypeOwner::World(id) => {
2923 let span = unresolved.type_spans[i];
2924 *id = self.map_world(*id, Some(span))
2925 .with_context(|| {
2926 "this type is not gated by a feature but its interface is gated by a feature"
2927 })?;
2928 }
2929 TypeOwner::Interface(_) | TypeOwner::None => {}
2930 }
2931 }
2932
2933 assert_eq!(self.worlds.len(), unresolved.world_spans.len());
2950 for (id, span) in self
2951 .worlds
2952 .iter()
2953 .zip(unresolved.world_spans.iter())
2954 .skip(foreign_worlds)
2955 {
2956 let Some(id) = *id else {
2957 continue;
2958 };
2959 self.process_world_includes(id, resolve, &pkgid, &span)?;
2960
2961 resolve.elaborate_world(id).with_context(|| {
2962 Error::new(
2963 span.span,
2964 format!(
2965 "failed to elaborate world imports/exports of `{}`",
2966 resolve.worlds[id].name
2967 ),
2968 )
2969 })?;
2970 }
2971
2972 for id in self.interfaces.iter().skip(foreign_interfaces) {
2974 let id = match id {
2975 Some(id) => *id,
2976 None => continue,
2977 };
2978 let iface = &mut resolve.interfaces[id];
2979 iface.package = Some(pkgid);
2980 if let Some(name) = &iface.name {
2981 let prev = resolve.packages[pkgid].interfaces.insert(name.clone(), id);
2982 assert!(prev.is_none());
2983 }
2984 }
2985 for id in self.worlds.iter().skip(foreign_worlds) {
2986 let id = match id {
2987 Some(id) => *id,
2988 None => continue,
2989 };
2990 let world = &mut resolve.worlds[id];
2991 world.package = Some(pkgid);
2992 let prev = resolve.packages[pkgid]
2993 .worlds
2994 .insert(world.name.clone(), id);
2995 assert!(prev.is_none());
2996 }
2997 Ok(pkgid)
2998 }
2999
3000 fn process_foreign_deps(
3001 &mut self,
3002 resolve: &mut Resolve,
3003 pkgid: PackageId,
3004 unresolved: &UnresolvedPackage,
3005 ) -> Result<()> {
3006 let mut world_to_package = HashMap::new();
3009 let mut interface_to_package = HashMap::new();
3010 for (i, (pkg_name, worlds_or_ifaces)) in unresolved.foreign_deps.iter().enumerate() {
3011 for (name, item) in worlds_or_ifaces {
3012 match item {
3013 AstItem::Interface(unresolved_interface_id) => {
3014 let prev = interface_to_package.insert(
3015 *unresolved_interface_id,
3016 (pkg_name, name, unresolved.foreign_dep_spans[i]),
3017 );
3018 assert!(prev.is_none());
3019 }
3020 AstItem::World(unresolved_world_id) => {
3021 let prev = world_to_package.insert(
3022 *unresolved_world_id,
3023 (pkg_name, name, unresolved.foreign_dep_spans[i]),
3024 );
3025 assert!(prev.is_none());
3026 }
3027 }
3028 }
3029 }
3030
3031 self.process_foreign_interfaces(unresolved, &interface_to_package, resolve)?;
3035
3036 self.process_foreign_worlds(unresolved, &world_to_package, resolve)?;
3040
3041 self.process_foreign_types(unresolved, pkgid, resolve)?;
3044
3045 for (id, span) in unresolved.required_resource_types.iter() {
3046 let Ok(mut id) = self.map_type(*id, Some(*span)) else {
3051 continue;
3052 };
3053 loop {
3054 match resolve.types[id].kind {
3055 TypeDefKind::Type(Type::Id(i)) => id = i,
3056 TypeDefKind::Resource => break,
3057 _ => bail!(Error::new(
3058 *span,
3059 format!("type used in a handle must be a resource"),
3060 )),
3061 }
3062 }
3063 }
3064
3065 #[cfg(debug_assertions)]
3066 resolve.assert_valid();
3067
3068 Ok(())
3069 }
3070
3071 fn process_foreign_interfaces(
3072 &mut self,
3073 unresolved: &UnresolvedPackage,
3074 interface_to_package: &HashMap<InterfaceId, (&PackageName, &String, Span)>,
3075 resolve: &mut Resolve,
3076 ) -> Result<(), anyhow::Error> {
3077 for (unresolved_iface_id, unresolved_iface) in unresolved.interfaces.iter() {
3078 let (pkg_name, interface, span) = match interface_to_package.get(&unresolved_iface_id) {
3079 Some(items) => *items,
3080 None => break,
3084 };
3085 let pkgid = resolve
3086 .package_names
3087 .get(pkg_name)
3088 .copied()
3089 .ok_or_else(|| {
3090 PackageNotFoundError::new(
3091 span,
3092 pkg_name.clone(),
3093 resolve.package_names.keys().cloned().collect(),
3094 )
3095 })?;
3096
3097 assert!(unresolved_iface.functions.is_empty());
3099
3100 let pkg = &resolve.packages[pkgid];
3101 let span = &unresolved.interface_spans[unresolved_iface_id.index()];
3102 let iface_id = pkg
3103 .interfaces
3104 .get(interface)
3105 .copied()
3106 .ok_or_else(|| Error::new(span.span, "interface not found in package"))?;
3107 assert_eq!(self.interfaces.len(), unresolved_iface_id.index());
3108 self.interfaces.push(Some(iface_id));
3109 }
3110 for (id, _) in unresolved.interfaces.iter().skip(self.interfaces.len()) {
3111 assert!(
3112 interface_to_package.get(&id).is_none(),
3113 "found foreign interface after local interface"
3114 );
3115 }
3116 Ok(())
3117 }
3118
3119 fn process_foreign_worlds(
3120 &mut self,
3121 unresolved: &UnresolvedPackage,
3122 world_to_package: &HashMap<WorldId, (&PackageName, &String, Span)>,
3123 resolve: &mut Resolve,
3124 ) -> Result<(), anyhow::Error> {
3125 for (unresolved_world_id, _) in unresolved.worlds.iter() {
3126 let (pkg_name, world, span) = match world_to_package.get(&unresolved_world_id) {
3127 Some(items) => *items,
3128 None => break,
3131 };
3132
3133 let pkgid = resolve
3134 .package_names
3135 .get(pkg_name)
3136 .copied()
3137 .ok_or_else(|| Error::new(span, "package not found"))?;
3138 let pkg = &resolve.packages[pkgid];
3139 let span = &unresolved.world_spans[unresolved_world_id.index()];
3140 let world_id = pkg
3141 .worlds
3142 .get(world)
3143 .copied()
3144 .ok_or_else(|| Error::new(span.span, "world not found in package"))?;
3145 assert_eq!(self.worlds.len(), unresolved_world_id.index());
3146 self.worlds.push(Some(world_id));
3147 }
3148 for (id, _) in unresolved.worlds.iter().skip(self.worlds.len()) {
3149 assert!(
3150 world_to_package.get(&id).is_none(),
3151 "found foreign world after local world"
3152 );
3153 }
3154 Ok(())
3155 }
3156
3157 fn process_foreign_types(
3158 &mut self,
3159 unresolved: &UnresolvedPackage,
3160 pkgid: PackageId,
3161 resolve: &mut Resolve,
3162 ) -> Result<(), anyhow::Error> {
3163 for ((unresolved_type_id, unresolved_ty), span) in
3164 unresolved.types.iter().zip(&unresolved.type_spans)
3165 {
3166 match unresolved_ty.kind {
3170 TypeDefKind::Unknown => {}
3171 _ => break,
3172 }
3173
3174 if !resolve.include_type(unresolved_ty, pkgid, *span)? {
3175 self.types.push(None);
3176 continue;
3177 }
3178
3179 let unresolved_iface_id = match unresolved_ty.owner {
3180 TypeOwner::Interface(id) => id,
3181 _ => unreachable!(),
3182 };
3183 let iface_id = self.map_interface(unresolved_iface_id, None)?;
3184 let name = unresolved_ty.name.as_ref().unwrap();
3185 let span = unresolved.unknown_type_spans[unresolved_type_id.index()];
3186 let type_id = *resolve.interfaces[iface_id]
3187 .types
3188 .get(name)
3189 .ok_or_else(|| {
3190 Error::new(span, format!("type `{name}` not defined in interface"))
3191 })?;
3192 assert_eq!(self.types.len(), unresolved_type_id.index());
3193 self.types.push(Some(type_id));
3194 }
3195 for (_, ty) in unresolved.types.iter().skip(self.types.len()) {
3196 if let TypeDefKind::Unknown = ty.kind {
3197 panic!("unknown type after defined type");
3198 }
3199 }
3200 Ok(())
3201 }
3202
3203 fn update_typedef(
3204 &mut self,
3205 resolve: &mut Resolve,
3206 ty: &mut TypeDef,
3207 span: Option<Span>,
3208 ) -> Result<()> {
3209 use crate::TypeDefKind::*;
3212 match &mut ty.kind {
3213 Handle(handle) => match handle {
3214 crate::Handle::Own(ty) | crate::Handle::Borrow(ty) => {
3215 self.update_type_id(ty, span)?
3216 }
3217 },
3218 Resource => {}
3219 Record(r) => {
3220 for field in r.fields.iter_mut() {
3221 self.update_ty(resolve, &mut field.ty, span)
3222 .with_context(|| format!("failed to update field `{}`", field.name))?;
3223 }
3224 }
3225 Tuple(t) => {
3226 for ty in t.types.iter_mut() {
3227 self.update_ty(resolve, ty, span)?;
3228 }
3229 }
3230 Variant(v) => {
3231 for case in v.cases.iter_mut() {
3232 if let Some(t) = &mut case.ty {
3233 self.update_ty(resolve, t, span)?;
3234 }
3235 }
3236 }
3237 Option(t) | List(t, ..) | FixedSizeList(t, ..) | Future(Some(t)) | Stream(Some(t)) => {
3238 self.update_ty(resolve, t, span)?
3239 }
3240 Result(r) => {
3241 if let Some(ty) = &mut r.ok {
3242 self.update_ty(resolve, ty, span)?;
3243 }
3244 if let Some(ty) = &mut r.err {
3245 self.update_ty(resolve, ty, span)?;
3246 }
3247 }
3248
3249 Type(crate::Type::Id(id)) => self.update_type_id(id, span)?,
3254 Type(_) => {}
3255
3256 Flags(_) | Enum(_) | Future(None) | Stream(None) => {}
3258
3259 Unknown => unreachable!(),
3260 }
3261
3262 Ok(())
3263 }
3264
3265 fn update_ty(
3266 &mut self,
3267 resolve: &mut Resolve,
3268 ty: &mut Type,
3269 span: Option<Span>,
3270 ) -> Result<()> {
3271 let id = match ty {
3272 Type::Id(id) => id,
3273 _ => return Ok(()),
3274 };
3275 self.update_type_id(id, span)?;
3276
3277 let mut cur = *id;
3282 let points_to_resource = loop {
3283 match resolve.types[cur].kind {
3284 TypeDefKind::Type(Type::Id(id)) => cur = id,
3285 TypeDefKind::Resource => break true,
3286 _ => break false,
3287 }
3288 };
3289
3290 if points_to_resource {
3291 *id = *self.own_handles.entry(*id).or_insert_with(|| {
3292 resolve.types.alloc(TypeDef {
3293 name: None,
3294 owner: TypeOwner::None,
3295 kind: TypeDefKind::Handle(Handle::Own(*id)),
3296 docs: Default::default(),
3297 stability: Default::default(),
3298 })
3299 });
3300 }
3301 Ok(())
3302 }
3303
3304 fn update_type_id(&self, id: &mut TypeId, span: Option<Span>) -> Result<()> {
3305 *id = self.map_type(*id, span)?;
3306 Ok(())
3307 }
3308
3309 fn update_interface(
3310 &mut self,
3311 resolve: &mut Resolve,
3312 iface: &mut Interface,
3313 spans: Option<&InterfaceSpan>,
3314 ) -> Result<()> {
3315 iface.types.retain(|_, ty| self.types[ty.index()].is_some());
3316 let iface_pkg_id = iface.package.as_ref().unwrap_or_else(|| {
3317 panic!(
3318 "unexpectedly missing package on interface [{}]",
3319 iface
3320 .name
3321 .as_ref()
3322 .map(String::as_str)
3323 .unwrap_or("<unknown>"),
3324 )
3325 });
3326
3327 for (_name, ty) in iface.types.iter_mut() {
3330 self.update_type_id(ty, spans.map(|s| s.span))?;
3331 }
3332 if let Some(spans) = spans {
3333 assert_eq!(iface.functions.len(), spans.funcs.len());
3334 }
3335 for (i, (func_name, func)) in iface.functions.iter_mut().enumerate() {
3336 let span = spans.map(|s| s.funcs[i]);
3337 if !resolve
3338 .include_stability(&func.stability, iface_pkg_id, span)
3339 .with_context(|| {
3340 format!(
3341 "failed to process feature gate for function [{func_name}] in package [{}]",
3342 resolve.packages[*iface_pkg_id].name,
3343 )
3344 })?
3345 {
3346 continue;
3347 }
3348 self.update_function(resolve, func, span)
3349 .with_context(|| format!("failed to update function `{}`", func.name))?;
3350 }
3351
3352 for (name, func) in mem::take(&mut iface.functions) {
3355 if resolve.include_stability(&func.stability, iface_pkg_id, None)? {
3356 iface.functions.insert(name, func);
3357 }
3358 }
3359
3360 Ok(())
3361 }
3362
3363 fn update_function(
3364 &mut self,
3365 resolve: &mut Resolve,
3366 func: &mut Function,
3367 span: Option<Span>,
3368 ) -> Result<()> {
3369 if let Some(id) = func.kind.resource_mut() {
3370 self.update_type_id(id, span)?;
3371 }
3372 for (_, ty) in func.params.iter_mut() {
3373 self.update_ty(resolve, ty, span)?;
3374 }
3375 if let Some(ty) = &mut func.result {
3376 self.update_ty(resolve, ty, span)?;
3377 }
3378
3379 if let Some(ty) = &func.result {
3380 if self.type_has_borrow(resolve, ty) {
3381 match span {
3382 Some(span) => {
3383 bail!(Error::new(
3384 span,
3385 format!(
3386 "function returns a type which contains \
3387 a `borrow<T>` which is not supported"
3388 )
3389 ))
3390 }
3391 None => unreachable!(),
3392 }
3393 }
3394 }
3395
3396 Ok(())
3397 }
3398
3399 fn update_world(
3400 &mut self,
3401 world: &mut World,
3402 resolve: &mut Resolve,
3403 pkg_id: &PackageId,
3404 spans: &WorldSpan,
3405 ) -> Result<()> {
3406 assert_eq!(world.imports.len(), spans.imports.len());
3407 assert_eq!(world.exports.len(), spans.exports.len());
3408
3409 let imports = mem::take(&mut world.imports).into_iter();
3413 let imports = imports.zip(&spans.imports).map(|p| (p, true));
3414 let exports = mem::take(&mut world.exports).into_iter();
3415 let exports = exports.zip(&spans.exports).map(|p| (p, false));
3416 for (((mut name, mut item), span), import) in imports.chain(exports) {
3417 if let WorldItem::Type(id) = &mut item {
3420 *id = self.map_type(*id, Some(*span))?;
3421 }
3422 let stability = item.stability(resolve);
3423 if !resolve
3424 .include_stability(stability, pkg_id, Some(*span))
3425 .with_context(|| format!("failed to process world item in `{}`", world.name))?
3426 {
3427 continue;
3428 }
3429 self.update_world_key(&mut name, Some(*span))?;
3430 match &mut item {
3431 WorldItem::Interface { id, .. } => {
3432 *id = self.map_interface(*id, Some(*span))?;
3433 }
3434 WorldItem::Function(f) => {
3435 self.update_function(resolve, f, Some(*span))?;
3436 }
3437 WorldItem::Type(_) => {
3438 }
3440 }
3441
3442 let dst = if import {
3443 &mut world.imports
3444 } else {
3445 &mut world.exports
3446 };
3447 let prev = dst.insert(name, item);
3448 assert!(prev.is_none());
3449 }
3450
3451 Ok(())
3452 }
3453
3454 fn process_world_includes(
3455 &self,
3456 id: WorldId,
3457 resolve: &mut Resolve,
3458 pkg_id: &PackageId,
3459 spans: &WorldSpan,
3460 ) -> Result<()> {
3461 let world = &mut resolve.worlds[id];
3462 assert_eq!(world.includes.len(), spans.includes.len());
3465 let includes = mem::take(&mut world.includes);
3466 let include_names = mem::take(&mut world.include_names);
3467 for (((stability, include_world), span), names) in includes
3468 .into_iter()
3469 .zip(&spans.includes)
3470 .zip(&include_names)
3471 {
3472 if !resolve
3473 .include_stability(&stability, pkg_id, Some(*span))
3474 .with_context(|| {
3475 format!(
3476 "failed to process feature gate for included world [{}] in package [{}]",
3477 resolve.worlds[include_world].name.as_str(),
3478 resolve.packages[*pkg_id].name
3479 )
3480 })?
3481 {
3482 continue;
3483 }
3484 self.resolve_include(id, include_world, names, *span, pkg_id, resolve)?;
3485 }
3486
3487 Ok(())
3488 }
3489
3490 fn update_world_key(&self, key: &mut WorldKey, span: Option<Span>) -> Result<()> {
3491 match key {
3492 WorldKey::Name(_) => {}
3493 WorldKey::Interface(id) => {
3494 *id = self.map_interface(*id, span)?;
3495 }
3496 }
3497 Ok(())
3498 }
3499
3500 fn resolve_include(
3501 &self,
3502 id: WorldId,
3503 include_world_id_orig: WorldId,
3504 names: &[IncludeName],
3505 span: Span,
3506 pkg_id: &PackageId,
3507 resolve: &mut Resolve,
3508 ) -> Result<()> {
3509 let world = &resolve.worlds[id];
3510 let include_world_id = self.map_world(include_world_id_orig, Some(span))?;
3511 let include_world = resolve.worlds[include_world_id].clone();
3512 let mut names_ = names.to_owned();
3513 let is_external_include = world.package != include_world.package;
3514
3515 for import in include_world.imports.iter() {
3517 self.remove_matching_name(import, &mut names_);
3518 }
3519 for export in include_world.exports.iter() {
3520 self.remove_matching_name(export, &mut names_);
3521 }
3522 if !names_.is_empty() {
3523 bail!(Error::new(
3524 span,
3525 format!("no import or export kebab-name `{}`. Note that an ID does not support renaming", names_[0].name),
3526 ));
3527 }
3528
3529 let mut cloner = clone::Cloner::new(
3530 resolve,
3531 TypeOwner::World(if is_external_include {
3532 include_world_id
3533 } else {
3534 include_world_id
3535 }),
3537 TypeOwner::World(id),
3538 );
3539 cloner.new_package = Some(*pkg_id);
3540
3541 for import in include_world.imports.iter() {
3543 self.resolve_include_item(
3544 &mut cloner,
3545 names,
3546 |resolve| &mut resolve.worlds[id].imports,
3547 import,
3548 span,
3549 "import",
3550 is_external_include,
3551 )?;
3552 }
3553
3554 for export in include_world.exports.iter() {
3555 self.resolve_include_item(
3556 &mut cloner,
3557 names,
3558 |resolve| &mut resolve.worlds[id].exports,
3559 export,
3560 span,
3561 "export",
3562 is_external_include,
3563 )?;
3564 }
3565 Ok(())
3566 }
3567
3568 fn resolve_include_item(
3569 &self,
3570 cloner: &mut clone::Cloner<'_>,
3571 names: &[IncludeName],
3572 get_items: impl Fn(&mut Resolve) -> &mut IndexMap<WorldKey, WorldItem>,
3573 item: (&WorldKey, &WorldItem),
3574 span: Span,
3575 item_type: &str,
3576 is_external_include: bool,
3577 ) -> Result<()> {
3578 match item.0 {
3579 WorldKey::Name(n) => {
3580 let n = if let Some(found) = names
3581 .into_iter()
3582 .find(|include_name| include_name.name == n.clone())
3583 {
3584 found.as_.clone()
3585 } else {
3586 n.clone()
3587 };
3588
3589 let mut new_item = item.1.clone();
3595 if let WorldItem::Function(f) = &mut new_item {
3596 f.name = n.clone();
3597 }
3598 let key = WorldKey::Name(n.clone());
3599 cloner.world_item(&key, &mut new_item);
3600
3601 let prev = get_items(cloner.resolve).insert(key, new_item);
3602 if prev.is_some() {
3603 bail!(Error::new(
3604 span,
3605 format!("{item_type} of `{n}` shadows previously {item_type}ed items"),
3606 ))
3607 }
3608 }
3609 key @ WorldKey::Interface(_) => {
3610 let prev = get_items(cloner.resolve)
3611 .entry(key.clone())
3612 .or_insert(item.1.clone());
3613 match (&item.1, prev) {
3614 (
3615 WorldItem::Interface {
3616 id: aid,
3617 stability: astability,
3618 },
3619 WorldItem::Interface {
3620 id: bid,
3621 stability: bstability,
3622 },
3623 ) => {
3624 assert_eq!(*aid, *bid);
3625 merge_include_stability(astability, bstability, is_external_include)?;
3626 }
3627 (WorldItem::Interface { .. }, _) => unreachable!(),
3628 (WorldItem::Function(_), _) => unreachable!(),
3629 (WorldItem::Type(_), _) => unreachable!(),
3630 }
3631 }
3632 };
3633
3634 Ok(())
3635 }
3636
3637 fn remove_matching_name(&self, item: (&WorldKey, &WorldItem), names: &mut Vec<IncludeName>) {
3638 match item.0 {
3639 WorldKey::Name(n) => {
3640 names.retain(|name| name.name != n.clone());
3641 }
3642 _ => {}
3643 }
3644 }
3645
3646 fn type_has_borrow(&mut self, resolve: &Resolve, ty: &Type) -> bool {
3647 let id = match ty {
3648 Type::Id(id) => *id,
3649 _ => return false,
3650 };
3651
3652 if let Some(Some(has_borrow)) = self.type_has_borrow.get(id.index()) {
3653 return *has_borrow;
3654 }
3655
3656 let result = self.typedef_has_borrow(resolve, &resolve.types[id]);
3657 if self.type_has_borrow.len() <= id.index() {
3658 self.type_has_borrow.resize(id.index() + 1, None);
3659 }
3660 self.type_has_borrow[id.index()] = Some(result);
3661 result
3662 }
3663
3664 fn typedef_has_borrow(&mut self, resolve: &Resolve, ty: &TypeDef) -> bool {
3665 match &ty.kind {
3666 TypeDefKind::Type(t) => self.type_has_borrow(resolve, t),
3667 TypeDefKind::Variant(v) => v
3668 .cases
3669 .iter()
3670 .filter_map(|case| case.ty.as_ref())
3671 .any(|ty| self.type_has_borrow(resolve, ty)),
3672 TypeDefKind::Handle(Handle::Borrow(_)) => true,
3673 TypeDefKind::Handle(Handle::Own(_)) => false,
3674 TypeDefKind::Resource => false,
3675 TypeDefKind::Record(r) => r
3676 .fields
3677 .iter()
3678 .any(|case| self.type_has_borrow(resolve, &case.ty)),
3679 TypeDefKind::Flags(_) => false,
3680 TypeDefKind::Tuple(t) => t.types.iter().any(|t| self.type_has_borrow(resolve, t)),
3681 TypeDefKind::Enum(_) => false,
3682 TypeDefKind::List(ty)
3683 | TypeDefKind::FixedSizeList(ty, ..)
3684 | TypeDefKind::Future(Some(ty))
3685 | TypeDefKind::Stream(Some(ty))
3686 | TypeDefKind::Option(ty) => self.type_has_borrow(resolve, ty),
3687 TypeDefKind::Result(r) => [&r.ok, &r.err]
3688 .iter()
3689 .filter_map(|t| t.as_ref())
3690 .any(|t| self.type_has_borrow(resolve, t)),
3691 TypeDefKind::Future(None) | TypeDefKind::Stream(None) => false,
3692 TypeDefKind::Unknown => unreachable!(),
3693 }
3694 }
3695}
3696
3697struct MergeMap<'a> {
3698 package_map: HashMap<PackageId, PackageId>,
3701
3702 interface_map: HashMap<InterfaceId, InterfaceId>,
3705
3706 type_map: HashMap<TypeId, TypeId>,
3709
3710 world_map: HashMap<WorldId, WorldId>,
3713
3714 interfaces_to_add: Vec<(String, PackageId, InterfaceId)>,
3722 worlds_to_add: Vec<(String, PackageId, WorldId)>,
3723
3724 from: &'a Resolve,
3726
3727 into: &'a Resolve,
3729}
3730
3731impl<'a> MergeMap<'a> {
3732 fn new(from: &'a Resolve, into: &'a Resolve) -> MergeMap<'a> {
3733 MergeMap {
3734 package_map: Default::default(),
3735 interface_map: Default::default(),
3736 type_map: Default::default(),
3737 world_map: Default::default(),
3738 interfaces_to_add: Default::default(),
3739 worlds_to_add: Default::default(),
3740 from,
3741 into,
3742 }
3743 }
3744
3745 fn build(&mut self) -> Result<()> {
3746 for from_id in self.from.topological_packages() {
3747 let from = &self.from.packages[from_id];
3748 let into_id = match self.into.package_names.get(&from.name) {
3749 Some(id) => *id,
3750
3751 None => {
3754 log::trace!("adding unique package {}", from.name);
3755 continue;
3756 }
3757 };
3758 log::trace!("merging duplicate package {}", from.name);
3759
3760 self.build_package(from_id, into_id).with_context(|| {
3761 format!("failed to merge package `{}` into existing copy", from.name)
3762 })?;
3763 }
3764
3765 Ok(())
3766 }
3767
3768 fn build_package(&mut self, from_id: PackageId, into_id: PackageId) -> Result<()> {
3769 let prev = self.package_map.insert(from_id, into_id);
3770 assert!(prev.is_none());
3771
3772 let from = &self.from.packages[from_id];
3773 let into = &self.into.packages[into_id];
3774
3775 for (name, from_interface_id) in from.interfaces.iter() {
3779 let into_interface_id = match into.interfaces.get(name) {
3780 Some(id) => *id,
3781 None => {
3782 log::trace!("adding unique interface {}", name);
3783 self.interfaces_to_add
3784 .push((name.clone(), into_id, *from_interface_id));
3785 continue;
3786 }
3787 };
3788
3789 log::trace!("merging duplicate interfaces {}", name);
3790 self.build_interface(*from_interface_id, into_interface_id)
3791 .with_context(|| format!("failed to merge interface `{name}`"))?;
3792 }
3793
3794 for (name, from_world_id) in from.worlds.iter() {
3795 let into_world_id = match into.worlds.get(name) {
3796 Some(id) => *id,
3797 None => {
3798 log::trace!("adding unique world {}", name);
3799 self.worlds_to_add
3800 .push((name.clone(), into_id, *from_world_id));
3801 continue;
3802 }
3803 };
3804
3805 log::trace!("merging duplicate worlds {}", name);
3806 self.build_world(*from_world_id, into_world_id)
3807 .with_context(|| format!("failed to merge world `{name}`"))?;
3808 }
3809
3810 Ok(())
3811 }
3812
3813 fn build_interface(&mut self, from_id: InterfaceId, into_id: InterfaceId) -> Result<()> {
3814 let prev = self.interface_map.insert(from_id, into_id);
3815 assert!(prev.is_none());
3816
3817 let from_interface = &self.from.interfaces[from_id];
3818 let into_interface = &self.into.interfaces[into_id];
3819
3820 for (name, from_type_id) in from_interface.types.iter() {
3834 let into_type_id = *into_interface
3835 .types
3836 .get(name)
3837 .ok_or_else(|| anyhow!("expected type `{name}` to be present"))?;
3838 let prev = self.type_map.insert(*from_type_id, into_type_id);
3839 assert!(prev.is_none());
3840
3841 self.build_type_id(*from_type_id, into_type_id)
3842 .with_context(|| format!("mismatch in type `{name}`"))?;
3843 }
3844
3845 for (name, from_func) in from_interface.functions.iter() {
3846 let into_func = match into_interface.functions.get(name) {
3847 Some(func) => func,
3848 None => bail!("expected function `{name}` to be present"),
3849 };
3850 self.build_function(from_func, into_func)
3851 .with_context(|| format!("mismatch in function `{name}`"))?;
3852 }
3853
3854 Ok(())
3855 }
3856
3857 fn build_type_id(&mut self, from_id: TypeId, into_id: TypeId) -> Result<()> {
3858 let _ = from_id;
3862 let _ = into_id;
3863 Ok(())
3864 }
3865
3866 fn build_type(&mut self, from_ty: &Type, into_ty: &Type) -> Result<()> {
3867 match (from_ty, into_ty) {
3868 (Type::Id(from), Type::Id(into)) => {
3869 self.build_type_id(*from, *into)?;
3870 }
3871 (from, into) if from != into => bail!("different kinds of types"),
3872 _ => {}
3873 }
3874 Ok(())
3875 }
3876
3877 fn build_function(&mut self, from_func: &Function, into_func: &Function) -> Result<()> {
3878 if from_func.name != into_func.name {
3879 bail!(
3880 "different function names `{}` and `{}`",
3881 from_func.name,
3882 into_func.name
3883 );
3884 }
3885 match (&from_func.kind, &into_func.kind) {
3886 (FunctionKind::Freestanding, FunctionKind::Freestanding) => {}
3887 (FunctionKind::AsyncFreestanding, FunctionKind::AsyncFreestanding) => {}
3888
3889 (FunctionKind::Method(from), FunctionKind::Method(into))
3890 | (FunctionKind::Static(from), FunctionKind::Static(into))
3891 | (FunctionKind::AsyncMethod(from), FunctionKind::AsyncMethod(into))
3892 | (FunctionKind::AsyncStatic(from), FunctionKind::AsyncStatic(into))
3893 | (FunctionKind::Constructor(from), FunctionKind::Constructor(into)) => {
3894 self.build_type_id(*from, *into)
3895 .context("different function kind types")?;
3896 }
3897
3898 (FunctionKind::Method(_), _)
3899 | (FunctionKind::Constructor(_), _)
3900 | (FunctionKind::Static(_), _)
3901 | (FunctionKind::Freestanding, _)
3902 | (FunctionKind::AsyncFreestanding, _)
3903 | (FunctionKind::AsyncMethod(_), _)
3904 | (FunctionKind::AsyncStatic(_), _) => {
3905 bail!("different function kind types")
3906 }
3907 }
3908
3909 if from_func.params.len() != into_func.params.len() {
3910 bail!("different number of function parameters");
3911 }
3912 for ((from_name, from_ty), (into_name, into_ty)) in
3913 from_func.params.iter().zip(&into_func.params)
3914 {
3915 if from_name != into_name {
3916 bail!("different function parameter names: {from_name} != {into_name}");
3917 }
3918 self.build_type(from_ty, into_ty)
3919 .with_context(|| format!("different function parameter types for `{from_name}`"))?;
3920 }
3921 match (&from_func.result, &into_func.result) {
3922 (Some(from_ty), Some(into_ty)) => {
3923 self.build_type(from_ty, into_ty)
3924 .context("different function result types")?;
3925 }
3926 (None, None) => {}
3927 (Some(_), None) | (None, Some(_)) => bail!("different number of function results"),
3928 }
3929 Ok(())
3930 }
3931
3932 fn build_world(&mut self, from_id: WorldId, into_id: WorldId) -> Result<()> {
3933 let prev = self.world_map.insert(from_id, into_id);
3934 assert!(prev.is_none());
3935
3936 let from_world = &self.from.worlds[from_id];
3937 let into_world = &self.into.worlds[into_id];
3938
3939 if from_world.imports.len() != into_world.imports.len() {
3948 bail!("world contains different number of imports than expected");
3949 }
3950 if from_world.exports.len() != into_world.exports.len() {
3951 bail!("world contains different number of exports than expected");
3952 }
3953
3954 for (from_name, from) in from_world.imports.iter() {
3955 let into_name = MergeMap::map_name(from_name, &self.interface_map);
3956 let name_str = self.from.name_world_key(from_name);
3957 let into = into_world
3958 .imports
3959 .get(&into_name)
3960 .ok_or_else(|| anyhow!("import `{name_str}` not found in target world"))?;
3961 self.match_world_item(from, into)
3962 .with_context(|| format!("import `{name_str}` didn't match target world"))?;
3963 }
3964
3965 for (from_name, from) in from_world.exports.iter() {
3966 let into_name = MergeMap::map_name(from_name, &self.interface_map);
3967 let name_str = self.from.name_world_key(from_name);
3968 let into = into_world
3969 .exports
3970 .get(&into_name)
3971 .ok_or_else(|| anyhow!("export `{name_str}` not found in target world"))?;
3972 self.match_world_item(from, into)
3973 .with_context(|| format!("export `{name_str}` didn't match target world"))?;
3974 }
3975
3976 Ok(())
3977 }
3978
3979 fn map_name(
3980 from_name: &WorldKey,
3981 interface_map: &HashMap<InterfaceId, InterfaceId>,
3982 ) -> WorldKey {
3983 match from_name {
3984 WorldKey::Name(s) => WorldKey::Name(s.clone()),
3985 WorldKey::Interface(id) => {
3986 WorldKey::Interface(interface_map.get(id).copied().unwrap_or(*id))
3987 }
3988 }
3989 }
3990
3991 fn match_world_item(&mut self, from: &WorldItem, into: &WorldItem) -> Result<()> {
3992 match (from, into) {
3993 (WorldItem::Interface { id: from, .. }, WorldItem::Interface { id: into, .. }) => {
3994 match (
3995 &self.from.interfaces[*from].name,
3996 &self.into.interfaces[*into].name,
3997 ) {
3998 (None, None) => self.build_interface(*from, *into)?,
4002
4003 _ => {
4008 if self.interface_map.get(&from) != Some(&into) {
4009 bail!("interfaces are not the same");
4010 }
4011 }
4012 }
4013 }
4014 (WorldItem::Function(from), WorldItem::Function(into)) => {
4015 let _ = (from, into);
4016 }
4019 (WorldItem::Type(from), WorldItem::Type(into)) => {
4020 let prev = self.type_map.insert(*from, *into);
4023 assert!(prev.is_none());
4024 }
4025
4026 (WorldItem::Interface { .. }, _)
4027 | (WorldItem::Function(_), _)
4028 | (WorldItem::Type(_), _) => {
4029 bail!("world items do not have the same type")
4030 }
4031 }
4032 Ok(())
4033 }
4034}
4035
4036fn update_stability(from: &Stability, into: &mut Stability) -> Result<()> {
4042 if from == into || from.is_unknown() {
4045 return Ok(());
4046 }
4047 if into.is_unknown() {
4050 *into = from.clone();
4051 return Ok(());
4052 }
4053
4054 bail!("mismatch in stability from '{:?}' to '{:?}'", from, into)
4057}
4058
4059fn merge_include_stability(
4060 from: &Stability,
4061 into: &mut Stability,
4062 is_external_include: bool,
4063) -> Result<()> {
4064 if is_external_include && from.is_stable() {
4065 log::trace!("dropped stability from external package");
4066 *into = Stability::Unknown;
4067 return Ok(());
4068 }
4069
4070 return update_stability(from, into);
4071}
4072
4073#[derive(Debug, Clone)]
4086pub struct InvalidTransitiveDependency(String);
4087
4088impl fmt::Display for InvalidTransitiveDependency {
4089 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
4090 write!(
4091 f,
4092 "interface `{}` transitively depends on an interface in \
4093 incompatible ways",
4094 self.0
4095 )
4096 }
4097}
4098
4099impl std::error::Error for InvalidTransitiveDependency {}
4100
4101#[cfg(test)]
4102mod tests {
4103 use crate::Resolve;
4104 use anyhow::Result;
4105
4106 #[test]
4107 fn select_world() -> Result<()> {
4108 let mut resolve = Resolve::default();
4109 resolve.push_str(
4110 "test.wit",
4111 r#"
4112 package foo:bar@0.1.0;
4113
4114 world foo {}
4115 "#,
4116 )?;
4117 resolve.push_str(
4118 "test.wit",
4119 r#"
4120 package foo:baz@0.1.0;
4121
4122 world foo {}
4123 "#,
4124 )?;
4125 resolve.push_str(
4126 "test.wit",
4127 r#"
4128 package foo:baz@0.2.0;
4129
4130 world foo {}
4131 "#,
4132 )?;
4133
4134 let dummy = resolve.push_str(
4135 "test.wit",
4136 r#"
4137 package foo:dummy;
4138
4139 world foo {}
4140 "#,
4141 )?;
4142
4143 assert!(resolve.select_world(dummy, None).is_ok());
4144 assert!(resolve.select_world(dummy, Some("xx")).is_err());
4145 assert!(resolve.select_world(dummy, Some("")).is_err());
4146 assert!(resolve.select_world(dummy, Some("foo:bar/foo")).is_ok());
4147 assert!(resolve
4148 .select_world(dummy, Some("foo:bar/foo@0.1.0"))
4149 .is_ok());
4150 assert!(resolve.select_world(dummy, Some("foo:baz/foo")).is_err());
4151 assert!(resolve
4152 .select_world(dummy, Some("foo:baz/foo@0.1.0"))
4153 .is_ok());
4154 assert!(resolve
4155 .select_world(dummy, Some("foo:baz/foo@0.2.0"))
4156 .is_ok());
4157 Ok(())
4158 }
4159}