fission 0.2.0 → 0.3.0.beta.1

data/lib/fission/command/stop.rb

@@ -7,13 +7,13 @@ module Fission
       end
 
       def execute
-        unless args.count == 1
+        unless @args.count == 1
           Fission.ui.output self.class.help
           Fission.ui.output ""
           Fission.ui.output_and_exit "Incorrect arguments for stop command", 1
         end
 
-        vm_name = args.first
+        vm_name = @args.first
 
         unless Fission::VM.exists? vm_name
           Fission.ui.output_and_exit "Unable to find the VM #{vm_name} (#{Fission::VM.path(vm_name)})", 1 

data/lib/fission/command/suspend.rb

@@ -4,34 +4,50 @@ module Fission
 
       def initialize(args=[])
         super
+        @options.all = false
       end
 
       def execute
-        unless args.count == 1
+        option_parser.parse! @args
+
+        if @args.count != 1 && !@options.all
           Fission.ui.output self.class.help
           Fission.ui.output ""
           Fission.ui.output_and_exit "Incorrect arguments for suspend command", 1
         end
 
-        vm_name = args.first
-
-        unless Fission::VM.exists? vm_name
-          Fission.ui.output_and_exit "Unable to find the VM #{vm_name} (#{Fission::VM.path(vm_name)})", 1 
+        vms_to_suspend.each do |vm_name|
+          Fission.ui.output "Suspending '#{vm_name}'"
+          Fission::VM.new(vm_name).suspend
         end
+      end
 
-        unless VM.all_running.include?(vm_name)
-          Fission.ui.output ''
-          Fission.ui.output_and_exit "VM '#{vm_name}' is not running", 0
+      def vms_to_suspend
+        if @options.all
+          vms_to_suspend = VM.all_running
+        else
+          vm_name = @args.first
+          unless Fission::VM.exists? vm_name
+            Fission.ui.output ''
+            Fission.ui.output_and_exit "Unable to find the VM #{vm_name} (#{Fission::VM.path(vm_name)})", 1
+          end
+
+          unless VM.all_running.include?(vm_name)
+            Fission.ui.output ''
+            Fission.ui.output_and_exit "VM '#{vm_name}' is not running", 1
+          end
+
+          vms_to_suspend = [vm_name]
         end
-
-        Fission.ui.output "suspending '#{vm_name}'"
-        @vm = Fission::VM.new vm_name
-        @vm.suspend
       end
 
       def option_parser
         optparse = OptionParser.new do |opts|
-          opts.banner = "\nsuspend usage: fission suspend vm"
+          opts.banner = "\nsuspend usage: fission suspend [vm | --all]"
+
+          opts.on '--all', 'Suspend all running VMs' do
+            @options.all = true
+          end
         end
 
         optparse

data/lib/fission/config.rb

@@ -14,6 +14,8 @@ module Fission
 
       @attributes['vmrun_bin'] = '/Library/Application Support/VMware Fusion/vmrun'
       @attributes['vmrun_cmd'] = "#{@attributes['vmrun_bin'].gsub(' ', '\ ')} -T fusion"
+      @attributes['plist_file'] = File.expand_path('~/Library/Preferences/com.vmware.fusion.plist')
+      @attributes['gui_bin'] = File.expand_path('/Applications/VMware Fusion.app/Contents/MacOS/vmware')
     end
 
     private

data/lib/fission/fusion.rb

@@ -0,0 +1,13 @@
+module Fission
+  class Fusion
+
+    def self.is_running?
+      command = "ps -ef | grep -v grep | grep -c "
+      command << "#{Fission.config.attributes['gui_bin'].gsub(' ', '\ ')} 2>&1"
+      output = `#{command}`
+
+      output.strip.to_i > 0 ? true : false
+    end
+
+  end
+end

data/lib/fission/metadata.rb

@@ -0,0 +1,39 @@
+module Fission
+  class Metadata
+
+    require 'cfpropertylist'
+
+    attr_accessor :content
+
+    def self.delete_vm_info(vm_path)
+      metadata = new
+      metadata.load
+      metadata.delete_vm_restart_document(vm_path)
+      metadata.delete_vm_favorite_entry(vm_path)
+      metadata.save
+    end
+
+    def load
+      raw_data = CFPropertyList::List.new :file => Fission.config.attributes['plist_file']
+      @content = CFPropertyList.native_types raw_data.value
+    end
+
+    def save
+      new_content = CFPropertyList::List.new
+      new_content.value = CFPropertyList.guess @content
+      new_content.save Fission.config.attributes['plist_file'],
+                       CFPropertyList::List::FORMAT_BINARY
+    end
+
+    def delete_vm_restart_document(vm_path)
+      if @content.has_key?('PLRestartDocumentPaths')
+        @content['PLRestartDocumentPaths'].delete_if { |p| p == vm_path }
+      end
+    end
+
+    def delete_vm_favorite_entry(vm_path)
+      @content['VMFavoritesListDefaults2'].delete_if { |vm| vm['path'] == vm_path }
+    end
+
+  end
+end

data/lib/fission/version.rb

@@ -1,3 +1,3 @@
 module Fission
-  VERSION = "0.2.0"
+  VERSION = "0.3.0.beta.1"
 end

data/lib/fission/vm.rb

@@ -6,8 +6,52 @@ module Fission
       @name = name
     end
 
-    def start
-      command = "#{Fission.config.attributes['vmrun_cmd']} start #{conf_file.gsub ' ', '\ '} gui 2>&1"
+    def create_snapshot(name)
+      command = "#{Fission.config.attributes['vmrun_cmd']} snapshot #{conf_file.gsub ' ', '\ '} \"#{name}\" 2>&1"
+      output = `#{command}`
+
+      if $?.exitstatus == 0
+        Fission.ui.output "Snapshot '#{name}' created"
+      else
+        Fission.ui.output "There was an error creating the snapshot."
+        Fission.ui.output_and_exit "The error was:\n#{output}", 1
+      end
+    end
+
+    def revert_to_snapshot(name)
+      command = "#{Fission.config.attributes['vmrun_cmd']} revertToSnapshot #{conf_file.gsub ' ', '\ '} \"#{name}\" 2>&1"
+      output = `#{command}`
+
+      if $?.exitstatus == 0
+        Fission.ui.output "Reverted to snapshot '#{name}'"
+      else
+        Fission.ui.output "There was an error reverting to the snapshot."
+        Fission.ui.output_and_exit "The error was:\n#{output}", 1
+      end
+    end
+
+    def snapshots
+      command = "#{Fission.config.attributes['vmrun_cmd']} listSnapshots #{conf_file.gsub ' ', '\ '} 2>&1"
+      output = `#{command}`
+
+      if $?.exitstatus == 0
+        snaps = output.split("\n").select { |s| !s.include? 'Total snapshots:' }
+        snaps.map { |s| s.strip }
+      else
+        Fission.ui.output "There was an error getting the list of snapshots."
+        Fission.ui.output_and_exit "The error was:\n#{output}", 1
+      end
+    end
+
+    def start(args={})
+      command = "#{Fission.config.attributes['vmrun_cmd']} start #{conf_file.gsub ' ', '\ '} "
+
+      if !args[:headless].blank? && args[:headless]
+        command << "nogui 2>&1"
+      else
+        command << "gui 2>&1"
+      end
+
       output = `#{command}`
 
       if $?.exitstatus == 0
@@ -40,7 +84,24 @@ module Fission
     end
 
     def conf_file
-      File.join self.class.path(@name), "#{@name}.vmx"
+      vmx_path = File.join(self.class.path(@name), "*.vmx")
+      conf_files = Dir.glob(vmx_path)
+
+      case conf_files.count
+      when 0
+        Fission.ui.output_and_exit "Unable to find a config file for VM '#{@name}' (in '#{vmx_path}')", 1
+      when 1
+        conf_files.first
+      else
+        if conf_files.include?(File.join(File.dirname(vmx_path), "#{@name}.vmx"))
+          File.join(File.dirname(vmx_path), "#{@name}.vmx")
+        else
+          output = "Multiple config files found for VM '#{@name}' ("
+          output << conf_files.sort.map { |f| "'#{File.basename(f)}'" }.join(', ')
+          output << " in '#{File.dirname(vmx_path)}')"
+          Fission.ui.output_and_exit output, 1
+        end
+      end
     end
 
     def self.all
@@ -56,21 +117,15 @@ module Fission
 
       output = `#{command}`
 
-      vms = []
-
       if $?.exitstatus == 0
