mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
ec1b146581
[IDE] Move primitive completion function label into CodeCompletionStringBuilder [IDE] NFC: Remove unneeded string builder methods on CodeCompletionResultBuilder [IDE] Move addValueBaseName into CodeCompletionStringBuilder [IDE] Make CodeCompletionResultBuilder a CodeCompletionStringBuilder [IDE] Explicitly pass DeclContext in CodeCompletionStringBuilder [IDE] Reduce includes in CodeCompletionStringBuilder.h
105 lines
3.1 KiB
C++
105 lines
3.1 KiB
C++
//===--- CodeCompletionStringPrinter.h ------------------------------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2022 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef SWIFT_IDE_CODECOMPLETIONRESULTSTRINGPRINTER_H
|
|
#define SWIFT_IDE_CODECOMPLETIONRESULTSTRINGPRINTER_H
|
|
|
|
#include "swift/AST/ASTPrinter.h"
|
|
#include "swift/IDE/CodeCompletionString.h"
|
|
|
|
namespace swift {
|
|
namespace ide {
|
|
|
|
/// 'ASTPrinter' printing to 'CodeCompletionString' with appropriate ChunkKind.
|
|
/// This is mainly used for printing types and override completions.
|
|
class CodeCompletionStringPrinter : public ASTPrinter {
|
|
protected:
|
|
using ChunkKind = CodeCompletionString::Chunk::ChunkKind;
|
|
|
|
private:
|
|
CodeCompletionStringBuilder &Builder;
|
|
SmallString<16> Buffer;
|
|
ChunkKind CurrChunkKind = ChunkKind::Text;
|
|
ChunkKind NextChunkKind = ChunkKind::Text;
|
|
SmallVector<PrintStructureKind, 2> StructureStack;
|
|
unsigned int TypeDepth = 0;
|
|
bool InPreamble = false;
|
|
|
|
bool isCurrentStructureKind(PrintStructureKind Kind) const {
|
|
return !StructureStack.empty() && StructureStack.back() == Kind;
|
|
}
|
|
|
|
bool isInType() const { return TypeDepth > 0; }
|
|
|
|
std::optional<ChunkKind>
|
|
getChunkKindForPrintNameContext(PrintNameContext context) const;
|
|
|
|
std::optional<ChunkKind>
|
|
getChunkKindForStructureKind(PrintStructureKind Kind) const;
|
|
|
|
void startNestedGroup(ChunkKind Kind);
|
|
|
|
void endNestedGroup();
|
|
|
|
protected:
|
|
void setNextChunkKind(ChunkKind Kind) { NextChunkKind = Kind; }
|
|
|
|
public:
|
|
CodeCompletionStringPrinter(CodeCompletionStringBuilder &Builder)
|
|
: Builder(Builder) {}
|
|
|
|
~CodeCompletionStringPrinter() {
|
|
// Flush the remainings.
|
|
flush();
|
|
}
|
|
|
|
void flush();
|
|
|
|
/// Start \c AttributeAndModifierListBegin group. This must be called before
|
|
/// any attributes/modifiers printed to the output when printing an override
|
|
/// compleion.
|
|
void startPreamble();
|
|
|
|
/// End the current \c AttributeAndModifierListBegin group if it's still open.
|
|
/// This is automatically called before the main part of the signature.
|
|
void endPremable();
|
|
|
|
/// Implement \c ASTPrinter .
|
|
public:
|
|
void printText(StringRef Text) override;
|
|
|
|
void printTypeRef(
|
|
Type T, const TypeDecl *TD, Identifier Name,
|
|
PrintNameContext NameContext = PrintNameContext::Normal) override;
|
|
|
|
void printDeclLoc(const Decl *D) override;
|
|
|
|
void printDeclNameEndLoc(const Decl *D) override;
|
|
|
|
void printNamePre(PrintNameContext context) override;
|
|
|
|
void printNamePost(PrintNameContext context) override;
|
|
|
|
void printTypePre(const TypeLoc &TL) override;
|
|
|
|
void printTypePost(const TypeLoc &TL) override;
|
|
|
|
void printStructurePre(PrintStructureKind Kind, const Decl *D) override;
|
|
|
|
void printStructurePost(PrintStructureKind Kind, const Decl *D) override;
|
|
};
|
|
|
|
} // end namespace ide
|
|
} // end namespace swift
|
|
|
|
#endif // SWIFT_IDE_CODECOMPLETIONRESULTSTRINGPRINTER_H
|