tecnh-chef 0.9.9.vpc

data/lib/chef/knife/slicehost_images_list.rb

@@ -0,0 +1,53 @@
+#
+# Author:: Ian Meyer (<ianmmeyer@gmail.com>)
+# Copyright:: Copyright (c) 2010 Ian Meyer
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife'
+require 'json'
+
+class Chef
+  class Knife
+    class SlicehostImagesList < Knife
+
+      banner "knife slicehost images list"
+
+      def highline
+        @highline ||= HighLine.new
+      end
+
+      def run
+        require 'fog'
+        require 'highline'
+
+        slicehost = Fog::Slicehost.new(
+          :slicehost_password => Chef::Config[:knife][:slicehost_password]
+        )
+
+        images  = slicehost.images.inject({}) { |h,i| h[i.id] = i.name; h }
+
+        image_list = [ highline.color('ID', :bold), highline.color('Name', :bold) ]
+
+        slicehost.images.each do |server|
+          image_list << server.id.to_s
+          image_list << server.name
+        end
+        puts highline.list(image_list, :columns_across, 2)
+
+      end
+    end
+  end
+end

data/lib/chef/knife/slicehost_server_create.rb

@@ -0,0 +1,103 @@
+#
+# Author:: Ian Meyer (<ianmmeyer@gmail.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife'
+require 'json'
+
+class Chef
+  class Knife
+    class SlicehostServerCreate < Knife
+
+      banner "knife slicehost server create [RUN LIST...] (options)"
+
+      option :flavor,
+        :short => "-f FLAVOR",
+        :long => "--flavor FLAVOR",
+        :description => "The flavor of server",
+        :proc => Proc.new { |f| f.to_i },
+        :default => 1
+
+      option :image,
+        :short => "-i IMAGE",
+        :long => "--image IMAGE",
+        :description => "The image of the server",
+        :proc => Proc.new { |i| i.to_i },
+        :default => 49
+
+      option :server_name,
+        :short => "-N NAME",
+        :long => "--server-name NAME",
+        :description => "The server name",
+        :default => "wtf"
+
+      option :slicehost_password,
+        :short => "-K KEY",
+        :long => "--slicehost-password password",
+        :description => "Your slicehost API password",
+        :proc => Proc.new { |password| Chef::Config[:knife][:slicehost_password] = password } 
+
+
+      def h
+        @highline ||= HighLine.new
+      end
+
+      def run 
+        require 'fog'
+        require 'highline'
+        require 'net/ssh/multi'
+        require 'readline'
+
+        slicehost = Fog::Slicehost.new(
+          :slicehost_password => Chef::Config[:knife][:slicehost_password]
+        )
+
+        flavors = slicehost.flavors.inject({}) { |h,f| h[f.id] = f.name; h }
+        images  = slicehost.images.inject({}) { |h,i| h[i.id] = i.name; h }
+
+        response = slicehost.create_slice(config[:flavor],
+                               config[:image],
+                               config[:server_name])
+        $stdout.sync = true
+
+        puts "#{h.color("Name", :cyan)}: #{response.body['name']}"
+        puts "#{h.color("Flavor", :cyan)}: #{flavors[response.body['flavor-id']]}"
+        puts "#{h.color("Image", :cyan)}: #{images[response.body['image-id']]}"
+        puts "#{h.color("Public Address", :cyan)}: #{response.body['addresses'][1]}"
+        puts "#{h.color("Private Address", :cyan)}: #{response.body['addresses'][0]}"
+        puts "#{h.color("Password", :cyan)}: #{response.body['root-password']}"
+     
+        print "\n#{h.color("Requesting status of #{response.body['name']}", :magenta)}"
+        saved_password = response.body['root-password']
+
+        # wait for it to be ready to do stuff
+        loop do
+          sleep 15
+          host = slicehost.get_slice(response.body['id'])
+          if host.body['status'] == 'active'
+            break
+          end
+        end
+
+        puts "\nServer ready!!"
+      
+      end
+    end
+  end
+end
+
+

data/lib/chef/knife/slicehost_server_delete.rb

@@ -0,0 +1,61 @@
+#
+# Author:: Ian Meyer (<ianmmeyer@gmail.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife'
+require 'json'
+
+class Chef
+  class Knife
+    class SlicehostServerDelete < Knife
+
+      banner "knife slicehost server delete SLICENAME"
+
+      def run 
+        require 'fog'
+        require 'readline'
+        
+        slicehost = Fog::Slicehost.new(
+          :slicehost_password => Chef::Config[:knife][:slicehost_password]
+        )
+      
+        # Build hash of slice => id  
+        servers = slicehost.servers.inject({}) { |h,f| h[f.name] = f.id; h }
+
+        unless servers.has_key?(@name_args[0])
+          Chef::Log.warn("I can't find a slice named #{@name_args[0]}")
+          return false
+        end
+ 
+        confirm("Do you really want to delete server ID #{servers[@name_args[0]]} named #{@name_args[0]}")
+
+        begin
+          response = slicehost.delete_slice(servers[@name_args[0]])
+          
+          if response.headers['status'] == "200 OK" 
+            Chef::Log.warn("Deleted server #{servers[@name_args[0]]} named #{@name_args[0]}")
+          end
+        rescue Excon::Errors::UnprocessableEntity
+          Chef::Log.warn("There was a problem deleting #{@name_args[0]}, check your slice manager")
+        end
+      end
+    end
+  end
+end
+
+
+

data/lib/chef/knife/slicehost_server_list.rb

@@ -0,0 +1,64 @@
+#
+# Author:: Ian Meyer (<ianmmeyer@gmail.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife'
+require 'json'
+
+class Chef
+  class Knife
+    class SlicehostServerList < Knife
+
+      banner "knife slicehost server list (options)"
+
+      def h
+        @highline ||= HighLine.new
+      end
+
+      def run 
+        require 'fog'
+        require 'highline'
+        require 'net/ssh/multi'
+        require 'readline'
+        
+        slicehost = Fog::Slicehost.new(
+          :slicehost_password => Chef::Config[:knife][:slicehost_password]
+        )
+       
+        # Make hash of flavor id => name and image id => name
+        flavors = slicehost.flavors.inject({}) { |h,f| h[f.id] = f.name; h }
+        images  = slicehost.images.inject({}) { |h,i| h[i.id] = i.name; h }
+ 
+        server_list = [ h.color('ID', :bold), h.color('Name', :bold), h.color('Public IP', :bold), h.color('Private IP', :bold), h.color('Image', :bold), h.color('Flavor', :bold) ]
+ 
+        slicehost.servers.each do |server|
+          server_list << server.id.to_s
+          server_list << server.name
+          server_list << server.addresses[1]
+          server_list << server.addresses[0]
+          server_list << images[server.image_id].to_s
+          server_list << flavors[server.flavor_id].to_s
+        end
+        puts h.list(server_list, :columns_across, 6)
+
+      end
+    end
+  end
+end
+
+
+

data/lib/chef/knife/ssh.rb

