Merge pull request #8 from sshaw/master

Add name, include-columns, and superclass options to type generators

Author: ajsharp

lib/generators/gql/gql_generator_base.rb

@@ -2,40 +2,71 @@
 
 module Gql
   module GqlGeneratorBase
-    extend ActiveSupport::Concern
+    protected
 
-    included do
-      protected
+    # Generate a namedspaced class name with the mutation prefix
+    def prefixed_class_name(prefix)
+      (class_path + ["#{prefix}_#{file_name}"]).map!(&:camelize).join("::")
+    end
 
-      # Generate a namedspaced class name with the mutation prefix
-      def prefixed_class_name(prefix)
-        (class_path + ["#{prefix}_#{file_name}"]).map!(&:camelize).join("::")
-      end
+    def type_map
+      {
+        integer: 'Int',
+        string: 'String',
+        boolean: 'Boolean',
+        decimal: 'Float',
+        datetime: 'GraphQL::Types::ISO8601DateTime',
+        date: 'GraphQL::Types::ISO8601Date',
+        hstore: 'GraphQL::Types::JSON',
+        text: 'String',
+        json: 'GraphQL::Types::JSON',
+        jsonb: 'GraphQL::Types::JSON'
+      }
+    end
 
-      def type_map
-        {
-          integer: 'Int',
-          string: 'String',
-          boolean: 'Boolean',
-          decimal: 'Float',
-          datetime: 'GraphQL::Types::ISO8601DateTime',
-          date: 'GraphQL::Types::ISO8601Date',
-          hstore: 'GraphQL::Types::JSON',
-          text: 'String',
-          json: 'GraphQL::Types::JSON'
-        }
-      end
-  
-      def map_model_types(model_name)
-        klass = model_name.constantize
-        associations = klass.reflect_on_all_associations(:belongs_to)
-        bt_columns = associations.map(&:foreign_key)
-  
-        klass.columns
-          .reject { |col| bt_columns.include?(col.name) }
-          .reject { |col| type_map[col.type].nil? }
-          .map { |col| {name: col.name, gql_type: type_map[col.type]} }
+    def map_model_types(model_name)
+      klass = model_name.constantize
+      associations = klass.reflect_on_all_associations(:belongs_to)
+      bt_columns = associations.map(&:foreign_key)
+
+      klass.columns
+        .reject { |col| bt_columns.include?(col.name) }
+        .reject { |col| type_map[col.type].nil? }
+        .map do |col|
+          {
+            name: col.name,
+            null: col.null,
+            gql_type: klass.primary_key == col.name ? 'GraphQL::Types::ID' : type_map[col.type]
+          }
+        end
+    end
+
+    def root_directory(namespace)
+      "app/graphql/#{namespace.underscore}"
+    end
+
+    def wrap_in_namespace(namespace)
+      namespace = namespace.split('::')
+      namespace.shift if namespace[0].empty?
+
+      code = namespace.each_with_index.map { |name, i| "  " * i + "module #{name}" }.join("\n")
+      code << "\n" << yield(namespace.size) << "\n"
+      code << (namespace.size - 1).downto(0).map { |i| "  " * i  + "end" }.join("\n")
+      code
+    end
+
+    def class_with_fields(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("%sfield :%s, %s, null: %s", "  " * (indent + 1), field[:name], field[:gql_type], field[:null])
+        end
+
+        klass << sprintf("%send", "  " * indent)
+        klass.join("\n")
       end
     end
   end
-end
+end

lib/generators/gql/input_generator.rb

@@ -3,15 +3,29 @@ module Gql
   class InputGenerator < Rails::Generators::Base
     include GqlGeneratorBase
     source_root File.expand_path('../templates', __FILE__)
+
     argument :model_name, type: :string
-  
+
+    class_option :name, type: :string
+    class_option :include_columns, type: :array, default: []
+    class_option :superclass, type: :string, default: 'Types::BaseInputObject'
+    class_option :namespace, type: :string, default: 'Types::Input'
+
     def generate_input_type
-      file_name = model_name
+      name = options['name'] || model_name
+      superclass = options['superclass']
 
       ignore = ['id', 'created_at', 'updated_at']
-      @fields = map_model_types(model_name).reject { |field| ignore.include?(field[:name]) }
+      fields = map_model_types(model_name)
+      fields.reject! { |field| ignore.include?(field[:name]) }
+      if options['include_columns'].any?
+        fields.reject! { |field| !options['include_columns'].include?(field[:name]) }
+      end
+
+      code = class_with_fields(options['namespace'], name, superclass, fields)
+      file_name = File.join(root_directory(options['namespace']), "#{name.underscore}_input.rb")
 
-      template('input_type.rb', "app/graphql/types/input/#{file_name.underscore}_input.rb")
+      create_file file_name, code
     end
   end
-end
+end

lib/generators/gql/model_type_generator.rb

@@ -3,12 +3,28 @@ module Gql
   class ModelTypeGenerator < Rails::Generators::Base
     include GqlGeneratorBase
     source_root File.expand_path('../templates', __FILE__)
+
     argument :model_name, type: :string
 
+    class_option :name, type: :string
+    class_option :include_columns, type: :array, default: []
+    class_option :superclass, type: :string, default: 'Types::BaseObject'
+    class_option :namespace, type: :string, default: 'Types'
+
     def type
-      file_name = "#{model_name.underscore}_type"
-      @fields = map_model_types(model_name)
-      template('model_type.rb', "app/graphql/types/#{file_name}.rb")
+      name = options['name'] || model_name
+
+      superclass = options['superclass']
+
+      fields = map_model_types(model_name)
+      if options['include_columns'].any?
+        fields.reject! { |field| !options['include_columns'].include?(field[:name]) }
+      end
+
+      code = class_with_fields(options['namespace'], name, superclass, fields)
+      file_name = File.join(root_directory(options['namespace']), "#{name.underscore}_type.rb")
+
+      create_file file_name, code
     end
   end
-end
+end