marklogic 0.0.1

data/lib/marklogic/queries/or_query.rb

@@ -0,0 +1,24 @@
+module MarkLogic
+  module Queries
+    class OrQuery< BaseQuery
+      def initialize(*args)
+        @queries = args.flat_map{ |i| i }
+      end
+
+      def to_json
+        {
+          "or-query" => {
+            "queries" => @queries.map do |q|
+              q.to_json
+            end
+          }
+        }
+      end
+
+      def to_xqy
+        sub_queries = @queries.map { |q| q.to_xqy }.join(', ')
+        %Q{cts:or-query((#{sub_queries}))}
+      end
+    end
+  end
+end

data/lib/marklogic/queries/properties_fragment_query.rb

@@ -0,0 +1,13 @@
+module MarkLogic
+  module Queries
+    class PropertiesFragmentQuery< BaseQuery
+      def initialize(query)
+        @query = query
+      end
+
+      def to_xqy
+        %Q{cts:properties-fragment-query(#{@query.to_xqy})}
+      end
+    end
+  end
+end

data/lib/marklogic/queries/range_query.rb

@@ -0,0 +1,67 @@
+module MarkLogic
+  module Queries
+    class RangeQuery < BaseQuery
+
+      attr_accessor :name, :range_type
+
+      def initialize(name, operator, range_type, value, options = {})
+        @name = name.to_s
+        @operator = operator.to_s.upcase
+        @range_type = range_type
+        @value = value
+        @options = options || {}
+        @weight = @options.delete(:weight) || 1.0
+      end
+
+      def operator=(op)
+        @operator = op
+      end
+
+      def operator
+        case @operator
+          when "LT"
+            "<"
+          when "LE"
+            "<="
+          when "GT"
+            ">"
+          when "GE"
+            ">="
+          when "EQ"
+            "="
+          when "NE"
+            "!="
+          else
+            @operator
+        end
+      end
+
+      def options=(opts)
+        @options = opts
+      end
+
+      def options
+        opts = []
+        @options.each do |k, v|
+          case k.to_s
+          when "collation", "min_occurs", "max_occurs", "score_function", "slope_factor"
+            opts << %Q{"#{k.to_s.gsub(/_/, '-')}=#{v}"}
+          when "cached"
+            opts << (v == true ? %Q{"cached"} : %Q{"uncached"})
+          when "synonym"
+            opts << %Q{"#{k}"}
+          else
+            opts << %Q{"#{v}"}
+          end
+        end
+
+        opts
+      end
+
+      def to_xqy
+        value = query_value(@value, @range_type)
+        %Q{cts:json-property-range-query("#{@name}","#{operator}",(#{value}),(#{options.join(',')}),#{@weight})}
+      end
+    end
+  end
+end

data/lib/marklogic/queries/value_query.rb

@@ -0,0 +1,44 @@
+module MarkLogic
+  module Queries
+    class ValueQuery< BaseQuery
+      def initialize(name, value, options = {})
+        @name = name.to_s
+        @value = value
+        @value = value.to_s if value.is_a?(ObjectId)
+        @options = options || {}
+        @weight = @options.delete(:weight) || 1.0
+        @options[:exact] = true if @options.length == 0
+      end
+
+      def options=(opts)
+        @options = opts
+      end
+
+      def options
+        opts = []
+        @options.each do |k, v|
+          dashed_key = k.to_s.gsub(/_/, '-')
+          case k.to_s
+          when "lang", "min_occurs", "max_occurs"
+            opts << %Q{"#{dashed_key}=#{v}"}
+          when /(case|diacritic|punctuation|whitespace)_sensitive/
+            opts << (v == true ? %Q{"#{$1}-sensitive"} : %Q{"#{$1}-insensitive"})
+          when "exact"
+            opts << %Q{"#{dashed_key}"}
+          when "stemmed", "wildcarded"
+            opts << (v == true ? %Q{"#{dashed_key}"} : %Q{"un#{dashed_key}"})
+          else
+            opts << %Q{"#{v}"}
+          end
+        end
+
+        opts
+      end
+
+      def to_xqy
+        value = query_value(@value)
+        %Q{cts:json-property-value-query("#{@name}",(#{value}),(#{options.join(',')}),#{@weight})}
+      end
+    end
+  end
+end

data/lib/marklogic/queries/word_query.rb

@@ -0,0 +1,38 @@
+module MarkLogic
+  module Queries
+    class WordQuery< BaseQuery
+      def initialize(values, options = {})
+        @values = values
+        @options = options || {}
+        @weight = @options.delete(:weight) || 1.0
+        @options[:exact] = true if @options.length == 0
+      end
+
+      def options
+        opts = []
+        @options.each do |k, v|
+          dashed_key = k.to_s.gsub(/_/, '-')
+          case k.to_s
+          when "lang", "distance_weight", "min_occurs", "max_occurs", "lexicon_expand"
+            opts << %Q{"#{dashed_key}=#{v}"}
+          when /(case|diacritic|punctuation|whitespace)_sensitive/
+            opts << (v == true ? %Q{"#{$1}-sensitive"} : %Q{"#{$1}-insensitive"})
+          when "exact"
+            opts << %Q{"#{dashed_key}"}
+          when "stemmed", "wildcarded"
+            opts << (v == true ? %Q{"#{dashed_key}"} : %Q{"un#{dashed_key}"})
+          else
+            opts << %Q{"#{v}"}
+          end
+        end
+
+        opts
+      end
+
+      def to_xqy
+        values = query_value(@values)
+        %Q{cts:word-query((#{values}),(#{options.join(',')}),#{@weight})}
+      end
+    end
+  end
+end

data/lib/marklogic/version.rb

@@ -0,0 +1,3 @@
+module Marklogic
+  VERSION = "0.0.1"
+end

data/marklogic.gemspec

@@ -0,0 +1,23 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'marklogic/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'marklogic'
+  spec.version       = Marklogic::VERSION
+  spec.authors       = ['Paxton Hare']
+  spec.email         = ['paxton@greenllama.com']
+  spec.summary       = %q{A Ruby Driver for MarkLogic}
+  spec.description   = %q{A Ruby Driver for MarkLogic}
+  spec.homepage      = ''
+  spec.license       = 'MIT'
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ['lib']
+  spec.has_rdoc        = 'yard'
+
+  spec.add_runtime_dependency 'activesupport', '~> 4.2', '>= 4.2.0'
+end

data/spec/marklogic/app_server_spec.rb

@@ -0,0 +1,21 @@
+require 'spec_helper'
+
+describe MarkLogic::AppServer do
+
+  describe "instance" do
+    let(:s) do
+      MarkLogic::AppServer.new(
+        "marklogic-gem-test",
+        8039,
+        "http",
+        "Default")
+    end
+
+    it "should create accessors" do
+      expect(s.server_name).to eq("marklogic-gem-test")
+      expect(s.server_type).to eq("http")
+      expect(s["content-database"]).to eq("marklogic-gem-test-content")
+    end
+
+  end
+end

data/spec/marklogic/application_spec.rb

@@ -0,0 +1,105 @@
+require 'spec_helper'
+
+describe MarkLogic::Application do
+
+  before do
+    @a = MarkLogic::Application.new("marklogic-gem-application-test2", :port => PORT + 1)
+    @content_db = MarkLogic::Database.new("marklogic-gem-application-test2-content")
+    @content_forest = MarkLogic::Forest.new("marklogic-gem-application-test2-content")
+    @modules_db = MarkLogic::Database.new("marklogic-gem-application-test2-modules")
+    @modules_forest = MarkLogic::Database.new("marklogic-gem-application-test2-modules")
+    @app_server = MarkLogic::AppServer.new("marklogic-gem-application-test2", PORT + 1, "http", "Default")
+  end
+
+  after do
+    @a.drop
+  end
+
+
+  describe "new" do
+    it "should create an instance" do
+      expect(@a.app_name).to eq("marklogic-gem-application-test2")
+      expect(@a.databases.length).to eq(0)
+    end
+  end
+
+  describe "exists?" do
+    it "should return if it exists or not" do
+      @a.drop if @a.exists?
+      expect(@a).to_not be_exists
+      @a.create
+      expect(@a).to be_exists
+      @a.drop
+      expect(@a).to_not be_exists
+    end
+  end
+
+  describe "create" do
+    it "should bootstrap" do
+      @a.drop if @a.exists?
+
+      expect(@content_forest).to_not be_exists
+      expect(@content_db).to_not be_exists
+      expect(@modules_db).to_not be_exists
+      expect(@modules_forest).to_not be_exists
+      expect(@app_server).to_not be_exists
+
+      @a.create
+      expect(@content_forest).to be_exists
+      expect(@content_db).to be_exists
+      expect(@modules_forest).to be_exists
+      expect(@modules_db).to be_exists
+      expect(@app_server).to be_exists
+
+      @a.drop
+    end
+  end
+
+  describe "drop" do
+    it "should drop" do
+      @a.drop if @a.exists?
+      @a.create
+      expect(@content_db).to be_exists
+      expect(@modules_db).to be_exists
+      expect(@app_server).to be_exists
+
+      @a.drop
+      expect(@content_db).to_not be_exists
+      expect(@modules_db).to_not be_exists
+      expect(@app_server).to_not be_exists
+    end
+  end
+
+  describe "stale?" do
+    it "should be stale when appropriate" do
+      @a.drop if @a.exists?
+
+      expect(@a).to be_stale
+
+      @a.create
+      expect(@a).to_not be_stale
+
+      @a.content_databases[0].add_range_element_index("stuff")
+      expect(@a).to be_stale
+
+      @a.drop
+      expect(@a).to be_stale
+    end
+
+    it "should be stale when indexes are added via the app" do
+      @a.drop if @a.exists?
+      expect(@a).to be_stale
+      @a.create
+      expect(@a).to_not be_stale
+
+      @a.add_index(MarkLogic::DatabaseSettings::RangeElementIndex.new(:_id, :type => 'string'))
+      @a.add_index(MarkLogic::DatabaseSettings::RangeElementIndex.new(:name, :type => 'string'))
+      @a.add_index(MarkLogic::DatabaseSettings::RangeElementIndex.new(:age, :type => 'int'))
+
+      expect(@a).to be_stale
+
+      @a.drop
+      expect(@a).to be_stale
+    end
+  end
+end

data/spec/marklogic/collection_spec.rb

@@ -0,0 +1,154 @@
+require 'spec_helper'
+
+describe MarkLogic::Collection do
+  before do
+    @collection = @database.collection("stuff")
+    @collection.drop
+  end
+
+  def save_ten
+    @collection.save((1..10).map do |n|
+      {:_id => n, :name => "John#{n}", :age => n, :stuff => "junk"}
+    end)
+  end
+
+  describe "count" do
+    it "returns the correct document count" do
+      expect(@collection.count).to eq 0
+      @collection.save( {:_id => "123", :name => "John", :age => 33, :stuff => "junk"} )
+      expect(@collection.count).to eq 1
+      @collection.save( {:_id => "456", :name => "Bill", :age => 33, :stuff => "junk"} )
+      expect(@collection.count).to eq 2
+      @collection.save( {:_id => "456", :name => "Bill", :age => 33, :stuff => "junk"} )
+      expect(@collection.count).to eq 2
+    end
+  end
+
+  describe "load" do
+    it "loads a document" do
+      expect(@collection.count).to eq 0
+      @collection.save( {:_id => "123", :name => "John", :age => 33, :stuff => "junk"} )
+      expect(@collection.count).to eq 1
+      doc = @collection.load("123")
+      expect(doc).to be_a_kind_of(Hash)
+      expect(doc["_id"]).to eq "123"
+      expect(doc['name']).to eq "John"
+    end
+  end
+
+  describe "save" do
+    it "saves a document" do
+      expect(@collection.count).to eq 0
+      @collection.save( {:_id => "123", :name => "John", :age => 33, :stuff => "junk"} )
+      expect(@collection.count).to eq 1
+    end
+  end
+
+  describe "#save multiples" do
+    it "saves a bunch of documents in one transaction" do
+      expect(@collection.count).to eq 0
+      @collection.save((1..10).map do |n|
+        {:_id => n.to_s, :name => "John#{n}", :age => 30 + n, :stuff => "junk#{n}"}
+      end)
+      expect(@collection.count).to eq 10
+    end
+  end
+
+  describe "remove" do
+    it "removes all document when no params given" do
+      expect(@collection.count).to eq 0
+      save_ten
+      expect(@collection.count).to eq 10
+      @collection.remove
+      expect(@collection.count).to eq 0
+    end
+
+    it "removes all document when an empty hash is given" do
+      expect(@collection.count).to eq 0
+      save_ten
+      expect(@collection.count).to eq 10
+      @collection.remove()
+      expect(@collection.count).to eq 0
+    end
+
+    it "removes documents matching the criteria given" do
+      expect(@collection.count).to eq 0
+      save_ten
+      expect(@collection.count).to eq 10
+      @collection.remove({:_id => 1})
+      expect(@collection.count).to eq 9
+
+      @collection.remove(@collection.from_criteria({:_id => 2}))
+      expect(@collection.count).to eq 8
+    end
+
+    it "removes documents matching the criteria given" do
+      expect(@collection.count).to eq 0
+      save_ten
+      expect(@collection.count).to eq 10
+      @collection.remove({:_id => [1, 2, 3]})
+      expect(@collection.count).to eq 7
+
+      @collection.remove(@collection.from_criteria({:_id => [4, 5, 6]}))
+      expect(@collection.count).to eq 4
+    end
+
+    it "removes documents matching the criteria given" do
+      expect(@collection.count).to eq 0
+      save_ten
+      expect(@collection.count).to eq 10
+      @collection.remove({ 'age' => { '$lt' => 7} })
+      expect(@collection.count).to eq 4
+
+      @collection.remove(@collection.from_criteria({ 'age' => { '$lt' => 8} }))
+      expect(@collection.count).to eq 3
+    end
+  end
+
+  describe "drop" do
+    it "removes all documents" do
+      expect(@collection.count).to eq 0
+      save_ten
+      expect(@collection.count).to eq 10
+      @collection.drop
+      expect(@collection.count).to eq 0
+    end
+  end
+
+  describe "find" do
+    it "finds documents matching the criteria given" do
+      expect(@collection.count).to eq 0
+      save_ten
+      expect(@collection.count).to eq 10
+      expect(@collection.find({:_id => 1}).count).to eq 1
+      expect(@collection.find(@collection.from_criteria({:_id => 1})).count).to eq 1
+    end
+
+    it "finds documents matching the criteria given" do
+      expect(@collection.count).to eq 0
+      save_ten
+      expect(@collection.count).to eq 10
+      expect(@collection.find({:_id => [1, 2, 3]}).count).to eq 3
+      expect(@collection.find(@collection.from_criteria({:_id => [1, 2, 3]})).count).to eq 3
+    end
+
+    it "finds documents matching the criteria given" do
+      expect(@collection.count).to eq 0
+      save_ten
+      expect(@collection.count).to eq 10
+      expect(@collection.find({ 'age' => { '$lt' => 7 } }).count).to eq 6
+      expect(@collection.find(@collection.from_criteria({ 'age' => { '$lt' => 7 } })).count).to eq 6
+    end
+  end
+
+  describe "#find_one" do
+    it "finds one document" do
+      expect(@collection.count).to eq 0
+      save_ten
+      expect(@collection.count).to eq 10
+      c = @collection.find_one({:_id => 1})
+      c = @collection.find_one(@collection.from_criteria({:_id => 1}))
+      expect(c['_id']).to eq(1)
+    end
+  end
+end