Add support for imports.

Author: dabelknap

Sources/PrettyPrint/TokenStreamCreator.swift

@@ -522,6 +522,9 @@ private final class TokenStreamCreator: SyntaxVisitor {
   }
 
   override func visit(_ node: ImportDeclSyntax) {
+    after(node.attributes?.lastToken, tokens: .space)
+    after(node.importTok, tokens: .space)
+    after(node.importKind, tokens: .space)
     super.visit(node)
   }
 

Tests/PrettyPrinterTests/ImportTests.swift

@@ -0,0 +1,25 @@
+public class ImportTests: PrettyPrintTestCase {
+  public func testImports() {
+    let input =
+      """
+      import someModule
+      import someLongerModule.withSubmodules
+      import class MyModule.MyClass
+      import struct MyModule.MyStruct
+      @testable import testModule
+      """
+
+    let expected =
+      """
+      import someModule
+      import someLongerModule.withSubmodules
+      import class MyModule.MyClass
+      import struct MyModule.MyStruct
+      @testable import testModule
+
+      """
+
+    // Imports should not wrap
+    assertPrettyPrintEqual(input: input, expected: expected, linelength: 5)
+  }
+}