weigh 0.0.1

data/README.md

@@ -0,0 +1,4 @@
+# Weigh
+
+A small ruby utility to wieigh directory contents.  Heavy stuff to the bottom.
+

data/bin/weigh

@@ -0,0 +1,6 @@
+#! /usr/bin/env ruby
+# Weigh directory contents
+
+require 'weigh/cli'
+
+Weigh::CLI.run ARGV

data/lib/weigh/cli.rb

@@ -0,0 +1,30 @@
+require 'weigh/flags'
+require 'weigh/runner'
+
+module Weigh
+  class CLI
+
+    attr_reader :flags
+    attr_reader :runner
+
+    def initialize(flags)
+      @flags  = flags
+      @runner = Weigh::Runner.new(@flags)
+    end
+
+    def run
+      if flags.help?
+        puts flags
+        exit
+      end
+
+      runner.run
+    end
+
+    def self.run(*args)
+      flags = Weigh::Flags.new args
+
+      return Weigh::CLI.new(flags).run
+    end
+  end
+end

data/lib/weigh/flags.rb

@@ -0,0 +1,48 @@
+require 'optparse'
+
+module Weigh
+  class Flags
+
+    attr_reader :args
+    attr_reader :verbose
+    attr_reader :depth
+    attr_reader :pathlist
+
+    def initialize(*args)
+      @args     = []
+      @verbose  = false
+      @depth    = 1
+      @pathlist = []
+
+      @options = OptionParser.new do|o|
+        o.banner = "Usage: #{File.basename $0} [options] [file|directory...]\n\n"
+
+        o.on( '--verbose', '-v', 'Speak up' ) do
+          @verbose = true
+        end
+
+        o.on( '--depth DEPTH', '-d', 'Sumarize deeper than DEPTH' ) do|d|
+          @depth = d
+        end
+
+        o.on( '-h', '--help', 'Display this screen' ) do
+          @help = true
+        end
+      end
+
+      @args = @options.parse!
+
+      if ARGV.size > 0
+        ARGV.each do |f|
+          @pathlist << f
+        end
+      else
+        @pathlist << "."
+      end
+
+      def help?
+        @help
+      end
+    end
+  end
+end

data/lib/weigh/runner.rb

@@ -0,0 +1,47 @@
+require 'weigh/util'
+require 'pp'
+require 'find'
+
+module Weigh
+  class Runner
+
+    attr_reader :flags
+
+    def initialize(flags)
+      @flags = flags
+    end
+
+    def run
+      total_size = 0
+      summary = {}
+      count = 0
+      sumdep = @flags.depth
+
+      @flags.pathlist.each do |p|
+        curdep = 0
+        Find.find(p) do |path|
+          count += 1
+          next if FileTest.symlink?(path)
+          if FileTest.directory?(path)
+            next if p == path
+            ret = Weigh::Util.sum_dir(path,@flags.verbose)
+            dir_size = ret[:dir_size]
+            next if dir_size == 0
+            count += ret[:count]
+            total_size += dir_size
+            summary["#{path}/"] = dir_size
+            Find.prune
+          else
+            size = FileTest.size(path)
+            next if size == 0
+            total_size += size
+            summary["#{path}"] = size
+          end
+        end
+      end
+
+      Weigh::Util.report(summary,total_size)
+    end
+
+  end
+end

data/lib/weigh/util.rb

@@ -0,0 +1,91 @@
+module Weigh
+  module Util
+
+    def self.neat_size(bytes)
+      # return a human readble size from the bytes supplied
+      bytes = bytes.to_f
+      if bytes > 2 ** 40       # TiB: 1024 GiB
+        neat = sprintf("%.2f TiB", bytes / 2**40)
+      elsif bytes > 2 ** 30    # GiB: 1024 MiB
+        neat = sprintf("%.2f GiB", bytes / 2**30)
+      elsif  bytes > 2 ** 20   # MiB: 1024 KiB
+        neat = sprintf("%.2f MiB", bytes / 2**20)
+      elsif bytes > 2 ** 10    # KiB: 1024 B
+        neat = sprintf("%.2f KiB", bytes / 2**10)
+      else                    # bytes
+        neat = sprintf("%.0f bytes", bytes)
+      end
+      neat
+    end
+
+    def self.sum_dir(dir,verbose=false)
+      # return the size of a given directory
+      #"Entering: #{dir}"
+      count=0
+      dir_size=0
+      data={}
+      Find.find(dir) do |path|
+        count += 1
+        next if FileTest.symlink?(path)
+        next if dir == path
+        if FileTest.directory?(path)
+          ret = sum_dir(path,verbose)
+          size = ret[:dir_size]
+          count += ret[:count]
+          dir_size += size
+          Find.prune
+        else
+          size = FileTest.size(path)
+          #puts "File: #{path} is #{size}"
+          puts "Found zero size file: #{path}" if verbose
+          dir_size += size
+        end
+      end
+      #puts "Exiting: #{dir} with #{dir_size}"
+      data[:dir_size] = dir_size
+      data[:count] = count
+      data
+    end
+
+    def self.report(summary,total_size)
+      summary.sort{|a,b| a[1]<=>b[1]}.each { |elem|
+        size     = elem[1]
+        filename = elem[0]
+        puts sprintf("%15s   %s\n", neat_size(size), filename)
+      }
+
+      puts sprintf("%16s %s\n", "---", "---")
+      puts sprintf("%15s   %s\n", self.neat_size(total_size), ":total size")
+      puts sprintf("%16s %s\n", "---", "---")
+    end
+
+    def sum_dir(dir,verbose=false)
+      # return the size of a given directory
+      #"Entering: #{dir}"
+      count=0
+      dir_size=0
+      data={}
+      Find.find(dir) do |path|
+        count += 1
+        next if FileTest.symlink?(path)
+        next if dir == path
+        if FileTest.directory?(path)
+          ret = sum_dir(path,verbose)
+          size = ret[:dir_size]
+          count += ret[:count]
+          dir_size += size
+          Find.prune
+        else
+          size = FileTest.size(path)
+          #puts "File: #{path} is #{size}"
+          puts "Found zero size file: #{path}" if verbose
+          dir_size += size
+        end
+      end
+      #puts "Exiting: #{dir} with #{dir_size}"
+      data[:dir_size] = dir_size
+      data[:count] = count
+      data
+    end
+  end
+end

metadata

@@ -0,0 +1,53 @@
+--- !ruby/object:Gem::Specification
+name: weigh
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Zach Leslie
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-05-28 00:00:00.000000000 Z
+dependencies: []
+description: Useful utility to weigh the directory contents and sort the output by
+  size in a neatly formatted list.
+email: xaque208@gmail.com
+executables:
+- weigh
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/weigh
+- lib/weigh/cli.rb
+- lib/weigh/flags.rb
+- lib/weigh/runner.rb
+- lib/weigh/util.rb
+- README.md
+homepage: https://github.com/xaque208/weigh
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Weigh directory contents
+test_files: []
+has_rdoc: