vagrant-smartos-zones 0.0.1.pre.21

data/lib/vagrant/smartos/zones/cap/create_gz_vnic.rb

@@ -0,0 +1,45 @@
+require 'vagrant/smartos/zones/cap/base'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Cap
+        class CreateGZVnic < Base
+          cap_method :create_gz_vnic
+
+          def execute
+            return unless machine.communicate.test('test -d %s' % vm_tmp_folder)
+
+            ui.info 'Installing vnic in global zone'
+            sudo = machine.config.smartos.suexec_cmd
+
+            machine.communicate.upload(create_vnic_script, '%s/create_gz_vnic' % vm_tmp_folder)
+            machine.communicate.upload(vnic_smf_manifest, '%s/create-gz-vnic.xml' % vm_tmp_folder)
+
+            machine.communicate.execute("#{sudo} mv %s/create_gz_vnic /opt/custom/method" % vm_tmp_folder)
+            machine.communicate.execute("#{sudo} mv %s/create-gz-vnic.xml /opt/custom/smf" % vm_tmp_folder)
+
+            machine.communicate.execute("#{sudo} svccfg import /opt/custom/smf/create-gz-vnic.xml")
+            machine.communicate.execute("#{sudo} svcadm enable -s create-gz-vnic")
+          end
+
+          def local_files_folder
+            File.expand_path('../../../../../../files', __FILE__)
+          end
+
+          def create_vnic_script
+            File.join(local_files_folder, 'gz_vnic', 'create_gz_vnic')
+          end
+
+          def vnic_smf_manifest
+            File.join(local_files_folder, 'smf', 'create-gz-vnic.xml')
+          end
+
+          def vm_tmp_folder
+            '/usbkey/vagrant'
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/cap/imgadm_import.rb

@@ -0,0 +1,30 @@
+require 'vagrant/smartos/zones/cap/base'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Cap
+        class ImgadmImport < Base
+          cap_method :imgadm_import
+
+          def execute
+            return ui.info 'No zone image set, skipping import' unless image
+
+            installed = machine.communicate.gz_test("#{sudo} imgadm get #{image}")
+
+            ui.info "Checking for zone image #{image}: #{installed ? 'installed' : 'not installed'}"
+            return if installed
+
+            ui.info '  Importing...'
+            machine.communicate.gz_execute("#{sudo} imgadm sources -a https://updates.joyent.com")
+            machine.communicate.gz_execute("#{sudo} imgadm import #{image}")
+          end
+
+          def image
+            machine.config.zone.image
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/cap/platform_image/install.rb

@@ -0,0 +1,17 @@
+require 'vagrant/smartos/zones/util/platform_images'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Cap
+        module PlatformImage
+          class Install
+            def self.platform_image_install(env, image)
+              Zones::Util::PlatformImages.new(env).install(image)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/cap/platform_image/latest.rb

@@ -0,0 +1,19 @@
+require 'vagrant/smartos/zones/util/platform_images'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Cap
+        module PlatformImage
+          class Latest
+            def self.platform_image_latest(env)
+              platform_image = Zones::Util::PlatformImages.new(env).latest
+              env.ui.info 'Unable to find platform image' unless platform_image
+              env.ui.info platform_image
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/cap/platform_image/list.rb

@@ -0,0 +1,17 @@
+require 'vagrant/smartos/zones/util/platform_images'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Cap
+        module PlatformImage
+          class List
+            def self.platform_image_list(env)
+              Zones::Util::PlatformImages.new(env).list
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/cap/zone/base.rb

@@ -0,0 +1,32 @@
+require 'vagrant/smartos/zones/cap/base'
+require 'vagrant/smartos/zones/models/zone'
+require 'vagrant/smartos/zones/util/zone_info'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Cap
+        module Zone
+          class Base < Vagrant::Smartos::Zones::Cap::Base
+            def zone_exists?
+              name = machine.config.zone.name
+              machine.communicate.gz_test("#{sudo} vmadm list -H | awk '{print $5}' | grep #{name}")
+            end
+
+            def zone_valid?
+              machine.config.zone && machine.config.zone.image && machine.config.zone.name
+            end
+
+            def zone_info
+              @zone_info ||= Util::ZoneInfo.new(machine)
+            end
+
+            def zone
+              @zone ||= Models::Zone.find(machine, machine.config.zone.name)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/cap/zone/create.rb

