neography-down 1.6.4

data/lib/neography/rest/batch.rb

@@ -0,0 +1,414 @@
+module Neography
+  class Rest
+    module Batch
+      include Neography::Rest::Helpers
+    
+      def batch(*args)
+        do_batch(*args)
+      end
+
+      def batch_no_streaming(*args)
+        do_batch_no_streaming(*args)
+      end
+
+      private
+
+      def do_batch(*args)
+        @connection.post("/batch", compute_batch_options(*args))
+      end
+
+      def do_batch_no_streaming(*args)
+        options = compute_batch_options(*args)
+        options[:headers].merge!({ 'X-Stream' => false })
+        @connection.post("/batch", options)
+      end
+
+      def compute_batch_options(*args)
+        batch = []
+        Array(args).each_with_index do |c, i|
+          batch << {:id => i }.merge(get_batch(c))
+        end
+        {:body => batch.to_json, :headers => json_content_type}
+      end
+
+      def get_batch(args)
+        begin
+          send("batch_#{args[0]}".to_sym, *args[1..-1])
+        rescue
+          raise UnknownBatchOptionException.new("Unknown option #{args[0]} - #{args}")
+        end
+      end
+
+      # Nodes
+
+      def batch_get_node(id)
+        get "/node/%{id}" % {:id => get_id(id)}
+      end
+
+      def batch_delete_node(id)
+        delete "/node/%{id}" % {:id => get_id(id)}
+      end
+
+      def batch_create_node(body)
+        post "/node" do
+          body
+        end
+      end
+
+      # NodeIndexes
+
+      def batch_create_node_index(name, type = "exact", provider = "lucene", extra_config = nil)
+        config = {
+          :type => type,
+          :provider => provider
+        }
+        config.merge!(extra_config) unless extra_config.nil?
+        post "/index/node" do
+          { :name => name,
+            :config => config
+          }
+        end
+      end
+
+      def batch_drop_node_index(index)
+        delete "/index/node/%{index}?unique" % {:index => index}
+      end
+
+      def batch_create_unique_node(index, key, value, properties)
+        post "/index/node/%{index}?unique" % {:index => index} do
+          {
+            :key        => key,
+            :value      => value,
+            :properties => properties
+          }
+        end
+      end
+      
+      def batch_create_or_fail_unique_node(index, key, value, properties = {})
+        post "/index/node/%{index}?uniqueness=%{function}" %  {:index => index, :function => 'create_or_fail'} do
+          {
+            :key        => key,
+            :value      => value,
+            :properties => properties
+          }
+        end
+      end
+
+      def batch_add_node_to_index(index, key, value, id, unique = false)
+        path = unique ? "/index/node/%{index}?unique" % {:index => index} : "/index/node/%{index}" % {:index => index}
+        post path do
+          {
+            :uri   => build_node_uri(id),
+            :key   => key,
+            :value => value
+          }
+        end
+      end
+
+      def batch_get_node_index(index, key, value)
+        get "/index/node/%{index}/%{key}/%{value}" % {:index => index, :key => key, :value => encode(value)}
+      end
+
+      def batch_remove_node_from_index(index, key_or_id, value_or_id = nil, id = nil)
+        delete remove_from_index_path("node", index, key_or_id, value_or_id, id)
+      end
+
+      # NodeProperties
+
+      def batch_set_node_property(id, property)
+        put "/node/%{id}/properties/%{property}" % {:id => get_id(id), :property => property.keys.first} do
+          property.values.first
+        end
+      end
+
+      def batch_reset_node_properties(id, body)
+        put "/node/%{id}/properties" % {:id => get_id(id)} do
+          body
+        end
+      end
+
+      def batch_remove_node_property(id, property)
+        delete "/node/%{id}/properties/%{property}" % {:id => get_id(id), :property => property}
+      end
+
+      # NodeLabel
+
+      def batch_add_label(id, body)
+        post build_node_uri(id) + "/labels" do
+          body
+        end
+      end
+
+      # NodeRelationships
+
+      def batch_get_node_relationships(id, direction = nil, types = nil)
+        if types.nil?
+          get "/node/%{id}/relationships/%{direction}" % {:id => get_id(id), :direction => direction || 'all'}
+        else
+          get "/node/%{id}/relationships/%{direction}/%{types}" % {:id => get_id(id), :direction => direction, :types => Array(types).join('&')}
+        end
+      end
+
+      # Relationships
+
+      def batch_get_relationship(id)
+        get "/relationship/%{id}" % {:id => get_id(id)}
+      end
+
+      def batch_delete_relationship(id)
+        delete "/relationship/%{id}" % {:id => get_id(id)}
+      end
+
+      def batch_create_relationship(type, from, to, data = nil)
+        post build_node_uri(from) + "/relationships" do
+          {
+            :to   => build_node_uri(to),
+            :type => type,
+            :data => data
+          }
+        end
+      end
+
+      # RelationshipIndexes
+
+      def batch_create_unique_relationship(index, key, value, type, from, to, props = nil)
+        post "/index/relationship/%{index}?unique" % {:index => index} do
+          {
+            :key   => key,
+            :value => value,
+            :type  => type,
+            :start => build_node_uri(from),
+            :end   => build_node_uri(to),
+            :properties => props
+          }
+        end
+      end
+
+      def batch_add_relationship_to_index(index, key, value, id)
+        post "/index/relationship/%{index}" % {:index => index} do
+          {
+            :uri   => build_relationship_uri(id),
+            :key   => key,
+            :value => value
+          }
+        end
+      end
+
+      def batch_get_relationship_index(index, key, value)
+        get "/index/relationship/%{index}/%{key}/%{value}" % {:index => index, :key => key, :value => encode(value)}
+      end
+
+      def batch_remove_relationship_from_index(index, key_or_id, value_or_id = nil, id = nil)
+        delete remove_from_index_path("relationship", index, key_or_id, value_or_id, id)
+      end
+
+      # RelationshipProperties
+
+      def batch_set_relationship_property(id, property)
+        put "/relationship/%{id}/properties/%{property}" % {:id => get_id(id), :property => property.keys.first} do
+          property.values.first
+        end
+      end
+
+      def batch_reset_relationship_properties(id, body)
+        put build_relationship_uri(id) + "/properties" do
+          body
+        end
+      end
+
+      # Cypher
+
+      def batch_execute_query(query, params = nil)
+        request = post @connection.cypher_path do
+          {
+            :query => query
+          }
+        end
+
+        request[:body].merge!({ :params => params }) if params
+
+        request
+      end
+
+      # Gremlin
+
+      def batch_execute_script(script, params = nil)
+        post @connection.gremlin_path do
+          {
+            :script => script,
+            :params => params
+          }
+        end
+      end
+
+      # Similar between nodes and relationships
+
+      def remove_from_index_path(klass, index, key_or_id, value_or_id = nil, id = nil)
+        if id
+          "/index/#{klass}/%{index}/%{key}/%{value}/%{id}" % {:index => index, :key => key_or_id, :value => value_or_id, :id => get_id(id)}
+        elsif value_or_id
+          "/index/#{klass}/%{index}/%{key}/%{id}" % {:index => index, :key => key_or_id, :id => get_id(value_or_id)}
+        else
+          "/index/#{klass}/%{index}/%{id}" % {:index => index, :id => get_id(key_or_id)}
+        end
+      end
+
+      # Spatial
+      
+      def batch_get_spatial
+        get "/ext/SpatialPlugin"
+      end
+    
+      def batch_add_point_layer(layer, lat = nil, lon = nil)
+         post "/ext/SpatialPlugin/graphdb/addSimplePointLayer" do
+          {
+              :layer => layer,
+              :lat => lat || "lat",
+              :lon => lon || "lon"
+            }
+          end
+      end
+
+      def batch_add_editable_layer(layer, format = "WKT", node_property_name = "wkt")
+        post "/ext/SpatialPlugin/graphdb/addEditableLayer" do 
+          {
+              :layer => layer,
+              :format => format,
+              :nodePropertyName => node_property_name
+            }
+        end
+      end
+
+      def batch_get_layer(layer)
+        post "/ext/SpatialPlugin/graphdb/getLayer" do
+          {
+              :layer => layer
+            }
+        end
+      end
+
+      def batch_add_geometry_to_layer(layer, geometry)
+        post "/ext/SpatialPlugin/graphdb/addGeometryWKTToLayer" do 
+          {
+              :layer => layer,
+              :geometry => geometry
+            }
+        end
+      end
+    
+      def batch_edit_geometry_from_layer(layer, geometry, node)
+        post "/ext/SpatialPlugin/graphdb/updateGeometryFromWKT" do 
+          {
+              :layer => layer,
+              :geometry => geometry,
+              :geometryNodeId => get_id(node)
+            }
+        end
+      end
+    
+      def batch_add_node_to_layer(layer, node)
+        post "/ext/SpatialPlugin/graphdb/addNodeToLayer" do 
+          {
+              :layer => layer,
+              :node => get_id(node)
+            }
+        end
+      end
+    
+      def batch_find_geometries_in_bbox(layer, minx, maxx, miny, maxy)
+        post "/ext/SpatialPlugin/graphdb/findGeometriesInBBox" do 
+          {
+              :layer => layer,
+              :minx => minx,
+              :maxx => maxx,
+              :miny => miny,
+              :maxy => maxy
+            }
+        end
+      end
+    
+      def batch_find_geometries_within_distance(layer, pointx, pointy, distance)
+        post "/ext/SpatialPlugin/graphdb/findGeometriesWithinDistance" do 
+          {
+              :layer => layer,
+              :pointX => pointx,
+              :pointY => pointy,
+              :distanceInKm => distance
+            }
+        end
+      end
+      
+      def batch_create_spatial_index(name, type, lat, lon)
+        post "/index/node" do
+          {
+            :name => name,
+            :config => {
+              :provider => "spatial",
+              :geometry_type => type || "point",
+              :lat => lat || "lat",
+              :lon => lon || "lon"
+              }
+          }
+        end
+      end
+
+      def batch_add_node_to_spatial_index(index, id)
+        post "/index/node/%{index}" % {:index => index} do
+          {
+            :uri   => build_node_uri(id),
+            :key   => "k",
+            :value => "v"
+          }
+        end
+      end
+
+      def get(to, &block)
+        request "GET", to, &block
+      end
+
+      def delete(to, &block)
+        request "DELETE", to, &block
+      end
+
+      def post(to, &block)
+        request "POST", to, &block
+      end
+
+      def put(to, &block)
+        request "PUT", to, &block
+      end
+
+      def request(method, to, &block)
+        request = {
+          :method => method,
+          :to     => to
+        }
+        request.merge!({ :body => yield }) if block_given?
+        request
+      end
+
+      # Helper methods
+
+      def build_node_uri(value)
+        build_uri(value, "node")
+      end
+
+      def build_relationship_uri(value)
+        build_uri(value, "relationship")
+      end
+
+      def build_uri(value, type)
+        path_or_variable(value, type) + "#{get_id(value)}"
+      end
+
+      def path_or_variable(value, type)
+        if value.is_a?(String) && value.start_with?("{")
+          ""
+        else
+          "/#{type}/"
+        end
+      end
+
+    end
+  end
+end

