volume_visualizer 0.0.1

data/lib/volume_visualizer/data_size_string.rb

@@ -0,0 +1,26 @@
+module VolumeVisualizer
+  class DataSizeString < String
+    ABBREV_MAP = {
+      "B" => 1024**0,
+      "K" => 1024**1,
+      "M" => 1024**2,
+      "G" => 1024**3,
+      "T" => 1024**4,
+      "P" => 1024**5,
+      "E" => 1024**6,
+      "Z" => 1024**7
+    }
+
+    def last_char
+      to_s[length - 1]
+    end
+
+    def multiplier
+      ABBREV_MAP[last_char] || 1
+    end
+
+    def bytes
+      (to_f * multiplier).to_i
+    end
+  end
+end

data/lib/volume_visualizer/version.rb

@@ -0,0 +1,3 @@
+module VolumeVisualizer
+  VERSION = "0.0.1"
+end

data/lib/volume_visualizer/zfs.rb

@@ -0,0 +1,12 @@
+module VolumeVisualizer
+  module ZFS
+    ATTRS = [ :name, :available, :used, :referenced, :used_by_snapshots, 
+      :used_by_dataset, :used_by_children, :compress_ratio, :type ]
+    NON_BYTE_ATTRS = [ :name, :compress_ratio, :type ]
+  end
+end
+
+require 'volume_visualizer/zfs/base'
+require 'volume_visualizer/zfs/pool'
+require 'volume_visualizer/zfs/filesystem'
+

data/lib/volume_visualizer/zfs/base.rb

@@ -0,0 +1,54 @@
+module VolumeVisualizer
+  module ZFS
+    class Base
+      attr_reader :filesystems, :root, :name
+
+      def initialize(name, data)
+        @name = name || "unknown volume"
+        @filesystems = {}
+
+        data.each do |line|
+          line.chomp!
+          fs = ZFS::Filesystem.new(line)
+          next if fs.snapshot? && fs.zero?
+          if filesystems.empty?
+            @root = fs
+          else
+            parent = filesystems[ fs.parent_name ]
+            fs.parent = parent
+            parent.children << fs
+          end
+          filesystems[fs.name] = fs
+        end
+      end
+
+      def walk(nodes = [ root ], &block)
+        nodes.each do |node|
+          yield node
+          walk node.children, &block
+        end
+      end
+
+      def to_h
+        {
+          :name => name,
+          :type => :root,
+          :children => [
+            {
+              :name => "free space",
+              :size => root.available.bytes,
+              :size_human => root.available,
+              :type => "free"
+            },
+            root.to_h
+          ]
+        }
+      end
+
+      def to_json
+        to_h.to_json(:max_nesting => 50)
+      end
+    end
+  end
+end
+

data/lib/volume_visualizer/zfs/filesystem.rb

@@ -0,0 +1,75 @@
+module VolumeVisualizer
+  module ZFS
+    class Filesystem
+      attr_accessor *ATTRS
+      attr_accessor :children, :parent
+
+      def initialize(string)
+        @children = []
+
+        params = string.split("\t")
+        ATTRS.each do |attr|
+          val = params.shift
+          unless NON_BYTE_ATTRS.include? attr
+            val = DataSizeString.new(val)
+          end
+          instance_variable_set "@#{attr.to_s}", val
+        end
+      end
+
+      def depth
+        depth = name.split("/").size
+        depth -= 1 unless snapshot?
+        depth
+      end
+
+      def parent_name
+        if snapshot?
+          name.split("@").first
+        else
+          File.dirname(name)
+        end
+      end
+
+      def snapshot?
+        type == "snapshot"
+      end
+
+      def zero?
+        used.bytes == 0
+      end
+
+      def dump
+        ATTRS.each do |attr|
+          puts "%18s %s" % [attr, self.send(attr)]
+        end
+      end
+
+      def to_h
+        hash = {
+          :name => name,
+          :type => type,
+          :size => used.bytes,
+          :size_human => used,
+          :type => type,
+        }
+        unless children.empty?
+          hash[:children] = [
+            {
+            :name => name + " (data)",
+            :type => "filesystem",
+            :size => referenced.bytes,
+            :size_human => referenced
+          }
+          ]
+          hash[:type] = "container"
+          children.each do |child|
+            hash[:children] << child.to_h
+          end
+        end
+
+        hash
+      end
+    end
+  end
+end

data/lib/volume_visualizer/zfs/pool.rb

@@ -0,0 +1,26 @@
+module VolumeVisualizer
+  module ZFS
+    class Pool
+      attr_reader :pool_name
+
+      def self.query_command(pool_name = "tank")
+        attrs = ATTRS.collect { |a| a.to_s.gsub("_","") }.join(",")
+        command = "zfs list -t all -H -r -o #{attrs} #{pool_name}"
+        if ENV["USER"] != "root"
+          command = "sudo " + command
+        else
+          command
+        end
+      end
+
+      def initialize(pool_name)
+        @pool_name = pool_name
+      end
+
+      def data
+        STDERR.puts "reading ZFS data"
+        File.popen(Pool.query_command pool_name).readlines
+      end
+    end
+  end
+end

data/volume_visualizer.gemspec

@@ -0,0 +1,26 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'volume_visualizer/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "volume_visualizer"
+  spec.version       = VolumeVisualizer::VERSION
+  spec.authors       = ["Rick Lee-Morlang"]
+  spec.email         = ["<rick@lee-morlang.com>"]
+  spec.summary       = %q{Visualize data distribution in your storage volumes.}
+  #spec.description   = %q{TODO: Write a longer description. Optional.}
+  spec.homepage      = "https://github.com/rleemorlang/volume_visualizer"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.5"
+  spec.add_development_dependency "rake"
+
+  spec.add_dependency "thor"
+  spec.add_dependency "json"
+end

metadata

@@ -0,0 +1,122 @@
+--- !ruby/object:Gem::Specification
+name: volume_visualizer
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Rick Lee-Morlang
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-03-21 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  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
+    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: json
+  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'
+description: 
+email:
+- <rick@lee-morlang.com>
+executables:
+- vv
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- assets/index.html
+- bin/vv
+- example/data
+- example/index.html
+- example/screenshot.jpg
+- example/screenshot_scaled.jpg
+- example/volviz.json
+- lib/volume_visualizer.rb
+- lib/volume_visualizer/cli.rb
+- lib/volume_visualizer/data_size_string.rb
+- lib/volume_visualizer/version.rb
+- lib/volume_visualizer/zfs.rb
+- lib/volume_visualizer/zfs/base.rb
+- lib/volume_visualizer/zfs/filesystem.rb
+- lib/volume_visualizer/zfs/pool.rb
+- volume_visualizer.gemspec
+homepage: https://github.com/rleemorlang/volume_visualizer
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+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.2.2
+signing_key: 
+specification_version: 4
+summary: Visualize data distribution in your storage volumes.
+test_files: []