@@ -0,0 +1,45 @@
+require 'vagrant/smartos/zones/cap/zone/base'
+require 'vagrant/smartos/zones/util/zone_json'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Cap
+        module Zone
+          class Create < Base
+            cap_method :zone__create
+
+            def execute
+              return warn_zone_config unless zone_valid?
+              return update_zone if zone_exists?
+              create_zone
+            end
+
+            def create_zone
+              zone_info.create(machine.config.zone.name)
+            end
+
+            def update_zone
+              ui.info "Zone #{machine.config.zone.name} exists"
+              zone_info.update(machine.config.zone.name)
+            end
+
+            def warn_zone_config
+              ui.info 'No zone configured, skipping'
+              ui.info '   add the following to your Vagrantfile to configure a local zone:'
+              ui.info "      config.zone.name      = 'my-zone'"
+              ui.info "      config.zone.image     = 'uuid'"
+              ui.info "      config.zone.brand     = 'joyent'"
+              ui.info '      config.zone.memory    = 2048'
+              ui.info '      config.zone.disk_size = 5'
+            end
+
+            def zone_json
+              Util::ZoneJson.new(machine).to_json
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/cap/zone/start.rb

@@ -0,0 +1,29 @@
+require 'vagrant/smartos/zones/cap/zone/base'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Cap
+        module Zone
+          class Start < Base
+            cap_method :zone__start
+
+            def execute
+              return unless zone_valid?
+              return unless zone_exists?
+
+              start_zone
+            end
+
+            def start_zone
+              return if zone.running?
+              name = machine.config.zone.name
+              ui.info "Starting zone #{name}"
+              zone.start
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/cap/zone/stop.rb

@@ -0,0 +1,29 @@
+require 'vagrant/smartos/zones/cap/zone/base'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Cap
+        module Zone
+          class Stop < Base
+            cap_method :zone__stop
+
+            def execute
+              return unless zone_valid?
+              return unless zone_exists?
+
+              stop_zone
+            end
+
+            def stop_zone
+              return unless zone.running?
+              name = machine.config.zone.name
+              ui.info "Stopping zone #{name}"
+              zone.stop
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/cap/zone_gate/enable.rb

@@ -0,0 +1,20 @@
+require 'vagrant/smartos/zones/cap/base'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Cap
+        module ZoneGate
+          class Enable < Cap::Base
+            cap_method :zone_gate_enable
+
+            def execute
+              ui.info 'Enabling ZoneGate'
+              machine.communicate.gz_execute("#{sudo} svcadm enable -s zonegate")
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/cap/zone_gate/install.rb

@@ -0,0 +1,46 @@
+require 'vagrant/smartos/zones/cap/base'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Cap
+        module ZoneGate
+          class Install < Cap::Base
+            cap_method :zone_gate_install
+
+            LOCAL_FILES_FOLDER = File.expand_path('../../../../../../../files', __FILE__)
+            LOCAL_ZONEGATE_FOLDER = File.join(LOCAL_FILES_FOLDER, 'zonegate')
+            LOCAL_SMF_MANIFEST = File.join(LOCAL_FILES_FOLDER, 'smf', 'zonegate.xml')
+            SMF_TMP_FOLDER = '/usbkey/vagrant'
+            SMF_FOLDER = '/opt/custom/smf'
+            ZONEGATE_FOLDER = '/opt/custom/method'
+
+            def execute
+              ui.info 'Installing ZoneGate'
+
+              create_zonegate_folder
+              upload_zonegate
+              install_zonegate
+            end
+
+            def create_zonegate_folder
+              machine.communicate.execute("#{sudo} mkdir -p %s/zonegate" % ZONEGATE_FOLDER)
+              machine.communicate.execute("#{sudo} chown vagrant:other %s/zonegate" % ZONEGATE_FOLDER)
+            end
+
+            def upload_zonegate
+              machine.communicate.upload(LOCAL_ZONEGATE_FOLDER, ZONEGATE_FOLDER)
+              machine.communicate.upload(LOCAL_SMF_MANIFEST, '%s/zonegate.xml' % SMF_TMP_FOLDER)
+            end
+
+            def install_zonegate
+              machine.communicate.execute("#{sudo} mv %s/zonegate.xml %s/zonegate.xml" % [SMF_TMP_FOLDER, SMF_FOLDER])
+              machine.communicate.execute("#{sudo} chown root:root %s/zonegate.xml" % SMF_FOLDER)
+              machine.communicate.execute("#{sudo} svccfg import %s/zonegate.xml" % SMF_FOLDER)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/commands/dataset.rb

