Fix generation not working for single language

This commit is contained in:
david-swift 2024-03-04 21:22:42 +01:00
parent edf10b111a
commit a00d5d4031

View File

@ -223,17 +223,25 @@ public enum Generation {
defaultLanguage: String defaultLanguage: String
) -> String { ) -> String {
var result = "func string(for language: String) -> String {\n" var result = "func string(for language: String) -> String {\n"
for language in getLanguages(dictionary: dictionary) where language != defaultLanguage { let languages = getLanguages(dictionary: dictionary)
for language in languages where language != defaultLanguage {
result += indent("if language.hasPrefix(\"\(language)\") {", by: indentTwo) result += indent("if language.hasPrefix(\"\(language)\") {", by: indentTwo)
result += indent("\nreturn \(language)", by: indentThree) result += indent("\nreturn \(language)", by: indentThree)
result += indent("\n} else", by: indentTwo) result += indent("\n} else", by: indentTwo)
} }
if languages.count <= 1 {
result += """
return \(defaultLanguage)
}
"""
} else {
result += """ result += """
{ {
return \(defaultLanguage) return \(defaultLanguage)
} }
} }
""" """
}
return result return result
} }