averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

[InterfacePrinting] Printing non-doc comments at the beginning of a file, rdar://21710187.

Browse Source 
When printing the interface extracted from a Swift file, we print every comments before
Swift code starts.

Swift SVN r29989
This commit is contained in:
Xi Ge
2015-07-08 20:10:57 +00:00
parent 76ec395dae
commit 6f33efd5f9
2 changed files with 53 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
lib/IDE/ModuleInterfacePrinting.cpp
View File

@@ -20,7 +20,9 @@
#include "swift/Basic/PrimitiveParsing.h"
#include "swift/ClangImporter/ClangImporter.h"
#include "swift/ClangImporter/ClangModule.h"
#include "swift/Parse/Token.h"
#include "swift/Serialization/ModuleFile.h"
#include "swift/Subsystems.h"
#include "swift/Serialization/SerializedModuleLoader.h"
#include "clang/AST/ASTContext.h"
#include "clang/AST/Decl.h"
@@ -396,6 +398,51 @@ void swift::ide::printSubmoduleInterface(
}
}
static SourceLoc getDeclStartPosition(SourceFile &File) {
llvm::Optional<SourceLoc> Winner;
SourceManager &SM = File.getASTContext().SourceMgr;
for (auto D : File.Decls) {
auto Start = !D->getRawComment().isEmpty() &&
SM.isBeforeInBuffer(D->getRawComment().Comments[0].Range.getStart(),
D->getSourceRange().Start) ?
D->getRawComment().Comments[0].Range.getStart() :
D->getSourceRange().Start;
if (Winner.hasValue()) {
if (SM.isBeforeInBuffer(Start, Winner.getValue())) {
Winner = Start;
}
} else {
Winner = Start;
}
}
return Winner.hasValue() ? Winner.getValue() : SourceLoc();
}
static void printUntilFirstDeclStarts(SourceFile &File, ASTPrinter &Printer) {
if (!File.getBufferID().hasValue())
return;
auto DeclStartLoc = getDeclStartPosition(File);
auto &SM = File.getASTContext().SourceMgr;
auto Tokens = swift::tokenize(File.getASTContext().LangOpts, SM,
File.getBufferID().getValue());
for (auto It = Tokens.begin(); It < Tokens.end() &&
(DeclStartLoc.isInvalid() || SM.isBeforeInBuffer(It->getLoc(),
DeclStartLoc)); ++ It) {
Printer << It->getText();
}
}
void swift::ide::printSwiftSourceInterface(SourceFile &File,
ASTPrinter &Printer,
const PrintOptions &Options) {
// We print all comments before the fist line of Swift code.
printUntilFirstDeclStarts(File, Printer);
File.print(Printer, Options);
}
void swift::ide::printHeaderInterface(
StringRef Filename,
ASTContext &Ctx,