du_pretty 0.3.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 61dc0af33ffecb01858522e4fa90309c5e04c6942bef5d8dbcbca1ec24f0aad3
-  data.tar.gz: 966e1e39727763f52a18a8a1fe418cfc845f1e3e37556adceb960c31d50e239e
+  metadata.gz: 39482a8862ce548bf8074ecf2564fda7aff0f48321a7ec6e106d26bf7c6953ff
+  data.tar.gz: 6213856a3b177f24bb6273b37e9ce9e7b70ce2610c96b80236f6779ed06779e8
 SHA512:
-  metadata.gz: f399efd4c2e46e23b22766a9d0c1cb589d6e7851064fafa969f9a68b6a36d90ff63c35f547efd10e8aef3a6d10023623358776db7ea2c3189efb85bbe14c0494
-  data.tar.gz: 65d2be03a70889c77f149f097995768875f20bf70665c36f4b1eee012cdad24d4280da6255e97ebcab68c228716c4b3395da8e57cb911e15e403f45e15dc0a0c
+  metadata.gz: f53e2724839bf1a5ed9e5e5e9e42d91db1d263fd31d5793ca3e221f8941e08cffd65f13f5e494add51dc967cb3b920a9870891fef245f698b45d16cf206a0d17
+  data.tar.gz: '08dd8ceae26022158f9b0febe7b5b30891ccdf5f3ebeb089ffb987ca894870c3a2db6ce99782f96f6a482d8ecc06033088d842573ca494706557a384b4af9d35'

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    du_pretty (0.3.0)
+    du_pretty (0.3.1)
       colorize
       thor
 

data/README.md

@@ -1,38 +1,44 @@
 # DuPretty
+Show pretty print of `du` results.
 
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/du_pretty`. To experiment with that code, run `bin/console` for an interactive prompt.
-
-TODO: Delete this and the text above, and describe your gem
+[![Image from Gyazo](https://i.gyazo.com/8a7911cf58bb4ac1aa586cc24c8a307e.png)](https://gyazo.com/8a7911cf58bb4ac1aa586cc24c8a307e)
 
 ## Installation
-
-Add this line to your application's Gemfile:
-
-```ruby
-gem 'du_pretty'
+```sh
+$ gem install 'du_pretty'
 ```
 
-And then execute:
+## Usage
+```sh
+$ du_pretty <path>
 
-    $ bundle
+# specify max depth
+$ du_pretty -d 2 <path>
 
-Or install it yourself as:
+# show directories if its size is 100KB+
+$ du_pretty --max-kbyte 100 <path>
 
-    $ gem install du_pretty
+# show directories sorted by its size
+$ du_pretty --sort <path>
 
-## Usage
+# help
+$ du_pretty --help
+```
 
-TODO: Write usage instructions here
 
 ## Development
