tty-tree 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: b7ed240a5a6239047b55b961fa170aaa28df6109
-  data.tar.gz: b5363c4033cf2218c69d4a0174d1348d2dca5ab7
+SHA256:
+  metadata.gz: 53e2981f489110b90e65e7205454e15b1bea811f29063fc5f165c68a2a4a6b5f
+  data.tar.gz: ca58a44574af6db9502106a7f6f292125f0b4f156401b3f69e6d4001b602831f
 SHA512:
-  metadata.gz: 0c873142481cc06042c4d6f1e3de114f91b372ddd52bca933a84f1a2ba3abedcaa86b0d35d75e4ed5cffe6a13ae5040576dc29e3d04a806a81730e82b6e372dc
-  data.tar.gz: 18c3d908b3a7c8e0e0205ccf908e9594237ccaece938c567ed5dd7bf86e0f300993cc8e139b5953d405b488cad2c36340c77f55dcd6245033d01d2b4e0c8a9c2
+  metadata.gz: bc7fa766eb270b8324b5c9241d7fa0869c4e63273ea57809a8d66c982b13e4db501173c88a1201003e71b7db6f629577a971e0bc97abe3feae9a1e9e6c7d6b63
+  data.tar.gz: 4508c2b82e085d0e3deca12fb62d74c9c20d4453367286add7e0e3b9a3b6e1f4aab27ae99fec2194957d6e0dcc2d5160172f7cbfa9d985099d9ba7589f9aedc9

data/CHANGELOG.md

@@ -1,7 +1,13 @@
 # Change log
 
+## [v0.2.0] - 2018-09-01
+
+### Changed
+* Change to Ruby >= 2.0
+
 ## [v0.1.0] - 2017-05-01
 
 * Initial implementation and release
 
+[v0.2.0]: https://github.com/piotrmurach/tty-tree/compare/v0.1.0...v0.2.0
 [v0.1.0]: https://github.com/piotrmurach/tty-tree/compare/v0.1.0

data/README.md

