Allow SourceEntityWalker to visit operator decls, so that they can get indented properly. rdar://25246628

Author: nkcsgexi

lib/IDE/SourceEntityWalker.cpp

@@ -107,6 +107,10 @@ bool SemaAnnotator::walkToDeclPre(Decl *D) {
     if (!handleImports(Import))
       return false;
 
+  } else if (auto OpD = dyn_cast<OperatorDecl>(D)) {
+    Loc = OpD->getLoc();
+    if (Loc.isValid())
+      NameLen = OpD->getName().getLength();
   } else {
     return true;
   }

test/SourceKit/CodeFormat/indent-basic.swift

@@ -13,6 +13,10 @@ foo(a: [
   3: 3
 ])
 
+operator <> {
+x
+}
+
 // RUN: %sourcekitd-test -req=format -line=1 -length=1 %s >%t.response
 // RUN: %sourcekitd-test -req=format -line=2 -length=1 %s >>%t.response
 // RUN: %sourcekitd-test -req=format -line=3 -length=1 %s >>%t.response
@@ -23,6 +27,7 @@ foo(a: [
 // RUN: %sourcekitd-test -req=format -line=8 -length=1 %s >>%t.response
 // RUN: %sourcekitd-test -req=format -line=9 -length=1 %s >>%t.response
 // RUN: %sourcekitd-test -req=format -line=13 -length=1 %s >>%t.response
+// RUN: %sourcekitd-test -req=format -line=17 -length=1 %s >>%t.response
 // RUN: FileCheck --strict-whitespace %s <%t.response
 
 // CHECK: key.sourcetext: "class Foo {"
@@ -36,3 +41,4 @@ foo(a: [
 // CHECK: key.sourcetext: "}"
 //                        "foo(a: ["
 // CHECK: key.sourcetext: "           3: 3"
+// CHECK: key.sourcetext: "    x"