datagrid 1.8.1 → 2.0.0.pre.alpha

data/lib/datagrid/columns.rb

@@ -1,13 +1,208 @@
+# frozen_string_literal: true
+
 require "datagrid/utils"
 require "active_support/core_ext/class/attribute"
 
 module Datagrid
-
+  # Defines a column to be used for displaying data in a Datagrid.
+  #
+  #     class UserGrid < ApplicationGrid
+  #       scope do
+  #         User.order("users.created_at desc").joins(:group)
+  #       end
+  #
+  #       column(:name)
+  #       column(:group, order: "groups.name") do
+  #         self.group.name
+  #       end
+  #       column(:active, header: "Activated") do |user|
+  #         !user.disabled
+  #       end
+  #     end
+  #
+  # Each column is used to generate data for the grid.
+  #
+  # To create a grid displaying all users:
+  #
+  #     grid = UserGrid.new
+  #     grid.header    # => ["Group", "Name", "Disabled"]
+  #     grid.rows      # => [
+  #                    #      ["Steve", "Spammers", true],
+  #                    #      ["John", "Spoilers", true],
+  #                    #      ["Berry", "Good people", false]
+  #                    #    ]
+  #     grid.data      # => Header & Rows
+  #     grid.data_hash # => [
+  #                    #      { name: "Steve", group: "Spammers", active: true },
+  #                    #      { name: "John", group: "Spoilers", active: true },
+  #                    #      { name: "Berry", group: "Good people", active: false },
+  #                    #    ]
+  #     }
+  #
+  # ## Column Value
+  #
+  # The value of a column can be defined by passing a block to `Datagrid.column`.
+  #
+  # ### Basic Column Value
+  #
+  # If no block is provided, the column value is generated automatically by sending the column name method to the model.
+  #
+  #     column(:name) # => asset.name
+  #
+  # Using <tt>instance_eval</tt>:
+  #
+  #     column(:completed) { completed? }
+  #
+  # Using the asset as an argument:
+  #
+  #     column(:completed) { |asset| asset.completed? }
+  #
+  # ### Advanced Column Value
+  #
+  # You can also pass the Datagrid object itself to define more complex column values.
+  #
+  # Using filters with columns:
+  #
+  #     filter(:category) do |value|
+  #       where("category LIKE '%#{value}%'")
+  #     end
+  #
+  #     column(:exactly_matches_category) do |asset, grid|
+  #       asset.category == grid.category
+  #     end
+  #
+  # Combining columns:
+  #
+  #     column(:total_sales) do |merchant|
+  #       merchant.purchases.sum(:subtotal)
+  #     end
+  #     column(:number_of_sales) do |merchant|
+  #       merchant.purchases.count
+  #     end
+  #     column(:average_order_value) do |_, _, row|
+  #       row.total_sales / row.number_of_sales
+  #     end
+  #
+  # ## Using Database Expressions
+  #
+  # Columns can use database expressions to directly manipulate data in the database.
+  #
+  #     column(:count_of_users, 'count(user_id)')
+  #     column(:uppercase_name, 'upper(name)')
+  #
+  # ## HTML Columns
+  #
+  # Columns can have different formats for HTML and non-HTML representations.
+  #
+  #     column(:name) do |asset|
+  #       format(asset.name) do |value|
+  #         content_tag(:strong, value)
+  #       end
+  #     end
+  #
+  # ## Column Value Cache
+  #
+  # Enables grid-level caching for column values.
+  #
+  #     self.cached = true
+  #
+  # ## Ordering
+  #
+  # Columns can specify SQL ordering expressions using the `:order` and `:order_desc` options.
+  #
+  # Basic ordering:
+  #
+  #     column(:group_name, order: "groups.name") { self.group.name }
+  #
+  # In example above descending order is automatically inherited from ascending order specified.
+  # When such default is not enough, specify both options together:
+  #
+  #     column(
+  #       :priority,
+  #       # models with null priority are always on bottom
+  #       # no matter if sorting ascending or descending
+  #       order: "priority is not null desc, priority",
+  #       order_desc: "priority is not null desc, priority desc"
+  #     )
+  #
+  # Disable order like this:
+  #
+  #     column(:title, order: false)
+  #
+  # Order by joined table
+  # Allows to join specified table only when order is enabled
+  # for performance:
+  #
+  #     column(:profile_updated_at, order: proc { |scope|
+  #       scope.joins(:profile).order("profiles.updated_at")
+  #     }) do |model|
+  #       model.profile.updated_at.to_date
+  #     end
+  #
+  # Order by a calculated value
+  #
+  #     column(
+  #       :duration_request,
+  #       order: "(requests.finished_at - requests.accepted_at)"
+  #     ) do |model|
+  #       Time.at(model.finished_at - model.accepted_at).strftime("%H:%M:%S")
+  #     end
+  #
+  # ## Default Column Options
+  #
+  # Default options for all columns in a grid can be set using `default_column_options`.
+  #
+  #     self.default_column_options = { order: false }
+  #
+  # ## Columns Visibility
+  #
+  # Columns can be dynamically shown or hidden based on the grid's `column_names` accessor.
+  #
+  #     grid.column_names = [:id, :name]
+  #
+  # ## Dynamic Columns
+  #
+  # Columns can be defined dynamically on a grid instance or based on data.
+  #
+  # Adding a dynamic column:
+  #
+  #     grid.column(:extra_data) do |model|
+  #       model.extra_data
+  #     end
+  #
+  # ## Localization
+  #
+  # Column headers can be localized using the `:header` option or through i18n files.
+  #
+  #     column(:active, header: Proc.new { I18n.t("activated") })
+  #
+  # ## Preloading Associations
+  #
+  # Preload database associations for better performance.
+  #
+  # Automatic association preloading:
+  #
+  #     column(:group) do |user|
+  #       user.group.name
+  #     end
+  #
+  # Custom preloading:
+  #
+  #     column(:account_name, preload: { |s| s.includes(:account) })
+  #
+  # ## Decorator
+  #
+  # A decorator or presenter class can be used around each object in the `scope`.
+  #
+  #     decorate { UserPresenter }
+  #     column(:created_at) do |presenter|
+  #       presenter.user.created_at
+  #     end
   module Columns
     require "datagrid/columns/column"
 
