helpers.rb in introspective_grape-0.5.0

- old
+ new
@@ -1,63 +1,70 @@
-module IntrospectiveGrape::Helpers
-  API_ACTIONS = [:index,:show,:create,:update,:destroy].freeze
-  def authentication_method=(method)
-    # IntrospectiveGrape::API.authentication_method=
-    @authentication_method = method
-  end
+module IntrospectiveGrape
+  module Helpers
+    API_ACTIONS = %i(index show create update destroy).freeze
 
-  def authentication_method(context)
-    # Default to "authenticate!" or as grape docs once suggested, "authorize!"
-    if @authentication_method
-      @authentication_method
-    elsif context.respond_to?('authenticate!')
-      'authenticate!'
-    elsif context.respond_to?('authorize!')
-      'authorize!'
+    def authentication_method=(method)
+      # IntrospectiveGrape::API.authentication_method=
+      @authentication_method = method
     end
-  end
 
-  def paginate(args={})
-    @pagination = args
-  end
+    def authentication_method(context)
+      # Default to "authenticate!" or as grape docs once suggested, "authorize!"
+      if @authentication_method
+        @authentication_method
+      elsif context.respond_to?('authenticate!')
+        'authenticate!'
+      elsif context.respond_to?('authorize!')
+        'authorize!'
+      end
+    end
 
-  def pagination
-    @pagination
-  end
+    def paginate(args={})
+      @pagination = args
+    end
 
-  def exclude_actions(model, *args)
-    @exclude_actions ||= {}; @exclude_actions[model.name] ||= []
-    args.flatten!
-    args = API_ACTIONS if args.include?(:all)
-    args = []          if args.include?(:none)
+    def pagination
+      @pagination
+    end
 
-    undefined_actions = args.compact-API_ACTIONS
-    raise "#{model.name} defines invalid actions: #{undefined_actions}" if undefined_actions.present?
+    def exclude_actions(model, *args)
+      args                           = all_or_none(args)
+      @exclude_actions             ||= {}
+      @exclude_actions[model.name] ||= []
 
-    @exclude_actions[model.name] = args.present? ? args.compact : @exclude_actions[model.name] || []
-  end
+      undefined_actions = args - API_ACTIONS
+      raise "#{model.name} defines invalid actions: #{undefined_actions}" if undefined_actions.present?
 
-  def include_actions(model, *args)
-    @exclude_actions ||= {}; @exclude_actions[model.name] ||= []
-    @exclude_actions[model.name] = API_ACTIONS-exclude_actions(model, args)
-  end
+      @exclude_actions[model.name] = args.present? ? args : @exclude_actions[model.name]
+    end
 
+    def all_or_none(args=[])
+      args.flatten!&.compact!
+      args = API_ACTIONS if args.include?(:all)
+      args = []          if args.include?(:none)
+      args
+    end
 
-  def default_includes(model, *args)
-    @default_includes ||= {}
-    @default_includes[model.name] = args.present? ? args.flatten : @default_includes[model.name] || []
-  end
+    def include_actions(model, *args)
+      @exclude_actions             ||= {}
+      @exclude_actions[model.name] ||= []
+      @exclude_actions[model.name] = API_ACTIONS - exclude_actions(model, args)
+    end
 
-  def whitelist(whitelist=nil)
-    return @whitelist if !whitelist
-    @whitelist = whitelist
-  end
+    def default_includes(model, *args)
+      @default_includes ||= {}
+      @default_includes[model.name] = args.present? ? args.flatten : @default_includes[model.name] || []
+    end
 
-  def skip_presence_validations(fields=nil)
-    return @skip_presence_fields||[] if !fields
-    @skip_presence_fields = [fields].flatten
-  end
+    def whitelist(whitelist=nil)
+      return @whitelist unless whitelist
 
+      @whitelist = whitelist
+    end
 
-end
+    def skip_presence_validations(fields=nil)
+      return @skip_presence_fields || [] unless fields
 
-
+      @skip_presence_fields = [fields].flatten
+    end
+  end
+end