-        output.split("\n").each do |vm|
-          if vm.include?('.vmx')
-            if File.exists?(vm) && (File.extname(vm) == '.vmx')
-              vms << File.basename(vm, '.vmx')
-            end
-          end
+        vms = output.split("\n").select do |vm|
+          vm.include?('.vmx') && File.exists?(vm) && File.extname(vm) == '.vmx'
         end
+
+        vms.map { |vm| File.basename(File.dirname(vm), '.vmwarevm') }
       else
         Fission.ui.output_and_exit "Unable to determine the list of running VMs", 1
       end
-
-      vms
     end
 
     def self.exists?(vm_name)
@@ -90,15 +145,45 @@ module Fission
       update_config source_vm, target_vm
     end
 
+    def self.delete(vm_name)
+      Fission.ui.output "Deleting vm #{vm_name}"
+      FileUtils.rm_rf path(vm_name)
+      Fission::Metadata.delete_vm_info(path(vm_name))
+    end
+
     private
     def self.rename_vm_files(from, to)
       files_to_rename(from, to).each do |file|
-        FileUtils.mv File.join(path(to), file), File.join(path(to), file.gsub(from, to))
+        text_to_replace = File.basename(file, File.extname(file))
+
+        if File.extname(file) == '.vmdk'
+          if file.match /\-s\d+\.vmdk/
+            text_to_replace = file.partition(/\-s\d+.vmdk/).first
+          end
+        end
+
+        unless File.exists?(File.join(path(to), file.gsub(text_to_replace, to)))
+          FileUtils.mv File.join(path(to), file),
+                       File.join(path(to), file.gsub(text_to_replace, to))
+        end
       end
     end
 
     def self.files_to_rename(from, to)
-      Dir.entries(path(to)).select { |f| f.include?(from) }
+      files_which_match_source_vm = []
+      other_files = []
+
+      Dir.entries(path(to)).each do |f|
+        unless f == '.' || f == '..'
+          f.include?(from) ? files_which_match_source_vm << f : other_files << f
+        end
+      end
+
+      files_which_match_source_vm + other_files
+    end
+
+    def self.vm_file_extensions
+      ['.nvram', '.vmdk', '.vmem', '.vmsd', '.vmss', '.vmx', '.vmxf']
     end
 
     def self.update_config(from, to)

data/spec/fission/cli_spec.rb

@@ -1,19 +1,20 @@
 require File.expand_path('../../spec_helper.rb', __FILE__)
 
 describe Fission::CLI do
-  before :each do
+  before do
     @string_io = StringIO.new
     Fission.stub!(:ui).and_return(Fission::UI.new(@string_io))
   end
 
   describe 'self.commands' do
     it 'should return the list of commands' do
-      Fission::CLI.commands.should == ['clone', 'start', 'status', 'stop', 'suspend']
+      Fission::CLI.commands.should == ['clone', 'delete', 'snapshot create',
+                                       'snapshot list', 'snapshot revert',
+                                       'start', 'status', 'stop', 'suspend']
     end
   end
 
   describe 'execute' do
