lib/soybean/generators/service_generator.rb in soybean-2.1.5 vs lib/soybean/generators/service_generator.rb in soybean-2.2.0

@@ -13,24 +13,28 @@
       def schemes
         @schemes.map { |url| TypesGenerator.new(url) }
       end
 
       def generate
-        (schemes + mappings + interface + model + specs).map do |generator|
+        (schemes + mappings + interface + model + specs + middleware).map do |generator|
           yield generator.fullpath(path), generator.generate
         end
       end
 
       def mappings
-        @mappings ||= schemes.map { |gen| MappingGenerator.new(gen.xsd) }
+        @mappings ||= schemes.map { |gen| MappingGenerator.new(gen.xsd, @wsdl) }
       end
 
       def interface
         @interface ||= [InterfaceGenerator.new(@wsdl)]
       end
 
       def model
         @model ||= [ModelGenerator.new(@wsdl)]
+      end
+
+      def middleware
+        @middleware ||= model.map { |m| MiddlewareGenerator.new(m) }
       end
 
       def specs
         @specs ||= model.map { |m| ModelSpecGenerator.new(m, @spec_dir) }
       end