[CodeCompletion] Move CompletionOverrideLookup to its own file

Author: ahoppen

include/swift/IDE/CompletionOverrideLookup.h

@@ -0,0 +1,128 @@
+//===--- CompletionOverrideLookup.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_COMPLETION_OVERRIDE_LOOKUP_H
+#define SWIFT_IDE_COMPLETION_OVERRIDE_LOOKUP_H
+
+#include "swift/AST/NameLookup.h"
+#include "swift/IDE/CodeCompletion.h"
+#include "swift/Sema/IDETypeChecking.h"
+
+namespace swift {
+namespace ide {
+
+class CompletionOverrideLookup : public swift::VisibleDeclConsumer {
+  CodeCompletionResultSink &Sink;
+  ASTContext &Ctx;
+  const DeclContext *CurrDeclContext;
+  SmallVectorImpl<StringRef> &ParsedKeywords;
+  SourceLoc introducerLoc;
+
+  bool hasFuncIntroducer = false;
+  bool hasVarIntroducer = false;
+  bool hasTypealiasIntroducer = false;
+  bool hasInitializerModifier = false;
+  bool hasAccessModifier = false;
+  bool hasOverride = false;
+  bool hasOverridabilityModifier = false;
+  bool hasStaticOrClass = false;
+
+public:
+  CompletionOverrideLookup(CodeCompletionResultSink &Sink, ASTContext &Ctx,
+                           const DeclContext *CurrDeclContext,
+                           SmallVectorImpl<StringRef> &ParsedKeywords,
+                           SourceLoc introducerLoc)
+      : Sink(Sink), Ctx(Ctx), CurrDeclContext(CurrDeclContext),
+        ParsedKeywords(ParsedKeywords), introducerLoc(introducerLoc) {
+    hasFuncIntroducer = isKeywordSpecified("func");
+    hasVarIntroducer = isKeywordSpecified("var") ||
+                       isKeywordSpecified("let");
+    hasTypealiasIntroducer = isKeywordSpecified("typealias");
+    hasInitializerModifier = isKeywordSpecified("required") ||
+                             isKeywordSpecified("convenience");
+    hasAccessModifier = isKeywordSpecified("private") ||
+                        isKeywordSpecified("fileprivate") ||
+                        isKeywordSpecified("internal") ||
+                        isKeywordSpecified("public") ||
+                        isKeywordSpecified("open");
+    hasOverride = isKeywordSpecified("override");
+    hasOverridabilityModifier =
+        isKeywordSpecified("final") || isKeywordSpecified("open");
+    hasStaticOrClass = isKeywordSpecified(getTokenText(tok::kw_class)) ||
+                       isKeywordSpecified(getTokenText(tok::kw_static));
+  }
+
+  bool isKeywordSpecified(StringRef Word) {
+    return std::find(ParsedKeywords.begin(), ParsedKeywords.end(), Word) !=
+           ParsedKeywords.end();
+  }
+
+  bool missingOverride(DeclVisibilityKind Reason) {
+    return !hasOverride && Reason == DeclVisibilityKind::MemberOfSuper &&
+           !CurrDeclContext->getSelfProtocolDecl();
+  }
+
+  /// Add an access modifier (i.e. `public`) to \p Builder is necessary.
+  /// Returns \c true if the modifier is actually added, \c false otherwise.
+  bool addAccessControl(const ValueDecl *VD,
+                        CodeCompletionResultBuilder &Builder);
+
+  /// Return type if the result type if \p VD should be represented as opaque
+  /// result type.
+  Type getOpaqueResultType(const ValueDecl *VD, DeclVisibilityKind Reason,
+                           DynamicLookupInfo dynamicLookupInfo);
+
+  void addValueOverride(const ValueDecl *VD, DeclVisibilityKind Reason,
+                        DynamicLookupInfo dynamicLookupInfo,
+                        CodeCompletionResultBuilder &Builder,
+                        bool hasDeclIntroducer);
+
+  void addMethodOverride(const FuncDecl *FD, DeclVisibilityKind Reason,
+                         DynamicLookupInfo dynamicLookupInfo);
+
+  void addVarOverride(const VarDecl *VD, DeclVisibilityKind Reason,
+                      DynamicLookupInfo dynamicLookupInfo);
+
+  void addSubscriptOverride(const SubscriptDecl *SD, DeclVisibilityKind Reason,
+                            DynamicLookupInfo dynamicLookupInfo);
+
+  void addTypeAlias(const AssociatedTypeDecl *ATD, DeclVisibilityKind Reason,
+                    DynamicLookupInfo dynamicLookupInfo);
+
+  void addConstructor(const ConstructorDecl *CD, DeclVisibilityKind Reason,
+                      DynamicLookupInfo dynamicLookupInfo);
+
+  // Implement swift::VisibleDeclConsumer.
+  void foundDecl(ValueDecl *D, DeclVisibilityKind Reason,
+                 DynamicLookupInfo dynamicLookupInfo) override;
+
+  void addDesignatedInitializers(NominalTypeDecl *NTD);
+
+  void addAssociatedTypes(NominalTypeDecl *NTD);
+
+  static StringRef
+  getResultBuilderDocComment(ResultBuilderBuildFunction function);
+
+  void addResultBuilderBuildCompletion(NominalTypeDecl *builder,
+                                       Type componentType,
+                                       ResultBuilderBuildFunction function);
+
+  /// Add completions for the various "build" functions in a result builder.
+  void addResultBuilderBuildCompletions(NominalTypeDecl *builder);
+
+  void getOverrideCompletions(SourceLoc Loc);
+};
+
+} // end namespace ide
+} // end namespace swift
+
+#endif // SWIFT_IDE_COMPLETION_OVERRIDE_LOOKUP_H

lib/IDE/CMakeLists.txt

@@ -13,6 +13,7 @@ add_swift_host_library(swiftIDE STATIC
   CompileInstance.cpp
   CompletionInstance.cpp
   CompletionLookup.cpp
+  CompletionOverrideLookup.cpp
   ConformingMethodList.cpp
   DependencyChecking.cpp
   ExprContextAnalysis.cpp