ascii_tree 1.0.1 → 1.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 98ccffa0747e6642b106241a0aa1cf9fe9d9a406
-  data.tar.gz: 72d58407cf20d8e0b73392a7a7171ede26429036
+  metadata.gz: 7e031ba9a1b346d19bccb1dada35c1221397fc70
+  data.tar.gz: e45c2c268c7bbfc2e7bc00ac646dc93d3ae2dc05
 SHA512:
-  metadata.gz: 245281f5e9393ca8358799e6b41abda704bbe8af808be7e4787a561da8ca0673467c45a21d5b9dfc83de2964198c3b6acd2271750976a00a8b04ad013073f75a
-  data.tar.gz: 91508434cd16cd1b9baa05ce183ade2f1e698f564e3a5233dcec784e279847ed64fcefebb6ffe4bed819caae3889cfcc59005321e0625479a9c9eed2f55c8abe
+  metadata.gz: 128c1e4d53c452c820c0b966a7d02fb1362b36e8a7a0d34ca1dd9294fbd4318d7921ae566ddd3d2c340967c6214c3b210bec7fb5073517df5c2cb044fb9a51ae
+  data.tar.gz: ee4ba71fe8e4bbe8866bef686fb87ccd9dda2550e7220f7dde4c52bf678d14f39b909206e6aa27fe1bce51a92d75c161568317cf4330d19f9536866737030d33

data/lib/ascii_tree/node_builder.rb

@@ -10,7 +10,11 @@ module AsciiTree
     end
 
     def build
-      build_for(root_word, nil)
+      if relationships.any?
+        build_for(root_word, nil)
+      else
+        nil
+      end
     end
 
     private
@@ -18,7 +22,11 @@ module AsciiTree
     attr_reader :relationships
 
     def root_word
-      relationships.first.parent_word
+      root_relationship.child_word
+    end
+
+    def root_relationship
+      relationships.detect { |r| r.parent_word.nil? }
     end
 
     def build_for(word, parent)

data/lib/ascii_tree/relationships_builder.rb

@@ -3,7 +3,12 @@ module AsciiTree
     class << self
 
       def build(words, edges)
-        relationships = edges.map do |edge|
+        relationships = []
+
+        root_relationship = root_relationship(words)
+        relationships << root_relationship if root_relationship
+
+        relationships += edges.map do |edge|
           parent = words.detect { |w| w.include?(edge.parent_coordinate) }
           child  = words.detect { |w| w.include?(edge.child_coordinate) }
 
@@ -18,6 +23,14 @@ module AsciiTree
 
       private
 
+      def root_relationship(words)
+        root_word = words.first
+
+        if root_word
+          Relationship.new(parent_word: nil, edge: nil, child_word: root_word)
+        end
+      end
+
       def validate_presence(parent, child, edge)
         if parent.nil? && child.nil?
           error = "No parent or child"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ascii_tree
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.0.2
 platform: ruby
 authors:
 - Chris Patuzzo
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-01-31 00:00:00.000000000 Z
+date: 2015-02-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec