lib/generate_models.rb in cqm-models-0.8.0 vs lib/generate_models.rb in cqm-models-0.8.1

@@ -282,9 +282,15 @@
   end
 
   # Add class
   contents.gsub!(/  _type: String,\n/, "  _type: { type: String, default: '#{dc_name.camelize}' },\n")
 
+  # Component and Facility types
+  contents.gsub!(/facilityLocations: \[\]/, 'facilityLocations: [FacilityLocationSchema]')
+  contents.gsub!(/facilityLocation: Code/, 'facilityLocation: FacilityLocationSchema')
+  contents.gsub!(/components: \[\]/, 'components: [ComponentSchema]')
+  contents.gsub!(/component: Code/, 'component: ComponentSchema')
+
   File.open(file_name, 'w') { |file| file.puts contents }
 end
 
 # Inject Ruby Patient model extensions
 template = File.read('templates/patient_extension.rb.erb')