du_pretty 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fbbeba90eac38b4c2ea8bd263de33b2a745d779f2e75d6117961cb39467d11b3
-  data.tar.gz: cfbebae6d5c7c99112dfaeb2227dd665328a5c607cb06d74e675bc8bdc6def93
+  metadata.gz: 77f03d1fc76d562c7f5e1f1be4d7fa5cebdad469b041818dc834b825d846ab1a
+  data.tar.gz: 2e1917543b0664bdbd66650b6d98112edc474c77026902e4984c9375626bb1ea
 SHA512:
-  metadata.gz: df2e22f3512ea0b158a89327f88aced1a1769250c40b6dc91bd1773453d6880de8cde363544bbebb94d32f192585f2b7a393f3a5e0e335d0384a6479ee7e7fa3
-  data.tar.gz: b32332c8b0a28d6e4900319ef0b6e13d6294c88c0fc2348932181c516e2f2158caad43ed5e600923be213cd4fe389a26d7a1358cb5d2b33b54a6993dd5b6a613
+  metadata.gz: fab99faf694934d97c69bf80ee37fc03ae140807382a9acc9eb36d0b62097f81f5165282927119606e5435861f6370b5585a9b43006fbd292556f3cca8ccf04a
+  data.tar.gz: 3695bdc6160ce233f5e734e01336bf98c11a2511a2bc32cfcd040e6cd13377cfc19c4841b0a45ff9614e10f9a458b8773ad851df9b8acf431ba43c162bde880b

data/Gemfile.lock

@@ -1,12 +1,14 @@
 PATH
   remote: .
   specs:
-    du_pretty (0.1.0)
+    du_pretty (0.2.0)
+      colorize
       thor
 
 GEM
   remote: https://rubygems.org/
   specs:
+    colorize (0.8.1)
     diff-lcs (1.3)
     rake (10.5.0)
     rspec (3.8.0)

data/du_pretty.gemspec

@@ -24,5 +24,6 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency 'bundler', '~> 1.16'
   spec.add_development_dependency 'rake', '~> 10.0'
   spec.add_development_dependency 'rspec', '~> 3.0'
+  spec.add_dependency 'colorize'
   spec.add_dependency 'thor'
 end

data/exe/du_pretty

@@ -1,4 +1,9 @@
 #!/usr/bin/env ruby
 
 require 'du_pretty'
-DuPretty::CLI.start
+if ['help', '--help'].include?(ARGV[0])
+  ARGV[1] = DuPretty::CLI.default_task
+else
+  ARGV.unshift(DuPretty::CLI.default_task)
+end
+DuPretty::CLI.start(ARGV)

data/lib/du_pretty/cli.rb

@@ -3,10 +3,20 @@ require 'thor'
 
 module DuPretty
   class CLI < Thor
-    default_command :search
-    desc 'search', 'search is my default_command'
-    def search
-      puts 'hello world!'
+    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
+    def path(path = '.')
+      min_kbyte = [(options[:min_gbyte] || 0) * 1024 * 1024, (options[:min_mbyte] || 0) * 1024, (options[:min_kbyte] || 0)].max
+      print DuWrapper.new(
+        path,
+        min_kbyte: min_kbyte,
+        depth: options[:depth]
+      ).pretty
+      print "\n"
     end
   end
 end

data/lib/du_pretty/version.rb

@@ -1,3 +1,3 @@
 module DuPretty
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/du_pretty.rb

@@ -1,6 +1,90 @@
+require 'colorize'
+
 require 'du_pretty/version'
 require 'du_pretty/cli'
 
 module DuPretty
-  # Your code goes here...
+  class DuWrapper
+    def initialize(path, min_kbyte: 0, depth: nil)
+      @path = File.expand_path(path, Pathname.pwd)
+      @min_kbyte = min_kbyte
+      @depth = depth
+    end
+
+    def pretty
+      du.split("\n")
+        .map { |line| DiskUsage.new(line, @path) }
+        .select { |x| x.kbyte >= @min_kbyte }
+        .reverse
+        .map(&:pretty)
+        .join("\n")
+    end
+
+    private
+
+    def du
+      options = @depth.nil? ? '' : "-d #{@depth}"
+      `du -k #{options} #{@path}`
+    end
+
+    class DiskUsage
+      attr_accessor :raw, :kbyte, :path
+
+      def initialize(raw, root)
+        @raw = raw
+        @root = root
+        @kbyte = raw.split("\t").compact[0].to_i
+        @path = raw.split("\t").compact[1]
+      end
+
+      def pretty
+        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".red
+          else
+            "#{gb}GB".light_red
+          end
+        elsif mb.positive?
+          if mb > 500
+            "#{mb}MB".yellow
+          else
+            "#{mb}MB".light_yellow
+          end
+        else
+          if @kbyte > 500
+            "#{@kbyte}KB".green
+          else
+            "#{@kbyte}KB".light_green
+          end
+        end
+      end
+
+      def pretty_path
+        return '.' if @path == @root
+        spaces = depth.positive? ? '  ' * depth : ''
+        tree_symbol = '└── '
+        spaces + tree_symbol + basename
+      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.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Katsuma Narisawa
@@ -52,6 +52,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: thor
   requirement: !ruby/object:Gem::Requirement