Support for Storyboard assets (#34)

Author: kaandedeoglu

Sources/Shark/SharkEnumBuilder.swift

@@ -14,9 +14,9 @@ enum SharkEnumBuilder {
         let localizationsString = try LocalizationEnumBuilder.localizationsEnumString(forFilesAtPaths: resourcePaths.localizationPaths, topLevelName: "L")
         let fontsString = try FontEnumBuilder.fontsEnumString(forFilesAtPaths: resourcePaths.fontPaths, topLevelName: "F")
         let dataAssetsString = try DataAssetEnumBuilder.dataAssetEnumString(forFilesAtPaths: resourcePaths.assetsPaths, topLevelName: "D")
-
+        let storyboardString = try StoryboardBuilder.storyboardEnumString(forFilesAtPaths: resourcePaths.storyboardPaths, topLevelName: "S")
         let declarationIndendationLevel = options.topLevelScope ? 0 : 1
-        let resourcesEnumsString = [imagesString, colorsString, fontsString, localizationsString, dataAssetsString]
+        let resourcesEnumsString = [imagesString, colorsString, fontsString, localizationsString, storyboardString, dataAssetsString]
             .compactMap({ $0?.indented(withLevel: declarationIndendationLevel) })
             .joined(separator: "\n\n")
 

Sources/Shark/StoryboardEnumBuilder.swift

@@ -0,0 +1,35 @@
+import Foundation
+
+private struct StoryboardValue: Equatable, Comparable {
+    let name: String
+
+    func declaration(indentLevel: Int) -> String {
+        return #"\#(String.indent(indentLevel))public static var \#(name.casenameSanitized): UIStoryboard { return UIStoryboard(name: "\#(name)", bundle: bundle) }"#
+    }
+
+    static func <(lhs: StoryboardValue, rhs: StoryboardValue) -> Bool {
+        return lhs.name < rhs.name
+    }
+}
+
+enum StoryboardBuilder {
+    private enum Constants {
+        static let storyboardExtension = "storyboard"
+    }
+
+    static func storyboardEnumString(forFilesAtPaths paths: [String], topLevelName: String) throws -> String? {
+        let storyboardPaths = Set(paths.map(\.lastPathComponent))
+
+        guard storyboardPaths.isEmpty == false else { return nil }
+
+        var result = "public enum \(topLevelName) {\n"
+        for name in storyboardPaths.map({ $0.deletingPathExtension }).sorted() {
+            result += StoryboardValue(name: name).declaration(indentLevel: 1)
+            result += "\n"
+        }
+
+        result += "}"
+
+        return result
+    }
+}

Sources/Shark/XcodeProjectHelper.swift

@@ -11,6 +11,7 @@ struct XcodeProjectHelper {
         fileprivate(set) var localizationPaths: [String] = []
         fileprivate(set) var assetsPaths: [String] = []
         fileprivate(set) var fontPaths: [String] = []
+        fileprivate(set) var storyboardPaths: [String] = []
     }
 
     private let projectPath: Path
@@ -62,6 +63,8 @@ struct XcodeProjectHelper {
                     result.localizationPaths.append(path)
                 case "ttf", "otf", "ttc":
                     result.fontPaths.append(path)
+                case "storyboard":
+                    result.storyboardPaths.append(path)
                 default:
                     break
                 }