require 'jsonapi/configuration' require 'jsonapi/resource_for' require 'jsonapi/association' module JSONAPI class Resource include ResourceFor @@resource_types = {} attr :context attr_reader :model def initialize(model, context = nil) @model = model @context = context end def remove @model.destroy end def id model.send(self.class._primary_key) end def create_has_many_link(association_type, association_key_value) association = self.class._associations[association_type] related_resource = self.class.resource_for(association.type).find_by_key(association_key_value, context: @context) # ToDo: Add option to skip relations that already exist instead of returning an error? relation = @model.send(association.type).where(association.primary_key => association_key_value).first if relation.nil? @model.send(association.type) << related_resource.model else raise JSONAPI::Exceptions::HasManyRelationExists.new(association_key_value) end end def replace_has_many_links(association_type, association_key_values) association = self.class._associations[association_type] send("#{association.foreign_key}=", association_key_values) end def create_has_one_link(association_type, association_key_value) association = self.class._associations[association_type] # ToDo: Add option to skip relations that already exist instead of returning an error? relation = @model.send("#{association.foreign_key}") if relation.nil? send("#{association.foreign_key}=", association_key_value) else raise JSONAPI::Exceptions::HasOneRelationExists.new end end def replace_has_one_link(association_type, association_key_value) association = self.class._associations[association_type] send("#{association.foreign_key}=", association_key_value) end def remove_has_many_link(association_type, key) association = self.class._associations[association_type] @model.send(association.type).delete(key) end def remove_has_one_link(association_type) association = self.class._associations[association_type] send("#{association.foreign_key}=", nil) end def replace_fields(field_data) field_data[:attributes].each do |attribute, value| begin send "#{attribute}=", value rescue ArgumentError # :nocov: Will be thrown if an enum value isn't allowed for an enum. Currently not tested as enums are a rails 4.1 and higher feature raise JSONAPI::Exceptions::InvalidFieldValue.new(attribute, value) # :nocov: end end field_data[:has_one].each do |association_type, value| if value.nil? remove_has_one_link(association_type) else replace_has_one_link(association_type, value) end end if field_data[:has_one] field_data[:has_many].each do |association_type, values| replace_has_many_links(association_type, values) end if field_data[:has_many] end def save @model.save! rescue ActiveRecord::RecordInvalid => e raise JSONAPI::Exceptions::ValidationErrors.new(e.record.errors.messages) end # Override this on a resource instance to override the fetchable keys def fetchable_fields self.class.fields end class << self def inherited(base) base._attributes = (_attributes || {}).dup base._associations = (_associations || {}).dup base._allowed_filters = (_allowed_filters || Set.new).dup type = base.name.demodulize.sub(/Resource$/, '').underscore base._type = type.pluralize.to_sym check_reserved_resource_name(base._type, base.name) # If eager loading is on this is how all the resource types are setup # If eager loading is off some resource types will be initialized in # _resource_name_from_type @@resource_types[base._type] ||= base.name.demodulize end attr_accessor :_attributes, :_associations, :_allowed_filters , :_type def create(context) self.new(self.create_model, context) end def create_model _model_class.new end def routing_options(options) @_routing_resource_options = options end def routing_resource_options @_routing_resource_options ||= {} end # Methods used in defining a resource class def attributes(*attrs) attrs.each do |attr| attribute(attr) end end def attribute(attr, options = {}) check_reserved_attribute_name(attr) @_attributes[attr] = options define_method attr do @model.send(attr) end unless method_defined?(attr) define_method "#{attr}=" do |value| @model.send "#{attr}=", value end unless method_defined?("#{attr}=") end def default_attribute_options {format: :default} end def has_one(*attrs) _associate(Association::HasOne, *attrs) end def has_many(*attrs) _associate(Association::HasMany, *attrs) end def model_name(model) @_model_name = model.to_sym end def filters(*attrs) @_allowed_filters.merge(attrs) end def filter(attr) @_allowed_filters.add(attr.to_sym) end def key(key) # :nocov: warn '[DEPRECATION] `key` is deprecated. Please use `primary_key` instead.' @_primary_key = key.to_sym # :nocov: end def primary_key(key) @_primary_key = key.to_sym end # Override in your resource to filter the updateable keys def updateable_fields(context = nil) _updateable_associations | _attributes.keys end # Override in your resource to filter the createable keys def createable_fields(context = nil) _updateable_associations | _attributes.keys end # Override in your resource to filter the sortable keys def sortable_fields(context = nil) _attributes.keys end def fields _associations.keys | _attributes.keys end def apply_filter(records, filter, value) records.where(filter => value) end # Override this method if you have more complex requirements than this basic find method provides def find(filters, options = {}) context = options[:context] sort_params = options.fetch(:sort_params) { [] } includes = [] records = records(options) filters.each do |filter, value| if _associations.include?(filter) if _associations[filter].is_a?(JSONAPI::Association::HasMany) includes.push(filter) records = apply_filter(records, "#{filter}.#{_associations[filter].primary_key}", value) else records = apply_filter(records, "#{_associations[filter].foreign_key}", value) end else records = apply_filter(records, filter, value) end end resources = [] order_options = construct_order_options(sort_params) records.order(order_options).includes(includes).each do |model| resources.push self.new(model, context) end return resources end def find_by_key(key, options = {}) context = options[:context] model = records(options).where({_primary_key => key}).first if model.nil? raise JSONAPI::Exceptions::RecordNotFound.new(key) end self.new(model, context) end def find_by_keys(keys, options = {}) context = options[:context] _models = records(options).where({_primary_key => keys}) unless _models.length == keys.length key = (keys - _models.pluck(_primary_key).map(&:to_s)).first raise JSONAPI::Exceptions::RecordNotFound.new(key) end _models.map { |model| self.new(model, context) } end # Override this method if you want to customize the relation for # finder methods (find, find_by_key, find_by_keys) def records(options = {}) _model_class end def verify_filters(filters, context = nil) verified_filters = {} filters.each do |filter, raw_value| verified_filter = verify_filter(filter, raw_value, context) verified_filters[verified_filter[0]] = verified_filter[1] end verified_filters end def is_filter_association?(filter) filter == _type || _associations.include?(filter) end def verify_filter(filter, raw, context = nil) filter_values = [] filter_values += CSV.parse_line(raw) unless raw.nil? || raw.empty? if is_filter_association?(filter) verify_association_filter(filter, filter_values, context) else verify_custom_filter(filter, filter_values, context) end end # override to allow for key processing and checking def verify_key(key, context = nil) return key end # override to allow for key processing and checking def verify_keys(keys, context = nil) return keys.collect do |key| verify_key(key, context) end end # override to allow for custom filters def verify_custom_filter(filter, value, context = nil) return filter, value end # override to allow for custom association logic, such as uuids, multiple keys or permission checks on keys def verify_association_filter(filter, raw, context = nil) return filter, raw end # quasi private class methods def _attribute_options(attr) default_attribute_options.merge(@_attributes[attr]) end def _updateable_associations associations = [] @_associations.each do |key, association| if association.is_a?(JSONAPI::Association::HasOne) || association.acts_as_set associations.push(key) end end associations end def _has_association?(type) type = type.to_s @_associations.has_key?(type.singularize.to_sym) || @_associations.has_key?(type.pluralize.to_sym) end def _association(type) type = type.to_sym @_associations[type] end def _model_name @_model_name ||= self.name.demodulize.sub(/Resource$/, '') end def _key # :nocov: warn '[DEPRECATION] `_key` is deprecated. Please use `_primary_key` instead.' _primary_key # :nocov: end def _primary_key @_primary_key ||= :id end def _as_parent_key @_as_parent_key ||= "#{_type.to_s.singularize}_#{_primary_key}" end def _allowed_filters !@_allowed_filters.nil? ? @_allowed_filters : Set.new([_primary_key]) end def _resource_name_from_type(type) class_name = @@resource_types[type] if class_name.nil? class_name = type.to_s.singularize.camelize + 'Resource' @@resource_types[type] = class_name end return class_name end # :nocov: if RUBY_VERSION >= '2.0' def _model_class @model ||= Object.const_get(_model_name.to_s) end else def _model_class @model ||= _model_name.to_s.safe_constantize end end # :nocov: def _allowed_filter?(filter) _allowed_filters.include?(filter) end def module_path @module_path ||= self.name =~ /::[^:]+\Z/ ? ($`.freeze.gsub('::', '/') + '/').downcase : '' end private def check_reserved_resource_name(type, name) if [:ids, :types, :hrefs, :links].include?(type) warn "[NAME COLLISION] `#{name}` is a reserved resource name." return end end def check_reserved_attribute_name(name) # Allow :id since it can be used to specify the format. Since it is a method on the base Resource # an attribute method won't be created for it. if [:type, :href, :links].include?(name.to_sym) warn "[NAME COLLISION] `#{name}` is a reserved key in #{@@resource_types[_type]}." end end def check_reserved_association_name(name) if [:id, :ids, :type, :types, :href, :hrefs, :link, :links].include?(name.to_sym) warn "[NAME COLLISION] `#{name}` is a reserved association name in #{@@resource_types[_type]}." end end def _associate(klass, *attrs) options = attrs.extract_options! attrs.each do |attr| check_reserved_association_name(attr) @_associations[attr] = klass.new(attr, options) foreign_key = @_associations[attr].foreign_key define_method foreign_key do @model.method(foreign_key).call end unless method_defined?(foreign_key) define_method "#{foreign_key}=" do |value| @model.method("#{foreign_key}=").call(value) end unless method_defined?("#{foreign_key}=") if @_associations[attr].is_a?(JSONAPI::Association::HasOne) define_method attr do type_name = self.class._associations[attr].type.to_s resource_class = self.class.resource_for(self.class.module_path + type_name) if resource_class associated_model = @model.send attr return associated_model ? resource_class.new(associated_model, @context) : nil end end unless method_defined?(attr) elsif @_associations[attr].is_a?(JSONAPI::Association::HasMany) define_method attr do type_name = self.class._associations[attr].type.to_s resource_class = self.class.resource_for(self.class.module_path + type_name) resources = [] if resource_class associated_models = @model.send attr associated_models.each do |associated_model| resources.push resource_class.new(associated_model, @context) end end return resources end unless method_defined?(attr) end end end def construct_order_options(sort_params) sort_params.each_with_object({}) { |sort_key, order_hash| if sort_key.starts_with?('-') order_hash[sort_key.slice(1..-1)] = :desc else order_hash[sort_key] = :asc end } end end end end