Merge pull request #2188 from benlangmuir/move-index-ast-walker-out-of-sourcekit

Author: swift-ci

include/swift/Index/Index.h

@@ -0,0 +1,29 @@
+//===--- Index.h - Swift Indexing -------------------------------*- C++ -*-===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2014 - 2016 Apple Inc. and the Swift project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See http://swift.org/LICENSE.txt for license information
+// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef SWIFT_INDEX_INDEX_H
+#define SWIFT_INDEX_INDEX_H
+
+#include "swift/Index/IndexDataConsumer.h"
+
+namespace swift {
+class ModuleDecl;
+
+namespace index {
+
+void indexModule(ModuleDecl *module, StringRef hash, unsigned bufferID,
+                 IndexDataConsumer &consumer);
+
+} // end namespace index
+} // end namespace swift
+
+#endif // SWIFT_INDEX_INDEX_H

include/swift/Index/IndexDataConsumer.h

@@ -0,0 +1,44 @@
+//===--- IndexDataConsumer.h - Consumer of indexing information -*- C++ -*-===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2014 - 2016 Apple Inc. and the Swift project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See http://swift.org/LICENSE.txt for license information
+// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef SWIFT_INDEX_INDEXDATACONSUMER_H
+#define SWIFT_INDEX_INDEXDATACONSUMER_H
+
+#include "swift/Index/IndexSymbol.h"
+
+namespace swift {
+namespace index {
+
+class IndexDataConsumer {
+  virtual void anchor();
+
+public:
+  virtual ~IndexDataConsumer() {}
+
+  virtual void failed(StringRef error) = 0;
+  virtual void warning(StringRef warning) {}
+  virtual bool enableWarnings() { return false; }
+
+  virtual bool recordHash(StringRef hash, bool isKnown) = 0;
+  virtual bool startDependency(SymbolKind kind, StringRef name, StringRef path,
+                               bool isSystem, StringRef hash) = 0;
+  virtual bool finishDependency(SymbolKind kind) = 0;
+  virtual bool startSourceEntity(const IndexSymbol &symbol) = 0;
+  virtual bool recordRelatedEntity(const IndexSymbol &symbol) = 0;
+  virtual bool finishSourceEntity(SymbolKind kind, SymbolSubKind subKind,
+                                  bool isRef) = 0;
+};
+
+} // end namespace index
+} // end namespace swift
+
+#endif // SWIFT_INDEX_INDEXDATACONSUMER_H

include/swift/Index/IndexSymbol.h

@@ -0,0 +1,119 @@
+//===--- IndexSymbol.h - Index symbol data types ----------------*- C++ -*-===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2014 - 2016 Apple Inc. and the Swift project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See http://swift.org/LICENSE.txt for license information
+// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef SWIFT_INDEX_INDEXSYMBOL_H
+#define SWIFT_INDEX_INDEXSYMBOL_H
+
+#include "swift/Basic/LLVM.h"
+#include "llvm/ADT/SmallString.h"
+
+namespace swift {
+class Decl;
+
+namespace index {
+
+enum class SymbolKind {
+  Unknown,
+
+  Module,
+  ClangModule, // FIXME: collapse into Module and use a separate Language field.
+  SourceFile,
+
+  Enum,
+  Struct,
+  Class,
+  Protocol,
+  Extension,
+
+  TypeAlias,
+  AssociatedType,
+  GenericTypeParam,
+
+  Function,
+  PrefixOperator,
+  PostfixOperator,
+  InfixOperator,
+
+  LocalVariable,
+  GlobalVariable,
+  ParamVariable,
+
+  Accessor,
+  Subscript,
+  EnumElement,
+
+  InstanceMethod,
+  ClassMethod,
+  StaticMethod,
+  InstanceProperty,
+  ClassProperty,
+  StaticProperty,
+
+  Constructor,
+  Destructor,
+};
+
+enum class SymbolSubKind {
+  None,
+
+  AccessorGetter,
+  AccessorSetter,
+  AccessorWillSet,
+  AccessorDidSet,
+  AccessorMaterializeForSet,
+  AccessorAddressor,
+  AccessorMutableAddressor,
+
+  ExtensionOfStruct,
+  ExtensionOfClass,
+  ExtensionOfEnum,
+  ExtensionOfProtocol,
+};
+
+struct IndexSymbol {
+  enum TypeKind { Base, FuncDecl, CallReference };
+  TypeKind entityType = Base;
+
+  SymbolKind kind;
+  SymbolSubKind subKind = SymbolSubKind::None;
+  bool isRef;
+  SmallString<32> name;
+  SmallString<64> USR;
+  SmallString<16> group;
+  unsigned line = 0;
+  unsigned column = 0;
+
+  IndexSymbol() = default;
+
+protected:
+  IndexSymbol(TypeKind TK) : entityType(TK) {}
+};
+
+struct FuncDeclIndexSymbol : public IndexSymbol {
+  bool IsTestCandidate = false;
+
+  FuncDeclIndexSymbol() : IndexSymbol(FuncDecl) {}
+};
+
+struct CallRefIndexSymbol : public IndexSymbol {
+  llvm::SmallString<64> ReceiverUSR;
+  bool IsDynamic = false;
+
+  CallRefIndexSymbol() : IndexSymbol(CallReference) {}
+};
+
+SymbolKind getSymbolKindForDecl(const Decl *D);
+
+} // end namespace index
+} // end namespace swift
+
+#endif // SWIFT_INDEX_INDEXSYMBOL_H

lib/CMakeLists.txt

@@ -4,6 +4,7 @@ add_subdirectory(Basic)
 add_subdirectory(ClangImporter)
 add_subdirectory(Driver)
 add_subdirectory(Frontend)
+add_subdirectory(Index)
 add_subdirectory(IDE)
 add_subdirectory(Immediate)
 add_subdirectory(IRGen)

lib/Index/CMakeLists.txt

@@ -0,0 +1,6 @@
+add_swift_library(swiftIndex
+  Index.cpp
+  IndexDataConsumer.cpp
+  IndexSymbol.cpp
+  LINK_LIBRARIES
+    swiftAST)