Merge pull request #14 from Pimentoso/master

Fix generation of input types

Author: ajsharp

lib/generators/gql/gql_generator_base.rb

@@ -68,5 +68,19 @@ def class_with_fields(namespace, name, superclass, fields)
         klass.join("\n")
       end
     end
+
+    def class_with_arguments(namespace, name, superclass, fields)
+      wrap_in_namespace(namespace) do |indent|
+        klass = []
+        klass << sprintf("%sclass %s < %s", "  " * indent, name, superclass)
+
+        fields.each do |field|
+          klass << sprintf("%sargument :%s, %s, required: %s", "  " * (indent + 1), field[:name], field[:gql_type], !field[:null])
+        end
+
+        klass << sprintf("%send", "  " * indent)
+        klass.join("\n")
+      end
+    end
   end
 end

lib/generators/gql/input_generator.rb

@@ -22,7 +22,7 @@ def generate_input_type
         fields.reject! { |field| !options['include_columns'].include?(field[:name]) }
       end
 
-      code = class_with_fields(options['namespace'], name, superclass, fields)
+      code = class_with_arguments(options['namespace'], name, superclass, fields)
       file_name = File.join(root_directory(options['namespace']), "#{name.underscore}.rb")
 
       create_file file_name, code