Merge pull request swiftlang#196 from dabelknap/overloads

Fix spacing for binary operator overloads

Author: allevato

Sources/SwiftFormatPrettyPrint/TokenStreamCreator.swift

@@ -189,6 +189,9 @@ private final class TokenStreamCreator: SyntaxVisitor {
   // MARK: - Function and function-like declaration nodes (initializers, deinitializers, subscripts)
 
   override func visit(_ node: FunctionDeclSyntax) {
+    if case .spacedBinaryOperator = node.identifier.tokenKind {
+      after(node.identifier.lastToken, tokens: .space)
+    }
     arrangeFunctionLikeDecl(
       node,
       attributes: node.attributes,

Tests/SwiftFormatPrettyPrintTests/FunctionDeclTests.swift

@@ -373,4 +373,22 @@ public class FunctionDeclTests: PrettyPrintTestCase {
       """
     assertPrettyPrintEqual(input: input, expected: wrapped, linelength: 12)
   }
+
+  public func testOperatorOverloads() {
+    let input =
+      """
+      func < (lhs: Position, rhs: Position) -> Bool {
+        // do stuff
+      }
+
+      func + (left: [Int], right: [Int]) -> [Int] {
+        // add two arrays
+      }
+
+      func ⊕ (left: Tensor, right: Tensor) -> Tensor {
+        // tensor addition
+      }
+      """
+    assertPrettyPrintEqual(input: input, expected: input + "\n", linelength: 50)
+  }
 }