pipette 0.1.0

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,22 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+.bundle
+
+## PROJECT::SPECIFIC

data/Gemfile

@@ -0,0 +1,15 @@
+# A sample Gemfile
+source :gemcutter
+#
+# gem "rails"
+
+gem "right_aws"
+gem "trollop"
+
+group :test do
+  gem "rspec"
+end
+
+group :dev do
+  gem "yard"
+end

data/Gemfile.lock

@@ -0,0 +1,18 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    right_aws (2.0.0)
+      right_http_connection (>= 1.2.1)
+    right_http_connection (1.2.4)
+    rspec (1.3.0)
+    trollop (1.16.2)
+    yard (0.5.8)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  right_aws
+  rspec
+  trollop
+  yard

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Paul Sadauskas
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,17 @@
+= pipette
+
+Description goes here.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2010 Paul Sadauskas. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,54 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "pipette"
+    gem.summary = %Q{pipette is a tool for managing volumes}
+    gem.description = %Q{Create, grow, and remove ebs+md+lvm volumes and disks}
+    gem.email = "psadauskas@gmail.com"
+    gem.homepage = "http://github.com/mongomachine/pipette"
+    gem.authors = ["Paul Sadauskas"]
+
+    gem.add_dependency "trollop"
+    gem.add_dependency "right_aws"
+
+    gem.add_development_dependency "rspec", ">= 1.2.9"
+    gem.add_development_dependency "yard", ">= 0"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+begin
+  require 'spec/rake/spectask'
+  Spec::Rake::SpecTask.new(:spec) do |spec|
+    spec.libs << 'lib' << 'spec'
+    spec.spec_files = FileList['spec/**/*_spec.rb']
+  end
+
+  Spec::Rake::SpecTask.new(:rcov) do |spec|
+    spec.libs << 'lib' << 'spec'
+    spec.pattern = 'spec/**/*_spec.rb'
+    spec.rcov = true
+  end
+
+  task :spec => :check_dependencies
+
+  task :default => :spec
+rescue LoadError
+  task :spec do
+    abort "RSpec is not available."
+  end
+end
+
+begin
+  require 'yard'
+  YARD::Rake::YardocTask.new
+rescue LoadError
+  task :yardoc do
+    abort "YARD is not available. In order to run yardoc, you must: sudo gem install yard"
+  end
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/bin/pipette