@@ -1,3 +1,7 @@
+<div align="center">
+  <a href="https://piotrmurach.github.io/tty" target="_blank"><img width="130" src="https://cdn.rawgit.com/piotrmurach/tty/master/images/tty.png" alt="tty logo" /></a>
+</div>
+
 # TTY::Tree [![Gitter](https://badges.gitter.im/Join%20Chat.svg)][gitter]
 
 [![Gem Version](https://badge.fury.io/rb/tty-tree.svg)][gem]
@@ -237,4 +241,4 @@ Bug reports and pull requests are welcome on GitHub at https://github.com/piotrm
 
 ## Copyright
 
-Copyright (c) 2017 Piotr Murach. See LICENSE for further details.
+Copyright (c) 2017-2018 Piotr Murach. See LICENSE for further details.

data/lib/tty-tree.rb

@@ -1,3 +1 @@
-# encoding: utf-8
-
 require_relative 'tty/tree'

data/lib/tty/tree.rb

@@ -1,4 +1,3 @@
-# encoding: utf-8
 # frozen_string_literal: true
 
 require_relative 'tree/node'

data/lib/tty/tree/hash_walker.rb

@@ -1,4 +1,3 @@
-# encoding: utf-8
 # frozen_string_literal: true
 
 require 'pathname'

data/lib/tty/tree/node.rb

@@ -1,4 +1,3 @@
-# encoding: utf-8
 # frozen_string_literal: true
 
 require 'forwardable'

data/lib/tty/tree/number_renderer.rb

@@ -1,4 +1,3 @@
-# encoding: utf-8
 # frozen_string_literal: true
 
 module TTY

data/lib/tty/tree/path_walker.rb

@@ -1,4 +1,3 @@
-# encoding: utf-8
 # frozen_string_literal: true
 
 require 'pathname'

data/lib/tty/tree/version.rb

@@ -1,7 +1,7 @@
-# encoding: utf-8
+# frozen_string_literal: true
 
 module TTY
   class Tree
-    VERSION = "0.1.0"
+    VERSION = "0.2.0"
   end # Tree
 end # TTY

data/spec/spec_helper.rb

@@ -0,0 +1,71 @@
+# encoding: utf-8
+
+if RUBY_VERSION > '1.9' and (ENV['COVERAGE'] || ENV['TRAVIS'])
+  require 'simplecov'
+  require 'coveralls'
+
+  SimpleCov.formatter = SimpleCov::Formatter::MultiFormatter[
+    SimpleCov::Formatter::HTMLFormatter,
+    Coveralls::SimpleCov::Formatter
+  ]
+
+  SimpleCov.start do
+    command_name 'spec'
+    add_filter 'spec'
+  end
+end
+
+require 'bundler/setup'
+require 'tty/tree'
+require 'fileutils'
+
+module TestHelpers
+  module Paths
+    def gem_root
+      File.expand_path("#{File.dirname(__FILE__)}/..")
+    end
+
+    def dir_path(*args)
+      path = File.join(gem_root, *args)
+      FileUtils.mkdir_p(path)
+      File.realpath(path)
+    end
+
+    def tmp_path(*args)
+      File.join(dir_path('tmp'), *args)
+    end
+
+    def fixtures_path(*args)
+      File.join(dir_path('spec/fixtures'), *args)
+    end
+
+    def within_dir(target, &block)
+      ::Dir.chdir(target, &block)
+    end
+  end
+end
+
+RSpec.configure do |config|
+  config.include(TestHelpers::Paths)
+  # config.example_status_persistence_file_path = ".rspec_status"
+
+  config.expect_with :rspec do |c|
+    c.syntax = :expect
+  end
+
+  config.mock_with :rspec do |mocks|
+    mocks.verify_partial_doubles = true
+  end
+
+  config.disable_monkey_patching!
+
+  config.warnings = true
+
+  if config.files_to_run.one?
+    config.default_formatter = 'doc'
+  end
+
+  config.order = :random
+
+  Kernel.srand config.seed
+end

data/spec/unit/directory_renderer_spec.rb

@@ -0,0 +1,173 @@
+# encoding: utf-8
+# frozen_string_literal: true
+
+RSpec.describe TTY::Tree::DirectoryRenderer do
+  it "renders directory as path" do
+    tree = within_dir(fixtures_path) do
+      TTY::Tree.new('dir1')
+    end
+
+    expect(tree.render).to eq([
+      "dir1\n",
+      "├── config.dat\n",
+      "├── dir2\n",
+      "│   ├── dir3\n",
+      "│   │   └── file3-1.txt\n",
+      "│   └── file2-1.txt\n",
+      "├── file1-1.txt\n",
+      "└── file1-2.txt\n",
+    ].join)
+  end
+
+  it "renders hash data as path " do
+    data = {
+      dir1: [
+        'config.dat',
+        { dir2: [
+          { dir3: [ 'file3-1.txt' ] },
+          'file2-1.txt'
+          ]
+        },
+        'file1-1.txt',
+        'file1-2.txt'
+      ]
+    }
+
+    tree = TTY::Tree.new(data)
+
+    expect(tree.render).to eq([
+      "dir1\n",
+      "├── config.dat\n",
+      "├── dir2\n",
+      "│   ├── dir3\n",
+      "│   │   └── file3-1.txt\n",
+      "│   └── file2-1.txt\n",
+      "├── file1-1.txt\n",
+      "└── file1-2.txt\n",
+    ].join)
+  end
+
+  it "correctly renders orphaned directory as path" do
+    tree = within_dir(fixtures_path) do
+      TTY::Tree.new('orphan_dir')
+    end
+
+    expect(tree.render).to eq([
+      "orphan_dir\n",
+      "├── a.txt\n",
+      "├── b.txt\n",
+      "└── data\n",
+      "    ├── data1.bin\n",
+      "    ├── data2.sql\n",
+      "    └── data3.inf\n",
+    ].join)
+  end
+
+  it "correctly renders hash with orphaned directory as path" do
+    data = {
+      orphan_dir: [
+        'a.txt',
+        'b.txt',
+        { data: [
+            'data1.bin',
+            'data2.sql',
+            'data3.inf'
+          ]
+        }
+      ]
+    }
+
+    tree = TTY::Tree.new(data)
+
+    expect(tree.render).to eq([
+      "orphan_dir\n",
+      "├── a.txt\n",
+      "├── b.txt\n",
+      "└── data\n",
+      "    ├── data1.bin\n",
+      "    ├── data2.sql\n",
+      "    └── data3.inf\n",
+    ].join)
+  end
+
+  it "renders directory as path without indentation" do
+    tree = within_dir(fixtures_path) do
+      TTY::Tree.new('dir1')
+    end
+
+    expect(tree.render(indent: 0)).to eq([
+      "dir1\n",
+      "├── config.dat\n",
+      "├── dir2\n",
+      "│├── dir3\n",
+      "││└── file3-1.txt\n",
+      "│└── file2-1.txt\n",
+      "├── file1-1.txt\n",
+      "└── file1-2.txt\n",
+    ].join)
+  end
+
+  it "renders directory without hidden files" do
+    tree = within_dir(fixtures_path) do
+      TTY::Tree.new('hidden_dir', show_hidden: true)
+    end
+
+    expect(tree.render).to eq([
+      "hidden_dir\n",
+      "├── .with_dot\n",
+      "│   └── config.dat\n",
+      "├── dir1\n",
+      "│   ├── .file1.2\n",
+      "│   ├── dir1.1\n",
+      "│   │   └── .file1.1.1\n",
+      "│   ├── dir1.2\n",
+      "│   │   └── file1.2.1\n",
+      "│   └── file1.1\n",
+      "└── dir2\n",
+      "    └── .file2.1\n",
+    ].join)
+  end
+
+  it "renders directory with only dirs" do
+    tree = within_dir(fixtures_path) do
+      TTY::Tree.new('hidden_dir', only_dirs: true)
+    end
+
+    expect(tree.render).to eq([
+      "hidden_dir\n",
+      "├── dir1\n",
+      "│   ├── dir1.1\n",
+      "│   └── dir1.2\n",
+      "└── dir2\n",
+    ].join)
+  end
+
+  it "renders directory with only dirs including hidden ones" do
+    tree = within_dir(fixtures_path) do
+      TTY::Tree.new('hidden_dir', only_dirs: true, show_hidden: true)
+    end
+
+    expect(tree.render).to eq([
+      "hidden_dir\n",
+      "├── .with_dot\n",
+      "├── dir1\n",
+      "│   ├── dir1.1\n",
+      "│   └── dir1.2\n",
+      "└── dir2\n",
+    ].join)
+  end
+
+  it "doesn't show dirs with files exceeding limit" do
+    tree = within_dir(fixtures_path) do
+      TTY::Tree.new('large_dir', file_limit: 4)
+    end
+
+    expect(tree.render).to eq([
+      "large_dir\n",
+      "├── huge_dir\n",
+      "├── large1\n",
+      "├── large2\n",
+      "└── large3\n",
+    ].join)
+  end
+end

data/spec/unit/hash_walker_spec.rb

@@ -0,0 +1,122 @@
+# encoding: utf-8
+# frozen_string_literal: true
+
+RSpec.describe TTY::Tree::HashWalker do
+  it "walks hash data and collects nodes" do
+    data = {
+      dir1: [
+        'config.dat',
+        { dir2: [
+          { dir3: [ 'file3-1.txt' ] },
+          'file2-1.txt'
+          ]
+        },
+        'file1-1.txt',
+        'file1-2.txt'
+      ]
+    }
+
+    walker = TTY::Tree::HashWalker.new
+
+    walker.traverse(data)
+
+    expect(walker.nodes).to eq([
+      TTY::Tree::Node.new('dir1', '', '', 0),
+      TTY::Tree::Node.new('config.dat', 'dir1', '', 1),
+      TTY::Tree::Node.new('dir2', 'dir1', '', 1),
+      TTY::Tree::Node.new('dir3', 'dir1/dir2', ':pipe', 2),
+      TTY::Tree::LeafNode.new('file3-1.txt', 'dir1/dir2/dir3', ':pipe:pipe', 3),
+      TTY::Tree::LeafNode.new('file2-1.txt', 'dir1/dir2', ':pipe', 2),
+      TTY::Tree::Node.new('file1-1.txt', 'dir1', '', 1),
+      TTY::Tree::LeafNode.new('file1-2.txt', 'dir1', '', 1),
+    ])
+
+    expect(walker.nodes.map(&:full_path).map(&:to_s)).to eq([
+      "dir1",
+      "dir1/config.dat",
+      "dir1/dir2",
+      "dir1/dir2/dir3",
+      "dir1/dir2/dir3/file3-1.txt",
+      "dir1/dir2/file2-1.txt",
+      "dir1/file1-1.txt",
+      "dir1/file1-2.txt",
+    ])
+  end
+
+  it "walks path tree and collects nodes up to max level" do
+    data = {
+      dir1: [
+        'config.dat',
+        { dir2: [
+          { dir3: [ 'file3-1.txt' ] },
+          'file2-1.txt'
+          ]
+        },
+        'file1-1.txt',
+        'file1-2.txt'
+      ]
+    }
+
+    walker = TTY::Tree::HashWalker.new(level: 2)
+
+    walker.traverse(data)
+
+    expect(walker.nodes.map(&:full_path).map(&:to_s)).to eq([
+      "dir1",
+      "dir1/config.dat",
+      "dir1/dir2",
+      "dir1/dir2/file2-1.txt",
+      "dir1/file1-1.txt",
+      "dir1/file1-2.txt",
+    ])
+  end
+
+  it "doesn't walks path tree exceeding file limit" do
+    data = {
+      orphan_dir: [
+        'a.txt',
+        'b.txt',
+        { data: [
+            'data1.bin',
+            'data2.sql',
+            'data3.inf',
+            'data4.csv'
+          ]
+        }
+      ]
+    }
+
+    walker = TTY::Tree::HashWalker.new(file_limit: 3)
+
+    walker.traverse(data)
+
+    expect(walker.nodes.map(&:full_path).map(&:to_s)).to eq([
+      "orphan_dir",
+      "orphan_dir/a.txt",
+      "orphan_dir/b.txt",
+      "orphan_dir/data"
+    ])
+  end
+
+  it "counts files & dirs" do
+    data = {
+      dir1: [
+        'config.dat',
+        { dir2: [
+          { dir3: [ 'file3-1.txt' ] },
+          'file2-1.txt'
+          ]
+        },
+        'file1-1.txt',
+        'file1-2.txt'
+      ]
+    }
+    walker = TTY::Tree::HashWalker.new
+
+    walker.traverse(data)
+
+    expect(walker.files_count).to eq(5)
+
+    expect(walker.dirs_count).to eq(2)
+  end
+end

data/spec/unit/new_spec.rb

@@ -0,0 +1,85 @@
+# encoding: utf-8
+# frozen_string_literal: true
+
+RSpec.describe TTY::Tree, '#new' do
+  it "exposes [] as instance creation & accepts Dir" do
+    tree = TTY::Tree[::File.join(Dir.pwd, 'lib')]
+
+    expect(tree.nodes).to_not be_empty
+  end
+
+  it "exposes [] as instance creation & accepts Pathname" do
+    tree = TTY::Tree[Pathname.pwd.join('lib')]
+
+    expect(tree.nodes).to_not be_empty
+  end
+
+  it "provides DSL for creating tree" do
+    tree = TTY::Tree.new do
+      node 'dir1' do
+        node 'config.dat'
+        node 'dir2' do
+          node 'dir3' do
+            leaf 'file3-1.txt'
+          end
+          leaf 'file2-1.txt'
+        end
+        node 'file1-1.txt'
+        leaf 'file1-2.txt'
+      end
+    end
+
+    expect(tree.nodes).to eq([
+      TTY::Tree::Node.new('dir1', '', '', 0),
+      TTY::Tree::Node.new('config.dat', 'dir1', '', 1),
+      TTY::Tree::Node.new('dir2', 'dir1', '', 1),
+      TTY::Tree::Node.new('dir3', 'dir1/dir2', ':pipe', 2),
+      TTY::Tree::LeafNode.new('file3-1.txt', 'dir1/dir2/dir3', ':pipe:pipe', 3),
+      TTY::Tree::LeafNode.new('file2-1.txt', 'dir1/dir2', ':pipe', 2),
+      TTY::Tree::Node.new('file1-1.txt', 'dir1', '', 1),
+      TTY::Tree::LeafNode.new('file1-2.txt', 'dir1', '', 1),
+    ])
+  end
+
+  it "provides DSL for creating orphaned tree" do
+    tree = TTY::Tree.new do
+      node 'orphan_dir' do
+        node 'a.txt'
+        node 'b.txt'
+        leaf 'data' do
+          node 'data1.bin'
+          node 'data2.sql'
+          leaf 'data3.inf'
+        end
+      end
+    end
+
+    expect(tree.nodes).to eq([
+      TTY::Tree::Node.new("orphan_dir", '', '', 0),
+      TTY::Tree::Node.new("a.txt", 'orphan_dir', '', 1),
+      TTY::Tree::Node.new("b.txt", 'orphan_dir', '', 1),
+      TTY::Tree::LeafNode.new("data", 'orphan_dir', '',  1),
+      TTY::Tree::Node.new("data1.bin", 'orphan_dir/data', ':space', 2),
+      TTY::Tree::Node.new("data2.sql", 'orphan_dir/data', ':space', 2),
+      TTY::Tree::LeafNode.new("data3.inf", 'orphan_dir/data', ':space', 2)
+    ])
+  end
+
+  it "yields current node to DSL tree node" do
+    yielded = []
+
+    TTY::Tree.new do
+      node 'dir1' do |dir1|
+        yielded << dir1
+        node 'dir2' do |dir2|
+          yielded << dir2
+        end
+      end
+    end
+
+    expect(yielded).to eq([
+      TTY::Tree::Node.new("dir1", '', '', 0),
+      TTY::Tree::Node.new("dir2", 'dir1', '', 1)
+    ])
+  end
+end

data/spec/unit/node_spec.rb

@@ -0,0 +1,36 @@
+# encoding: utf-8
+# frozen_string_literal: true
+
+RSpec.describe TTY::Tree::Node do
+  it "provides directory stats" do
+    dir = fixtures_path('dir1')
+    node = TTY::Tree::Node.new(dir, '', '',  0)
+
+    expect(node.directory?).to eq(true)
+    expect(node.file?).to eq(false)
+    expect(node.symlink?).to eq(false)
+    expect(node.socket?).to eq(false)
+    expect(node.pipe?).to eq(false)
+  end
+
+  it "informs if node is hidden" do
+    file = fixtures_path('hidden', 'dir1', '.file1.2')
+    node = TTY::Tree::Node.new(file, '', '', 0)
+
+    expect(node.hidden?).to eq(true)
+  end
+
+  it "compares two nodes for equality" do
+    node_a = TTY::Tree::Node.new('dir1', '', '', 0)
+    node_b = TTY::Tree::Node.new('dir1', '', '', 0)
+
+    expect(node_a).to eq(node_b)
+  end
+
+  it "compares two nodes for inequlity" do
+    node_a = TTY::Tree::Node.new('dir1', '', '', 0)
+    node_b = TTY::Tree::Node.new('dir2', '', '', 0)
+
+    expect(node_a).to_not eq(node_b)
+  end
+end

data/spec/unit/number_renderer_spec.rb

@@ -0,0 +1,65 @@
+# encoding: utf-8
+
+RSpec.describe TTY::Tree::NumberRenderer do
+  it "renders directory as numbers" do
+    tree = within_dir(fixtures_path) do
+      TTY::Tree.new('dir1')
+    end
+
+    expect(tree.render(as: :number)).to eq([
+      "dir1\n",
+      "1.1 config.dat\n",
+      "1.2 dir2\n",
+      "    2.3 dir3\n",
+      "        3.4 file3-1.txt\n",
+      "    2.5 file2-1.txt\n",
+      "1.6 file1-1.txt\n",
+      "1.7 file1-2.txt\n",
+    ].join)
+  end
+
+  it "renders hash data as numbers" do
+    data = {
+      dir1: [
+        'config.dat',
+        { dir2: [
+          { dir3: [ 'file3-1.txt' ] },
+          'file2-1.txt'
+          ]
+        },
+        'file1-1.txt',
+        'file1-2.txt'
+      ]
+    }
+
+    tree = TTY::Tree.new(data)
+
+    expect(tree.render(as: :number)).to eq([
+      "dir1\n",
+      "1.1 config.dat\n",
+      "1.2 dir2\n",
+      "    2.3 dir3\n",
+      "        3.4 file3-1.txt\n",
+      "    2.5 file2-1.txt\n",
+      "1.6 file1-1.txt\n",
+      "1.7 file1-2.txt\n",
+    ].join)
+  end
+
+  it "renders directory as numbers without indentation" do
+    tree = within_dir(fixtures_path) do
+      TTY::Tree.new('dir1')
+    end
+
+    expect(tree.render(as: :number, indent: 0)).to eq([
+      "dir1\n",
+      "1.1 config.dat\n",
+      "1.2 dir2\n",
+      "2.3 dir3\n",
+      "3.4 file3-1.txt\n",
+      "2.5 file2-1.txt\n",
+      "1.6 file1-1.txt\n",
+      "1.7 file1-2.txt\n",
+    ].join)
+  end
+end

data/spec/unit/path_walker_spec.rb

@@ -0,0 +1,127 @@
+# encoding: utf-8
+# frozen_string_literal: true
+
+RSpec.describe TTY::Tree::PathWalker do
+  it "walks path tree and collects nodes" do
+    walker = TTY::Tree::PathWalker.new
+
+    within_dir(fixtures_path) do
+      walker.traverse('dir1')
+    end
+
+    expect(walker.nodes).to eq([
+      TTY::Tree::Node.new('dir1', '', '', 0),
+      TTY::Tree::Node.new('dir1/config.dat', 'dir1', '', 1),
+      TTY::Tree::Node.new('dir2', 'dir1', '', 1),
+      TTY::Tree::Node.new('dir3', 'dir1/dir2', ':pipe', 2),
+      TTY::Tree::LeafNode.new('dir1/dir2/dir3/file3-1.txt', 'dir1/dir2/dir3', ':pipe:pipe', 3),
+      TTY::Tree::LeafNode.new('dir1/dir2/file2-1.txt', 'dir1/dir2', ':pipe', 2),
+      TTY::Tree::Node.new('dir1/file1-1.txt', 'dir1', '', 1),
+      TTY::Tree::LeafNode.new('dir1/file1-2.txt', 'dir1', '', 1),
+    ])
+
+    expect(walker.nodes.map(&:full_path).map(&:to_s)).to eq([
+      "dir1",
+      "dir1/config.dat",
+      "dir1/dir2",
+      "dir1/dir2/dir3",
+      "dir1/dir2/dir3/file3-1.txt",
+      "dir1/dir2/file2-1.txt",
+      "dir1/file1-1.txt",
+      "dir1/file1-2.txt",
+    ])
+  end
+
+  it "walks orphaned path tree and collects nodes" do
+    walker = TTY::Tree::PathWalker.new
+
+    within_dir(fixtures_path) do
+      walker.traverse('orphan_dir')
+    end
+
+    expect(walker.nodes).to eq([
+      TTY::Tree::Node.new("orphan_dir", '', '', 0),
+      TTY::Tree::Node.new("orphan_dir/a.txt", 'orphan_dir', '', 1),
+      TTY::Tree::Node.new("orphan_dir/b.txt", 'orphan_dir', '', 1),
+      TTY::Tree::LeafNode.new("data", 'orphan_dir', '',  1),
+      TTY::Tree::Node.new("orphan_dir/data/data1.bin", 'orphan_dir/data', ':space', 2),
+      TTY::Tree::Node.new("orphan_dir/data/data2.sql", 'orphan_dir/data', ':space', 2),
+      TTY::Tree::LeafNode.new("orphan_dir/data/data3.inf", 'orphan_dir/data', ':space', 2)
+    ])
+
+    expect(walker.nodes.map(&:full_path).map(&:to_s)).to eq([
+      "orphan_dir",
+      "orphan_dir/a.txt",
+      "orphan_dir/b.txt",
+      "orphan_dir/data",
+      "orphan_dir/data/data1.bin",
+      "orphan_dir/data/data2.sql",
+      "orphan_dir/data/data3.inf"
+    ])
+  end
+
+  it "walks path tree and collects nodes limited by the level" do
+    walker = TTY::Tree::PathWalker.new(level: 2)
+
+    within_dir(fixtures_path) do
+      walker.traverse('dir1')
+    end
+
+    expect(walker.nodes.map(&:full_path).map(&:to_s)).to eq([
+      "dir1",
+      "dir1/config.dat",
+      "dir1/dir2",
+      "dir1/dir2/file2-1.txt",
+      "dir1/file1-1.txt",
+      "dir1/file1-2.txt",
+    ])
+  end
+
+  it "doesn't walk paths exceeding file limit" do
+    walker = TTY::Tree::PathWalker.new(file_limit: 4)
+
+    within_dir(fixtures_path) do
+      walker.traverse('large_dir')
+    end
+
+    expect(walker.nodes.map(&:full_path).map(&:to_s)).to eq([
+      "large_dir",
+      "large_dir/huge_dir",
+      "large_dir/large1",
+      "large_dir/large2",
+      "large_dir/large3"
+    ])
+  end
+
+  it "counts files & dirs" do
+    walker = TTY::Tree::PathWalker.new
+
+    within_dir(fixtures_path) do
+      walker.traverse('dir1')
+    end
+
+    expect(walker.files_count).to eq(5)
+
+    expect(walker.dirs_count).to eq(2)
+  end
+
+  it "counts files & dirs including max level limit" do
+    walker = TTY::Tree::PathWalker.new(level: 2)
+
+    within_dir(fixtures_path) do
+      walker.traverse('dir1')
+    end
+
+    expect(walker.files_count).to eq(4)
+
+    expect(walker.dirs_count).to eq(1)
+  end
+
+  it "raises when walking non-directory" do
+    walker = TTY::Tree::PathWalker.new
+
+    expect {
+      walker.traverse('unknown-dir')
+    }.to raise_error(ArgumentError, /unknown-dir is not a directory path/)
+  end
+end

data/tty-tree.gemspec

@@ -1,4 +1,3 @@
-# coding: utf-8
 lib = File.expand_path('../lib', __FILE__)
 $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 require 'tty/tree/version'
@@ -14,13 +13,14 @@ Gem::Specification.new do |spec|
   spec.homepage      = "https://piotrmurach.github.io/tty"
   spec.license       = "MIT"
 
-  spec.files         = `git ls-files -z`.split("\x0").reject do |f|
-    f.match(%r{^(test|spec|features)/})
-  end
+  spec.files         = Dir['{lib,spec}/**/*.rb', '{bin,tasks}/*', 'tty-tree.gemspec']
+  spec.files        += Dir['README.md', 'CHANGELOG.md', 'LICENSE.txt', 'Rakefile']
   spec.bindir        = "exe"
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
 
+  spec.required_ruby_version = '>= 2.0.0'
+
   spec.add_development_dependency "bundler", "~> 1.14"
   spec.add_development_dependency "rake", "~> 10.0"
   spec.add_development_dependency "rspec", "~> 3.0"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tty-tree
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Piotr Murach
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-05-01 00:00:00.000000000 Z
+date: 2018-09-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -59,16 +59,10 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".gitignore"
-- ".rspec"
-- ".travis.yml"
 - CHANGELOG.md
-- CODE_OF_CONDUCT.md
-- Gemfile
 - LICENSE.txt
 - README.md
 - Rakefile
-- appveyor.yml
 - bin/console
 - bin/setup
 - lib/tty-tree.rb
@@ -79,6 +73,13 @@ files:
 - lib/tty/tree/number_renderer.rb
 - lib/tty/tree/path_walker.rb
 - lib/tty/tree/version.rb
+- spec/spec_helper.rb
+- spec/unit/directory_renderer_spec.rb
+- spec/unit/hash_walker_spec.rb
+- spec/unit/new_spec.rb
+- spec/unit/node_spec.rb
+- spec/unit/number_renderer_spec.rb
+- spec/unit/path_walker_spec.rb
 - tasks/console.rake
 - tasks/coverage.rake
 - tasks/spec.rake
@@ -95,7 +96,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 2.0.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
@@ -103,9 +104,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.5.1
+rubygems_version: 2.7.3
 signing_key: 
 specification_version: 4
 summary: Print directory or structured data in a tree like format.
 test_files: []
-has_rdoc: 

data/.gitignore

@@ -1,12 +0,0 @@
-/.bundle/
-/.yardoc
-/Gemfile.lock
-/_yardoc/
-/coverage/
-/doc/
-/pkg/
-/spec/reports/
-/tmp/
-
-# rspec failure tracking
-.rspec_status

data/.rspec

@@ -1,3 +0,0 @@
---color
---require spec_helper
---warnings

data/.travis.yml

@@ -1,24 +0,0 @@
----
-language: ruby
-sudo: false
-before_install: gem install bundler -v 1.14.6
-bundler_args: --without tools
-script: "bundle exec rake ci"
-rvm:
-  - 2.0.0
-  - 2.1.10
-  - 2.2.6
-  - 2.3.3
-  - 2.4.0
-  - ruby-head
-  - jruby-9000
-  - jruby-head
-matrix:
-  allow_failures:
-    - rvm: ruby-head
-    - rvm: jruby-head
-  fast_finish: true
-branches:
-  only: master
-notifications:
-  email: false

data/CODE_OF_CONDUCT.md

@@ -1,74 +0,0 @@
-# Contributor Covenant Code of Conduct
-
-## Our Pledge
-
-In the interest of fostering an open and welcoming environment, we as
-contributors and maintainers pledge to making participation in our project and
-our community a harassment-free experience for everyone, regardless of age, body
-size, disability, ethnicity, gender identity and expression, level of experience,
-nationality, personal appearance, race, religion, or sexual identity and
-orientation.
-
-## Our Standards
-
-Examples of behavior that contributes to creating a positive environment
-include:
-
-* Using welcoming and inclusive language
-* Being respectful of differing viewpoints and experiences
-* Gracefully accepting constructive criticism
-* Focusing on what is best for the community
-* Showing empathy towards other community members
-
-Examples of unacceptable behavior by participants include:
-
-* The use of sexualized language or imagery and unwelcome sexual attention or
-advances
-* Trolling, insulting/derogatory comments, and personal or political attacks
-* Public or private harassment
-* Publishing others' private information, such as a physical or electronic
-  address, without explicit permission
-* Other conduct which could reasonably be considered inappropriate in a
-  professional setting
-
-## Our Responsibilities
-
-Project maintainers are responsible for clarifying the standards of acceptable
-behavior and are expected to take appropriate and fair corrective action in
-response to any instances of unacceptable behavior.
-
-Project maintainers have the right and responsibility to remove, edit, or
-reject comments, commits, code, wiki edits, issues, and other contributions
-that are not aligned to this Code of Conduct, or to ban temporarily or
-permanently any contributor for other behaviors that they deem inappropriate,
-threatening, offensive, or harmful.
-
-## Scope
-
-This Code of Conduct applies both within project spaces and in public spaces
-when an individual is representing the project or its community. Examples of
-representing a project or community include using an official project e-mail
-address, posting via an official social media account, or acting as an appointed
-representative at an online or offline event. Representation of a project may be
-further defined and clarified by project maintainers.
-
-## Enforcement
-
-Instances of abusive, harassing, or otherwise unacceptable behavior may be
-reported by contacting the project team at [email]. All
-complaints will be reviewed and investigated and will result in a response that
-is deemed necessary and appropriate to the circumstances. The project team is
-obligated to maintain confidentiality with regard to the reporter of an incident.
-Further details of specific enforcement policies may be posted separately.
-
-Project maintainers who do not follow or enforce the Code of Conduct in good
-faith may face temporary or permanent repercussions as determined by other
-members of the project's leadership.
-
-## Attribution
-
-This Code of Conduct is adapted from the [Contributor Covenant][homepage], version 1.4,
-available at [http://contributor-covenant.org/version/1/4][version]
-
-[homepage]: http://contributor-covenant.org
-[version]: http://contributor-covenant.org/version/1/4/

data/Gemfile

@@ -1,19 +0,0 @@
-source 'https://rubygems.org'
-
-gemspec
-
-group :test do
-  gem 'benchmark-ips', '~> 2.0.0'
-  gem 'simplecov', '~> 0.10.0'
-  gem 'coveralls', '~> 0.8.2'
-  gem 'term-ansicolor', '=1.3.2'
-end
-
-group :tools do
-  gem 'byebug', platform: :mri
-end
-
-group :metrics do
-  gem 'yard',      '~> 0.8.7'
-  gem 'yardstick', '~> 0.9.9'
-end

data/appveyor.yml

@@ -1,23 +0,0 @@
----
-install:
-  - SET PATH=C:\Ruby%ruby_version%\bin;%PATH%
-  - ruby --version
-  - gem --version
-  - bundle install
-build: off
-test_script:
-  - bundle exec rake ci
-environment:
-  matrix:
-    - ruby_version: "193"
-    - ruby_version: "200"
-    - ruby_version: "200-x64"
-    - ruby_version: "21"
-    - ruby_version: "21-x64"
-    - ruby_version: "22"
-    - ruby_version: "22-x64"
-    - ruby_version: "23"
-    - ruby_version: "23-x64"
-matrix:
-  allow_failures:
-    - ruby_version: "193"