-
     describe 'with no arguments' do
       it 'should output the usage info' do
         lambda {
@@ -35,7 +36,6 @@ describe Fission::CLI do
 
         end
       end
-
     end
 
     describe '-h or --help' do
@@ -48,11 +48,10 @@ describe Fission::CLI do
           @string_io.string.should match /Usage/
         end
       end
-
     end
 
     describe 'with the sub command' do
-      before :each do
+      before do
         @cmd_mock = mock('cmd')
         @cmd_mock.should_receive(:execute)
       end
@@ -60,23 +59,74 @@ describe Fission::CLI do
       describe 'clone' do
         it "should try to clone the vm" do
           Fission::Command::Clone.stub!(:help).and_return('')
-          Fission::Command::Clone.should_receive(:new).with(['foo', 'bar']).and_return(@cmd_mock)
+          Fission::Command::Clone.should_receive(:new).with(['foo', 'bar']).
+                                                       and_return(@cmd_mock)
           Fission::CLI.execute ['clone', 'foo', 'bar']
         end
+
+        it 'should try to clone the vm and start it' do
+          Fission::Command::Clone.stub!(:help).and_return('')
+          Fission::Command::Clone.should_receive(:new).with(['foo', 'bar', '--start']).
+                                                       and_return(@cmd_mock)
+          Fission::CLI.execute ['clone', 'foo', 'bar', '--start']
+        end
+      end
+
+      describe 'snapshot create' do
+        it "should create a snapshot" do
+          Fission::Command::SnapshotCreate.stub!(:help).and_return('')
+          Fission::Command::SnapshotCreate.should_receive(:new).
+                                           with(['foo', 'snap1']).
+                                           and_return(@cmd_mock)
+          Fission::CLI.execute ['snapshot', 'create', 'foo', 'snap1']
+        end
+      end
+
+      describe 'snapshot list' do
+        it "should list the snapshots" do
+          Fission::Command::SnapshotList.stub!(:help).and_return('')
+          Fission::Command::SnapshotList.should_receive(:new).
+                                         with([]).
+                                         and_return(@cmd_mock)
+          Fission::CLI.execute ['snapshot', 'list']
+        end
+      end
+
+      describe 'snapshot revert' do
+        it "should revert to the snapshots" do
+          Fission::Command::SnapshotRevert.stub!(:help).and_return('')
+          Fission::Command::SnapshotRevert.should_receive(:new).
+                                           with(['foo', 'snap1']).
+                                           and_return(@cmd_mock)
+          Fission::CLI.execute ['snapshot', 'revert', 'foo', 'snap1']
+        end
       end
 
       describe 'start' do
         it "should try to start the vm" do
           Fission::Command::Start.stub!(:help).and_return('')
-          Fission::Command::Start.should_receive(:new).with(['foo']).and_return(@cmd_mock)
+          Fission::Command::Start.should_receive(:new).
+                                  with(['foo']).
+                                  and_return(@cmd_mock)
           Fission::CLI.execute ['start', 'foo']
         end
+
+        it 'should try to start the vm headless' do
+          Fission::Command::Start.stub!(:help).and_return('')
+          Fission::Command::Start.should_receive(:new).
+                                  with(['foo', '--headless']).
+                                  and_return(@cmd_mock)
+          Fission::CLI.execute ['start', 'foo', '--headless']
+
+        end
       end
 
       describe 'status' do
         it "should try to get the status for all VMs" do
           Fission::Command::Status.stub!(:help).and_return('')
-          Fission::Command::Status.should_receive(:new).and_return(@cmd_mock)
+          Fission::Command::Status.should_receive(:new).
+                                   with([]).
+                                   and_return(@cmd_mock)
           Fission::CLI.execute ['status']
         end
       end
@@ -84,7 +134,9 @@ describe Fission::CLI do
       describe 'stop' do
         it "should try to stop the vm" do
           Fission::Command::Stop.stub!(:help).and_return('')
-          Fission::Command::Stop.should_receive(:new).with(['foo']).and_return(@cmd_mock)
+          Fission::Command::Stop.should_receive(:new).
+                                 with(['foo']).
+                                 and_return(@cmd_mock)
           Fission::CLI.execute ['stop', 'foo']
         end
       end
@@ -92,9 +144,37 @@ describe Fission::CLI do
       describe 'suspend' do
         it "should try to suspend the vm" do
           Fission::Command::Suspend.stub!(:help).and_return('')
-          Fission::Command::Suspend.should_receive(:new).with(['foo']).and_return(@cmd_mock)
+          Fission::Command::Suspend.should_receive(:new).
+                                    with(['foo']).
+                                    and_return(@cmd_mock)
           Fission::CLI.execute ['suspend', 'foo']
         end
+
+        it 'should try to suspend all of vms' do
+          Fission::Command::Suspend.stub!(:help).and_return('')
+          Fission::Command::Suspend.should_receive(:new).
+                                    with(['--all']).
+                                    and_return(@cmd_mock)
+          Fission::CLI.execute ['suspend', '--all']
+        end
+      end
+
+      describe 'delete' do
+        it "should try to delete the vm" do
+          Fission::Command::Delete.stub!(:help).and_return('')
+          Fission::Command::Delete.should_receive(:new).
+                                   with(['foo']).
+                                   and_return(@cmd_mock)
+          Fission::CLI.execute ['delete', 'foo']
+        end
+
+        it 'should try to force delete the vm' do
+          Fission::Command::Delete.stub!(:help).and_return('')
+          Fission::Command::Delete.should_receive(:new).
+                                   with(['foo', '--force']).
+                                   and_return(@cmd_mock)
+          Fission::CLI.execute ['delete', 'foo', '--force']
+        end
       end
     end