groonga-command 1.0.0

data/lib/groonga/command/select.rb

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2012  Kouhei Sutou <kou@clear-code.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+require "groonga/command/base"
+
+module Groonga
+  module Command
+    class Select < Base
+      Command.register("select", self)
+
+      class << self
+        def parameter_names
+          # TODO: Fill me
+          [
+            :table
+          ]
+        end
+      end
+
+      def sortby
+        self[:sortby]
+      end
+
+      def scorer
+        self[:scorer]
+      end
+
+      def query
+        self[:query]
+      end
+
+      def filter
+        self[:filter]
+      end
+
+      def conditions
+        @conditions ||= filter.split(/(?:&&|&!|\|\|)/).collect do |condition|
+          condition = condition.strip
+          condition = condition.gsub(/\A[\s\(]*/, '')
+          condition = condition.gsub(/[\s\)]*\z/, '') unless /\(/ =~ condition
+          condition
+        end
+      end
+
+      def drilldowns
+        @drilldowns ||= (self[:drilldown] || "").split(/\s*,\s*/)
+      end
+
+      def output_columns
+        self[:output_columns]
+      end
+    end
+  end
+end

data/lib/groonga/command/suggest.rb

@@ -0,0 +1,47 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2012  Haruka Yoshihara <yoshihara@clear-code.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+require "groonga/command/base"
+
+module Groonga
+  module Command
+    class Suggest < Base
+      Command.register("suggest", self)
+
+      class << self
+        def parameter_names
+          [
+            :types,
+            :table,
+            :column,
+            :query,
+            :sortby,
+            :output_columns,
+            :offset,
+            :limit,
+            :frequency_threshold,
+            :conditional_probability_threshold,
+            :prefix_search,
+            :similar_search
+          ]
+        end
+      end
+    end
+  end
+end
+

data/lib/groonga/command/table-create.rb

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2012  Kouhei Sutou <kou@clear-code.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+require "groonga/command/base"
+
+module Groonga
+  module Command
+    class TableCreate < Base
+      Command.register("table_create", self)
+
+      class << self
+        def parameter_names
+          [
+            :name,
+            :flags,
+            :key_type,
+            :value_type,
+            :default_tokenizer,
+          ]
+        end
+      end
+    end
+  end
+end

data/lib/groonga/command/table-remove.rb

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2012  Kouhei Sutou <kou@clear-code.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+require "groonga/command/base"
+
+module Groonga
+  module Command
+    class TableRemove < Base
+      Command.register("table_remove", self)
+
+      class << self
+        def parameter_names
+          [
+            :name,
+          ]
+        end
+      end
+    end
+  end
+end

data/lib/groonga/command/table-rename.rb

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2012  Kouhei Sutou <kou@clear-code.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+require "groonga/command/base"
+
+module Groonga
+  module Command
+    class TableRename < Base
+      Command.register("table_rename", self)
+
+      class << self
+        def parameter_names
+          [
+            :name,
+            :new_name,
+          ]
+        end
+      end
+    end
+  end
+end

data/lib/groonga/command/truncate.rb

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2012  Kouhei Sutou <kou@clear-code.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+require "groonga/command/base"
+
+module Groonga
+  module Command
+    class Truncate < Base
+      Command.register("truncate", self)
+
+      class << self
+        def parameter_names
+          [
+            :table,
+          ]
+        end
+      end
+    end
+  end
+end

data/lib/groonga/command/version.rb

@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2012  Kouhei Sutou <kou@clear-code.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+module Groonga
+  module Command
+    VERSION = "1.0.0"
+  end
+end

data/test/command/test-base.rb

@@ -0,0 +1,114 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2011-2012  Kouhei Sutou <kou@clear-code.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+class BaseCommandTest < Test::Unit::TestCase
+  class CovnertToURIFormatTest < self
+    def test_no_special_characters
+      select = Groonga::Command::Base.new("select",
+                                          :table => "Users",
+                                          :filter => "age<=30",
+                                          :output_type => "json")
+      assert_equal("/d/select.json?filter=age%3C%3D30&table=Users",
+                   select.to_uri_format)
+    end
+
+    def test_double_quote
+      filter = 'geo_in_rectangle(location,' +
+                                '"35.73360x139.7394","62614x139.7714") && ' +
+                 '((type == "たいやき" || type == "和菓子")) && ' +
+                 'keyword @ "たいやき" &! keyword @ "白" &! keyword @ "養殖"'
+      select = Groonga::Command::Base.new("select",
+                                          :table => "Users",
+                                          :filter => filter,
+                                          :output_type => "json")
+      assert_equal("/d/select.json?filter=geo_in_rectangle%28location%2C" +
+                   "%2235.73360x139.7394%22%2C%2262614x139.7714%22%29+" +
+                   "%26%26+%28%28type+" +
+                   "%3D%3D+%22%E3%81%9F%E3%81%84%E3%82%84%E3%81%8D%22+" +
+                   "%7C%7C+type+%3D%3D+" +
+                   "%22%E5%92%8C%E8%8F%93%E5%AD%90%22%29%29+" +
+                   "%26%26+keyword+%40+" +
+                   "%22%E3%81%9F%E3%81%84%E3%82%84%E3%81%8D%22+%26%21+" +
+                   "keyword+%40+%22%E7%99%BD%22+%26%21+" +
+                   "keyword+%40+%22%E9%A4%8A%E6%AE%96%22&table=Users",
+                   select.to_uri_format)
+    end
+  end
+
+  class CovnertToCommandFormatTest < self
+    def test_no_special_characters
+      select = Groonga::Command::Base.new("select",
+                                          :table => "Users",
+                                          :filter => "age<=30",
+                                          :output_type => "json")
+      assert_equal("select --filter \"age<=30\" " +
+                     "--output_type \"json\" --table \"Users\"",
+                   select.to_command_format)
+    end
+
+    def test_double_quote
+      filter = 'geo_in_rectangle(location,' +
+                                '"35.73360x139.7394","62614x139.7714") && ' +
+                 '((type == "たいやき" || type == "和菓子")) && ' +
+                 'keyword @ "たいやき" &! keyword @ "白" &! keyword @ "養殖"'
+      select = Groonga::Command::Base.new("select",
+                                          :table => "Users",
+                                          :filter => filter,
+                                          :output_type => "json")
+      assert_equal("select " +
+                     "--filter " +
+                       "\"geo_in_rectangle(location," +
+                       "\\\"35.73360x139.7394\\\",\\\"62614x139.7714\\\") && " +
+                       "((type == \\\"たいやき\\\" || " +
+                         "type == \\\"和菓子\\\")) && " +
+                       "keyword @ \\\"たいやき\\\" &! keyword @ \\\"白\\\" &! " +
+                       "keyword @ \\\"養殖\\\"\" " +
+                     "--output_type \"json\" --table \"Users\"",
+                   select.to_command_format)
+    end
+  end
+
+  sub_test_case("#[]") do
+    def setup
+      @table = "Users"
+      @select = Groonga::Command::Base.new("select", :table => @table)
+    end
+
+    def test_symbol
+      assert_equal(@table, @select[:table])
+    end
+
+    def test_string
+      assert_equal(@table, @select["table"])
+    end
+  end
+
+  sub_test_case("#has_key?") do
+    def setup
+      @select = Groonga::Command::Base.new("select", :table => "Users")
+    end
+
+    def test_symbol
+      assert_true(@select.has_key?(:table))
+    end
+
+    def test_string
+      assert_true(@select.has_key?("table"))
+    end
+  end
+end

data/test/command/test-column-create.rb

@@ -0,0 +1,47 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2012  Kouhei Sutou <kou@clear-code.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+class ColumnCreateCommandTest < Test::Unit::TestCase
+  class CommandLineTest < self
+    include GroongaCommandTestUtils::CommandLineCommandParser
+
+    def test_ordered_arguments
+      table    = "Lexicon"
+      name     = "content_index"
+      flags    = "COLUMN_INDEX"
+      type     = "Posts"
+      source   = "content"
+
+      command = parse(table, name, flags, type, source)
+      assert_instance_of(Groonga::Command::ColumnCreate, command)
+      assert_equal({
+                     :table    => table,
+                     :name     => name,
+                     :flags    => flags,
+                     :type     => type,
+                     :source   => source,
+                   },
+                   command.arguments)
+    end
+
+    private
+    def parse(*arguments)
+      super("column_create", arguments, :output_type => false)
+    end
+  end
+end