active-record-data-tables 0.0.0 → 0.0.1

data/Gemfile

@@ -3,6 +3,8 @@ source "http://rubygems.org"
 # Example:
 #   gem "activesupport", ">= 2.3.5"
 
+gem "string-cases"
+
 # Add dependencies to develop your gem here.
 # Include everything needed to run rake, tests, features, etc.
 group :development do

data/Gemfile.lock

@@ -20,6 +20,7 @@ GEM
     rspec-expectations (2.8.0)
       diff-lcs (~> 1.1.2)
     rspec-mocks (2.8.0)
+    string-cases (0.0.0)
 
 PLATFORMS
   ruby
@@ -29,3 +30,4 @@ DEPENDENCIES
   jeweler (~> 1.8.4)
   rdoc (~> 3.12)
   rspec (~> 2.8.0)
+  string-cases

data/VERSION

@@ -1 +1 @@
-0.0.0
+0.0.1

data/active-record-data-tables.gemspec

@@ -0,0 +1,63 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = "active-record-data-tables"
+  s.version = "0.0.1"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["kaspernj"]
+  s.date = "2013-06-04"
+  s.description = "Helping functionality for filtering, sorting and stuff like that with data-tables and ActiveRecord."
+  s.email = "k@spernj.org"
+  s.extra_rdoc_files = [
+    "LICENSE.txt",
+    "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+    ".rspec",
+    "Gemfile",
+    "Gemfile.lock",
+    "LICENSE.txt",
+    "README.rdoc",
+    "Rakefile",
+    "VERSION",
+    "active-record-data-tables.gemspec",
+    "lib/active-record-data-tables.rb",
+    "spec/active-record-data-tables_spec.rb",
+    "spec/spec_helper.rb"
+  ]
+  s.homepage = "http://github.com/kaspernj/active-record-data-tables"
+  s.licenses = ["MIT"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = "1.8.23"
+  s.summary = "Helping functionality for filtering, sorting and stuff like that with data-tables and ActiveRecord."
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<string-cases>, [">= 0"])
+      s.add_development_dependency(%q<rspec>, ["~> 2.8.0"])
+      s.add_development_dependency(%q<rdoc>, ["~> 3.12"])
+      s.add_development_dependency(%q<bundler>, [">= 1.0.0"])
+      s.add_development_dependency(%q<jeweler>, ["~> 1.8.4"])
+    else
+      s.add_dependency(%q<string-cases>, [">= 0"])
+      s.add_dependency(%q<rspec>, ["~> 2.8.0"])
+      s.add_dependency(%q<rdoc>, ["~> 3.12"])
+      s.add_dependency(%q<bundler>, [">= 1.0.0"])
+      s.add_dependency(%q<jeweler>, ["~> 1.8.4"])
+    end
+  else
+    s.add_dependency(%q<string-cases>, [">= 0"])
+    s.add_dependency(%q<rspec>, ["~> 2.8.0"])
+    s.add_dependency(%q<rdoc>, ["~> 3.12"])
+    s.add_dependency(%q<bundler>, [">= 1.0.0"])
+    s.add_dependency(%q<jeweler>, ["~> 1.8.4"])
+  end
+end
+

data/lib/active-record-data-tables.rb

@@ -1,21 +1,85 @@
+require "string-cases"
+
 class ActiveRecordDataTables
   def initialize(args)
     @args = args
     @params = @args[:params]
+    @filter_params = @args[:filter_params]
+    @force_filter_for = @args[:force_filter_for]
+    @dts = @params[:dataTables]
     @columns = @args[:columns]
+    
+    @model = @args[:model]
+    raise "No ':model' was given." unless @model
+    @model_method_name = StringCases.camel_to_snake(@model.name)
+    
+    @joins = {}
+    @sort_columns = [:title, :name]
+    @executed = false
   end
   
   def execute
-    @obj = @args[:model]
+    @executed = true
+    @query = @model
+    
+    filter
     sorting
     limit
+    
+    return @query
+  end
+  
+  def json(final_query = nil)
+    unless final_query
+      execute unless @executed
+      final_query = @query
+    end
+    
+    dis_rec_count = final_query.limit(nil).count    
+    dis_rec_count = dis_rec_count.length if dis_rec_count.is_a?(Hash)
+    
+    res = {
+      :sEcho => @dts[:sEcho] ? @dts[:sEcho].to_i : 1,
+      :iTotalRecords => @model.count,
+      :iTotalDisplayRecords => dis_rec_count,
+      :aaData => []
+    }
   end
   
   private
   
