Merge pull request #4 from ajsharp/cud-mutations

Create, update and delete mutation generators

Author: ajsharp

README.md

@@ -66,6 +66,27 @@ module Types
 end
 ```
 
+### gql:mutations MODEL_CLASS
+
+Generate create, update and delete mutations for a model.
+
+```
+rails generate gql:mutations Post
+```
+
+Result:
+```ruby
+# app/graphql/types/post_input.rb
+module Types
+  module Input
+    class PostInput < Types::BaseInputObject
+      argument :title, String, required: false
+      argument :body, String, required: false
+    end
+  end
+end
+```
+
 ### gql:mutation MUTATION_PREFIX MODEL_NAME
 
 Generate a mutation class from a model.

lib/generators/gql/gql_generator_base.rb

@@ -1,11 +1,17 @@
 require 'rails/generators/base'
-# require 'active_support/extend'
+
 module Gql
   module GqlGeneratorBase
     extend ActiveSupport::Concern
 
     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
+
       def type_map
         {
           integer: 'Int',

lib/generators/gql/mutation_generator.rb

@@ -1,7 +1,9 @@
 require "rails/generators/named_base"
+require_relative 'gql_generator_base'
 
 module Gql
   class MutationGenerator < Rails::Generators::NamedBase
+    include GqlGeneratorBase
     remove_argument :name # remove name base default arg
 
     argument :mutation_prefix, type: :string
@@ -12,15 +14,13 @@ class MutationGenerator < Rails::Generators::NamedBase
     def name
       model_name
     end
-
-    # Generate a namedspaced class name with the mutation prefix
-    def prefixed_class_name
-      (class_path + ["#{mutation_prefix}_#{file_name}"]).map!(&:camelize).join("::")
-    end
 
     def mutation
       file_name = "#{mutation_prefix}_#{singular_name}"
       template('model_mutation.rb', "app/graphql/mutations/#{class_path.join('/')}/#{file_name.underscore}.rb")
+      insert_into_file("app/graphql/types/mutation_type.rb", after: "  class MutationType < Types::BaseObject\n") do
+        "\t\tfield :#{file_name.camelcase(:lower)}, mutation: Mutations::#{prefixed_class_name(mutation_prefix)}\n"
+      end
     end
   end
 

lib/generators/gql/mutations_generator.rb

@@ -0,0 +1,28 @@
+require "rails/generators/named_base"
+require_relative 'gql_generator_base'
+
+module Gql
+  class MutationsGenerator < Rails::Generators::NamedBase
+    include GqlGeneratorBase
+    source_root File.expand_path('../templates', __FILE__)
+    desc "Generate create, update and delete generators for a model."
+
+    def mutations
+      insert_into_file("app/graphql/mutations/base_mutation.rb", before: "\tend\nend") do
+        "def model_errors!(model)\n# define me!\n"
+      end
+      generate_mutation('update')
+      generate_mutation('create')
+      generate_mutation('delete')
+    end
+
+    protected
+    def generate_mutation(prefix)
+      file_name = "#{prefix}_#{singular_name}"
+      template("#{prefix}_mutation.rb", "app/graphql/mutations/#{class_path.join('/')}/#{file_name.underscore}.rb")
+      insert_into_file("app/graphql/types/mutation_type.rb", after: "  class MutationType < Types::BaseObject\n") do
+        "\t\tfield :#{file_name.camelcase(:lower)}, mutation: Mutations::#{prefixed_class_name(prefix)}\n"
+      end
+    end
+  end
+end

lib/generators/gql/templates/create_mutation.rb

@@ -0,0 +1,17 @@
+module Mutations
+  class <%= prefixed_class_name('Create') %> < Mutations::BaseMutation
+    field :<%= singular_name %>, Types::<%= name %>Type, null: true
+
+    argument :attributes, Types::Input::<%= name %>Input, required: true
+
+    def resolve(attributes:)
+      model = <%= name %>.new(attributes.to_h)
+
+      if model.save
+        {<%= singular_name %>: model}
+      else
+        model_errors!(model)
+      end
+    end
+  end
+end

lib/generators/gql/templates/delete_mutation.rb

@@ -0,0 +1,14 @@
+module Mutations
+  class <%= prefixed_class_name('Delete') %> < Mutations::BaseMutation
+    field :<%= singular_name %>, Types::<%= name %>Type, null: true
+
+    argument :id, Int, required: true
+
+    def resolve(id:)
+      model = <%= class_name %>.find(id)
+
+      model.destroy
+      {<%= singular_name %>: model}
+    end
+  end
+end

lib/generators/gql/templates/model_mutation.rb

@@ -1,5 +1,5 @@
 module Mutations
-  class <%= prefixed_class_name %> < Mutations::BaseMutation
+  class <%= prefixed_class_name(mutation_prefix) %> < Mutations::BaseMutation
     field :<%= singular_name %>, Types::<%= @model_name %>Type, null: true
 
     argument :attributes, Types::Input::<%= @model_name %>Input, required: true
@@ -8,6 +8,7 @@ class <%= prefixed_class_name %> < Mutations::BaseMutation
     def resolve(attributes:, id: nil)
       model = find_or_build_model(id)
       model.attributes = attributes.to_h
+
       if model.save
         {<%= singular_name %>: model}
       else

lib/generators/gql/templates/update_mutation.rb

@@ -0,0 +1,18 @@
+module Mutations
+  class <%= prefixed_class_name('Update') %> < Mutations::BaseMutation
+    field :<%= singular_name %>, Types::<%= name %>Type, null: true
+
+    argument :id, Int, required: true
+    argument :attributes, Types::Input::<%= name %>Input, required: true
+    
+    def resolve(attributes:, id:)
+      model = <%= class_name %>.find(id)
+
+      if model.update_attributes(attributes.to_h)
+        {<%= singular_name %>: model}
+      else
+        model_errors!(model)
+      end
+    end
+  end
+end