Update to add 'access modifier' for generated swift file

Author: Wayne Wanbok Choi

Sources/POEditorParser/Lib/LiteralsParser.swift

@@ -156,17 +156,19 @@ enum SwiftCodeGeneratorConstants {
         return formatter
     }()
 
-    static let rootObjectHeader = """
-    // Generated using POEditorParser
-    // DO NOT EDIT
-    // Generated: \(SwiftCodeGeneratorConstants.dateFormatter.string(from: Date()))
-    
-    // swiftlint:disable all
+    static func rootObjectHeader(accessModifier: AccessModifier = .public) -> String {
+        """
+        // Generated using POEditorParser
+        // DO NOT EDIT
+        // Generated: \(SwiftCodeGeneratorConstants.dateFormatter.string(from: Date()))
+        
+        // swiftlint:disable all
 
-    import Foundation
-    
-    enum Literals {
-    """
+        import Foundation
+        
+        \(accessModifier.rawValue) enum Literals {
+        """
+    }
     static let rootObjectFooter = "\n}\n// swiftlint:enable all\n"
 
     static let methodOrVariableHeader = "\n"
@@ -181,7 +183,7 @@ class StringCodeGenerator: SwiftCodeGenerator {
     var generatedResult = ""
 
     func generateCode(translations: [Translation]) {
-        generatedResult += SwiftCodeGeneratorConstants.rootObjectHeader
+        generatedResult += SwiftCodeGeneratorConstants.rootObjectHeader()
 
         for t in translations {
             generatedResult += SwiftCodeGeneratorConstants.methodOrVariableHeader
@@ -195,13 +197,18 @@ class StringCodeGenerator: SwiftCodeGenerator {
 public class FileCodeGenerator: SwiftCodeGenerator {
 
     let fileHandle: FileHandle
-    public init(fileHandle: FileHandle) {
+    let accessModifier: AccessModifier
+    public init(
+        fileHandle: FileHandle,
+        access: String
+    ) {
         self.fileHandle = fileHandle
+        self.accessModifier = AccessModifier(accessString: access)
     }
 
     // TODO: Generalize!!! += (same code as in string)
     public func generateCode(translations: [Translation]) {
-        fileHandle += SwiftCodeGeneratorConstants.rootObjectHeader
+        fileHandle += SwiftCodeGeneratorConstants.rootObjectHeader()
 
         for t in translations {
             fileHandle += SwiftCodeGeneratorConstants.methodOrVariableHeader
@@ -365,6 +372,15 @@ enum TranslationValueParser {
 }
 
 
+enum AccessModifier: String {
+    case `public`, `private`, `open`, `internal`
+    
+    init(accessString: String) {
+        self = AccessModifier(rawValue: accessString) ?? .public
+    }
+}
+
+
 extension String {
     var first: String {
         return String(prefix(1))

Sources/POEditorParser/main.swift

@@ -10,8 +10,9 @@ command(
     Argument<Int>("id", description: "The id of the project"),
     Argument<String>("language", description: "The language code"),
     Option<String>("swiftfile", default: "${SRCROOT}/${TARGET_NAME}/Literals.swift", description: "The output Swift file directory."),
-    Option<String>("stringsfile", default: "${SRCROOT}/${TARGET_NAME}/Localizable.strings", description: "The output Strings file directory.")
-) { (token: String, id: Int, language: String, swiftfile: String, stringsfile: String) in
+    Option<String>("stringsfile", default: "${SRCROOT}/${TARGET_NAME}/Localizable.strings", description: "The output Strings file directory."),
+    Option<String>("access", default: "public", description: "The access modifier.")
+) { (token: String, id: Int, language: String, swiftfile: String, stringsfile: String, access: String) in
 
     print("Fetching contents of strings at POEditor...".blue)
 
@@ -60,7 +61,7 @@ command(
                 print("Fatal error: Couldn't write to file located at \(swiftfile)".red)
                 return
             }
-            let fileCodeGenerator = FileCodeGenerator(fileHandle: swiftHandle)
+            let fileCodeGenerator = FileCodeGenerator(fileHandle: swiftHandle, access: access)
             fileCodeGenerator.generateCode(translations: translations)
             print("Success! Literals generated at \(swiftfile)".green)