+```sh
+# run
+$ bundle install
+$ bundle exec exe/du_pretty
 
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` 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).
+# test
+$ bundle exec rspec spec
+```
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/du_pretty. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.
+Bug reports and pull requests are welcome on GitHub at https://github.com/nullnull/du_pretty. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.
 
 ## License
 

data/lib/du_pretty/cli.rb

@@ -1,33 +1,48 @@
 require 'du_pretty'
+require 'du_pretty/utils'
 require 'thor'
 
 module DuPretty
   class CLI < Thor
     default_task :path
-    desc 'path', ''
-    option :min_gbyte, type: :numeric
-    option :min_mbyte, type: :numeric
-    option :min_kbyte, type: :numeric
-    option :depth, type: :numeric, aliases: :d
-    option :sort, aliases: :s, type: :boolean
+    desc 'path', <<~EOT
+Details:
+  -a, --all
+      Display an entry for each file in a file hierarchy.
+
+  -d depth
+      Display an entry for all files and directories depth directories deep.
+
+  --size n[kMGT]
+      Filter results of `du` by its filesize. if n is followed by a scale indicator then the file's size is compared to n scaled as:
+      k(default)   kilobytes (1024 bytes)
+      M            megabytes (1024 kilobytes)
+      G            gigabytes (1024 megabytes)
+      T            terabytes (1024 gigabytes)
+
+  --sort
+      Show results sorted by the file's size.
+    EOT
     option :all, aliases: :a, type: :boolean
+    option :depth, type: :numeric, aliases: :d
+    option :size, aliases: :s
+    option :sort, type: :boolean
     option :tree, type: :boolean, default: true
 
     def path(path = '.')
-      min_kbyte = [(options[:min_gbyte] || 0) * 1024 * 1024, (options[:min_mbyte] || 0) * 1024, (options[:min_kbyte] || 0)].max
-      du_wrapper = DuWrapper.new(
+      prettier = DuPretty::Prettier.new(
         path,
-        min_kbyte: min_kbyte,
+        min_kbyte: DuPretty::Utils.size_to_kbyte(options[:size]),
         depth: options[:depth],
-        with_files: options[:all],
+        with_files: options[:all]
       )
       result = if options[:sort]
-        du_wrapper.sorted
-      elsif options[:tree] == false
-        du_wrapper.original
-      else
-        du_wrapper.tree
-      end
+                 prettier.sorted
+               elsif options[:tree] == false
+                 prettier.original
+               else
+                 prettier.tree
+               end
       print result + "\n"
     end
   end

data/lib/du_pretty/prettier.rb

@@ -0,0 +1,116 @@
+require 'colorize'
+require 'pathname'
+
+module DuPretty
+  class Prettier
+    def initialize(path, min_kbyte: 0, depth: nil, with_files: false)
+      @path = File.expand_path(path, Pathname.pwd)
+      @min_kbyte = min_kbyte
+      @depth = depth
+      @with_files = with_files
+    end
+
+    def original
+      filtered_disk_usages.map(&:pretty).join("\n")
+    end
+
+    def sorted
+      filtered_disk_usages.sort_by(&:kbyte).map(&:pretty).join("\n")
+    end
+
+    def tree
+      filtered_disk_usages.reverse.map(&:tree_format).join("\n")
+    end
+
+    private
+
+    def du
+      options = [
+        @depth.nil? ? nil : "-d #{@depth}",
+        @with_files ? '-a' : nil,
+        '-k'
+      ]
+      `du #{options.join(' ')} #{@path}`
+    end
+
+    def disk_usages
+      xs = du.split("\n").map { |line| DiskUsage.new(line, @path) }
+      total = xs.map(&:kbyte).max
+      xs.map { |x| DiskUsage.new(x.raw, @path, total: total) }
+    end
+
+    def filtered_disk_usages
+      disk_usages.select { |x| x.kbyte >= @min_kbyte }
+    end
+
+    class DiskUsage
+      attr_accessor :raw, :kbyte, :path
+
+      def initialize(raw, root, total: nil)
+        @raw = raw
+        @root = root
+        @kbyte = raw.split("\t").compact[0].to_i
+        @path = raw.split("\t").compact[1]
+        @total = total
+      end
+
+      def pretty
+        pretty_byte + "\t." + relative_path
+      end
+
+      def tree_format
+        pretty_path + ' ' + pretty_byte
+      end
+
+      private
+
+      def basename
+        Pathname.new(@path).basename.to_s
+      end
+
+      def depth
+        Pathname.new(relative_path).each_filename.to_a.size - 1
+      end
+
+      def pretty_byte
+        mb = @kbyte / 1024
+        gb = mb / 1024
+        if gb.positive?
+          if gb > 10
+            "#{gb}GB (#{percentage})".red
+          else
+            "#{gb}GB (#{percentage})".light_red
+          end
+        elsif mb.positive?
+          if mb > 500
+            "#{mb}MB (#{percentage})".yellow
+          else
+            "#{mb}MB (#{percentage})".light_yellow
+          end
+        else
+          if @kbyte > 500
+            "#{@kbyte}KB (#{percentage})".green
+          else
+            "#{@kbyte}KB (#{percentage})".light_green
+          end
+        end
+      end
+
+      def pretty_path
+        return '.' if @path == @root
+        spaces = depth.positive? ? '  ' * depth : ''
+        tree_symbol = '└── '
+        spaces + tree_symbol + basename
+      end
+
+      def percentage
+        percentage = @kbyte * 1.0 / @total * 100
+        "#{percentage.ceil}%"
+      end
+
+      def relative_path
+        @path.sub(/^#{@root}/, '')
+      end
+    end
+  end
+end

data/lib/du_pretty/utils.rb

@@ -0,0 +1,17 @@
+module DuPretty
+  module Utils
+    def self.size_to_kbyte(size)
+      return 0 if size.nil?
+      size = size.upcase
+      if size.end_with?('T')
+        size.to_i * (1024**3)
+      elsif size.end_with?('G')
+        size.to_i * (1024**2)
+      elsif size.end_with?('M')
+        size.to_i * 1024
+      else
+        size.to_i
+      end
+    end
+  end
+end

data/lib/du_pretty/version.rb

@@ -1,3 +1,3 @@
 module DuPretty
-  VERSION = "0.3.0"
+  VERSION = '0.3.1'.freeze
 end

data/lib/du_pretty.rb

@@ -1,119 +1,6 @@
-require 'colorize'
-require 'pathname'
-
 require 'du_pretty/version'
 require 'du_pretty/cli'
+require 'du_pretty/prettier'
 
 module DuPretty
-  class DuWrapper
-    def initialize(path, min_kbyte: 0, depth: nil, with_files: false)
-      @path = File.expand_path(path, Pathname.pwd)
-      @min_kbyte = min_kbyte
-      @depth = depth
-      @with_files = with_files
-    end
-
-    def original
-      filtered_disk_usages.map(&:pretty).join("\n")
-    end
-
-    def sorted
-      filtered_disk_usages.sort_by(&:kbyte).map(&:pretty).join("\n")
-    end
-
-    def tree
-      filtered_disk_usages.reverse.map(&:tree_format).join("\n")
-    end
-
-    private
-
-    def du
-      options = [
-        @depth.nil? ? nil : "-d #{@depth}",
-        @with_files ? '-a' : nil,
-        '-k'
-      ]
-      `du #{options.join(' ')} #{@path}`
-    end
-
-    def disk_usages
-      xs = du.split("\n").map { |line| DiskUsage.new(line, @path) }
-      total = xs.map(&:kbyte).max
-      xs.map { |x| DiskUsage.new(x.raw, @path, total: total) }
-    end
-
-    def filtered_disk_usages
-      disk_usages.select { |x| x.kbyte >= @min_kbyte }
-    end
-
-    class DiskUsage
-      attr_accessor :raw, :kbyte, :path
-
-      def initialize(raw, root, total: nil)
-        @raw = raw
-        @root = root
-        @kbyte = raw.split("\t").compact[0].to_i
-        @path = raw.split("\t").compact[1]
-        @total = total
-      end
-
-      def pretty
-        pretty_byte + "\t." + relative_path
-      end
-
-      def tree_format
-        pretty_path + ' ' + pretty_byte
-      end
-
-      private
-
-      def basename
-        Pathname.new(@path).basename.to_s
-      end
-
-      def depth
-        Pathname.new(relative_path).each_filename.to_a.size - 1
-      end
-
-      def pretty_byte
-        mb = @kbyte / 1024
-        gb = mb / 1024
-        if gb.positive?
-          if gb > 10
-            "#{gb}GB (#{percentage})".red
-          else
-            "#{gb}GB (#{percentage})".light_red
-          end
-        elsif mb.positive?
-          if mb > 500
-            "#{mb}MB (#{percentage})".yellow
-          else
-            "#{mb}MB (#{percentage})".light_yellow
-          end
-        else
-          if @kbyte > 500
-            "#{@kbyte}KB (#{percentage})".green
-          else
-            "#{@kbyte}KB (#{percentage})".light_green
-          end
-        end
-      end
-
-      def pretty_path
-        return '.' if @path == @root
-        spaces = depth.positive? ? '  ' * depth : ''
-        tree_symbol = '└── '
-        spaces + tree_symbol + basename
-      end
-
-      def percentage
-        percentage = @kbyte * 1.0 / @total * 100
-        "#{percentage.ceil}%"
-      end
-
-      def relative_path
-        @path.sub(/^#{@root}/, '')
-      end
-    end
-  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: du_pretty
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.1
 platform: ruby
 authors:
 - Katsuma Narisawa
@@ -102,6 +102,8 @@ files:
 - exe/du_pretty
 - lib/du_pretty.rb
 - lib/du_pretty/cli.rb
+- lib/du_pretty/prettier.rb
+- lib/du_pretty/utils.rb
 - lib/du_pretty/version.rb
 homepage: https://github.com/nullnull/du_pretty
 licenses: