@@ -34,9 +34,11 @@ struct Service {
3434 generatorSettings. onlyCreateInitializer = SettingsManager . isSettingEnabled ( . OnlyCreateInitializer)
3535
3636 do {
37- // Try to generate the code
38- guard let bodies = try modelBodies ( fromSource: source, generatorSettings: generatorSettings) else { throw ModelParserError . NoModelNameFound }
39- let code = extensionCode ( fromBodies: bodies)
37+ // Try to generate the code bodies
38+ guard let extensions = try extensionBodies ( fromSource: source, generatorSettings: generatorSettings) else { throw ModelParserError . NoModelNameFound }
39+
40+ //Concatenate the extensions
41+ let code = extensionCode ( fromBodies: extensions)
4042
4143 // Play success sound
4244 playSound ( Service . successSound)
@@ -74,7 +76,7 @@ struct Service {
7476 return retVal
7577 }
7678
77- static func modelBodies ( fromSource source: String , generatorSettings: ModelGeneratorSettings ) throws -> [ String ] ? {
79+ static func extensionBodies ( fromSource source: String , generatorSettings: ModelGeneratorSettings ) throws -> [ String ] ? {
7880 if let codes = try codeStrings ( fromSourceCode: source) {
7981 var retVal = [ String] ( )
8082 var outerModelPrefix = " "
@@ -90,8 +92,6 @@ struct Service {
9092 }
9193 codeToParse. append ( character)
9294 }
93- //For some reason this doesn't work in current Swift4???
94- // codeToParse = codeToParse.insert(contentsOf: outerModelPrefix, at: codeToParse.startIndex(range.length))
9595 }
9696 let newCode = try ModelGenerator . modelCode ( fromSourceCode: codeToParse, withSettings: generatorSettings)
9797 retVal. append ( newCode)
0 commit comments