gremlin2dot 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0a1c658eedf3c5ff39c687cbe21b3fd7e407a7285b67e720a46ef010b9796683
-  data.tar.gz: f88dfd7d98dbd4a0ae5c866c20709bc5098191a1369379b56943e3e279e8358a
+  metadata.gz: 9f3caf040c14213db68f758fb1120d2ba2fb617d4e2189c0bafee75a0a27e749
+  data.tar.gz: e080468b7856ad111121cab2b20816d73b2d7ee6771a33fcf871d648e3856acf
 SHA512:
-  metadata.gz: 11d45d23294c8dbfaa5e590b06d38f920a9bf623c5d03080c907dea4b6341ca6e8c154e0d3bd6759c63b3448aa3b38011d7a9fb5359ff620771d308fae0caff9
-  data.tar.gz: d8cbc265a3cb5ad24de2f6abfc4c73a995d9a3c92d09ba9547feed8fae4cdca60f1f3a89e796de7b2ecd088334b4980b833c058bc0bbb6702aa00a9bbae21c36
+  metadata.gz: c8c3d4342e60d5e322501d58c26ec2c341b9112982de6644ea554228b3f90b3d267afd6e053f6bc514695cb6a8fdd7feaf6129abfcf328593f7364b4d2aa14a3
+  data.tar.gz: da9d52a2977719bda217b2b3aabdc1640c1eb7e4b45a2705ae1146ab0d73e75aaf87c7ef9a45f97473e19d746c6ac876a07cc4460602d12eb14bcfb52d009c92

data/lib/gremlin2dot/graph_builder.rb

@@ -14,41 +14,46 @@ module Gremlin2Dot
     end
 
     def build(unmangled_data)
-      trees = unmangled_data['result']['data']
-
-      trees.each do |t0|
-        t0.each do |t1|
-          add_thing t1
-        end
-      end
-
+      add_thing unmangled_data['result']['data']
       @g
     end
 
+    private
+
     def add_thing(thing)
+      # $stderr.puts "add_thing #{thing.class} #{thing}"
       case thing
       when Vertex
-        node = thing
-        unless @seen.include? node.id
-          label = node.label + node.properties.entries.sort_by(&:first).map {|k, v| "\n#{k}=#{v.first}" }.join("")
-          @g.add_nodes(node.id, label: label)
-          @seen << node.id
-        end
+        once_only(thing.id) { add_node thing }
       when Edge
-        edge = thing
-        unless @seen.include? edge.id
-          label = edge.label + edge.properties.entries.sort_by(&:first).map {|k, v| "\n#{k}=#{v}" }.join("")
-          @g.add_edges(edge.outV, edge.inV, label: label)
-          @seen << edge.id
-        end
+        once_only(thing.id) { add_edge thing }
       when Tree
         add_thing thing.key
         thing.items.each {|child| add_thing child}
+      when Array
+        thing.each {|child| add_thing child}
       else
         $stderr.puts "Not adding #{thing.class} to the graph"
       end
     end
 
+    def once_only(id)
+      unless @seen.include? id
+        yield
+        @seen << id
+      end
+    end
+
+    def add_node(node)
+      label = node.id + "\n" + node.label + node.properties.entries.sort_by(&:first).map {|k, v| "\n#{k}=#{v.first}" }.join("")
+      @g.add_nodes(node.id, label: label)
+    end
+
+    def add_edge(edge)
+      label = edge.id + "\n" + edge.label + edge.properties.entries.sort_by(&:first).map {|k, v| "\n#{k}=#{v}" }.join("")
+      @g.add_edges(edge.outV, edge.inV, label: label)
+    end
+
   end
 
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gremlin2dot
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Rachel Evans
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-06-28 00:00:00.000000000 Z
+date: 2018-07-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rest-client