@@ -0,0 +1,311 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2009 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife'
+require 'chef/data_bag_item'
+
+begin
+  gem "net-ssh", ">= 2.0.23"
+rescue LoadError
+end
+
+class Chef
+  class Knife
+    class Ssh < Knife
+
+      attr_writer :password
+
+      banner "knife ssh QUERY COMMAND (options)"
+
+      option :concurrency,
+        :short => "-C NUM",
+        :long => "--concurrency NUM",
+        :description => "The number of concurrent connections",
+        :default => nil,
+        :proc => lambda { |o| o.to_i }
+
+      option :attribute,
+        :short => "-a ATTR",
+        :long => "--attribute ATTR",
+        :description => "The attribute to use for opening the connection - default is fqdn",
+        :default => "fqdn" 
+
+      option :manual,
+        :short => "-m",
+        :long => "--manual-list",
+        :boolean => true,
+        :description => "QUERY is a space separated list of servers",
+        :default => false
+
+      option :ssh_user,
+        :short => "-x USERNAME",
+        :long => "--ssh-user USERNAME",
+        :description => "The ssh username"
+
+      option :ssh_password,
+        :short => "-P PASSWORD",
+        :long => "--ssh-password PASSWORD",
+        :description => "The ssh password"
+
+      option :identity_file,
+        :short => "-i IDENTITY_FILE",
+        :long => "--identity-file IDENTITY_FILE",
+        :description => "The SSH identity file used for authentication"
+
+      option :vpc_mode,
+        :short => "-V {TRUE/FALSE}",
+        :long => "--vpc_mode {TRUE/FALSE}",
+        :description => "Enable or disable features needed for Virtual Private Cloud mode",
+        :default => false
+
+      def session
+        @session ||= Net::SSH::Multi.start(:concurrent_connections => config[:concurrency])
+      end
+
+      def h
+        @highline ||= HighLine.new
+      end
+
+      def configure_session
+        list = case config[:manual]
+               when true
+                 @name_args[0].split(" ")
+               when false
+                 r = Array.new
+                 q = Chef::Search::Query.new
+                 q.search(:node, @name_args[0]) do |item|
+                   r << format_for_display(item)[config[:attribute]]
+                 end
+                 r
+               end
+        (Chef::Log.fatal("No nodes returned from search!"); exit 10) if list.length == 0
+        session_from_list(list)
+      end
+
+      def session_from_list(list)        
+        options_hash = (config[:vpc_mode] == "true") ? {:paranoid => false, :user_known_hosts_file => "/dev/null"} : {}
+        
+        list.each do |item|
+          Chef::Log.debug("Adding #{item}")
+          item = "#{config[:ssh_user]}@#{item}" if config[:ssh_user]
+          Chef::Log.debug("Using user #{config[:ssh_user]}")
+          if config[:identity_file]
+            Chef::Log.debug("Using identity file #{config[:identity_file]}")
+            options_hash.merge!(:keys => File.expand_path(config[:identity_file]))
+          elsif config[:password]
+            Chef::Log.debug("Using password #{config[:password]}")
+            options_hash.merge(:password => config[:password])
+          end
+          Chef::Log.debug("options_hash = #{options_hash.inspect}")
+          session.use item, options_hash
+          @longest = item.length if item.length > @longest
+        end
+        
+        session
+      end
+
+      def fixup_sudo(command)
+        command.sub(/^sudo/, 'sudo -p \'knife sudo password: \'')
+      end
+
+      def print_data(host, data)
+        if data =~ /\n/
+          data.split(/\n/).each { |d| print_data(host, d) }
+        else
+          padding = @longest - host.length
+          print h.color(host, :cyan)
+          padding.downto(0) { print " " }
+          puts data
+        end
+      end
+
+      def ssh_command(command, subsession=nil)
+        subsession ||= session
+        command = fixup_sudo(command)
+        subsession.open_channel do |ch|
+          ch.request_pty
+          ch.exec command do |ch, success|
+            raise ArgumentError, "Cannot execute #{command}" unless success
+            ch.on_data do |ichannel, data|
+              print_data(ichannel[:host], data)
+              if data =~ /^knife sudo password: /
+                ichannel.send_data("#{get_password}\n")
+              end
+            end
+          end
+        end
+        session.loop
+      end
+
+      def get_password
+        @password ||= h.ask("Enter your password: ") { |q| q.echo = false }
+      end
+
+      # Present the prompt and read a single line from the console. It also
+      # detects ^D and returns "exit" in that case. Adds the input to the
+      # history, unless the input is empty. Loops repeatedly until a non-empty
+      # line is input.
+      def read_line
+        loop do
+          command = reader.readline("#{h.color('knife-ssh>', :bold)} ", true)
+
+          if command.nil?
+            command = "exit"
+            puts(command)
+          else
+            command.strip!
+          end
+
+          unless command.empty?
+            return command
+          end
+        end
+      end
+
+      def reader
+        Readline
+      end
+
+      def interactive
+        puts "Connected to #{h.list(session.servers_for.collect { |s| h.color(s.host, :cyan) }, :inline, " and ")}"
+        puts
+        puts "To run a command on a list of servers, do:"
+        puts "  on SERVER1 SERVER2 SERVER3; COMMAND"
+        puts "  Example: on latte foamy; echo foobar"
+        puts
+        puts "To exit interactive mode, use 'quit!'"
+        puts
+        while 1
+          command = read_line
+          case command
+          when 'quit!'
+            puts 'Bye!'
+            break
+          when /^on (.+?); (.+)$/
+            raw_list = $1.split(" ")
+            server_list = Array.new
+            session.servers.each do |session_server|
+              server_list << session_server if raw_list.include?(session_server.host) 
+            end
+            command = $2
+            ssh_command(command, session.on(*server_list))
+          else
+            ssh_command(command)
+          end
+        end
+      end
+
+      def screen
+        tf = Tempfile.new("knife-ssh-screen")
+        if File.exist? "#{ENV["HOME"]}/.screenrc"
+          tf.puts("source #{ENV["HOME"]}/.screenrc")
+        end
+        tf.puts("caption always '%-Lw%{= BW}%50>%n%f* %t%{-}%+Lw%<'")
+        tf.puts("hardstatus alwayslastline 'knife ssh #{@name_args[0]}'")
+        window = 0
+        session.servers_for.each do |server|
+          tf.print("screen -t \"#{server.host}\" #{window} ssh ")
+          server.user ? tf.puts("#{server.user}@#{server.host}") : tf.puts(server.host)
+          window += 1
+        end
+        tf.close
+        exec("screen -c #{tf.path}")
+      end
+
+      def tmux
+        begin
+          Chef::Mixin::Command.run_command(:command => "tmux new-session -d -s 'knife'")
+        rescue Chef::Exceptions::Exec
+        end
+        session.servers_for.each do |server|
+          begin
+            Chef::Mixin::Command.run_command(:command => "tmux new-window -d -n '#{server.host}' -t 'knife' 'ssh #{server.user ? "#{server.user}@#{server.host}" : server.host}'")
+          rescue Chef::Exceptions::Exec
+          end
+        end
+        exec("tmux attach-session -t knife")
+      end
+
+      def macterm
+        require 'appscript'
+        Appscript.app("/Applications/Utilities/Terminal.app").windows.first.activate  
+        Appscript.app("System Events").application_processes["Terminal.app"].keystroke("n", :using=>:command_down)
+        term = Appscript.app('Terminal')  
+        window = term.windows.first.get
+        session.servers_for.each do |server|
+          Appscript.app("System Events").application_processes["Terminal.app"].keystroke("t", :using=>:command_down)
+          cmd = "unset PROMPT_COMMAND; echo -e \"\\033]0;#{server.host}\\007\"; ssh #{server.user ? "#{server.user}@#{server.host}" : server.host}"
+          Appscript.app('Terminal').do_script(cmd, :in => window.tabs.last.get)
+          sleep 1
+        end
+      end
+
+      def run 
+        @longest = 0
+
+        load_late_dependencies
+
+        configure_session
+
+        case @name_args[1]
+        when "interactive"
+          interactive 
+        when "screen"
+          screen
+        when "tmux"
+          tmux
+        when "macterm"
+          macterm
+        else
+          ssh_command(@name_args[1..-1].join(" "))
+        end
+
+        session.close
+      end
+
+      def load_late_dependencies
+        require 'net/ssh/multi'
+        require 'readline'
+        require 'highline'
+
+        assert_net_ssh_version_acceptable!
+      end
+
+      # :nodoc:
+      # TODO: remove this stuff entirely and package knife ssh as a knife plugin. (Dan - 08 Jul 2010)
+      #
+      # The correct way to specify version deps is in the gemspec or other packaging.
+      # However, we don't want to have a gem dep on net-ssh, because it's a hassle
+      # when you only need the chef-client (e.g., on a managed node). So we have to
+      # check here that you have a decent version of Net::SSH.
+      #
+      # net-ssh of lower versions has a bug that causes 'knife ssh (searchterm) (commandname)" 
+      # to loop infinitely and consume all the CPU of one core.
+      def assert_net_ssh_version_acceptable!
+        netssh_version = Net::SSH::Version
+        # we want version 2.0.23 and higher:
+        unless (netssh_version::MAJOR == 2) && (netssh_version::TINY >= 23 || netssh_version::MINOR >= 1)
+          STDERR.puts "ERROR: Please install net-ssh version 2.0.23 or higher, as lower versions cause issues."
+          exit 1
+        end
+      end
+
+    end
+  end
+end
+