@@ -0,0 +1,91 @@
+require 'vagrant'
+require_relative 'multi_command'
+require 'vagrant/smartos/zones/errors'
+require 'vagrant/smartos/zones/models/dataset'
+require 'vagrant/smartos/zones/models/zone'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Command
+        class Dataset < Vagrant.plugin('2', :command)
+          include MultiCommand
+
+          attr_accessor :host, :ui
+
+          COMMANDS = %w(list create delete install).freeze
+
+          OPTION_PARSER = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant dataset [subcommand] [options]'
+            o.separator ''
+            o.separator 'Commands:'
+            o.separator '  create [zone] [name]         create a local dataset [name] from running zone [zone]'
+            o.separator '  delete [name]                delete a local dataset [name]'
+            o.separator '  list                         show all locally installed datasets'
+            o.separator '  install [url]                download a dataset to the local machine'
+            o.separator ''
+            o.separator 'Options:'
+            o.separator ''
+          end
+
+          def self.synopsis
+            'Manage local SmartOS datasets'
+          end
+
+          def execute
+            process_subcommand
+          end
+
+          private
+
+          def host
+            @env.host
+          end
+
+          def create(zonename, dataset)
+            with_zone(zonename) do |zone|
+              Models::Dataset.create(dataset, zone)
+            end
+          end
+
+          def delete(dataset_name)
+            return ui.warn('Unable to delete datasets') unless host.capability?(:dataset_delete)
+            host.capability(:dataset_delete, dataset_name)
+          end
+
+          def list(*_args)
+            return ui.warn('Unable to list datasets') unless host.capability?(:dataset_list)
+            host.capability(:dataset_list)
+          end
+
+          def install(url)
+            if url.nil?
+              ui.warn 'No url given'
+              ui.warn ''
+              ui.warn OPTION_PARSER.to_s
+              exit 1
+            end
+
+            return ui.warn('Unable to install dataset') unless host.capability?(:dataset_install)
+            host.capability(:dataset_install, url)
+          end
+
+          def ui
+            @env.ui
+          end
+
+          def with_zone(name)
+            with_target_vms('default', single_target: true) do |machine|
+              Models::Zone.find(machine, name).tap do |zone|
+                yield zone
+              end
+            end
+          rescue ZoneNotFound
+            ui.warn(I18n.t('vagrant.smartos.zones.warning.zone_not_found',
+                           name: name), prefix: false)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/commands/global_zone.rb

@@ -0,0 +1,43 @@
+require 'vagrant'
+require 'vagrant/smartos/zones/util/global_zone/ssh_info'
+require 'vagrant/util/ssh'
+require_relative 'multi_command'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Command
+        class GlobalZone < Vagrant.plugin('2', :command)
+          include MultiCommand
+
+          COMMANDS = %w(ssh)
+
+          OPTION_PARSER = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant global-zone [command]'
+            o.separator ''
+            o.separator 'Commands:'
+            o.separator '  ssh        ssh into the global zone'
+            o.separator ''
+            o.separator 'Options:'
+            o.separator ''
+          end
+
+          def self.synopsis
+            'View and interact with the SmartOS global zone'
+          end
+
+          def execute
+            process_subcommand
+          end
+
+          def ssh(*_args)
+            with_target_vms('default', single_target: true) do |machine|
+              ssh_info = Util::GlobalZone::SSHInfo.new(machine.provider, machine.config, machine.env).to_hash
+              Vagrant::Util::SSH.exec(ssh_info)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/commands/multi_command.rb