data/lib/neography/rest/clean.rb

@@ -0,0 +1,17 @@
+module Neography
+  class Rest
+    module Clean
+      include Neography::Rest::Helpers
+
+      def clean_database(sanity_check = "not_really")
+        if sanity_check == "yes_i_really_want_to_clean_the_database"
+          @connection.delete("/cleandb/secret-key")
+          true
+        else
+          false
+        end          
+      end
+
+    end
+  end
+end

data/lib/neography/rest/constraints.rb

@@ -0,0 +1,38 @@
+module Neography
+  class Rest
+    module Constraints
+      include Neography::Rest::Helpers
+    
+      def drop_constraint(label, property)
+        @connection.delete("/schema/constraint/%{label}/uniqueness/%{property}" % {:label => label, :property => property})
+      end
+
+      def get_constraints(label=nil)
+        #if label.nil?
+        #  @connection.get(base_path)          
+        #else
+          @connection.get("/schema/constraint/%{label}" % {:label => label})
+        #end
+      end
+
+      def get_uniqueness(label)
+        @connection.get("/schema/constraint/%{label}/uniqueness/" % {:label => label})
+      end
+
+      def get_unique_constraint(label, property)
+        @connection.get("/schema/constraint/%{label}/uniqueness/%{property}" % {:label => label, :property => property})
+      end
+
+      def create_unique_constraint(label, property)
+        options = {
+          :body => {
+            :property_keys => [property]
+          }.to_json,
+          :headers => json_content_type
+        }
+        @connection.post("/schema/constraint/%{label}/uniqueness/" % {:label => label}, options)
+      end
+
+    end
+  end
+end