resource_proxy.rb

module Graphiti
  class ResourceProxy
    include Enumerable

    attr_reader :resource, :query, :scope, :payload, :cache_expires_in, :cache

    def initialize(resource, scope, query,
      payload: nil,
      single: false,
      raise_on_missing: false,
      cache: nil,
      cache_expires_in: nil)

      @resource = resource
      @scope = scope
      @query = query
      @payload = payload
      @single = single
      @raise_on_missing = raise_on_missing
      @cache = cache
      @cache_expires_in = cache_expires_in
    end

    def cache?
      !!@cache
    end

    alias_method :cached?, :cache?

    def single?
      !!@single
    end

    def raise_on_missing?
      !!@raise_on_missing
    end

    def errors
      data.errors
    end

    def [](val)
      data[val]
    end

    def jsonapi_render_options(opts = {})
      opts[:expose] ||= {}
      opts[:expose][:context] = Graphiti.context[:object]
      opts
    end

    def to_jsonapi(options = {})
      options = jsonapi_render_options(options)
      Renderer.new(self, options).to_jsonapi
    end

    def to_json(options = {})
      Renderer.new(self, options).to_json
    end

    def as_json(options = {})
      Renderer.new(self, options).as_json
    end

    def to_xml(options = {})
      Renderer.new(self, options).to_xml
    end

    def to_graphql(options = {})
      Renderer.new(self, options).to_graphql
    end

    def as_graphql(options = {})
      Renderer.new(self, options).as_graphql
    end

    def data
      @data ||= begin
        records = @scope.resolve
        raise Graphiti::Errors::RecordNotFound if records.empty? && raise_on_missing?

        records = records[0] if single?
        records
      end
    end
    alias_method :to_a, :data
    alias_method :resolve_data, :data

    def future_resolve_data
      @scope.future_resolve.then do |records|
        raise Graphiti::Errors::RecordNotFound if records.empty? && raise_on_missing?

        records = records[0] if single?
        @data = records
      end
    end

    def meta
      @meta ||= data.respond_to?(:meta) ? data.meta : {}
    end

    def each(&blk)
      to_a.each(&blk)
    end

    def stats
      @stats ||= if @query.hash[:stats]
        scope = @scope.unpaginated_object
        if resource.adapter.can_group?
          if (group = @query.hash[:stats].delete(:group_by))
            scope = resource.adapter.group(scope, group[0])
          end
        end
        payload = Stats::Payload.new @resource,
          @query,
          scope,
          data
        payload.generate
      else
        {}
      end
    end

    def pagination
      @pagination ||= Delegates::Pagination.new(self)
    end

    def save(action: :create)
      # TODO: remove this. Only used for persisting many-to-many with AR
      # (see activerecord adapter)
      original = Graphiti.context[:namespace]
      begin
        Graphiti.context[:namespace] = action
        ::Graphiti::RequestValidator.new(@resource, @payload.params, action).validate!
        validator = persist {
          @resource.persist_with_relationships \
            @payload.meta(action: action),
            @payload.attributes,
            @payload.relationships
        }
      ensure
        Graphiti.context[:namespace] = original
      end
      @data, success = validator.to_a

      if success
        # If the context namespace is `update` or `create`, certain
        # adapters will cause N+1 validation calls, so lets explicitly
        # switch to a lookup context.
        Graphiti.with_context(Graphiti.context[:object], :show) do
          @scope.resolve_sideloads([@data])
        end
      end

      success
    end

    def destroy
      resolve_data
      transaction_response = @resource.transaction do
        metadata = {method: :destroy}
        model = @resource.destroy(@query.filters[:id], metadata)
        model.instance_variable_set(:@__serializer_klass, @resource.serializer)
        @resource.after_graph_persist(model, metadata)
        validator = ::Graphiti::Util::ValidationResponse.new \
          model, @payload
        validator.validate!
        @resource.before_commit(model, metadata)

        {result: validator}
      end
      @data, success = transaction_response[:result].to_a
      success
    end

    def update
      resolve_data
      save(action: :update)
    end

    alias update_attributes update # standard:disable Style/Alias

    def include_hash
      @include_hash ||= begin
        base = @payload ? @payload.include_hash : {}
        base.deep_merge(@query.include_hash)
      end
    end

    def fields
      query.fields
    end

    def extra_fields
      query.extra_fields
    end

    def debug_requested?
      query.debug_requested?
    end

    def updated_at
      @scope.updated_at
    end

    def etag
      "W/#{ActiveSupport::Digest.hexdigest(cache_key_with_version.to_s)}"
    end

    def cache_key
      ActiveSupport::Cache.expand_cache_key([@scope.cache_key, @query.cache_key])
    end

    def cache_key_with_version
      ActiveSupport::Cache.expand_cache_key([@scope.cache_key_with_version, @query.cache_key])
    end

    private

    def persist
      transaction_response = @resource.transaction do
        ::Graphiti::Util::TransactionHooksRecorder.record do
          model = yield
          ::Graphiti::Util::TransactionHooksRecorder.run_graph_persist_hooks
          validator = ::Graphiti::Util::ValidationResponse.new \
            model, @payload
          validator.validate!
          validator
        end
      end

      _data, success = transaction_response[:result].to_a
      if success
        transaction_response[:after_commit_hooks].each do |hook|
          hook.call
        end
      end

      transaction_response[:result]
    end
  end
end