vagrant-newdisk 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 6e6dc15fece47b2c2b659960109e4fc37fb31299
+  data.tar.gz: c201c5df7066de9a64065999ebff5174cf8d75d0
+SHA512:
+  metadata.gz: 0a3c564a7bf83b3b775415fefedb04fb1e3eb70a50bc53848325dffc9759b2215770be54b936c3733d812a29cd26a70982a5a08c12861689605cc59dfb1ab1bd
+  data.tar.gz: 695058b2614bc0a42efebe6ebef9db3d4102522dd663188fe6fe013046ef99215a72c1bcc06de48d9475bfb70ad59825de702cf6cd2fae6c9b8691f78d5ace81

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,11 @@
+source 'https://rubygems.org'
+
+group :development do
+  gem 'vagrant', git: 'https://github.com/mitchellh/vagrant.git'
+end
+
+group :plugins do
+  gem 'vagrant-newdisk', path: '.'
+end
+
+gem "rake", "~> 10.0"

data/LICENSE.txt

@@ -0,0 +1,19 @@
+The MIT License (MIT)
+
+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,31 @@
+# vagrant-newdisk
+
+A Vagrant plugin to add a disks in VirtualBox
+
+
+## Installation
+
+
+```shell
+vagrant plugin install vagrant-newdisk
+```
+
+## Usage
+
+Set the size you want for your disk in your Vagrantfile. For example
+
+```ruby
+Vagrant.configure('2') do |config|
+  config.vm.box = 'ubuntu/xenial64'
+  config.newdisk.size = 10 * 1024 # size in megabytes
+  config.newdisk.path = "/tmp/your-file.vdi"
+end
+```
+## Limitations
+
+At present only one disk can be added to the first controller.
+
+## 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,3 @@
+require "rubygems"
+require "bundler/setup"
+Bundler::GemHelper.install_tasks

data/Vagrantfile

@@ -0,0 +1,74 @@
+# -*- 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.newdisk.path = "test-disk.vdi"
+  config.newdisk.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/newdisk"
+
+# 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/newdisk/actions.rb

@@ -0,0 +1,101 @@
+module Vagrant
+  module Newdisk
+    class Action
+
+      class NewDisk
+        def initialize(app, env)
+          @app = app
+          @machine = env[:machine]
+          @enabled = true
+          @ui = env[:ui]
+          if @machine.provider.to_s !~ /VirtualBox/
+            @enabled = false
+            env[:ui].error "The vagrant-newdisk plugin only supports VirtualBox at present."
+          end
+        end
+
+        def call(env)
+          # Create the disk before boot
+          if @enabled
+            size = @machine.config.newdisk.size
+            path = @machine.config.newdisk.path
+            env[:ui].error("size = #{size}")
+            env[:ui].error("path = #{path}")
+            new_disk(env, path, size)
+          end
+
+          # Allow middleware chain to continue so VM is booted
+          @app.call(env)
+        end
+
+        private
+
+        def new_disk(env, path, size)
+          driver = @machine.provider.driver
+          create_disk(driver, path, size)
+          attach_disk(driver, path)
+        end
+
+        def attach_disk(driver, path)
+          disk = find_place_for_new_disk(driver)
+          driver.execute('storageattach', @machine.id, '--storagectl', disk[:controller],
+                         '--port', disk[:port].to_s, '--device', disk[:device].to_s, '--type', 'hdd',
+                         '--medium', path)
+        end
+
+        def find_place_for_new_disk(driver)
+          disks = get_disks(driver)
+          @ui.error "disks = #{disks.to_s}"
+          controller = disks.first[:controller]
+          disks = disks.select { |disk| disk[:controller] == controller }
+          port = disks.map { |disk| disk[:port] }.max
+          disks = disks.select { |disk| disk[:port] == port }
+          max_device = disks.map { |disk| disk[:device] }.max
+
+          @ui.error "last disks: #{disks.to_s}"
+          new_disk = {:controller => controller, :port => port.to_i, :device => max_device.to_i + 1}
+          @ui.error "new disk: #{new_disk.to_s}"
+          new_disk
+        end
+
+        def get_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-/,'-')
+            parts = disk_name.split('-')
+            disks << {
+              controller: parts[0],
+              port: parts[1].to_i,
+              device: parts[2].to_i
+            }
+          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 create_disk(driver, path, size)
+          driver.execute('createhd', '--filename', path, '--size', size.to_s)
+        end
+
+        def unquoted(s)
+          s.gsub(/\A"(.*)"\Z/,'\1')
+        end
+      end
+
+    end
+  end
+end

data/lib/vagrant/newdisk/config.rb

@@ -0,0 +1,23 @@
+module Vagrant
+  module Newdisk
+    class Config < Vagrant.plugin('2', :config)
+      attr_accessor :size
+      attr_accessor :path
+
+      def initialize
+        @size = UNSET_VALUE
+        @path = UNSET_VALUE
+      end
+
+      def finalize!
+        return if @size == UNSET_VALUE
+      end
+
+      def validate(machine)
+        errors = []
+        return { 'Newdisk configuration' => errors }
+      end
+    end
+  end
+end
+

data/lib/vagrant/newdisk/version.rb

@@ -0,0 +1,5 @@
+module Vagrant
+  module Newdisk
+    VERSION = "0.0.1"
+  end
+end

data/lib/vagrant/newdisk.rb

@@ -0,0 +1,29 @@
+begin
+  require 'vagrant'
+rescue LoadError
+  raise 'The vagrant-newdisk plugin must be run within vagrant.'
+end
+
+
+module Vagrant
+  module Newdisk
+    class Plugin < Vagrant.plugin('2')
+
+      name 'vagrant-newdisk'
+
+      description <<-DESC
+      Provides the ability to add new VirtualBox disk at creation time.
+      DESC
+
+      config 'newdisk' do
+        require_relative 'newdisk/config'
+        Config
+      end
+
+      action_hook(:newdisk, :machine_action_up) do |hook|
+        require_relative 'newdisk/actions'
+        hook.before(VagrantPlugins::ProviderVirtualBox::Action::Boot, Action::NewDisk)
+      end
+    end
+  end
+end

data/vagrant-newdisk.gemspec

@@ -0,0 +1,36 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'vagrant/newdisk/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "vagrant-newdisk"
+  spec.version       = Vagrant::Newdisk::VERSION
+  spec.authors       = ["Yavor Nikolov"]
+  spec.email         = ["nikolov.javor@gmail.com"]
+
+  spec.summary       = %q{Vagrant plugin to add VirtualBox disk}
+  spec.description   = %q{Vagrant plugin to add VirtualBox disk at creation time}
+  spec.homepage      = "https://github.com/dbfit/vagrant-newdisk"
+  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-newdisk
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Yavor Nikolov
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2016-10-22 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 add VirtualBox disk at creation time
+email:
+- nikolov.javor@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- Vagrantfile
+- bin/console
+- bin/setup
+- lib/vagrant/newdisk.rb
+- lib/vagrant/newdisk/actions.rb
+- lib/vagrant/newdisk/config.rb
+- lib/vagrant/newdisk/version.rb
+- vagrant-newdisk.gemspec
+homepage: https://github.com/dbfit/vagrant-newdisk
+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.6.4
+signing_key: 
+specification_version: 4
+summary: Vagrant plugin to add VirtualBox disk
+test_files: []