-    # @!method default_column_options=
-    # @param value [Hash] default options passed to #column method call
+    # @!method default_column_options=(value)
+    # @param [Hash] value default options passed to #column method call
     # @return [Hash] default options passed to #column method call
     # @example
     #   # Disable default order
@@ -19,8 +214,8 @@ module Datagrid
     # @return [Hash]
     # @see #default_column_options=
 
-    # @!method batch_size=
-    # @param value [Integer] Specify a default batch size when generating CSV or just data. Default: 1000
+    # @!method batch_size=(value)
+    # @param [Integer] value Specify a default batch size when generating CSV or just data. Default: 1000
     # @return [Integer] Specify a default batch size when generating CSV or just data.
     # @example
     #   self.batch_size = 500
@@ -33,8 +228,9 @@ module Datagrid
     # @see #batch_size=
 
     # @visibility private
+    # @param [Object] base
     def self.included(base)
-      base.extend         ClassMethods
+      base.extend ClassMethods
       base.class_eval do
         include Datagrid::Core
 
@@ -44,11 +240,9 @@ module Datagrid
         class_attribute :cached, default: false
         class_attribute :decorator, instance_writer: false
       end
-      base.include InstanceMethods
     end
 
     module ClassMethods
-
       # @param data [Boolean] if true returns only columns with data representation. Default: false.
       # @param html [Boolean] if true returns only columns with html columns. Default: false.
       # @param column_names [Array<String>] list of column names if you want to limit data only to specified columns
@@ -63,7 +257,7 @@ module Datagrid
       #
       # @param name [Symbol] column name
       # @param query [String, nil] a string representing the query to select this column (supports only ActiveRecord)
