command_search 0.11.1 → 0.11.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 647f1b1fa534c1187bd63333ae986e357fb122a777e9f56d9af2d7d6f34ee6c8
-  data.tar.gz: d070864c2e1bba273a8fa837b20c7732efcbb5768149046cf2e68ee666e6f0b3
+  metadata.gz: c9a7f59d4e3aefbd46963b94d52d18d49d63263ab51ceb31a207ee17ca23bf1b
+  data.tar.gz: 43fe311f65ec44dd89dcb997f0b61daa1347466f63fae1721c0168eb62bfa377
 SHA512:
-  metadata.gz: 3b37775511b9cb93c556c9ce766bc76b3df6f15fb8a2671c814fa0809d32c96a5ffff0371cce4f9fb53dd9de0123f477c233611be04280883c811635a0775208
-  data.tar.gz: e9a636dd33c9e53211e381a00e6af3cf2aa0901fdd5e6a12079053285465eb8b4775206f48b295146b2a2e08e4e440b3888e8bff0c1bd8270edff520ad00e0b5
+  metadata.gz: f9751ac591a874cf99009d28fc691b6543b5fc038dbbb6d7a7b063cf47f134da0e8ee500300877565e5c2b4164f82a446d983c9a6730c2eab35a86a6ae6a92ed
+  data.tar.gz: 47173f4d0904edcb9ddcbd1f2a24e00f9cd70d528c9234cd1840c861cb4cb7e85681bc6b3362fb73fefff095d16cb6aa277f83027185bc5068ed43c3b5b9b757

data/lib/command_search.rb

@@ -8,6 +8,7 @@ load(__dir__ + '/command_search/backends/memory.rb')
 load(__dir__ + '/command_search/backends/mongoer.rb')
 load(__dir__ + '/command_search/backends/postgres.rb')
 load(__dir__ + '/command_search/backends/sqlite.rb')
+load(__dir__ + '/command_search/backends/mysql.rb')
 
 class Boolean; end
 
@@ -29,6 +30,10 @@ module CommandSearch
       Normalizer.normalize!(ast, fields, false)
       return Sqlite.build_query(ast)
     end
+    if type == :mysql
+      Normalizer.normalize!(ast, fields, false)
+      return Mysql.build_query(ast)
+    end
     Normalizer.normalize!(ast, fields)
     return Mongoer.build_query(ast) if type == :mongo
     ast
@@ -47,6 +52,10 @@ module CommandSearch
       ast = CommandSearch.build(:sqlite, query, options)
       return source.where(ast)
     end
+    if source.respond_to?(:mysql2_connection)
+      ast = CommandSearch.build(:mysql, query, options)
+      return source.where(ast)
+    end
     ast = CommandSearch.build(:other, query, options)
     source.select { |x| CommandSearch::Memory.check(x, ast) }
   end

data/lib/command_search/backends/mysql.rb

@@ -0,0 +1,106 @@
+module CommandSearch
+  module Mysql
+    module_function
+
+    def quote_string(str)
+      # activerecord/lib/active_record/connection_adapters/abstract/quoting.rb:62
+      str.gsub('\\', '\&\&').gsub("'", "''")
+    end
+
+    def build_quoted_regex(str)
+      str = Regexp.escape(str)
+      str = quote_string(str)
+      if str[/(^\W)|(\W$)/]
+        head_border = '(^|[^:+[[:alnum:]]])'
+        tail_border = '($|[^:+[[:alnum:]]])'
+        return head_border + str + tail_border
+      end
+      '\\\\b' + str + '\\\\b'
+    end
+
+    def command_search(node)
+      field_node = node[:value].first
+      field = field_node[:value]
+      search_node = node[:value].last
+      val = search_node[:value]
+      type = search_node[:type]
+      return '0 = 1' if field == '__CommandSearch_dummy_key__'
+      if type == Boolean || type == :existence
+        if val
+          return "(NOT ((#{field} IS NULL) OR (#{field} LIKE '0')))"
+        end
+        return "((#{field} IS NULL) OR (#{field} LIKE '0'))"
+      end
+      if type == Time
+        return '0 = 1' unless val
+        return "
+          (
+            (#{field} > '#{val[0] - 1}') AND
+            (#{field} <= '#{val[1] - 1}') AND
+            (#{field} IS NOT NULL)
+          )
+        "
+      end
+      if type == :quote
+        op = 'RLIKE BINARY'
+        val = "'#{build_quoted_regex(val)}'"
+      elsif type == :str
+        op = 'LIKE'
+        val = quote_string(val)
+        val.gsub!('%', '\%')
+        val.gsub!('_', '\_')
+        val = "'%#{val}%'"
+      elsif type == :number
+        op = '='
+      end
+      "((#{field} #{op} #{val}) AND (#{field} IS NOT NULL))"
+    end
+
+    def compare_search(node)
+      field_node = node[:value].first
+      field = field_node[:value]
+      search_node = node[:value].last
+      val = search_node[:value]
+      type = search_node[:type]
+      op = node[:nest_op]
+      if node[:compare_across_fields]
+        "
+          (
+            (#{field} #{op} #{val}) AND
+            (#{field} IS NOT NULL) AND
+            (#{val} IS NOT NULL)
+          )
+        "
+      elsif type == Time && val
+        "(#{field} #{op} '#{val}') AND (#{field} IS NOT NULL)"
+      elsif val.is_a?(Numeric) || val == val.to_i.to_s || val == val.to_f.to_s
+        "(#{field} #{op} #{val}) AND (#{field} IS NOT NULL)"
+      else
+        '0 = 1'
+      end
+    end
+
+    def build_query(ast)
+      out = []
+      ast = [ast] unless ast.is_a?(Array)
+      ast.each do |node|
+        type = node[:type]
+        if type == :colon
+          out.push(command_search(node))
+        elsif type == :compare
+          out.push(compare_search(node))
+        elsif type == :and
+          out.push(build_query(node[:value]))
+        elsif type == :or
+          clauses = node[:value].map { |x| build_query(x) }
+          clause = clauses.join(' OR ')
+          out.push("(#{clause})")
+        elsif type == :not
+          clause = build_query(node[:value])
+          out.push("NOT (#{clause})")
+        end
+      end
+      out.join(' AND ')
+    end
+  end
+end

data/lib/command_search/backends/postgres.rb

@@ -45,7 +45,7 @@ module CommandSearch
       end
       if type == :quote
         op = '~'
-        val =  "'#{build_quoted_regex(val)}'"
+        val = "'#{build_quoted_regex(val)}'"
       elsif type == :str
         op = '~~*'
         val = quote_string(val)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: command_search
 version: !ruby/object:Gem::Version
-  version: 0.11.1
+  version: 0.11.2
 platform: ruby
 authors:
 - zumbalogy
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-12-18 00:00:00.000000000 Z
+date: 2019-12-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: chronic
@@ -34,6 +34,7 @@ files:
 - lib/command_search/aliaser.rb
 - lib/command_search/backends/memory.rb
 - lib/command_search/backends/mongoer.rb
+- lib/command_search/backends/mysql.rb
 - lib/command_search/backends/postgres.rb
 - lib/command_search/backends/sqlite.rb
 - lib/command_search/lexer.rb