import.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619
  1. // Part of the Carbon Language project, under the Apache License v2.0 with LLVM
  2. // Exceptions. See /LICENSE for license information.
  3. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  4. #include "toolchain/check/import.h"
  5. #include "common/check.h"
  6. #include "common/map.h"
  7. #include "toolchain/base/kind_switch.h"
  8. #include "toolchain/check/context.h"
  9. #include "toolchain/check/import_ref.h"
  10. #include "toolchain/check/merge.h"
  11. #include "toolchain/parse/node_ids.h"
  12. #include "toolchain/sem_ir/file.h"
  13. #include "toolchain/sem_ir/ids.h"
  14. #include "toolchain/sem_ir/import_ir.h"
  15. #include "toolchain/sem_ir/inst.h"
  16. #include "toolchain/sem_ir/name_scope.h"
  17. #include "toolchain/sem_ir/typed_insts.h"
  18. namespace Carbon::Check {
  19. // Returns name information for an EntityWithParamsBase.
  20. template <typename T>
  21. static auto GetImportNameForEntity(const T& entity)
  22. -> std::pair<SemIR::NameId, SemIR::NameScopeId> {
  23. return {entity.name_id, entity.parent_scope_id};
  24. }
  25. template <>
  26. auto GetImportNameForEntity(const SemIR::NameScope& entity)
  27. -> std::pair<SemIR::NameId, SemIR::NameScopeId> {
  28. return {entity.name_id(), entity.parent_scope_id()};
  29. }
  30. // Returns name information for the entity, corresponding to IDs in the import
  31. // IR rather than the current IR.
  32. static auto GetImportName(const SemIR::File& import_sem_ir,
  33. SemIR::Inst import_inst)
  34. -> std::pair<SemIR::NameId, SemIR::NameScopeId> {
  35. CARBON_KIND_SWITCH(import_inst) {
  36. case SemIR::BindAlias::Kind:
  37. case SemIR::BindName::Kind:
  38. case SemIR::BindSymbolicName::Kind:
  39. case SemIR::ExportDecl::Kind: {
  40. auto bind_inst = import_inst.As<SemIR::AnyBindNameOrExportDecl>();
  41. return GetImportNameForEntity(
  42. import_sem_ir.entity_names().Get(bind_inst.entity_name_id));
  43. }
  44. case CARBON_KIND(SemIR::ClassDecl class_decl): {
  45. return GetImportNameForEntity(
  46. import_sem_ir.classes().Get(class_decl.class_id));
  47. }
  48. case CARBON_KIND(SemIR::FunctionDecl function_decl): {
  49. return GetImportNameForEntity(
  50. import_sem_ir.functions().Get(function_decl.function_id));
  51. }
  52. case CARBON_KIND(SemIR::InterfaceDecl interface_decl): {
  53. return GetImportNameForEntity(
  54. import_sem_ir.interfaces().Get(interface_decl.interface_id));
  55. }
  56. case CARBON_KIND(SemIR::Namespace ns): {
  57. return GetImportNameForEntity(
  58. import_sem_ir.name_scopes().Get(ns.name_scope_id));
  59. }
  60. default:
  61. CARBON_FATAL("Unsupported export kind: {0}", import_inst);
  62. }
  63. }
  64. // Translate the name to the current IR. It will usually be an identifier, but
  65. // could also be a builtin name ID which is equivalent cross-IR.
  66. static auto CopyNameFromImportIR(Context& context,
  67. const SemIR::File& import_sem_ir,
  68. SemIR::NameId import_name_id) {
  69. if (auto import_identifier_id = import_name_id.AsIdentifierId();
  70. import_identifier_id.has_value()) {
  71. auto name = import_sem_ir.identifiers().Get(import_identifier_id);
  72. return SemIR::NameId::ForIdentifier(context.identifiers().Add(name));
  73. }
  74. return import_name_id;
  75. }
  76. namespace {
  77. struct NamespaceResult {
  78. SemIR::NameScopeId name_scope_id;
  79. SemIR::InstId inst_id;
  80. bool is_duplicate_of_namespace_in_current_package;
  81. };
  82. } // namespace
  83. // Adds a namespace to the IR. The bool on return is true if there was a name
  84. // conflict. diagnose_duplicate_namespace is used when handling a cross-package
  85. // import, where an existing namespace is in the current package and the new
  86. // namespace is a different package.
  87. static auto AddNamespace(Context& context, SemIR::TypeId namespace_type_id,
  88. SemIR::NameId name_id,
  89. SemIR::NameScopeId parent_scope_id,
  90. bool diagnose_duplicate_namespace,
  91. llvm::function_ref<SemIR::InstId()> make_import_id)
  92. -> NamespaceResult {
  93. auto* parent_scope = &context.name_scopes().Get(parent_scope_id);
  94. auto [inserted, entry_id] = parent_scope->LookupOrAdd(
  95. name_id,
  96. // This InstId is temporary and would be overridden if used.
  97. SemIR::InstId::None, SemIR::AccessKind::Public);
  98. if (!inserted) {
  99. const auto& prev_entry = parent_scope->GetEntry(entry_id);
  100. if (!prev_entry.is_poisoned) {
  101. auto prev_inst_id = prev_entry.inst_id;
  102. if (auto namespace_inst =
  103. context.insts().TryGetAs<SemIR::Namespace>(prev_inst_id)) {
  104. if (diagnose_duplicate_namespace) {
  105. auto import_id = make_import_id();
  106. CARBON_CHECK(import_id.has_value());
  107. context.DiagnoseDuplicateName(import_id, prev_inst_id);
  108. }
  109. return {namespace_inst->name_scope_id, prev_inst_id, true};
  110. }
  111. }
  112. }
  113. auto import_id = make_import_id();
  114. CARBON_CHECK(import_id.has_value());
  115. auto import_loc_id = context.insts().GetLocId(import_id);
  116. auto namespace_inst =
  117. SemIR::Namespace{namespace_type_id, SemIR::NameScopeId::None, import_id};
  118. auto namespace_inst_and_loc =
  119. import_loc_id.is_import_ir_inst_id()
  120. ? context.MakeImportedLocAndInst(import_loc_id.import_ir_inst_id(),
  121. namespace_inst)
  122. // TODO: Check that this actually is an `AnyNamespaceId`.
  123. : SemIR::LocIdAndInst(Parse::AnyNamespaceId(import_loc_id.node_id()),
  124. namespace_inst);
  125. auto namespace_id =
  126. context.AddPlaceholderInstInNoBlock(namespace_inst_and_loc);
  127. context.import_ref_ids().push_back(namespace_id);
  128. namespace_inst.name_scope_id =
  129. context.name_scopes().Add(namespace_id, name_id, parent_scope_id);
  130. context.ReplaceInstBeforeConstantUse(namespace_id, namespace_inst);
  131. // Note we have to get the parent scope freshly, creating the imported
  132. // namespace may invalidate the pointer above.
  133. parent_scope = &context.name_scopes().Get(parent_scope_id);
  134. // Diagnose if there's a name conflict, but still produce the namespace to
  135. // supersede the name conflict in order to avoid repeat diagnostics. Names are
  136. // poisoned optimistically by name lookup before checking for imports, so we
  137. // may be overwriting a poisoned entry here.
  138. auto& entry = parent_scope->GetEntry(entry_id);
  139. if (entry.is_poisoned) {
  140. entry.is_poisoned = false;
  141. } else if (!inserted) {
  142. context.DiagnoseDuplicateName(namespace_id, entry.inst_id);
  143. entry.access_kind = SemIR::AccessKind::Public;
  144. }
  145. entry.inst_id = namespace_id;
  146. return {namespace_inst.name_scope_id, namespace_id, false};
  147. }
  148. // Adds a copied namespace to the cache.
  149. static auto CacheCopiedNamespace(
  150. Map<SemIR::NameScopeId, SemIR::NameScopeId>& copied_namespaces,
  151. SemIR::NameScopeId import_scope_id, SemIR::NameScopeId to_scope_id)
  152. -> void {
  153. auto result = copied_namespaces.Insert(import_scope_id, to_scope_id);
  154. CARBON_CHECK(result.is_inserted() || result.value() == to_scope_id,
  155. "Copy result for namespace changed from {0} to {1}",
  156. import_scope_id, to_scope_id);
  157. }
  158. // Copies a namespace from the import IR, returning its ID. This may diagnose
  159. // name conflicts, but that won't change the result because namespaces supersede
  160. // other names in conflicts. The bool on return is true if there was a name
  161. // conflict. copied_namespaces is optional.
  162. static auto CopySingleNameScopeFromImportIR(
  163. Context& context, SemIR::TypeId namespace_type_id,
  164. Map<SemIR::NameScopeId, SemIR::NameScopeId>* copied_namespaces,
  165. SemIR::ImportIRId ir_id, SemIR::InstId import_inst_id,
  166. SemIR::NameScopeId import_scope_id, SemIR::NameScopeId parent_scope_id,
  167. SemIR::NameId name_id) -> NamespaceResult {
  168. // Produce the namespace for the entry.
  169. auto make_import_id = [&]() {
  170. auto entity_name_id = context.entity_names().Add(
  171. {.name_id = name_id,
  172. .parent_scope_id = parent_scope_id,
  173. .bind_index = SemIR::CompileTimeBindIndex::None});
  174. auto import_ir_inst_id = context.import_ir_insts().Add(
  175. {.ir_id = ir_id, .inst_id = import_inst_id});
  176. auto inst_id = context.AddInstInNoBlock(
  177. context.MakeImportedLocAndInst<SemIR::ImportRefLoaded>(
  178. import_ir_inst_id, {.type_id = namespace_type_id,
  179. .import_ir_inst_id = import_ir_inst_id,
  180. .entity_name_id = entity_name_id}));
  181. context.import_ref_ids().push_back(inst_id);
  182. return inst_id;
  183. };
  184. NamespaceResult result =
  185. AddNamespace(context, namespace_type_id, name_id, parent_scope_id,
  186. /*diagnose_duplicate_namespace=*/false, make_import_id);
  187. auto namespace_const_id = context.constant_values().Get(result.inst_id);
  188. context.import_ir_constant_values()[ir_id.index].Set(import_inst_id,
  189. namespace_const_id);
  190. if (copied_namespaces) {
  191. CacheCopiedNamespace(*copied_namespaces, import_scope_id,
  192. result.name_scope_id);
  193. }
  194. return result;
  195. }
  196. // Copies ancestor name scopes from the import IR. Handles the parent traversal.
  197. // Returns the NameScope corresponding to the copied import_parent_scope_id.
  198. static auto CopyAncestorNameScopesFromImportIR(
  199. Context& context, SemIR::TypeId namespace_type_id,
  200. const SemIR::File& import_sem_ir, SemIR::ImportIRId ir_id,
  201. SemIR::NameScopeId import_parent_scope_id,
  202. Map<SemIR::NameScopeId, SemIR::NameScopeId>& copied_namespaces)
  203. -> SemIR::NameScopeId {
  204. // Package-level names don't need work.
  205. if (import_parent_scope_id == SemIR::NameScopeId::Package) {
  206. return import_parent_scope_id;
  207. }
  208. // The scope to add namespaces to. Note this may change while looking at
  209. // parent scopes, if we encounter a namespace that's already added.
  210. auto scope_cursor = SemIR::NameScopeId::Package;
  211. // Build a stack of ancestor namespace names, with the immediate parent first.
  212. llvm::SmallVector<SemIR::NameScopeId> new_namespaces;
  213. while (import_parent_scope_id != SemIR::NameScopeId::Package) {
  214. // If the namespace was already copied, reuse the results.
  215. if (auto result = copied_namespaces.Lookup(import_parent_scope_id)) {
  216. // We inject names at the provided scope, and don't need to keep
  217. // traversing parents.
  218. scope_cursor = result.value();
  219. break;
  220. }
  221. // The namespace hasn't been copied yet, so add it to our list.
  222. const auto& scope = import_sem_ir.name_scopes().Get(import_parent_scope_id);
  223. auto scope_inst =
  224. import_sem_ir.insts().GetAs<SemIR::Namespace>(scope.inst_id());
  225. new_namespaces.push_back(scope_inst.name_scope_id);
  226. import_parent_scope_id = scope.parent_scope_id();
  227. }
  228. // Add ancestor namespace names, starting with the outermost.
  229. for (auto import_scope_id : llvm::reverse(new_namespaces)) {
  230. auto import_scope = import_sem_ir.name_scopes().Get(import_scope_id);
  231. auto name_id =
  232. CopyNameFromImportIR(context, import_sem_ir, import_scope.name_id());
  233. scope_cursor =
  234. CopySingleNameScopeFromImportIR(
  235. context, namespace_type_id, &copied_namespaces, ir_id,
  236. import_scope.inst_id(), import_scope_id, scope_cursor, name_id)
  237. .name_scope_id;
  238. }
  239. return scope_cursor;
  240. }
  241. // Adds an ImportRef for an entity, handling merging if needed.
  242. static auto AddImportRefOrMerge(Context& context, SemIR::ImportIRId ir_id,
  243. const SemIR::File& import_sem_ir,
  244. SemIR::InstId import_inst_id,
  245. SemIR::NameScopeId parent_scope_id,
  246. SemIR::NameId name_id) -> void {
  247. // Leave a placeholder that the inst comes from the other IR.
  248. auto& parent_scope = context.name_scopes().Get(parent_scope_id);
  249. auto [inserted, entry_id] = parent_scope.LookupOrAdd(
  250. name_id,
  251. // This InstId is temporary and would be overridden if used.
  252. SemIR::InstId::None, SemIR::AccessKind::Public);
  253. auto& entry = parent_scope.GetEntry(entry_id);
  254. if (inserted) {
  255. auto entity_name_id = context.entity_names().Add(
  256. {.name_id = name_id,
  257. .parent_scope_id = parent_scope_id,
  258. .bind_index = SemIR::CompileTimeBindIndex::None});
  259. entry.inst_id = AddImportRef(
  260. context, {.ir_id = ir_id, .inst_id = import_inst_id}, entity_name_id);
  261. return;
  262. }
  263. auto inst_id = entry.inst_id;
  264. auto prev_ir_inst = GetCanonicalImportIRInst(context, inst_id);
  265. VerifySameCanonicalImportIRInst(context, inst_id, prev_ir_inst, ir_id,
  266. &import_sem_ir, import_inst_id);
  267. }
  268. namespace {
  269. // A scope in the API file that still needs to be copied to the implementation
  270. // file. Only used for API file imports.
  271. struct TodoScope {
  272. // The scope's instruction in the API file.
  273. SemIR::InstId api_inst_id;
  274. // The scope in the API file.
  275. SemIR::NameScopeId api_scope_id;
  276. // The already-translated scope name in the implementation file.
  277. SemIR::NameId impl_name_id;
  278. // The already-copied parent scope in the implementation file.
  279. SemIR::NameScopeId impl_parent_scope_id;
  280. };
  281. } // namespace
  282. // Adds an ImportRef to a name scope.
  283. static auto AddScopedImportRef(Context& context,
  284. SemIR::NameScopeId parent_scope_id,
  285. SemIR::NameScope& parent_scope,
  286. SemIR::NameId name_id,
  287. SemIR::ImportIRInst import_inst,
  288. SemIR::AccessKind access_kind) -> SemIR::InstId {
  289. // Add an ImportRef for other instructions.
  290. auto impl_entity_name_id = context.entity_names().Add(
  291. {.name_id = name_id,
  292. .parent_scope_id = parent_scope_id,
  293. .bind_index = SemIR::CompileTimeBindIndex::None});
  294. auto import_ref_id = AddImportRef(context, import_inst, impl_entity_name_id);
  295. parent_scope.AddRequired({.name_id = name_id,
  296. .inst_id = import_ref_id,
  297. .access_kind = access_kind});
  298. return import_ref_id;
  299. }
  300. // Imports entries in a specific scope into the current file.
  301. static auto ImportScopeFromApiFile(Context& context,
  302. const SemIR::File& api_sem_ir,
  303. SemIR::NameScopeId api_scope_id,
  304. SemIR::NameScopeId impl_scope_id,
  305. llvm::SmallVector<TodoScope>& todo_scopes)
  306. -> void {
  307. const auto& api_scope = api_sem_ir.name_scopes().Get(api_scope_id);
  308. auto& impl_scope = context.name_scopes().Get(impl_scope_id);
  309. for (const auto& api_entry : api_scope.entries()) {
  310. if (api_entry.is_poisoned) {
  311. continue;
  312. }
  313. auto impl_name_id =
  314. CopyNameFromImportIR(context, api_sem_ir, api_entry.name_id);
  315. if (auto ns =
  316. api_sem_ir.insts().TryGetAs<SemIR::Namespace>(api_entry.inst_id)) {
  317. // Ignore cross-package imports. These will be handled through
  318. // ImportLibrariesFromOtherPackage.
  319. if (api_scope_id == SemIR::NameScopeId::Package) {
  320. const auto& ns_scope = api_sem_ir.name_scopes().Get(ns->name_scope_id);
  321. if (!ns_scope.import_ir_scopes().empty()) {
  322. continue;
  323. }
  324. }
  325. // Namespaces will be recursed into. Name scope creation is delayed in
  326. // order to avoid invalidating api_scope/impl_scope.
  327. todo_scopes.push_back({.api_inst_id = api_entry.inst_id,
  328. .api_scope_id = ns->name_scope_id,
  329. .impl_name_id = impl_name_id,
  330. .impl_parent_scope_id = impl_scope_id});
  331. } else {
  332. // Add an ImportRef for other instructions.
  333. AddScopedImportRef(context, impl_scope_id, impl_scope, impl_name_id,
  334. {.ir_id = SemIR::ImportIRId::ApiForImpl,
  335. .inst_id = api_entry.inst_id},
  336. api_entry.access_kind);
  337. }
  338. }
  339. }
  340. auto ImportApiFile(Context& context, SemIR::TypeId namespace_type_id,
  341. const SemIR::File& api_sem_ir) -> void {
  342. context.import_ir_constant_values()[SemIR::ImportIRId::ApiForImpl.index].Set(
  343. SemIR::Namespace::PackageInstId,
  344. context.constant_values().Get(SemIR::Namespace::PackageInstId));
  345. llvm::SmallVector<TodoScope> todo_scopes = {};
  346. ImportScopeFromApiFile(context, api_sem_ir, SemIR::NameScopeId::Package,
  347. SemIR::NameScopeId::Package, todo_scopes);
  348. while (!todo_scopes.empty()) {
  349. auto todo_scope = todo_scopes.pop_back_val();
  350. auto impl_scope_id =
  351. CopySingleNameScopeFromImportIR(
  352. context, namespace_type_id, /*copied_namespaces=*/nullptr,
  353. SemIR::ImportIRId::ApiForImpl, todo_scope.api_inst_id,
  354. todo_scope.api_scope_id, todo_scope.impl_parent_scope_id,
  355. todo_scope.impl_name_id)
  356. .name_scope_id;
  357. ImportScopeFromApiFile(context, api_sem_ir, todo_scope.api_scope_id,
  358. impl_scope_id, todo_scopes);
  359. }
  360. }
  361. auto ImportLibrariesFromCurrentPackage(
  362. Context& context, SemIR::TypeId namespace_type_id,
  363. llvm::ArrayRef<SemIR::ImportIR> import_irs) -> void {
  364. for (auto import_ir : import_irs) {
  365. auto ir_id = AddImportIR(context, import_ir);
  366. context.import_ir_constant_values()[ir_id.index].Set(
  367. SemIR::Namespace::PackageInstId,
  368. context.constant_values().Get(SemIR::Namespace::PackageInstId));
  369. for (const auto import_inst_id :
  370. import_ir.sem_ir->inst_blocks().Get(SemIR::InstBlockId::Exports)) {
  371. auto import_inst = import_ir.sem_ir->insts().Get(import_inst_id);
  372. auto [import_name_id, import_parent_scope_id] =
  373. GetImportName(*import_ir.sem_ir, import_inst);
  374. Map<SemIR::NameScopeId, SemIR::NameScopeId> copied_namespaces;
  375. auto name_id =
  376. CopyNameFromImportIR(context, *import_ir.sem_ir, import_name_id);
  377. SemIR::NameScopeId parent_scope_id = CopyAncestorNameScopesFromImportIR(
  378. context, namespace_type_id, *import_ir.sem_ir, ir_id,
  379. import_parent_scope_id, copied_namespaces);
  380. if (auto import_namespace_inst = import_inst.TryAs<SemIR::Namespace>()) {
  381. // Namespaces are always imported because they're essential for
  382. // qualifiers, and the type is simple.
  383. CopySingleNameScopeFromImportIR(
  384. context, namespace_type_id, &copied_namespaces, ir_id,
  385. import_inst_id, import_namespace_inst->name_scope_id,
  386. parent_scope_id, name_id);
  387. } else {
  388. AddImportRefOrMerge(context, ir_id, *import_ir.sem_ir, import_inst_id,
  389. parent_scope_id, name_id);
  390. }
  391. }
  392. // If an import of the current package caused an error for the imported
  393. // file, it transitively affects the current file too.
  394. if (import_ir.sem_ir->name_scopes()
  395. .Get(SemIR::NameScopeId::Package)
  396. .has_error()) {
  397. context.name_scopes().Get(SemIR::NameScopeId::Package).set_has_error();
  398. }
  399. }
  400. }
  401. auto ImportLibrariesFromOtherPackage(Context& context,
  402. SemIR::TypeId namespace_type_id,
  403. SemIR::InstId import_decl_id,
  404. IdentifierId package_id,
  405. llvm::ArrayRef<SemIR::ImportIR> import_irs,
  406. bool has_load_error) -> void {
  407. CARBON_CHECK(has_load_error || !import_irs.empty(),
  408. "There should be either a load error or at least one IR.");
  409. auto name_id = SemIR::NameId::ForIdentifier(package_id);
  410. NamespaceResult result = AddNamespace(
  411. context, namespace_type_id, name_id, SemIR::NameScopeId::Package,
  412. /*diagnose_duplicate_namespace=*/true, [&] { return import_decl_id; });
  413. auto namespace_const_id = context.constant_values().Get(result.inst_id);
  414. auto& scope = context.name_scopes().Get(result.name_scope_id);
  415. scope.set_is_closed_import(
  416. !result.is_duplicate_of_namespace_in_current_package);
  417. for (auto import_ir : import_irs) {
  418. auto ir_id = AddImportIR(context, import_ir);
  419. scope.AddImportIRScope({ir_id, SemIR::NameScopeId::Package});
  420. context.import_ir_constant_values()[ir_id.index].Set(
  421. SemIR::Namespace::PackageInstId, namespace_const_id);
  422. }
  423. if (has_load_error) {
  424. scope.set_has_error();
  425. }
  426. }
  427. // Looks up a name in a scope imported from another package. An `identifier` is
  428. // provided if `name_id` corresponds to an identifier in the current file;
  429. // otherwise, `name_id` is file-agnostic and can be used directly.
  430. static auto LookupNameInImport(const SemIR::File& import_ir,
  431. SemIR::NameScopeId import_scope_id,
  432. SemIR::NameId name_id,
  433. llvm::StringRef identifier)
  434. -> const Carbon::SemIR::NameScope::Entry* {
  435. // Determine the NameId in the import IR.
  436. SemIR::NameId import_name_id = name_id;
  437. if (!identifier.empty()) {
  438. auto import_identifier_id = import_ir.identifiers().Lookup(identifier);
  439. if (!import_identifier_id.has_value()) {
  440. // Name doesn't exist in the import IR.
  441. return nullptr;
  442. }
  443. import_name_id = SemIR::NameId::ForIdentifier(import_identifier_id);
  444. }
  445. // Look up the name in the import scope.
  446. const auto& import_scope = import_ir.name_scopes().Get(import_scope_id);
  447. auto import_scope_entry_id = import_scope.Lookup(import_name_id);
  448. if (!import_scope_entry_id) {
  449. // Name doesn't exist in the import scope.
  450. return nullptr;
  451. }
  452. const auto& import_scope_entry =
  453. import_scope.GetEntry(*import_scope_entry_id);
  454. if (import_scope_entry.access_kind != SemIR::AccessKind::Public) {
  455. // Ignore cross-package non-public names.
  456. return nullptr;
  457. }
  458. return &import_scope_entry;
  459. }
  460. // Adds a namespace that points to one in another package.
  461. static auto AddNamespaceFromOtherPackage(Context& context,
  462. SemIR::ImportIRId import_ir_id,
  463. SemIR::InstId import_inst_id,
  464. SemIR::Namespace import_ns,
  465. SemIR::NameScopeId parent_scope_id,
  466. SemIR::NameId name_id)
  467. -> SemIR::InstId {
  468. auto namespace_type_id =
  469. context.GetSingletonType(SemIR::NamespaceType::SingletonInstId);
  470. NamespaceResult result = CopySingleNameScopeFromImportIR(
  471. context, namespace_type_id, /*copied_namespaces=*/nullptr, import_ir_id,
  472. import_inst_id, import_ns.name_scope_id, parent_scope_id, name_id);
  473. auto& scope = context.name_scopes().Get(result.name_scope_id);
  474. scope.set_is_closed_import(
  475. !result.is_duplicate_of_namespace_in_current_package);
  476. scope.AddImportIRScope({import_ir_id, import_ns.name_scope_id});
  477. return result.inst_id;
  478. }
  479. auto ImportNameFromOtherPackage(
  480. Context& context, SemIRLoc loc, SemIR::NameScopeId scope_id,
  481. llvm::ArrayRef<std::pair<SemIR::ImportIRId, SemIR::NameScopeId>>
  482. import_ir_scopes,
  483. SemIR::NameId name_id) -> SemIR::InstId {
  484. // If the name is an identifier, get the string first so that it can be shared
  485. // when there are multiple IRs.
  486. llvm::StringRef identifier;
  487. if (auto identifier_id = name_id.AsIdentifierId();
  488. identifier_id.has_value()) {
  489. identifier = context.identifiers().Get(identifier_id);
  490. CARBON_CHECK(!identifier.empty());
  491. }
  492. // Annotate diagnostics as occurring during this name lookup.
  493. DiagnosticAnnotationScope annotate_diagnostics(
  494. &context.emitter(), [&](auto& builder) {
  495. CARBON_DIAGNOSTIC(InNameLookup, Note, "in name lookup for `{0}`",
  496. SemIR::NameId);
  497. builder.Note(loc, InNameLookup, name_id);
  498. });
  499. // Although we track the result here and look in each IR, we pretty much use
  500. // the first result.
  501. auto result_id = SemIR::InstId::None;
  502. // The canonical IR and inst_id for where `result_id` came from, which may be
  503. // indirectly imported. This is only resolved on a conflict, when it can be
  504. // used to determine the conflict is actually the same instruction.
  505. std::optional<SemIR::ImportIRInst> canonical_result_inst;
  506. for (auto [import_ir_id, import_scope_id] : import_ir_scopes) {
  507. auto& import_ir = context.import_irs().Get(import_ir_id);
  508. const auto* import_scope_entry = LookupNameInImport(
  509. *import_ir.sem_ir, import_scope_id, name_id, identifier);
  510. if (!import_scope_entry) {
  511. continue;
  512. }
  513. auto import_inst =
  514. import_ir.sem_ir->insts().Get(import_scope_entry->inst_id);
  515. if (import_inst.Is<SemIR::AnyImportRef>()) {
  516. // This entity was added to name lookup by using an import, and is not
  517. // exported.
  518. continue;
  519. }
  520. // Add the first result found.
  521. if (!result_id.has_value()) {
  522. // If the imported instruction is a namespace, we add it directly instead
  523. // of as an ImportRef.
  524. if (auto import_ns = import_inst.TryAs<SemIR::Namespace>()) {
  525. result_id = AddNamespaceFromOtherPackage(context, import_ir_id,
  526. import_scope_entry->inst_id,
  527. *import_ns, scope_id, name_id);
  528. } else {
  529. result_id = AddScopedImportRef(
  530. context, scope_id, context.name_scopes().Get(scope_id), name_id,
  531. {.ir_id = import_ir_id, .inst_id = import_scope_entry->inst_id},
  532. SemIR::AccessKind::Public);
  533. LoadImportRef(context, result_id);
  534. }
  535. continue;
  536. }
  537. // When namespaces collide between files, merge lookup in the scopes.
  538. if (auto import_ns = import_inst.TryAs<SemIR::Namespace>()) {
  539. if (auto ns = context.insts().TryGetAs<SemIR::Namespace>(result_id)) {
  540. auto& name_scope = context.name_scopes().Get(ns->name_scope_id);
  541. name_scope.AddImportIRScope({import_ir_id, import_ns->name_scope_id});
  542. continue;
  543. }
  544. }
  545. // When there's a name collision, they need to either be the same canonical
  546. // instruction, or we'll diagnose.
  547. if (!canonical_result_inst) {
  548. canonical_result_inst = GetCanonicalImportIRInst(context, result_id);
  549. }
  550. VerifySameCanonicalImportIRInst(context, result_id, *canonical_result_inst,
  551. import_ir_id, import_ir.sem_ir,
  552. import_scope_entry->inst_id);
  553. }
  554. return result_id;
  555. }
  556. } // namespace Carbon::Check