-      # @param options [Hash<Symbol, Object>] hash of options
+      # @param options [Hash{Symbol => Object}] hash of options
       # @param block [Block] proc to calculate a column value
       # @return [Datagrid::Columns::Column]
       #
@@ -72,7 +266,7 @@ module Datagrid
       # * <tt>html</tt> - determines if current column should be present in html table and how is it formatted
       # * <tt>order</tt> - determines if this column could be sortable and how.
       #   The value of order is explicitly passed to ORM ordering method.
-      #   Ex: <tt>"created_at, id"</tt> for ActiveRecord, <tt>[:created_at, :id]</tt> for Mongoid
+      #   Example: <tt>"created_at, id"</tt> for ActiveRecord, <tt>[:created_at, :id]</tt> for Mongoid
       # * <tt>order_desc</tt> - determines a descending order for given column
       #   (only in case when <tt>:order</tt> can not be easily reversed by ORM)
       # * <tt>order_by_value</tt> - used in case it is easier to perform ordering at ruby level not on database level.
@@ -86,6 +280,8 @@ module Datagrid
       # * <tt>if</tt> - the column is shown if the reult of calling this argument is true
       # * <tt>unless</tt> - the column is shown unless the reult of calling this argument is true
       # * <tt>preload</tt> - spefies which associations of the scope should be preloaded for this column
+      # * `tag_options` - specify HTML attributes to be set for `<td>` or `<th>` of a column
+      #   Example: `{ class: "content-align-right", "data-group": "statistics" }`
       #
       # @see https://github.com/bogdan/datagrid/wiki/Columns
       def column(name, query = nil, **options, &block)
@@ -116,7 +312,7 @@ module Datagrid
       # @example
       #   column(:name) do |model|
       #     format(model.name) do |value|
-      #       content_tag(:strong, value)
+      #       tag.strong(value)
       #     end
       #   end
       def format(value, &block)
@@ -147,20 +343,25 @@ module Datagrid
         end
         return self.decorator = block unless model
         return model unless decorator
+
         presenter = ::Datagrid::Utils.apply_args(model, &decorator)
-        presenter = presenter.is_a?(Class) ?  presenter.new(model) : presenter
+        presenter = presenter.new(model) if presenter.is_a?(Class)
         block_given? ? yield(presenter) : presenter
       end
 
       # @!visibility private
       def inherited(child_class)
-        super(child_class)
-        child_class.columns_array = self.columns_array.clone
+        super
+        child_class.columns_array = columns_array.clone
       end
 
       # @!visibility private
       def filter_columns(columns_array, *names, data: false, html: false)
         names.compact!
+        if names.size >= 1 && names.all? { |n| n.is_a?(Datagrid::Columns::Column) && n.grid_class == self.class }
+          return names
+        end
+
         names.map!(&:to_sym)
         columns_array.select do |column|
           (!data || column.data?) &&
@@ -173,7 +374,7 @@ module Datagrid
       def define_column(columns, name, query = nil, **options, &block)
         check_scope_defined!("Scope should be defined before columns")
         block ||= lambda do |model|
