pennyworth-tool 0.0.1

data/lib/pennyworth/ssh_keys_importer.rb

@@ -0,0 +1,107 @@
+# Copyright (c) 2013-2014 SUSE LLC
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of version 3 of the GNU General Public License as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.   See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, contact SUSE LLC.
+#
+# To contact SUSE about this file by physical or electronic mail,
+# you may find current contact information at www.suse.com
+
+require "tempfile"
+
+module Pennyworth
+  class SshKeysImporter
+    def self.import(ip, username, password, identity_file = nil)
+      tmp_file = Tempfile.new("pennyworth-ssh-key-importer")
+
+      cmd = "ssh-copy-id"
+      if identity_file
+        cmd << " -i \"#{identity_file}\""
+        cmd << " -o \"IdentityFile=#{identity_file}\""
+      end
+      cmd << " -o \"UserKnownHostsFile=/dev/null\""
+      cmd << " -o \"StrictHostKeyChecking=no\""
+      cmd << " #{username}@#{ip}"
+
+      # This temporary bash script copies the SSH keys to the host using
+      # ssh-copy-id.
+      #
+      # In order to drive this interactive command it is executed by expect.
+      tmp_file.write(<<-EOF
+        #!/usr/bin/expect -f
+
+        spawn #{cmd}
+        expect {
+          -re ".*assword.*" {
+            send "#{password}\\r"
+            expect {
+              -re ".*assword.*" {
+                exit 1
+              }
+              -re ".*Number of key.*" {
+                exit 0
+              }
+            }
+          }
+          -re ".*Number of key.*" {
+            exit 0
+          }
+          "skipped" {
+            exit 2
+          }
+          "authenticity" {
+            send "yes\\r"
+            exp_continue
+          }
+          "Connection refused" {
+            exit 3
+          }
+        }
+        exit 4
+      EOF
+      )
+      tmp_file.close
+
+      attempt = 1
+      begin
+        Cheetah.run("expect", tmp_file.path)
+      rescue Cheetah::ExecutionFailed => e
+        case e.status.exitstatus
+          when 1
+            raise WrongPasswordException, "Error: Could not upload SSH keys because" \
+              " the password is wrong."
+          when 2
+            raise SshKeysAlreadyExistsException, "SSH keys were not uploaded because they" \
+              " were already on the host."
+          when 3
+            # We give sshd 30 seconds to start up
+            if attempt > 30
+              raise SshConnectionFailed, "SSH connection failed."
+            else
+              attempt += 1
+              sleep 1
+              retry
+            end
+          when 4
+            raise RuntimeError, "An unknown error occured while trying to upload " \
+              "the SSH keys."
+          when 127
+            raise CommandNotFoundError, "Error: Please install expect before running" \
+              " pennyworth copy-ssh-keys."
+          else
+            raise e
+        end
+      end
+    ensure
+      tmp_file.unlink
+    end
+  end
+end

data/lib/pennyworth/urls.rb