+  def filter
+    @filter_params.each do |key, val|
+      if key.start_with?("#{@model_method_name}_")
+        key = key.slice(@model_method_name.length + 1, key.length)
+      end
+      
+      if @force_filter_for && @force_filter_for.include?(key)
+        ret = @args[:filter].call(:key => key, :val => val, :query => @query)
+        @query = ret if ret
+      elsif @model.column_names.include?(key)
+        @query = @query.where(key => val)
+      elsif match = key.to_s.match(/^(.+)_like$/) and @model.column_names.include?(match[1])
+        next if val.blank?
+        table = @model.arel_table
+        @query = @query.where(table[match[1].to_sym].matches("%#{escape(val)}%"))
+      elsif @args[:filter]
+        ret = @args[:filter].call(:key => key, :val => val, :query => @query)
+        @query = ret if ret
+      else
+        raise "Dont know what to do regarding filter with key: '#{key}'."
+      end
+    end
+  end
+  
   def limit
-    disp_start = @params["iDisplayStart"].to_i
-    disp_length = @params["iDisplayLength"].to_i
+    raise "'iDisplayStart' was not given? #{@dts}" unless @dts.key?("iDisplayStart")
+    raise "'iDisplayEnd' was not given? #{@dts}" unless @dts.key?("iDisplayLength")
+    
+    disp_start = @dts["iDisplayStart"].to_i
+    disp_length = @dts["iDisplayLength"].to_i
+    
+    @query = @query.page((disp_start / disp_length) + 1).per(disp_length)
   end
   
   def sorting
@@ -23,30 +87,63 @@ class ActiveRecordDataTables
     sorts = []
     
     loop do
+      sorted = false
       name_col = "iSortCol_#{sort_no}"
       name_mode = "sSortDir_#{sort_no}"
-      sort_col = @params[name]
+      sort_col = @dts[name_col]
       break if !sort_col
       
-      col_name = @columns[name_col]
+      col_name = @columns[sort_col.to_i]
       next if !col_name
       
-      if name_mode == "desc"
+      if @dts[name_mode] == "desc"
         sort_mode = "DESC"
       else
         sort_mode = "ASC"
       end
       
-      sorts << "`#{escape_col(col_name)}` #{sort_mode}"
+      if match = col_name.to_s.match(/^(.+)_id$/)
+        method_name = match[1]
+        sub_model_name = StringCases.snake_to_camel(col_name.slice(0, col_name.length - 3))
+        
+        if Kernel.const_defined?(sub_model_name)
+          sub_model_const = Kernel.const_get(sub_model_name)
+          unless @joins.key?(method_name)
+            @query = @query.includes(method_name)
+            @joins[method_name] = true
+          end
+          
+          @sort_columns.each do |sort_col_name|
+            if sub_model_const.column_names.include?(sort_col_name.to_s)
+              sorts << "`#{sub_model_const.table_name}`.`#{escape_col(sort_col_name)}` #{sort_mode}"
+              sorted = true
+              break
+            end
+          end
+        end
+      end
+      
+      if @model.column_names.include?(col_name.to_s)
+        sorts << "`#{@model.table_name}`.`#{escape_col(col_name)}` #{sort_mode}"
+      elsif @args[:sort]
+        res = @args[:sort].call(:key => col_name, :sort_mode => sort_mode, :query => @query)
+        @query = res if res
+      else
+        raise "Unknown sort-column: '#{col_name}'."
+      end
       
       sort_no += 1
     end
     
-    @obj = @obj.order(sorts.join(" "))
+    @query = @query.order(sorts.join(", "))
+  end
+  
+  def escape(str)
+    return ActiveRecord::Base.connection.quote(str).slice(1, str.length - 1)
   end
   
   def escape_col(name)
-    raise "Possible SQL injection hack: '#{name}'." unless name.to_s.match(/\A[A-z\d_+]\Z/)
+    raise "Possible SQL injection hack: '#{name}'." unless name.to_s.match(/\A[A-z\d_]+\Z/)
     return name
   end
 end

metadata

@@ -2,15 +2,31 @@
 name: active-record-data-tables
 version: !ruby/object:Gem::Version
   prerelease: 
-  version: 0.0.0
+  version: 0.0.1
 platform: ruby
 authors:
 - kaspernj
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-06-03 00:00:00.000000000 Z
+date: 2013-06-04 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: string-cases
+  type: :runtime
+  prerelease: false
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   version_requirements: !ruby/object:Gem::Requirement
     none: false
@@ -92,6 +108,7 @@ files:
 - README.rdoc
 - Rakefile
 - VERSION
+- active-record-data-tables.gemspec
 - lib/active-record-data-tables.rb
 - spec/active-record-data-tables_spec.rb
 - spec/spec_helper.rb
@@ -109,7 +126,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       segments:
       - 0
-      hash: 344081574259786733
+      hash: 2526805225709928991
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false