RubyOnRails GraphQL Type Generator

graphql_generator.rb

# lib/generators/rails/graphql/graphql_generator.rb
# run with "rails generate graphql my_model"

class Rails::GraphqlGenerator < Rails::Generators::NamedBase
  def create_graphql_type_file
    attributes = ""
    
    begin
      klass = Object.const_get(class_name)
      obj = klass.new
      obj.attributes.each do |attr, i|
        if attr != "id"
          attr_type = klass.columns_hash[attr].type.to_s.capitalize
          attr_type = 'DateTime' if attr_type == 'Datetime'
          attr_type = 'String' if attr_type == 'Binary'
          attributes += "field :#{attr}, #{attr_type}\n    "
        end
      end
    rescue
    end

    create_file "app/graphql/types/#{file_name}_type.rb", <<-FILE
module Types
  class #{class_name}Type < BaseObject
    field :id, ID, null: false
    #{ attributes }
  end
end
    FILE
  end
end