@@ -0,0 +1,46 @@
+module Vagrant
+  module Smartos
+    module Zones
+      module Command
+        # Requires
+        #
+        # Define a COMMANDS constant, which is an array of method names:
+        #
+        #   COMMANDS = %w(subcommand other_subcommand)
+        #
+        # Define an OPTION_PARSER constant, which is an instance of an
+        # OptionParser
+        #
+        module MultiCommand
+          def process_subcommand
+            args = parse_options(option_parser)
+            exit unless args
+
+            command = args.shift
+            command_method = subcommands.find { |c| c == command }
+
+            unless command_method
+              ui.warn option_parser.to_s, prefix: false
+              exit 1
+            end
+
+            send command, *args
+          end
+
+          def fail_options!
+            ui.warn option_parser.to_s, prefix: false
+            exit 1
+          end
+
+          def option_parser
+            self.class.const_get('OPTION_PARSER')
+          end
+
+          def subcommands
+            self.class.const_get('COMMANDS')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/commands/smartos.rb

@@ -0,0 +1,70 @@
+require 'vagrant'
+require_relative 'multi_command'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Command
+        class Smartos < Vagrant.plugin('2', :command)
+          include MultiCommand
+
+          attr_accessor :host, :ui
+
+          COMMANDS = %w(latest list install).freeze
+
+          OPTION_PARSER = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant smartos [name]'
+            o.separator ''
+            o.separator 'Commands:'
+            o.separator '  latest             show the latest remote SmartOS platform image'
+            o.separator '  list               show installed SmartOS platform images'
+            o.separator '  install [image]    install SmartOS platform image'
+            o.separator ''
+            o.separator 'Options:'
+            o.separator ''
+          end
+
+          def self.synopsis
+            'Manage SmartOS platform images'
+          end
+
+          def execute
+            process_subcommand
+          end
+
+          private
+
+          def host
+            @env.host
+          end
+
+          def install(image)
+            if image.nil?
+              ui.warn 'No image given'
+              ui.warn ''
+              ui.warn OPTION_PARSER.to_s
+              exit 1
+            end
+
+            return ui.warn('Unable to install platform image') unless host.capability?(:platform_image_install)
+            host.capability(:platform_image_install, image)
+          end
+
+          def latest(*_args)
+            return ui.warn('Unable to show latest platform image') unless host.capability?(:platform_image_latest)
+            host.capability(:platform_image_latest)
+          end
+
+          def list(*_args)
+            return ui.warn('Unable to list platform image') unless host.capability?(:platform_image_list)
+            host.capability(:platform_image_list)
+          end
+
+          def ui
+            @env.ui
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant/smartos/zones/commands/zlogin.rb

@@ -0,0 +1,35 @@
+require 'vagrant'
+require 'vagrant/smartos/zones/util/global_zone/ssh_info'
+require 'vagrant/smartos/zones/models/zone'
+require 'vagrant/util/ssh'
+
+module Vagrant
+  module Smartos
+    module Zones
+      module Command
+        class Zlogin < Vagrant.plugin('2', :command)
+          OPTION_PARSER = OptionParser.new do |o|
+            o.banner = 'Usage: vagrant zlogin [name]'
+          end
+
+          def self.synopsis
+            'Log into a SmartOS local zone'
+          end
+
+          def execute
+            argv = parse_options(OPTION_PARSER)
+            return unless argv
+
+            zone_alias = argv.shift
+
+            with_target_vms('default', single_target: true) do |machine|
+              ssh_info = Util::GlobalZone::SSHInfo.new(machine.provider, machine.config, machine.env).to_hash
+              zone = Models::Zone.find(machine, zone_alias)
+              Vagrant::Util::SSH.exec(ssh_info, extra_args: ['-t', "pfexec zlogin -l vagrant #{zone.uuid}"])
+            end
+          end
+        end
+      end
+    end
+  end
+end