Module: But::HomeKeeper

Defined in:
lib/but/home_keeper.rb

Constant Summary collapse

BLOCK_SIZE = 
1024

Class Method Summary collapse

Class Method Details

.all_eof(files) ⇒ Object 



26
27
28
 
# File 'lib/but/home_keeper.rb', line 26

def self.all_eof(files)
  files.find { |f| !f.eof }.nil?
end

.create_disk_usage_file(options = {}) ⇒ Object

Q: is $HOME always set?



31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
 
# File 'lib/but/home_keeper.rb', line 31

def self.create_disk_usage_file(options = {})
  tmpfile = ::Tempfile.new("diskusage")
  dir = options[:dir] || ENV['HOME']

  total_lines = `ls -d ~/.*/ ~/*/ | wc -l`.to_i
  #puts total_lines
  progressbar = ProgressBar.create(title: "inspecting #{dir}", total: total_lines , 
                                   :format         => "%a %c/%C %b\u{15E7}%i %p%% %t",
                                   :progress_mark  => ' ',
                                   :remainder_mark => "\u{FF65}",
                                  )

  ::Open3.popen3("du","-d 1", dir) do |stdin, stdout, stderr, wait_thr|
    # inspired by from https://gist.github.com/chrisn/7450808

    stdin.close_write
    begin
      files = [stdout, stderr]
      until all_eof(files) do
        ready = IO.select(files)
        progressbar.refresh

        if ready
          readable = ready[0]
          readable.each do |f|
            fileno = f.fileno

            begin
              data = f.read_nonblock(BLOCK_SIZE)

              #puts "fileno: #{fileno}, data: #{data.inspect}"
              progressbar.progress += data.lines.count
              progressbar.title = "just completed: %s" % data.split(/\s+/).last
              tmpfile.write(data)
            rescue EOFError => e
              puts "fileno: #{fileno} EOF"
              files.delete f
            end
          end
        end
      end
    rescue IOError => e
      puts "IOError: #{e}"
    end


    #if wait_thr.value != 0
    #  STDERR.write "Exitcode: #{wait_thr.value}\n".light_black
    #  STDERR.write stderr.gets(nil).light_black
    #end

    #tmpfile.write(stdout.gets(nil))
  end
  progressbar.finish
  tmpfile.rewind
  return tmpfile.path
end

.info(options = {}) ⇒ Object 



10
11
12
13
 
# File 'lib/but/home_keeper.rb', line 10

def self.info(options = {})
  file = ENV['BUTHOMEKEEPER'] || create_disk_usage_file(options)
  file_info = File.open(file).readlines.map(&:chomp)
end

.judge(options = {}) ⇒ Object 



15
16
17
18
19
20
21
22
23
24
 
# File 'lib/but/home_keeper.rb', line 15

def self.judge(options = {})
  total_disc_usage, dir = info(options).last.split(/\s+/)
  total_disc_usage = total_disc_usage.to_i

  if total_disc_usage >= But::HOMELIMIT
    return "Dare you! Your directory #{dir} contains #{total_disc_usage}, which is #{total_disc_usage - But::HOMELIMIT} above the limit of #{But::HOMELIMIT}. Dare you!".red
  else
    return "Good job! Your directory #{dir} takes only #{total_disc_usage} disk storage. Good job!".light_green
  end
end