tree_graph 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 457d1b759ef73e437da805f9fea94aef374bf8b0
-  data.tar.gz: e38b29a657054c54d9026bff943b93c2b0fb079a
+  metadata.gz: ae297d2b5265ef94e73403efb35c9fd1a7348737
+  data.tar.gz: 4fa3803abbc5a6d6509313dc5f2e3ff631c1db89
 SHA512:
-  metadata.gz: 5595b438ce21764d26bb2ea65471c1beb87a6f277403959231535d307d4c92a93d6e465c5458b68273b3e92f04da6990f60f0304ce8bd73d6ad5ba0992aa897f
-  data.tar.gz: 600015270d0f10ba8016daf6d22f48d250c1fbe4e0498379b28c719dbaee350894d08309f98908bdae1ed5f409df31902d2c07402d4f744c570994a8939b80bf
+  metadata.gz: d89095d952e52da07dbae183425cb73ae3223597a9a67800e24d97a60ea7aec3b68af1233eb0d7aaa2bc1785f0bedcd48c76966c027e019c471ca59b974b2d43
+  data.tar.gz: f868060e3bc62449d4e3bff34631c0a20df41de532678f4e2a93a5830822bb4fb856c032586682b65a2362b92197e3c94be8dd23a28a9d1dccc652d0d8ade22d

data/README.md

@@ -20,22 +20,9 @@ Or install it yourself as:
 
 ## Usage
 
-In Node class, `include TreeGraph`, then implement `parent_for_tree_graph`, `label_for_tree_graph`, `is_last_for_tree_graph`, you get `tree_graph_level`. To have `tree_graph`, also implement `children_for_tree_graph`.
+In Node class, `include TreeGraph`, then implement two methods, `label_for_tree_graph` and `children_for_tree_graph`(which return thing responds to `each`). Then you can call `tree_graph` on that Node object.
 
-Gems depending on this: [constree](https://github.com/turnon/constree), [trace_tree](https://github.com/turnon/trace_tree).
+Or checkout [test/tree_graph_test.rb](https://github.com/turnon/tree_graph/blob/master/test/tree_graph_test.rb) to see how to use.
 
-## Development
-
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
-
-To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
-
-## Contributing
-
-Bug reports and pull requests are welcome on GitHub at https://github.com/ken/tree_graph.
-
-
-## License
-
-The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).
+Gems depending on tree_graph: [constree](https://github.com/turnon/constree), [trace_tree](https://github.com/turnon/trace_tree).
 

data/lib/tree_graph/version.rb

@@ -1,3 +1,3 @@
 module TreeGraph
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

data/lib/tree_graph.rb

@@ -3,34 +3,57 @@ require "tree_graph/version"
 module TreeGraph
 
   def tree_graph
-    ([tree_graph_level] +
-     children_for_tree_graph.map(&:tree_graph)
-    ).join("\n")
+    Node.new(self).tree_graph
   end
 
-  def tree_graph_level
-    tree_graph_indent +
-      tree_graph_branch +
-      label_for_tree_graph
-  end
+  class Node
 
-  def tree_graph_ancestors
-    return [] unless parent_for_tree_graph
-    parent_for_tree_graph.tree_graph_ancestors + [parent_for_tree_graph]
-  end
+    attr_accessor :is_last
+    attr_reader :raw_node, :parent
 
-  def tree_graph_branch
-    return '' unless parent_for_tree_graph
-    is_last_for_tree_graph ? '└─' : '├─'
-  end
+    def initialize raw_node, parent=nil
+      @raw_node, @parent, @is_last = raw_node, parent, false
+    end
 
-  def tree_graph_indent
-    tree_graph_ancestors.map do |a|
-      unless a.parent_for_tree_graph
-        ''
-      else
-        a.is_last_for_tree_graph ? '  ' : '│ '
+    def tree_graph
+      ([level] +
+       children_nodes.map(&:tree_graph)
+      ).join("\n")
+    end
+
+    def children_nodes
+      children = []
+      raw_node.children_for_tree_graph.each do |c|
+        children << Node.new(c, self)
       end
-    end.join
+      return children if children.empty?
+      children.last.is_last = true
+      children
+    end
+
+    def level
+      [indent, branch, raw_node.label_for_tree_graph].join
+    end
+
+    def ancestors
+      return [] unless parent
+      parent.ancestors + [parent]
+    end
+
+    def branch
+      return '' unless parent
+      is_last ? '└─' : '├─'
+    end
+
+    def indent
+      ancestors.map do |a|
+        unless a.parent
+          ''
+        else
+          a.is_last ? '  ' : '│ '
+        end
+      end.join
+    end
+
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tree_graph
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - ken
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-02-04 00:00:00.000000000 Z
+date: 2017-02-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -90,7 +90,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.5.2
+rubygems_version: 2.6.8
 signing_key: 
 specification_version: 4
 summary: A mixin to help you generate tree graph