Skip to content

Commit 7198441

Browse files
author
Wayne Wanbok Choi
committed
Make it handle the access modifier from the functions and variables
1 parent d8b0a54 commit 7198441

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

Sources/POEditorParser/Lib/LiteralsParser.swift

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -111,24 +111,24 @@ public struct Translation {
111111
return rawKey.capitalized.replacingOccurrences(of: "_", with: "")
112112
}
113113

114-
var swiftCode: String {
114+
func swiftCode(accessModifier: AccessModifier = .public) -> String {
115115
if variables.isEmpty {
116-
return generateVariableLessSwiftCode()
116+
return generateVariableLessSwiftCode(accessModifier: accessModifier)
117117
} else {
118-
return generateVariableSwiftCode()
118+
return generateVariableSwiftCode(accessModifier: accessModifier)
119119
}
120120
}
121121

122-
private func generateVariableLessSwiftCode() -> String {
122+
private func generateVariableLessSwiftCode(accessModifier: AccessModifier = .public) -> String {
123123
/*
124124
static var Welcome: String {
125125
return NSLocalizedString()
126126
}
127127
*/
128-
return "\tstatic var \(prettyKey): String {\n\t\treturn \(localizedStringFunction)(\"\(rawKey)\", comment: \"\")\n\t}\n"
128+
return "\t\(accessModifier.rawValue) static var \(prettyKey): String {\n\t\treturn \(localizedStringFunction)(\"\(rawKey)\", comment: \"\")\n\t}\n"
129129
}
130130

131-
private func generateVariableSwiftCode() -> String {
131+
private func generateVariableSwiftCode(accessModifier: AccessModifier = .public) -> String {
132132
/*
133133
static func ReadBooksKey(readNumber: Int) -> String {
134134
return ""
@@ -141,7 +141,7 @@ public struct Translation {
141141
.map { $0.parameterKey }
142142
.map { $0.snakeCased() }
143143
.joined(separator: ", ")
144-
return "\tstatic func \(prettyKey)(\(parameters)) -> String {\n\t\treturn String(format: \(localizedStringFunction)(\"\(rawKey)\", comment: \"\"), \(localizedArguments))\n\t}"
144+
return "\t\(accessModifier.rawValue) static func \(prettyKey)(\(parameters)) -> String {\n\t\treturn String(format: \(localizedStringFunction)(\"\(rawKey)\", comment: \"\"), \(localizedArguments))\n\t}"
145145
}
146146

147147
}
@@ -208,11 +208,11 @@ public class FileCodeGenerator: SwiftCodeGenerator {
208208

209209
// TODO: Generalize!!! += (same code as in string)
210210
public func generateCode(translations: [Translation]) {
211-
fileHandle += SwiftCodeGeneratorConstants.rootObjectHeader()
211+
fileHandle += SwiftCodeGeneratorConstants.rootObjectHeader(accessModifier: accessModifier)
212212

213213
for t in translations {
214214
fileHandle += SwiftCodeGeneratorConstants.methodOrVariableHeader
215-
fileHandle += t.swiftCode
215+
fileHandle += t.swiftCode(accessModifier: accessModifier)
216216
}
217217

218218
fileHandle += SwiftCodeGeneratorConstants.rootObjectFooter

0 commit comments

Comments
 (0)