neo4apis 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 30ba847cad3af0957ed3e603dd236b5037c80f20
-  data.tar.gz: eb86312649a77abe55ae3c668b178709cb5ac803
+  metadata.gz: 435f18b35183016cf989780b44ca0e551f53c4b2
+  data.tar.gz: 127f94b29ba58b55709691b6c3823c384daf0dbd
 SHA512:
-  metadata.gz: 1ac9fcb8bbe6b8f1f934df3f81ad6dbeb267fcc502bff457fe3b7a6698963d0de9dff9f51874f1f0125585c1ee62af9975384b3ad3231832ae882d4c5fbfdf7c
-  data.tar.gz: bd4597cad7cf2adf0a6c4dab7e2235e6ce3b7c86b45b30f1e07fa564179fbc0930eda0c6134b67ed30553c64bebce29fd9edc3c9b2d8eee78d1fb8a7b5aa6758
+  metadata.gz: ea2f6f6f40d3a224fe39c57c6ca25b6a248b51aac3f3e71c2e759bc1952f97d65ae79228430e73aaabd78a6d8c818984ad83fcde4ed6ab097e3b3d9e0395586c
+  data.tar.gz: 240362413f2b4bfb21efb5ee9b73e3a0423dd1be6b78d5a5a7632f167af382f1e09af167c9462ada6f451ee389410cd053ae0851fbfa0ab4c0eb664ac993f8fa

data/README.md

@@ -8,18 +8,8 @@ In the below example we assume that a variable `awesome_client` is passed in whi
 
   module Neo4Apis
     class AwesomeSite < Base
-      PREFIX = 'awesome_site'
-
-      def initialize(neo4j_client, options = {})
-        @client = options[:awesome_client]
-
-        options[:uuids] = (options[:uuids] || {}).merge({
-          User: :id,
-          Widget: :uuid
-        })
-
-        super(neo4j_client, options)
-      end
+      uuid :User, :id
+      uuid :Widget, :uuid
 
       def import_widget_search(*args)
         @client.widget_search(*args).each do |widget|
@@ -27,9 +17,7 @@ In the below example we assume that a variable `awesome_client` is passed in whi
         end
       end
 
-      private
-      
-      def add_widget(widget)
+      importer :Widget do |widget|
         user_node = add_user(widget.owner)
 
         # add_node comes from From Neo4Apis::Base
@@ -44,7 +32,7 @@ In the below example we assume that a variable `awesome_client` is passed in whi
         node
       end
 
-      def add_user(user)
+      importer :User do |user|
         add_node :User, {
           id: user.id,
           username: user.username,
@@ -63,10 +51,12 @@ Then somebody else could use your gem in the following manner:
 neo4j_session = Neo4j::Session.open # From the neo4j-core gem
 awesome_client = Awesome.open # From a theoretical API wrapping gem
 
-neo4apis_awesome = Neo4Apis::AwesomeSite.new(neo4j_session, awesome_client: awesome_client)
+neo4apis_awesome = Neo4Apis::AwesomeSite.new(neo4j_session)
 
 neo4apis_awesome.batch do
-  neo4apis_awesome.import_widget_search('cool') # Does a search for 'cool' via the Awesome gem and imports to the neo4j database
+  awesome_client.widget_search('cool').each do |widget|
+    import :Widget, widget # import is provided by neo4apis
+  end
 end
 
 ```

data/lib/neo4apis/base.rb

@@ -3,25 +3,14 @@ module Neo4Apis
     DEFAULT_FLUSH_SIZE = 500
 
     NODE_PROXIES = {}
+    IMPORTERS = {}
+
+    attr_reader :options
 
     def initialize(neo4j_session, options = {})
       @buffer = QueryBuffer.new(neo4j_session)
+      @options = options
       @flush_size = DEFAULT_FLUSH_SIZE
-      (options[:uuids] || {}).each do |label, uuid_field|
-        NODE_PROXIES[label] = Struct.new(:label, :props) do
-          const_set(:UUID_FIELD, uuid_field)
-
-          def uuid_value
-            raise ArgumentError, "props does not have UUID field `#{uuid_field}` for #{self.inspect}" if not props.has_key?(uuid_field)
-
-            props[uuid_field]
-          end
-
-          def uuid_field
-            self.class::UUID_FIELD
-          end
-        end
-      end
     end
 
     def add_node(label, props = {})
@@ -44,16 +33,44 @@ module Neo4Apis
       @buffer << create_relationship_query(type, source, target, props)
     end
 
-    def batch
+    def batch(&block)
       @in_batch = true
 
-      yield
+      instance_eval &block
 
       @buffer.flush
     ensure
       @in_batch = false
     end
 
+    def import(label, object)
+      self.instance_exec object, &IMPORTERS[label.to_sym]
+    end
+
+    def self.importer(label, &block)
+      IMPORTERS[label.to_sym] = block
+    end
+
+    def self.uuid(label, uuid_field)
+      NODE_PROXIES[label.to_sym] = node_proxy_from_uuid(label.to_sym, uuid_field.to_sym)
+    end
+
+    def self.node_proxy_from_uuid(label, uuid_field)
+      Struct.new(:label, :props) do
+        const_set(:UUID_FIELD, uuid_field)
+
+        def uuid_value
+          raise ArgumentError, "props does not have UUID field `#{uuid_field}` for #{self.inspect}" if not props.has_key?(uuid_field)
+
+          props[uuid_field]
+        end
+
+        def uuid_field
+          self.class::UUID_FIELD
+        end
+      end
+    end
+
     private
 
     def create_node_query(node_proxy)

data/lib/neo4apis/version.rb

@@ -1,3 +1,3 @@
 module Neo4Apis
-  VERSION = '0.0.1'
+  VERSION = '0.0.2'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: neo4apis
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Brian Underwood
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-10-29 00:00:00.000000000 Z
+date: 2014-10-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday