vagrant-disksize 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 9c0e70788c6b8f747354f486d19a1e85b3ada6be
+  data.tar.gz: dc08aab0dfc7dcd908e6cd2dc6b0a33626f8c26c
+SHA512:
+  metadata.gz: 6b6b0e2782b10a6b855f93f8b3af99abe8a6c5aae58c8eed0a1c114e09b06a603a0c0ea3beda2652759822085748b1f8cab0701c032fbc5887f584ae9a4d013c
+  data.tar.gz: 925ecacc060eebb5399c722bc53f6b380ffa0d621280a0139cd699726e8ca0f26af516784bf67fcc6cfdd96497bcaf32cf870ca11da4fac87a103f7047df0b50

data/.gitignore

@@ -0,0 +1,14 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+
+.ruby-version
+.ruby-gemset
+
+.*.swp

data/Gemfile

@@ -0,0 +1,15 @@
+source 'https://rubygems.org'
+
+group :development do
+  gem 'vagrant', git: 'https://github.com/mitchellh/vagrant.git', tag: 'v1.8.5'
+end
+
+group :plugins do
+  gem 'vagrant-disksize', path: '.'
+end
+
+
+# Specify your gem's dependencies in vagrant-disksize.gemspec
+# Actually, we can't do this as we need to be in the plugins group
+# for vagrant to load us during development
+#gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2016 Simon Protheroe
+
+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.md

@@ -0,0 +1,41 @@
+# Vagrant::Disksize
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/vagrant/disksize`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+TODO: Delete this and the text above, and describe your gem
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'vagrant-disksize'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install vagrant-disksize
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/vagrant-disksize.
+
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).
+

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+task :default => :spec

data/Vagrantfile

@@ -0,0 +1,73 @@
+# -*- mode: ruby -*-
+# vi: set ft=ruby :
+
+# All Vagrant configuration is done below. The "2" in Vagrant.configure
+# configures the configuration version (we support older styles for
+# backwards compatibility). Please don't change it unless you know what
+# you're doing.
+Vagrant.configure("2") do |config|
+  # The most common configuration options are documented and commented below.
+  # For a complete reference, please see the online documentation at
+  # https://docs.vagrantup.com.
+
+  # Every Vagrant development environment requires a box. You can search for
+  # boxes at https://atlas.hashicorp.com/search.
+  config.vm.box = "ubuntu/xenial64"
+
+  # Disable automatic box update checking. If you disable this, then
+  # boxes will only be checked for updates when the user runs
+  # `vagrant box outdated`. This is not recommended.
+  # config.vm.box_check_update = false
+
+  # Create a forwarded port mapping which allows access to a specific port
+  # within the machine from a port on the host machine. In the example below,
+  # accessing "localhost:8080" will access port 80 on the guest machine.
+  # config.vm.network "forwarded_port", guest: 80, host: 8080
+
+  # Create a private network, which allows host-only access to the machine
+  # using a specific IP.
+  # config.vm.network "private_network", ip: "192.168.33.10"
+
+  # Create a public network, which generally matched to bridged network.
+  # Bridged networks make the machine appear as another physical device on
+  # your network.
+  # config.vm.network "public_network"
+
+  # Share an additional folder to the guest VM. The first argument is
+  # the path on the host to the actual folder. The second argument is
+  # the path on the guest to mount the folder. And the optional third
+  # argument is a set of non-required options.
+  # config.vm.synced_folder "../data", "/vagrant_data"
+
+  # Provider-specific configuration so you can fine-tune various
+  # backing providers for Vagrant. These expose provider-specific options.
+  # Example for VirtualBox:
+  #
+  # config.vm.provider "virtualbox" do |vb|
+  #   # Display the VirtualBox GUI when booting the machine
+  #   vb.gui = true
+  #
+  #   # Customize the amount of memory on the VM:
+  #   vb.memory = "1024"
+  # end
+  #
+  # View the documentation for the provider you are using for more
+  # information on available options.
+
+  config.disksize.size = '20GB' 
+
+  # Define a Vagrant Push strategy for pushing to Atlas. Other push strategies
+  # such as FTP and Heroku are also available. See the documentation at
+  # https://docs.vagrantup.com/v2/push/atlas.html for more information.
+  # config.push.define "atlas" do |push|
+  #   push.app = "YOUR_ATLAS_USERNAME/YOUR_APPLICATION_NAME"
+  # end
+
+  # Enable provisioning with a shell script. Additional provisioners such as
+  # Puppet, Chef, Ansible, Salt, and Docker are also available. Please see the
+  # documentation for more information about their specific syntax and use.
+  # config.vm.provision "shell", inline: <<-SHELL
+  #   apt-get update
+  #   apt-get install -y apache2
+  # SHELL
+end

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "vagrant/disksize"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/vagrant/disksize.rb

@@ -0,0 +1,34 @@
+#require "vagrant/disksize/version"
+begin
+  require 'vagrant'
+rescue LoadError
+  raise 'The vagrant-disksize plugin must be run within vagrant.'
+end
+
+
+module Vagrant
+  module Disksize
+    class Plugin < Vagrant.plugin('2')
+
+      name 'vagrant-disksize'
+
+      description <<-DESC
+      Provides the ability to resize VirtualBox disks at creation time,
+      so they don't need to be the same size as the default for the box.
+      Filesystems are not resized by this code.
+      DESC
+
+      config 'disksize' do
+        require_relative 'disksize/config'
+        Config
+      end
+
+      action_hook(:disksize, :machine_action_up) do |hook|
+        require_relative 'disksize/actions'
+
+        # TODO Ensure we are using the VirtualBox provider
+        hook.before(VagrantPlugins::ProviderVirtualBox::Action::Boot, Action::ResizeDisk)
+      end
+    end
+  end
+end

data/lib/vagrant/disksize/actions.rb

@@ -0,0 +1,140 @@
+module Vagrant
+  module Disksize
+    class Action
+
+      class ResizeDisk
+
+        # NOTE - size is represented in MB throughout this class, for ease of using with VirtualBox
+
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @enabled = true
+          if @machine.provider.to_s !~ /VirtualBox/
+            @enabled = false
+            env[:ui].error "The vagrant-disksize plugin only supports VirtualBox at present. Disk size will not be changed."
+          end
+        end
+
+        def call(env)
+          # Resize disk itself before boot
+          if @enabled
+            size = @machine.config.disksize.size
+            ensure_disk_resizable(env)
+            resize_disk(env, size)
+          end
+
+          # Allow middleware chain to continue so VM is booted
+          @app.call(env)
+
+          # TODO Possibly resize partition and filesystem here if needed
+        end
+
+        private
+
+        def ensure_disk_resizable(env)
+          driver = @machine.provider.driver
+          disks = identify_disks(driver)
+          # TODO Shouldn't assume that the first disk is the one we want to resize
+          unless disk_resizeable? disks.first
+            old_disk = disks.first
+            new_disk = generate_resizable_disk(old_disk)
+            unless File.exist? new_disk[:file]
+              clone_as_vdi(driver, old_disk, new_disk)
+              attach_disk(driver, new_disk)
+              File.delete(old_disk[:file])
+            end
+          end
+        end
+
+        def resize_disk(env, req_size)
+          driver = @machine.provider.driver
+          disks = identify_disks(driver)
+          target = disks.first    # TODO Shouldn't assume that the first disk is the one we want to resize
+
+          old_size = get_disk_size(driver, target)
+          if old_size < req_size
+            grow_vdi(driver, target, req_size)
+            new_size = get_disk_size(driver, target)
+            env[:ui].success "Resized disk: old #{old_size} MB, req #{req_size} MB, new #{new_size} MB"
+            env[:ui].success "You may need to resize the filesystem from within the guest."
+          elsif old_size > req_size
+            env[:ui].error "Disk cannot be decreased in size. #{req_size} MB requested but disk is already #{old_size} MB."
+          end
+        end
+
+        def clone_as_vdi(driver, src, dst)
+          driver.execute('clonemedium', src[:file], dst[:file], '--format', 'VDI')
+        end
+
+        def grow_vdi(driver, disk, size)
+          driver.execute('modifymedium', disk[:file], '--resize', size.to_s)
+        end
+
+        def attach_disk(driver, disk)
+          parts = disk[:name].split('-')
+          controller = parts[0]
+          port = parts[1]
+          device = parts[2]
+          driver.execute('storageattach', @machine.id, '--storagectl', controller, '--port', port, '--device', device, '--type', 'hdd',  '--medium', disk[:file])
+        end
+
+        def get_disk_size(driver, disk)
+          size = nil
+          driver.execute('showmediuminfo', disk[:file]).each_line do |line|
+            if line =~ /Capacity:\s+([0-9]+)\s+MB/
+              size = $1.to_i
+            end
+          end
+          size
+        end
+
+        def identify_disks(driver)
+          vminfo = get_vminfo(driver)
+          disks = []
+          disk_keys = vminfo.keys.select { |k| k =~ /-ImageUUID-/ }
+          disk_keys.each do |key|
+            uuid = vminfo[key]
+            disk_name = key.gsub(/-ImageUUID-/,'-')
+            disk_file = vminfo[disk_name]
+            disks << {
+              uuid: uuid,
+              name: disk_name,
+              file: disk_file
+            }
+          end
+          disks
+        end
+
+        def get_vminfo(driver)
+          vminfo = {}
+          driver.execute('showvminfo', @machine.id, '--machinereadable', retryable: true).split("\n").each do |line|
+            parts = line.partition('=')
+            key = unquoted(parts.first)
+            value = unquoted(parts.last)
+            vminfo[key] = value
+          end
+          vminfo
+        end
+
+        def generate_resizable_disk(disk)
+          src = disk[:file]
+          src_extn = File.extname(src)
+          src_path = File.dirname(src)
+          src_base = File.basename(src, src_extn)
+          dst = File.join(src_path, src_base) + '.vdi'
+          disk.merge({ uuid: "(undefined)", file: dst })
+        end
+
+        def disk_resizeable?(disk)
+          disk[:file].end_with? '.vdi'
+        end
+
+        def unquoted(s)
+          s.gsub(/\A"(.*)"\Z/,'\1')
+        end
+      end
+
+    end
+  end
+end

data/lib/vagrant/disksize/config.rb

@@ -0,0 +1,44 @@
+module Vagrant
+  module Disksize
+    class Config < Vagrant.plugin('2', :config)
+      attr_accessor :size
+
+      SIZE_REGEX = /^(?<number>[0-9]+)\s?(?<scale>KB|MB|GB|TB)?$/
+
+      def initialize
+        @size = UNSET_VALUE
+      end
+
+      def finalize!
+        # Convert from human to machine readable
+        size_str = @size.to_s.strip
+        matches = SIZE_REGEX.match(size_str)
+        if matches
+          number = matches[:number]
+          scale = matches[:scale]
+          @size = number.to_i
+          if scale
+            pos = %w(KB MB GB TB).index(scale)
+            mult = 1 << 10*(pos+1)
+            @size *= mult
+          end
+        end
+        # Convert size from bytes to MB
+        size_in_mb = (@size.to_i + (1<<20)-1) / (1<<20)
+        @size = size_in_mb
+      end
+
+      def validate(machine)
+        errors = []
+
+        unless @size.to_s =~ SIZE_REGEX
+          errors << "'#{@size}' is not a valid specification of disk size"
+        end
+
+        return { 'Disksize configuration' => errors }
+      end
+
+    end
+  end
+end
+

data/lib/vagrant/disksize/version.rb

@@ -0,0 +1,5 @@
+module Vagrant
+  module Disksize
+    VERSION = "0.1.0"
+  end
+end

data/vagrant-disksize.gemspec

@@ -0,0 +1,35 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'vagrant/disksize/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "vagrant-disksize"
+  spec.version       = Vagrant::Disksize::VERSION
+  spec.authors       = ["Simon Protheroe"]
+  spec.email         = ["protheroe@gmail.com"]
+
+  spec.summary       = %q{Vagrant plugin to resize VirtualBox disks}
+  spec.description   = %q{Vagrant plugin to resize VirtualBox disks at creation time}
+  spec.homepage      = "https://github.com/sprotheroe/vagrant-disksize"
+  spec.license       = "MIT"
+
+  # Prevent pushing this gem to RubyGems.org. To allow pushes either set the 'allowed_push_host'
+  # to allow pushing to a single host or delete this section to allow pushing to any host.
+  if spec.respond_to?(:metadata)
+    spec.metadata['allowed_push_host'] = "https://rubygems.org"
+  else
+    raise "RubyGems 2.0 or newer is required to protect against " \
+      "public gem pushes."
+  end
+
+  spec.files         = `git ls-files -z`.split("\x0").reject do |f|
+    f.match(%r{^(test|spec|features)/})
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.13"
+  spec.add_development_dependency "rake", "~> 10.0"
+end

metadata

@@ -0,0 +1,86 @@
+--- !ruby/object:Gem::Specification
+name: vagrant-disksize
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Simon Protheroe
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2016-09-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.13'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.13'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: Vagrant plugin to resize VirtualBox disks at creation time
+email:
+- protheroe@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- Vagrantfile
+- bin/console
+- bin/setup
+- lib/vagrant/disksize.rb
+- lib/vagrant/disksize/actions.rb
+- lib/vagrant/disksize/config.rb
+- lib/vagrant/disksize/version.rb
+- vagrant-disksize.gemspec
+homepage: https://github.com/sprotheroe/vagrant-disksize
+licenses:
+- MIT
+metadata:
+  allowed_push_host: https://rubygems.org
+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.4.8
+signing_key: 
+specification_version: 4
+summary: Vagrant plugin to resize VirtualBox disks
+test_files: []