From 65f18dd9637f9a9602fd480a09a9b666c5944bbc Mon Sep 17 00:00:00 2001
From: Noah McCann <Noah@Noahs-iMac.local>
Date: Tue, 20 Jan 2015 21:44:02 -0500
Subject: [PATCH] Add UIKit import to generated file

Add UIKit import to generated file to avoid build errors
---
 R.swift/func.swift | 4 ++++
 R.swift/main.swift | 5 ++++-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/R.swift/func.swift b/R.swift/func.swift
index 4d40b0ce..faf9323c 100644
--- a/R.swift/func.swift
+++ b/R.swift/func.swift
@@ -139,6 +139,10 @@ func writeResourceFile(code: String, toFolderURL folderURL: NSURL) {
 
 // MARK: Code generator functions
 
+func swiftImports() -> String {
+    return "import UIKit"
+}
+
 func swiftImageStructWithAssetFolders(assetFolders: [AssetFolder]) -> String {
   return distinct(assetFolders.flatMap { $0.imageAssets })
     .reduce("struct image {\n") {
diff --git a/R.swift/main.swift b/R.swift/main.swift
index 3f2b9015..129b0ffa 100644
--- a/R.swift/main.swift
+++ b/R.swift/main.swift
@@ -15,6 +15,9 @@ let findAllStoryboardURLsInDirectory = filterDirectoryContentsRecursively(defaul
 
 inputDirectories(NSProcessInfo.processInfo())
   .each { directory in
+    // Imports
+    let imports = swiftImports()
+    
     // Storyboards
     let storyboards = findAllStoryboardURLsInDirectory(url: directory)
       .map { Storyboard(url: $0) }
@@ -37,6 +40,6 @@ inputDirectories(NSProcessInfo.processInfo())
 
     // Write out the code
     let code = [imageStruct, segueStruct, storyboardStructs, validateAllStoryboardsFunction]
-      .reduce("struct R {") { $0 + "\n" + indent(string: $1) } + "}\n"
+      .reduce("\(imports)\n\nstruct R {") { $0 + "\n" + indent(string: $1) } + "}\n"
     writeResourceFile(code, toFolderURL: directory)
   }