mez 0.8.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 9229e10affa135c3ce6542006affa29d1cf148da
+  data.tar.gz: e0559af63b3b33677d6b907f70e4e8f971d42cc2
+SHA512:
+  metadata.gz: 98b194786f55dfed6770da15fcb8d2c01fcf974538503aea4799a8a34ed7ea34b9f9bb33d4edfc72898a7c5573f43b3dc445daec8839775cc78c3d1f0b8aee06
+  data.tar.gz: b00f64269b83b2a72202fac60890946dec3719d1d2786f370672bf999f05458bb5006ed153f269204fb02ac8a08126580a21aeb34e22a93b975bf6057ad448dc

data/bin/mez

@@ -0,0 +1,33 @@
+#!/usr/bin/env ruby
+require 'mez'
+
+unless File.exist?(File.join(Mez::APP_DIR, 'folders.json'))
+  os_prefix = OS.windows? ? 'c:/files/' : '~/'
+  puts 'You are missing your configuration file!'
+  puts "It should be at #{Mez::CONFIG}, and should look like this:
+
+        {
+            \"projects\": [
+                \"#{os_prefix}work projects\",
+                \"#{os_prefix}personal projects\" ],
+            \"books\": [
+                \"#{os_prefix}fiction\",
+                \"#{os_prefix}nonfiction\" ]
+        }"
+  exit
+end
+
+total_size = 0
+total_difference = 0
+
+Mez.intro_line
+JSON.parse(File.read(Mez::CONFIG)).each do |name, folderset|
+  size = Mez.setsize(folderset)
+  Mez.update(name, size)
+  total_size += size
+  yesterday = Mez.yesterday_size(name)
+  difference = yesterday ? size - yesterday : 0
+  total_difference += difference
+  Mez.report(name, size, difference)
+end
+Mez.closing_line(total_size, total_difference)

data/lib/mez/VERSION.rb

@@ -0,0 +1,3 @@
+module Mez
+  VERSION = '0.8.1'.freeze
+end

data/lib/mez.rb

@@ -0,0 +1,116 @@
+# Configuration and the SQLite database are stored in ~/.mez/ on macOS
+# and Linux, and in %localappdata%/mez on Windows.
+require 'sqlite3'
+require 'date'
+require 'os'
+require 'json'
+require 'fileutils'
+require 'win32ole' if OS.windows?
+
+module Mez
+  APP_DIR = if OS.windows?
+              File.join(File.expand_path('~'), 'appdata', 'local', 'mez')
+            else
+              File.join(File.expand_path('~'), '.mez')
+            end
+  CONFIG = File.join(APP_DIR, 'folders.json')
+  DB = if File.exist?(File.join(APP_DIR, 'folders.db'))
+         SQLite3::Database.new(File.join(APP_DIR, 'folders.db'))
+       else
+         FileUtils.mkdir_p(APP_DIR) unless File.exist?(APP_DIR)
+         db = SQLite3::Database.new(File.join(APP_DIR, 'folders.db'))
+         db.execute('CREATE TABLE folders (name string, date string, size int,
+                     PRIMARY KEY ("name", "date"))')
+         db
+       end
+
+  # Calculates the size of a folder and its contents as an integer.
+  # On Windows, uses the Win32 API, which caches folder information and makes
+  # repeated calls to this function very fast. On other systems, sums individual
+  # file sizes.
+  #
+  # @param folder [String] a path to a folder, eg '~/books' or 'c:\books'
+  # @return [Integer] the folder's size in bytes
+  def self.folder_size(folder)
+    if OS.windows?
+      # Much faster for repeated calls than calculating sizes in Ruby.
+      WIN32OLE.new('Scripting.FileSystemObject').getFolder(folder).size.to_i
+    else
+      size = 0
+      Dir.glob(File.join(folder, '**', '*')) { |file| size += File.size(file) }
+      size
+    end
+  end
+
+  # Uses commas as separators to make large numbers human-readable.
+  #
+  # @param num [Integer] any integer, eg -48151
+  # @return [String] a human readable number, eg "-48,151"
+  def self.humanise(num)
+    (prefix(num) == '-' ? '-' : '') + num
+                                      .abs
+                                      .to_s
+                                      .split('')
+                                      .reverse
+                                      .each_slice(3)
+                                      .map(&:join)
+                                      .join(',')
+                                      .reverse
+  end
+
+  # Provide a "+" or "-" prefix based on whether `n` is positive or negative.
+  #
+  # @param n [Number]
+  # @return [String] "+" or "-"
+  def self.prefix(n)
+    n >= 0 ? '+' : '-'
+  end
+
+  # Format a number for the 'change' column, meaning a + or - prefix and a
+  # number in bytes converted to a comma-separated number in IEC megabytes.
+  #
+  # @param n [Integer] size differential in bytes
+  # @return [String] "+48,151"
+  def self.difference_report(n)
+    n.zero? ? '' : prefix(n) + humanise(n.abs / 1_000_000)
+  end
+
+  def self.yesterday_size(name)
+    DB.get_first_value('SELECT size FROM folders
+                        WHERE name = ? AND date <> ?
+                        ORDER BY date DESC
+                        LIMIT 1', name, Date.today.to_s)
+  end
+
+  def self.report(name, size, difference)
+    puts name.ljust(54) +
+         humanise(size / 1_000_000).rjust(10) +
+         difference_report(difference).rjust(16)
+  end
+
+  def self.intro_line
+    puts "\nFOLDERSET".ljust(56) + 'SIZE (MB)' + 'CHANGE (MB)'.rjust(16)
+    puts('-' * 80)
+  end
+
+  # Take an array of folder names, and return their total size in bytes.
+  #
+  # @param [Array<String>] array of folder names
+  # @return [Integer] total size of all folders and their contents
+  def self.setsize(folderset)
+    folderset.reduce(0) { |acc, elem| acc + Mez.folder_size(elem) }
+  end
+
+  def self.closing_line(total_size, total_difference)
+    puts('-' * 80)
+    puts 'TOTAL SIZE: ' +
+         humanise(total_size / 1_000_000) +
+         ('TOTAL CHANGE: ' + humanise(total_difference / 1_000_000)).rjust(59)
+  end
+
+  def self.update(name, size)
+    DB.execute('INSERT OR REPLACE INTO folders
+                (name, size, date)
+                VALUES (?,?,?)', name, size, Date.today.to_s)
+  end
+end

metadata

@@ -0,0 +1,89 @@
+--- !ruby/object:Gem::Specification
+name: mez
+version: !ruby/object:Gem::Version
+  version: 0.8.1
+platform: ruby
+authors:
+- Ryan Plant
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-01-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: os
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.9.6
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.9.6
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.5'
+description: 
+email: ryan@ryanplant.net
+executables:
+- mez
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/mez
+- lib/mez.rb
+- lib/mez/VERSION.rb
+homepage: https://github.com/ryantriangles/mez
+licenses:
+- MIT
+metadata: {}
+post_install_message: Please configure your folders.json!
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: A CLI utility to track directory sizes over time
+test_files: []