@@ -0,0 +1,28 @@
+# Copyright (c) 2013-2014 SUSE LLC
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of version 3 of the GNU General Public License as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.   See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, contact SUSE LLC.
+#
+# To contact SUSE about this file by physical or electronic mail,
+# you may find current contact information at www.suse.com
+
+module Pennyworth
+  # URL manipulation utilities.
+  module URLs
+    def self.join(*parts)
+      parts = parts.reject(&:empty?)
+      parts = [parts.first] + parts[1..-1].map { |p| p.sub(/^\//, "") }
+      parts = parts[0..-2].map { |p| p.sub(/\/$/, "") } + [parts.last]
+      parts.join("/")
+    end
+  end
+end

data/lib/pennyworth/vagrant.rb

@@ -0,0 +1,81 @@
+# Copyright (c) 2013-2014 SUSE LLC
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of version 3 of the GNU General Public License as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.   See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, contact SUSE LLC.
+#
+# To contact SUSE about this file by physical or electronic mail,
+# you may find current contact information at www.suse.com
+
+module Pennyworth
+  class Vagrant
+    def initialize(vagrant_dir)
+      @vagrant_dir = vagrant_dir
+    end
+
+    def run(*args)
+      # Many vagrant commands don't have a --provider option, but they fail
+      # because they can't find VirtualBox. The following code is a crude
+      # workaround.
+      Dir.chdir(@vagrant_dir) do
+        with_env "VAGRANT_DEFAULT_PROVIDER" => "libvirt" do
+          Cheetah.run "vagrant", *args
+        end
+      end
+    end
+
+    def ssh_config(vm_name)
+      output = Dir.chdir(@vagrant_dir) do
+        if vm_name
+          run "ssh-config", vm_name, :stdout => :capture
+        else
+          config = ""
+          status = run "status", :stdout => :capture
+          status.scan(/\n(.*?)\s*running/).each do |vm_name|
+            config += run "ssh-config", vm_name.first, :stdout => :capture
+          end
+          config
+        end
+      end
+
+      parse_ssh_config_output(output)
+    end
+
+    private
+
+    def parse_ssh_config_output(output)
+      # See http://linux.die.net/man/5/ssh_config for description of the format.
+
+      config = {}
+      host = nil
+
+      output.each_line do |line|
+        line.chomp!("\n")
+
+        next if line.empty? || line.start_with?("#")
+
+        m = /^\s*(?<key>\w+)(\s*=\s*|\s+)(?<value>.*)$/.match(line)
+        raise "Invalid line in SSH config: #{line.inspect}." unless m
+
+        if m[:key] == "Host"
+          host = m[:value]
+          config[host] = {}
+        else
+          raise "Missing Host keyword before #{m[:key]}." unless host
+
+          config[host][m[:key]] = m[:value]
+        end
+      end
+
+      config
+    end
+  end
+end

data/lib/pennyworth/vagrant_command.rb

@@ -0,0 +1,120 @@
+# Copyright (c) 2013-2014 SUSE LLC
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of version 3 of the GNU General Public License as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.   See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, contact SUSE LLC.
+#
+# To contact SUSE about this file by physical or electronic mail,
+# you may find current contact information at www.suse.com
+
+module Pennyworth
+  class VagrantCommand < Command
+    attr_reader :vagrant
+
+    def initialize
+      @vagrant = Pennyworth::Vagrant.new(Pennyworth::Cli.settings.vagrant_dir)
+    end
+
+    def self.parse_status output
+      result = []
+      parsing_vms = false
+      output.each_line do |line|
+        line.strip!
+        if line.empty?
+          break if parsing_vms
+          parsing_vms = true
+        elsif parsing_vms
+          line =~ /^(.*?)\s+(.*)$/
+          vm_name = $1
+          vm_state = $2
+          if vm_state =~ /running/
+            vm_name += " (running)"
+          end
+          result.push vm_name
+        end
+      end
+      result
+    end
+
+    def self.setup_environment(dir)
+      if !File.exists?(File.join(dir, "Vagrantfile"))
+        vagrant = Pennyworth::Vagrant.new(dir)
+        FileUtils.mkdir_p(dir)
+        vagrant.run("init")
+      end
+    end
+
+    def up(vm_name, destroy)
+      Pennyworth::Libvirt.ensure_libvirt_env_started
+
+      if vm_name
+        log "Starting VM #{vm_name}..."
+        @vagrant.run "destroy", vm_name if destroy
+        @vagrant.run "up", vm_name
+      else
+        log "Starting test environment..."
+        @vagrant.run "destroy" if destroy
+        @vagrant.run "up"
+      end
+    end
+
+    def down(vm_name)
+      if vm_name
+        log "Stopping VM #{vm_name}..."
+        @vagrant.run "halt", vm_name
+      else
+        log "Stopping test environment..."
+        @vagrant.run "halt"
+      end
+    end
+
+    def env_clean?
+      status = @vagrant.run "status", :stdout => :capture
+
+      return status.index("running").nil?
+    end
+
+    def reset(vm_name)
+      if vm_name
+        log "Resetting VM #{vm_name}..."
+        @vagrant.run "destroy", vm_name
+      else
+        log "Resetting test environment..."
+        @vagrant.run "destroy"
+      end
+    end
+
+    def ssh(vm_name, cmd, stdout = nil)
+      if vm_name
+        @vagrant.run "ssh", vm_name, "-c", cmd, :stdout => stdout
+      else
+        @vagrant.run "ssh", "-c", cmd, :stdout => stdout
+      end
+    end
+
+    def list
+      boxes = @vagrant.run "box", "list", :stdout => :capture
+      boxes.split("\n")
+    end
+
+    def status
+      Pennyworth::VagrantCommand.parse_status( @vagrant.run "status", :stdout => :capture )
+    end
+
+    def destroy
+      @vagrant.run "destroy"
+    end
+
+    def add_box box, box_path
+      @vagrant.run "box", "add", box, box_path, "--force"
+    end
+  end
+end

data/lib/pennyworth/vagrant_runner.rb

@@ -0,0 +1,44 @@
+# Copyright (c) 2013-2014 SUSE LLC
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of version 3 of the GNU General Public License as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.   See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, contact SUSE LLC.
+#
+# To contact SUSE about this file by physical or electronic mail,
+# you may find current contact information at www.suse.com
+
+module Pennyworth
+  class VagrantRunner < Runner
+    def initialize(box, vagrant_dir, username)
+      @box = box
+      @username = username
+      @vagrant = Vagrant.new(vagrant_dir)
+    end
+
+    def start
+      @vagrant.run "destroy", @box
+      @vagrant.run "up", @box
+
+      ip = @vagrant.ssh_config(@box)[@box]["HostName"]
+      @command_runner = RemoteCommandRunner.new(ip, @username)
+
+      ip
+    end
+
+    def stop
+      @vagrant.run "halt", @box
+    end
+
+    def cleanup_directory(_dir)
+      # The machine will be reset anyway after the tests, so this is is a NOP
+    end
+  end
+end

data/lib/pennyworth/version.rb

@@ -0,0 +1,22 @@
+# Copyright (c) 2013-2014 SUSE LLC
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of version 3 of the GNU General Public License as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.   See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, contact SUSE LLC.
+#
+# To contact SUSE about this file by physical or electronic mail,
+# you may find current contact information at www.suse.com
+
+module Pennyworth
+
+  VERSION = "0.0.1"
+
+end

data/lib/pennyworth/vm.rb

@@ -0,0 +1,62 @@
+# Copyright (c) 2013-2014 SUSE LLC
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of version 3 of the GNU General Public License as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.   See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, contact SUSE LLC.
+#
+# To contact SUSE about this file by physical or electronic mail,
+# you may find current contact information at www.suse.com
+
+module Pennyworth
+  # Represents a virtual machine that can be started, stopped, and interacted
+  # with.
+  class VM
+    attr_accessor :ip, :runner
+
+    def initialize(runner)
+      @runner = runner
+    end
+
+    def start
+      @ip = @runner.start
+    end
+
+    def stop
+      @runner.stop
+    end
+
+    def run_command(*args)
+      command_runner.run(*args)
+    end
+
+    def inject_file(source, destination, opts = {})
+      command_runner.inject_file(source, destination, opts)
+    end
+
+    def extract_file(source, destination)
+      command_runner.extract_file(source, destination)
+    end
+
+    def inject_directory(source, destination, opts = {})
+      command_runner.inject_directory(source, destination, opts)
+    end
+
+    def cleanup_directory(dir)
+      runner.cleanup_directory(dir)
+    end
+
+    private
+
+    def command_runner
+      @runner.command_runner
+    end
+  end
+end