-          model.send(name)
+          model.public_send(name)
         end
         position = Datagrid::Utils.extract_position_from_options(columns, options)
         column = Datagrid::Columns::Column.new(
@@ -184,344 +385,356 @@ module Datagrid
       end
 
       # @!visibility private
-      def find_column_by_name(columns,name)
+      def find_column_by_name(columns, name)
         return name if name.is_a?(Datagrid::Columns::Column)
+
         columns.find do |col|
           col.name.to_sym == name.to_sym
         end
       end
-
     end
 
-    module InstanceMethods
+    # @!visibility private
+    def assets
+      append_column_preload(
+        driver.append_column_queries(
+          super, columns.select(&:query),
+        ),
+      )
+    end
 
-      # @!visibility private
-      def assets
-        append_column_preload(
-          driver.append_column_queries(
-            super, columns.select(&:query)
-          )
-        )
-      end
+    # @param column_names [Array<String>] list of column names if you want to limit data only to specified columns
+    # @return [Array<String>] human readable column names. See also "Localization" section
+    def header(*column_names)
+      data_columns(*column_names).map(&:header)
+    end
 
-      # @param column_names [Array<String>] list of column names if you want to limit data only to specified columns
-      # @return [Array<String>] human readable column names. See also "Localization" section
-      def header(*column_names)
-        data_columns(*column_names).map(&:header)
+    # @param asset [Object] asset from datagrid scope
+    # @param column_names [Array<String>] list of column names if you want to limit data only to specified columns
+    # @return [Array<Object>] column values for given asset
+    def row_for(asset, *column_names)
+      data_columns(*column_names).map do |column|
+        data_value(column, asset)
       end
+    end
 
-      # @param asset [Object] asset from datagrid scope
-      # @param column_names [Array<String>] list of column names if you want to limit data only to specified columns
-      # @return [Array<Object>] column values for given asset
-      def row_for(asset, *column_names)
-        data_columns(*column_names).map do |column|
-          data_value(column, asset)
-        end
+    # @param asset [Object] asset from datagrid scope
+    # @return [Hash] A mapping where keys are column names and values are column values for the given asset
+    def hash_for(asset)
+      result = {}
+      data_columns.each do |column|
+        result[column.name] = data_value(column, asset)
       end
+      result
+    end
 
-      # @param asset [Object] asset from datagrid scope
-      # @return [Hash] A mapping where keys are column names and values are column values for the given asset
-      def hash_for(asset)
-        result = {}
-        self.data_columns.each do |column|
-          result[column.name] = data_value(column, asset)
-        end
-        result
+    # @param column_names [Array<String>] list of column names if you want to limit data only to specified columns
+    # @return [Array<Array<Object>>] with data for each row in datagrid assets without header
+    def rows(*column_names)
+      map_with_batches do |asset|
+        row_for(asset, *column_names)
       end
+    end
 
-      # @param column_names [Array<String>] list of column names if you want to limit data only to specified columns
-      # @return [Array<Array<Object>>] with data for each row in datagrid assets without header
-      def rows(*column_names)
-        map_with_batches do |asset|
-          self.row_for(asset, *column_names)
-        end
-      end
+    # @param column_names [Array<String>] list of column names if you want to limit data only to specified columns
+    # @return [Array<Array<Object>>] data for each row in datagrid assets with header.
+    def data(*column_names)
+      rows(*column_names).unshift(header(*column_names))
+    end
 
-      # @param column_names [Array<String>] list of column names if you want to limit data only to specified columns
-      # @return [Array<Array<Object>>] data for each row in datagrid assets with header.
-      def data(*column_names)
-        self.rows(*column_names).unshift(self.header(*column_names))
+    # @return [Array<Hash{Symbol => Object}>] an array of hashes representing the rows in the filtered datagrid relation
+    #
+    # @example
+    #   class MyGrid
+    #     scope { Model }
+    #     column(:id)
+    #     column(:name)
+    #   end
+    #
+    #   Model.create!(name: "One")
+    #   Model.create!(name: "Two")
+    #
+    #   MyGrid.new.data_hash # => [{name: "One"}, {name: "Two"}]
+    def data_hash
+      map_with_batches do |asset|
+        hash_for(asset)
       end
+    end
 
-      # Return Array of Hashes where keys are column names and values are column values
-      # for each row in filtered datagrid relation.
-      #
-      # @example
-      #   class MyGrid
-      #     scope { Model }
-      #     column(:id)
-      #     column(:name)
-      #   end
-      #
-      #   Model.create!(name: "One")
-      #   Model.create!(name: "Two")
-      #
-      #   MyGrid.new.data_hash # => [{name: "One"}, {name: "Two"}]
-      def data_hash
-        map_with_batches do |asset|
-          hash_for(asset)
+    # @param column_names [Array<String>]
+    # @param options [Hash] CSV generation options
+    # @return [String] a CSV representation of the data in the grid
+    #
+    # @example
+    #   grid.to_csv
+    #   grid.to_csv(:id, :name)
+    #   grid.to_csv(col_sep: ';')
+    def to_csv(*column_names, **options)
+      require "csv"
+      CSV.generate(
+        headers: header(*column_names),
+        write_headers: true,
+        **options,
+      ) do |csv|
+        each_with_batches do |asset|
+          csv << row_for(asset, *column_names)
         end
       end
+    end
 
-      # @param column_names [Array<String>]
-      # @param options [Hash] CSV generation options
-      # @return [String] a CSV representation of the data in the grid
-      #
-      # @example
-      #   grid.to_csv
-      #   grid.to_csv(:id, :name)
-      #   grid.to_csv(col_sep: ';')
-      def to_csv(*column_names, **options)
-        require "csv"
-        CSV.generate(
-          headers: self.header(*column_names),
-          write_headers: true,
-          **options
-        ) do |csv|
-          each_with_batches do |asset|
-            csv << row_for(asset, *column_names)
-          end
-        end
+    # @param column_names [Array<Symbol, String>]
+    # @param [Boolean] data return only data columns
+    # @param [Boolean] html return only HTML columns
+    # @return [Array<Datagrid::Columns::Column>] all columns selected in grid instance
+    # @example
+    #   MyGrid.new.columns # => all defined columns
+    #   grid = MyGrid.new(column_names: [:id, :name])
+    #   grid.columns # => id and name columns
+    #   grid.columns(:id, :category) # => id and category column
+    def columns(*column_names, data: false, html: false)
+      self.class.filter_columns(
+        columns_array, *column_names, data: data, html: html,
+      ).select do |column|
+        column.enabled?(self)
       end
+    end
 
+    # @param column_names [Array<String, Symbol>] list of column names
+    #   if you want to limit data only to specified columns
+    # @param [Boolean] html return only HTML columns
+    # @return [Array<Datagrid::Columns::Column>] columns that can be represented in plain data(non-html) way
+    def data_columns(*column_names, html: false)
+      columns(*column_names, html: html, data: true)
+    end
 
-      # @param column_names [Array<Symbol, String>]
-      # @return [Array<Datagrid::Columns::Column>] all columns selected in grid instance
-      # @example
-      #   MyGrid.new.columns # => all defined columns
-      #   grid = MyGrid.new(column_names: [:id, :name])
-      #   grid.columns # => id and name columns
-      #   grid.columns(:id, :category) # => id and category column
-      def columns(*column_names, data: false, html: false)
-        self.class.filter_columns(
-          columns_array, *column_names, data: data, html: html
-        ).select do |column|
-          column.enabled?(self)
-        end
-      end
+    # @param column_names [Array<String>] list of column names if you want to limit data only to specified columns
+    # @param [Boolean] data return only data columns
+    # @return all columns that can be represented in HTML table
+    def html_columns(*column_names, data: false)
+      columns(*column_names, data: data, html: true)
+    end
 
-      # @param column_names [Array<String, Symbol>] list of column names if you want to limit data only to specified columns
-      # @return columns that can be represented in plain data(non-html) way
-      def data_columns(*column_names, **options)
-        self.columns(*column_names, **options, data: true)
-      end
+    # Finds a column definition by name
+    # @param name [String, Symbol] column name to be found
+    # @return [Datagrid::Columns::Column, nil]
+    def column_by_name(name)
+      self.class.find_column_by_name(columns_array, name)
+    end
 
-      # @param column_names [Array<String>] list of column names if you want to limit data only to specified columns
-      # @return all columns that can be represented in HTML table
-      def html_columns(*column_names, **options)
-        self.columns(*column_names, **options, html: true)
+    # Gives ability to have a different formatting for CSV and HTML column value.
+    #
+    # @example
+    #   column(:name) do |model|
+    #     format(model.name) do |value|
+    #       tag.strong(value)
+    #     end
+    #   end
+    #
+    #   column(:company) do |model|
+    #     format(model.company.name) do
+    #       render partial: "company_with_logo", locals: {company: model.company }
+    #     end
+    #   end
+    # @return [Datagrid::Columns::Column::ResponseFormat] Format object
+    def format(value, &block)
+      if block_given?
+        self.class.format(value, &block)
+      else
+        # don't override Object#format method
+        super
       end
+    end
 
-      # Finds a column definition by name
-      # @param name [String, Symbol] column name to be found
-      # @return [Datagrid::Columns::Column, nil]
-      def column_by_name(name)
-        self.class.find_column_by_name(columns_array, name)
-      end
+    # @return [Datagrid::Columns::DataRow] an object representing a grid row.
+    # @example
+    #  class MyGrid
+    #    scope { User }
+    #    column(:id)
+    #    column(:name)
+    #    column(:number_of_purchases) do |user|
+    #      user.purchases.count
+    #    end
+    #  end
+    #
+    #  row = MyGrid.new.data_row(User.last)
+    #  row.id # => user.id
+    #  row.number_of_purchases # => user.purchases.count
+    def data_row(asset)
+      ::Datagrid::Columns::DataRow.new(self, asset)
+    end
 
-      # Gives ability to have a different formatting for CSV and HTML column value.
-      #
-      # @example
-      #   column(:name) do |model|
-      #     format(model.name) do |value|
-      #       content_tag(:strong, value)
-      #     end
-      #   end
-      #
-      #   column(:company) do |model|
-      #     format(model.company.name) do
-      #       render partial: "company_with_logo", locals: {company: model.company }
-      #     end
-      #   end
-      # @return [Datagrid::Columns::Column::ResponseFormat] Format object
-      def format(value, &block)
-        if block_given?
-          self.class.format(value, &block)
-        else
-          # don't override Object#format method
-          super
-        end
-      end
+    # Defines a column at instance level
+    #
+    # @see Datagrid::Columns::ClassMethods#column
+    def column(name, query = nil, **options, &block)
+      self.class.define_column(columns_array, name, query, **options, &block)
+    end
 
-      # @return [Datagrid::Columns::DataRow] an object representing a grid row.
-      # @example
-      #  class MyGrid
-      #    scope { User }
-      #    column(:id)
-      #    column(:name)
-      #    column(:number_of_purchases) do |user|
-      #      user.purchases.count
-      #    end
-      #  end
-      #
-      #  row = MyGrid.new.data_row(User.last)
-      #  row.id # => user.id
-      #  row.number_of_purchases # => user.purchases.count
-      def data_row(asset)
-        ::Datagrid::Columns::DataRow.new(self, asset)
-      end
+    # @!visibility private
+    def initialize(*)
+      self.columns_array = self.class.columns_array.clone
+      super
+    end
 
-      # Defines a column at instance level
-      #
-      # @see Datagrid::Columns::ClassMethods#column
-      def column(name, query = nil, **options, &block)
-        self.class.define_column(columns_array, name, query, **options, &block)
+    # @return [Array<Datagrid::Columns::Column>] all columns
+    #   that are possible to be displayed for the current grid object
+    #
+    # @example
+    #   class MyGrid
+    #     filter(:search) {|scope, value| scope.full_text_search(value)}
+    #     column(:id)
+    #     column(:name, mandatory: true)
+    #     column(:search_match, if: proc {|grid| grid.search.present? }) do |model, grid|
+    #       search_match_line(model.searchable_content, grid.search)
+    #     end
+    #   end
+    #
+    #   grid = MyGrid.new
+    #   grid.columns # => [ #<Column:name> ]
+    #   grid.available_columns # => [ #<Column:id>, #<Column:name> ]
+    #   grid.search = "keyword"
+    #   grid.available_columns # => [ #<Column:id>, #<Column:name>, #<Column:search_match> ]
+    def available_columns
+      columns_array.select do |column|
+        column.enabled?(self)
       end
+    end
 
-      # @!visibility private
-      def initialize(*)
-        self.columns_array = self.class.columns_array.clone
-        super
-      end
+    # @return [Object] a cell data value for given column name and asset
+    def data_value(column_name, asset)
+      column = column_by_name(column_name)
+      cache(column, asset, :data_value) do
+        raise "no data value for #{column.name} column" unless column.data?
 
-      # @return [Array<Datagrid::Columns::Column>] all columns that are possible to be displayed for the current grid object
-      #
-      # @example
-      #   class MyGrid
-      #     filter(:search) {|scope, value| scope.full_text_search(value)}
-      #     column(:id)
-      #     column(:name, mandatory: true)
-      #     column(:search_match, if: proc {|grid| grid.search.present? }) do |model, grid|
-      #       search_match_line(model.searchable_content, grid.search)
-      #     end
-      #   end
-      #
-      #   grid = MyGrid.new
-      #   grid.columns # => [ #<Column:name> ]
-      #   grid.available_columns # => [ #<Column:id>, #<Column:name> ]
-      #   grid.search = "keyword"
-      #   grid.available_columns # => [ #<Column:id>, #<Column:name>, #<Column:search_match> ]
-      def available_columns
-        columns_array.select do |column|
-          column.enabled?(self)
-        end
+        result = generic_value(column, asset)
+        result.is_a?(Datagrid::Columns::Column::ResponseFormat) ? result.call_data : result
       end
+    end
 
-      # @return [Object] a cell data value for given column name and asset
-      def data_value(column_name, asset)
-        column = column_by_name(column_name)
-        cache(column, asset, :data_value) do
-          raise "no data value for #{column.name} column" unless column.data?
+    # @return [Object] a cell HTML value for given column name and asset and view context
+    def html_value(column_name, context, asset)
+      column = column_by_name(column_name)
+      cache(column, asset, :html_value) do
+        if column.html? && column.html_block
+          value_from_html_block(context, asset, column)
+        else
           result = generic_value(column, asset)
-          result.is_a?(Datagrid::Columns::Column::ResponseFormat) ? result.call_data : result
+          result.is_a?(Datagrid::Columns::Column::ResponseFormat) ? result.call_html(context) : result
         end
       end
+    end
 
-      # @return [Object] a cell HTML value for given column name and asset and view context
-      def html_value(column_name, context, asset)
-        column  = column_by_name(column_name)
-        cache(column, asset, :html_value) do
-          if column.html? && column.html_block
-            value_from_html_block(context, asset, column)
-          else
-            result = generic_value(column, asset)
-            result.is_a?(Datagrid::Columns::Column::ResponseFormat) ? result.call_html(context) : result
-          end
-        end
-      end
-
-      # @return [Object] a decorated version of given model if decorator is specified or the model otherwise.
-      def decorate(model)
-        self.class.decorate(model)
-      end
-
-      # @!visibility private
-      def generic_value(column, model)
-        cache(column, model, :generic_value) do
-          presenter = decorate(model)
-          unless column.enabled?(self)
-            raise Datagrid::ColumnUnavailableError, "Column #{column.name} disabled for #{inspect}"
-          end
+    # @return [Object] a decorated version of given model if decorator is specified or the model otherwise.
+    def decorate(model)
+      self.class.decorate(model)
+    end
 
-          if column.data_block.arity >= 1
-            Datagrid::Utils.apply_args(presenter, self, data_row(model), &column.data_block)
-          else
-            presenter.instance_eval(&column.data_block)
-          end
+    # @!visibility private
+    def generic_value(column, model)
+      cache(column, model, :generic_value) do
+        presenter = decorate(model)
+        unless column.enabled?(self)
+          raise Datagrid::ColumnUnavailableError, "Column #{column.name} disabled for #{inspect}"
         end
 
+        if column.data_block.arity >= 1
+          Datagrid::Utils.apply_args(presenter, self, data_row(model), &column.data_block)
+        else
+          presenter.instance_eval(&column.data_block)
+        end
       end
+    end
 
-      protected
+    # @!visibility private
+    def reset
+      super
+      @cache = {}
+    end
 
-      def append_column_preload(relation)
-        columns.inject(relation) do |current, column|
-          column.append_preload(current)
-        end
-      end
+    protected
 
-      def cache(column, asset, type)
-        @cache ||= {}
-        unless cached?
-          @cache.clear
-          return yield
-        end
-        key = cache_key(asset)
-        unless key
-          raise(Datagrid::CacheKeyError, "Datagrid Cache key is #{key.inspect} for #{asset.inspect}.")
-        end
-        @cache[column.name] ||= {}
-        @cache[column.name][key] ||= {}
-        @cache[column.name][key][type] ||= yield
+    def append_column_preload(relation)
+      columns.inject(relation) do |current, column|
+        column.append_preload(current)
       end
+    end
 
-      def cache_key(asset)
-        if cached.respond_to?(:call)
-          cached.call(asset)
-        else
-          driver.default_cache_key(asset)
-        end
-      rescue NotImplementedError
-        raise Datagrid::ConfigurationError, "#{self} is setup to use cache. But there was appropriate cache key found for #{asset.inspect}. Please set cached option to block with asset as argument and cache key as returning value to resolve the issue."
+    def cache(column, asset, type)
+      @cache ||= {}
+      unless cached?
+        @cache.clear
+        return yield
       end
+      key = cache_key(asset)
+      raise(Datagrid::CacheKeyError, "Datagrid Cache key is #{key.inspect} for #{asset.inspect}.") unless key
 
+      @cache[column.name] ||= {}
+      @cache[column.name][key] ||= {}
+      @cache[column.name][key][type] ||= yield
+    end
 
-      def map_with_batches(&block)
-        result = []
-        each_with_batches do |asset|
-          result << block.call(asset)
-        end
-        result
+    def cache_key(asset)
+      if cached.respond_to?(:call)
+        cached.call(asset)
+      else
+        driver.default_cache_key(asset)
+      end
+    rescue NotImplementedError
+      raise Datagrid::ConfigurationError,
+        <<~MSG
+          #{self} is setup to use cache. But there was appropriate cache key found for #{asset.inspect}.
+        MSG
+    end
+
+    def map_with_batches(&block)
+      result = []
+      each_with_batches do |asset|
+        result << block.call(asset)
       end
+      result
+    end
 
-      def each_with_batches(&block)
-        if batch_size && batch_size > 0
-          driver.batch_each(assets, batch_size, &block)
-        else
-          assets.each(&block)
-        end
+    def each_with_batches(&block)
+      if batch_size&.positive?
+        driver.batch_each(assets, batch_size, &block)
+      else
+        assets.each(&block)
       end
+    end
 
-      def value_from_html_block(context, asset, column)
-        args = []
-        remaining_arity = column.html_block.arity
-        remaining_arity = 1 if remaining_arity < 0
+    def value_from_html_block(context, asset, column)
+      args = []
+      remaining_arity = column.html_block.arity
+      remaining_arity = 1 if remaining_arity.negative?
 
-        asset = decorate(asset)
+      asset = decorate(asset)
 
-        if column.data?
-          args << data_value(column, asset)
-          remaining_arity -= 1
-        end
+      if column.data?
+        args << data_value(column, asset)
+        remaining_arity -= 1
+      end
 
-        args << asset if remaining_arity > 0
-        args << self if remaining_arity > 1
+      args << asset if remaining_arity.positive?
+      args << self if remaining_arity > 1
 
-        context.instance_exec(*args, &column.html_block)
-      end
+      context.instance_exec(*args, &column.html_block)
     end
 
     # Object representing a single row of data when building a datagrid table
-    # @see Datagrid::Columns::InstanceMethods#data_row
+    # @see Datagrid::Columns#data_row
     class DataRow < BasicObject
       def initialize(grid, model)
         @grid = grid
         @model = model
       end
 
-      def method_missing(meth, *args, &blk)
+      def method_missing(meth, *_args)
         @grid.data_value(meth, @model)
       end
+
+      def respond_to_missing?(meth, include_private = false)
+        !!@grid.column_by_name(meth) || super
+      end
     end
   end
 end