mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
7f8a0e8a6c
Add ModuleImplicitImportsRequest, which computes the modules that should be implicitly imported by each file of a given module. Use this request in import resolution to add all the necessary implicit imports. The request computes the implicit imports by consulting the ImplicitImportInfo, which ModuleDecl can now be created with. This allows us to remove uses of `SourceFile::addImports` in favor of adding modules needed to be implicitly imported to the ImplicitImportInfo.
60 lines
2.5 KiB
C++
60 lines
2.5 KiB
C++
//===--- TestContext.cpp - Helper for setting up ASTContexts --------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "TestContext.h"
|
|
#include "swift/AST/Module.h"
|
|
#include "swift/AST/ParseRequests.h"
|
|
#include "swift/Strings.h"
|
|
#include "swift/Subsystems.h"
|
|
|
|
using namespace swift;
|
|
using namespace swift::unittest;
|
|
|
|
static Decl *createOptionalType(ASTContext &ctx, SourceFile *fileForLookups,
|
|
Identifier name) {
|
|
auto wrapped = new (ctx) GenericTypeParamDecl(fileForLookups,
|
|
ctx.getIdentifier("Wrapped"),
|
|
SourceLoc(), /*depth*/0,
|
|
/*index*/0);
|
|
auto params = GenericParamList::create(ctx, SourceLoc(), wrapped,
|
|
SourceLoc());
|
|
auto decl = new (ctx) EnumDecl(SourceLoc(), name, SourceLoc(),
|
|
/*inherited*/{}, params, fileForLookups);
|
|
wrapped->setDeclContext(decl);
|
|
return decl;
|
|
}
|
|
|
|
TestContext::TestContext(ShouldDeclareOptionalTypes optionals)
|
|
: Ctx(*ASTContext::get(LangOpts, TypeCheckerOpts, SearchPathOpts, SourceMgr,
|
|
Diags)) {
|
|
registerParseRequestFunctions(Ctx.evaluator);
|
|
registerTypeCheckerRequestFunctions(Ctx.evaluator);
|
|
auto stdlibID = Ctx.getIdentifier(STDLIB_NAME);
|
|
auto *module = ModuleDecl::create(stdlibID, Ctx);
|
|
Ctx.LoadedModules[stdlibID] = module;
|
|
|
|
FileForLookups = new (Ctx) SourceFile(*module, SourceFileKind::Library,
|
|
/*buffer*/ None, /*keeps token*/ false);
|
|
module->addFile(*FileForLookups);
|
|
|
|
if (optionals == DeclareOptionalTypes) {
|
|
SmallVector<Decl *, 2> optionalTypes;
|
|
optionalTypes.push_back(createOptionalType(
|
|
Ctx, FileForLookups, Ctx.getIdentifier("Optional")));
|
|
optionalTypes.push_back(createOptionalType(
|
|
Ctx, FileForLookups, Ctx.getIdentifier("ImplicitlyUnwrappedOptional")));
|
|
|
|
Ctx.evaluator.cacheOutput(ParseSourceFileRequest{FileForLookups},
|
|
Ctx.AllocateCopy(optionalTypes));
|
|
}
|
|
}
|