@@ -0,0 +1,75 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'trollop'
+
+SUB_COMMANDS = %w[ init append add grow remove ]
+
+opts = Trollop::options do
+  banner <<-EOS
+This script will manage mirrored EBS volumes and lvm volume groups.
+
+Usage:
+  pipette [global options] subcommand [options]
+
+  Where subcommand is one of #{SUB_COMMANDS.inspect}
+
+Global Options:
+EOS
+  opt :access_key, "AWS Access Key", :short => "-a", :default => ENV['ACCESS_KEY']
+  opt :secret_access_key, "AWS Secret Access Key", :short => "-c", :default => ENV['SECRET_ACCESS_KEY']
+  stop_on SUB_COMMANDS
+end
+
+cmd = ARGV.shift
+cmd_opts = case cmd
+           when "init"
+             Trollop::options do
+               banner "Create an initial volume group of EBS mirrored volumes"
+               opt :name          , "Name of volume group to create"                                  , :short => "-n", :required => true, :type => :string
+               opt :size          , "Size of initial volume in GB"                                    , :short => "-s", :required => true, :type => :int
+               opt :ebs_per_mirror, "Number of EBS volumes to put in each mirrored disk"              , :default => 2 , :type => :int
+               opt :num_mirrors   , "Number of mirrors to create; will be spanned by the volume group", :default => 2 , :type => :int
+             end
+           when "append"
+             Trollop::options do
+               banner "Add another set of mirrored EBS volumes to the volume group"
+               opt :name          , "Name of volume group to append to"                               , :short => "-n", :required => true, :type => :string
+               opt :size          , "Size of initial volume in GB"                                    , :short => "-s", :required => true, :type => :int
+               opt :ebs_per_mirror, "Number of EBS volumes to put in each mirrored disk"              , :default => 2 , :type => :int
+               opt :num_mirrors   , "Number of mirrors to create; will be spanned by the volume group", :default => 1 , :type => :int
+             end
+           when "add"
+             Trollop::options do
+               banner "Create a new logical volume in the volume group"
+               opt :name  , "Name of new logical volume"  , :short => "-n", :required => true, :type => :string
+               opt :volume, "Name of volume group to use" , :short => "-v", :required => true, :type => :string
+               opt :size  , "Size of new logical volume",   :short => "-s", :required => true, :type => :int
+             end
+           when "grow"
+             Trollop::options do
+               banner "Increase the size of a logical volume"
+               opt :name  , "Name of logical volume"                                            , :short => "-n", :required => true, :type => :string
+               opt :volume, "Name of volume group to use"                                       , :short => "-v", :required => true, :type => :string
+               opt :size  , "Size to increase logical volume to or '+n' to increase size by nGB", :short => "-s", :required => true, :type => :string
+             end
+           when "remove"
+             Trollop::options do
+               banner "Remove a logical volume"
+               opt :name  , "Name of logical volume"     , :short => "-n", :required => true, :type => :string
+               opt :volume, "Name of volume group to use", :short => "-v", :required => true, :type => :string
+             end
+           else
+             Trollop::die "unknown subcommand #{cmd.inspect}"
+           end
+
+begin
+  require 'pipette'
+rescue LoadError => ex
+  # Not in load path
+  require File.join(File.dirname(__FILE__), '..', 'lib', 'pipette')
+end
+
+Pipette.new(opts).send(cmd, cmd_opts)
+
+# vim: ft=ruby

data/lib/pipette/commands.rb

@@ -0,0 +1,93 @@
+
+class Pipette
+  module Commands
+
+    # cmd_opts:
+    #
+    #   size: Size of each volume, in GB
+    #   name: Name of the volume group to create
+    #   ebs_per_mirror: Number of EBS volumes to put in each mirrored disk. Default 2.
+    #   num_mirrors: Number of mirrors to create, to be spanned by the volume group. Default 2.
+    def init(cmd_opts)
+      raise "No volume group name specified" unless cmd_opts[:name]
+      raise "No EBS volume size specified" unless cmd_opts[:size]
+
+      md_devs = create_mirrors(cmd_opts[:size],
+                               cmd_opts[:num_mirrors],
+                               cmd_opts[:ebs_per_mirror])
+
+
+      vg_create cmd_opts[:name], md_devs
+    end
+
+    def append(cmd_opts)
+      raise "No volume group name specified" unless cmd_opts[:name]
+      raise "No EBS volume size specified" unless cmd_opts[:size]
+
+      md_devs = create_mirrors(cmd_opts[:size],
+                               cmd_opts[:num_mirrors],
+                               cmd_opts[:ebs_per_mirror])
+
+      vg_extend cmd_opts[:name], md_devs
+    end
+
+    def add(cmd_opts)
+      raise "No logical volume name specified" unless cmd_opts[:name]
+      raise "No logical volume size specified" unless cmd_opts[:size]
+      raise "No volume group name specified" unless cmd_opts[:volume]
+
+      dev = lv_create(cmd_opts[:name], cmd_opts[:volume], cmd_opts[:size])
+      format(dev)
+      mount(dev, "/data/#{cmd_opts[:name]}")
+    end
+
+    def grow(cmd_opts)
+      raise "No logical volume name specified" unless cmd_opts[:name]
+      raise "No logical volume size specified" unless cmd_opts[:size]
+      raise "No volume group name specified" unless cmd_opts[:volume]
+
+      dev = lv_dev(cmd_opts[:volume], cmd_opts[:name])
+      lv_extend(dev, cmd_opts[:size])
+      growfs(dev)
+    end
+
+		def remove(cmd_opts)
+      raise "No logical volume name specified" unless cmd_opts[:name]
+      raise "No volume group name specified" unless cmd_opts[:volume]
+
+			dev = lv_dev(cmd_opts[:volume], cmd_opts[:name])
+			umount(dev)
+			lv_remove(dev)
+		end
+
+    protected
+
+    def create_mirrors(size, num_mirrors = 2, num_disks = 2)
+      md_devs = []
+      num_mirrors.times do
+        volumes = []
+        num_disks.times do
+          volumes << create_volume(size)
+        end
+
+        devices = []
+        volumes.each do |v_id|
+          mp = next_avail_disk
+          mount_volume v_id, mp
+          devices << mp
+        end
+
+        wait_for devices
+
+        md_dev = next_md
+        create_md_mirror md_dev, *devices
+
+        pv_create md_dev
+        md_devs << md_dev
+      end
+
+      md_devs
+    end
+
+  end
+end

data/lib/pipette/ec2.rb

@@ -0,0 +1,36 @@
+require 'open-uri'
+
+class Pipette
+  module Ec2
+
+    def ec2
+      @ec2 ||= RightAws::Ec2.new(opts[:access_key], opts[:secret_access_key])
+    end
+
+    def create_volume(size)
+      say_with_time "Creating new volume of size #{size}G" do
+        res = ec2.create_volume(nil, size, zone)
+        res[:aws_id]
+      end
+    end
+
+    def mount_volume(volume_id, mount_point)
+      say_with_time "Mounting #{volume_id} at #{mount_point}" do
+        res = ec2.attach_volume(volume_id, instance_id, mount_point)
+        res[:aws_device]
+      end
+    end
+
+    def zone
+      @zone ||= meta("placement/availability-zone")
+    end
+
+    def instance_id
+      @instance_id ||= meta("instance-id")
+    end
+
+    def meta(path)
+      open("http://169.254.169.254/latest/meta-data/#{path}").read.strip
+    end
+  end
+end

data/lib/pipette/lvm.rb

@@ -0,0 +1,49 @@
+
+class Pipette
+  module Lvm
+
+    def pv_create(device)
+      run_command "pvcreate", device
+    end
+
+    def vg_create(name, *devices)
+      run_command "vgcreate",
+                  "--physicalextentsize 16M",
+                  name,
+                  *devices
+    end
+
+    def vg_extend(name, *devices)
+      run_command "vgextend",
+                  name,
+                  *devices
+    end
+
+		def lv_create(name, vg, size)
+			run_command "lvcreate",
+									"--size #{size}G",
+									"--name #{name}",
+									vg
+
+      lv_dev(vg, name)
+		end
+
+    def lv_extend(device, size)
+      run_command "lvextend",
+                  "--size #{size}G",
+                  device
+    end
+
+		def lv_remove(device)
+			run_command "lvremove", "--force", device
+		end
+
+    protected
+
+    def lv_dev(vg, name)
+      "/dev/#{vg}/#{name}"
+    end
+
+
+  end
+end

data/lib/pipette/md.rb

@@ -0,0 +1,16 @@
+
+class Pipette
+  module Md
+
+    def create_md_mirror(md_device, *devices)
+      run_command "mdadm",
+                  "--create", md_device,
+                  "--level=mirror",
+                  "--raid-devices=#{devices.size}",
+                  *devices
+
+      wait_for md_device
+    end
+
+  end
+end

data/lib/pipette/util.rb

@@ -0,0 +1,76 @@
+
+class Pipette
+  module Util
+
+    def run_command(cmd, *args)
+      args.unshift(cmd)
+
+      command = args.flatten.join(' ')
+
+      say_with_time command do
+        `#{command}`
+      end
+    end
+
+    def next_avail_disk
+      @last_disk ||= Dir["/dev/sd[d-p]{1,2,3,4,5,6,7,8,9,10,11,12,13,14,15}"].sort { |a,b|
+        disk_parts(a) <=> disk_parts(b)
+      }.last
+
+      disk, part = disk_parts(@last_disk)
+
+      if disk
+        if part.to_i >= 15
+          disk = disk.succ
+          part = 1
+        else
+          part += 1
+        end
+      else
+        disk, part = "d", "1"
+      end
+
+      @last_disk = "/dev/sd#{disk}#{part}"
+    end
+
+    def next_md
+      md = Dir["/dev/md[0-9]*"].sort.last
+      if md && match = md.match(%r{/dev/md([\d+])})
+        i = match[1].to_i + 1
+      else
+        i = 0
+      end
+      "/dev/md#{i}"
+    end
+
+    def disk_parts(dev)
+      if dev && match = dev.match(%r{/dev/sd([d-p])([0-1][0-9]|[0-9])})
+        disk, part = match[1], match[2].to_i
+      else
+        nil
+      end
+    end
+
+    def say_with_time(msg = "", &block)
+      @depth ||= 0
+      @depth += 1
+      indent = '  ' * @depth
+      puts "#{indent}#{msg}"
+      t = Time.now
+      res = yield
+      puts "#{indent}-> %0.4f" % (Time.now - t)
+      @depth -= 1
+      res
+    end
+
+    def wait_for(*paths)
+      paths.flatten.each do |path|
+        puts "Waiting for #{path} to appear..."
+        until File.exist?(path)
+          sleep 0.1
+        end
+      end
+    end
+
+  end
+end

data/lib/pipette/xfs.rb

@@ -0,0 +1,22 @@
+
+class Pipette
+	module Xfs
+
+		def format(device)
+			run_command "mkfs.xfs", device
+		end
+
+		def mount(device, mountpoint)
+			FileUtils.mkdir_p(mountpoint)
+			run_command "mount", device, mountpoint
+		end
+
+		def growfs(device)
+			run_command "xfs_growfs", device
+		end
+
+		def umount(device)
+			run_command "umount", device
+		end
+	end
+end

data/lib/pipette.rb

@@ -0,0 +1,24 @@
+require 'right_aws'
+
+dir = File.dirname(__FILE__)
+
+%w[ util ec2 md lvm xfs commands ].each do |file|
+  require File.join(dir, 'pipette', file)
+end
+
+class Pipette
+  include Util
+  include Ec2
+  include Md
+  include Lvm
+	include Xfs
+  include Commands
+
+  attr_reader :opts
+
+  def initialize(opts)
+    @opts = opts
+  end
+
+end
+

data/spec/pipette_spec.rb

@@ -0,0 +1,7 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Pipette" do
+  it "fails" do
+    fail "hey buddy, you should probably rename this file and start specing for real"
+  end
+end

data/spec/spec.opts

@@ -0,0 +1 @@
+--color

data/spec/spec_helper.rb

@@ -0,0 +1,9 @@
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'pipette'
+require 'spec'
+require 'spec/autorun'
+
+Spec::Runner.configure do |config|
+  
+end

metadata

@@ -0,0 +1,146 @@
+--- !ruby/object:Gem::Specification 
+name: pipette
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Paul Sadauskas
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-08-18 00:00:00 -06:00
+default_executable: pipette
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: trollop
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: right_aws
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 13
+        segments: 
+        - 1
+        - 2
+        - 9
+        version: 1.2.9
+  type: :development
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: yard
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id004
+description: Create, grow, and remove ebs+md+lvm volumes and disks
+email: psadauskas@gmail.com
+executables: 
+- pipette
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- README
+- README.rdoc
+- Rakefile
+- VERSION
+- bin/pipette
+- lib/pipette.rb
+- lib/pipette/commands.rb
+- lib/pipette/ec2.rb
+- lib/pipette/lvm.rb
+- lib/pipette/md.rb
+- lib/pipette/util.rb
+- lib/pipette/xfs.rb
+- spec/pipette_spec.rb
+- spec/spec.opts
+- spec/spec_helper.rb
+has_rdoc: true
+homepage: http://github.com/mongomachine/pipette
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: pipette is a tool for managing volumes
+test_files: 
+- spec/pipette_spec.rb
+- spec/spec_helper.rb