cuken 0.1.4 → 0.1.7

data/lib/cuken/api/ssh-forever.rb

@@ -0,0 +1,237 @@
+require 'pathname'
+require 'open4'
+
+module SshForever
+  ::SshForever::VERSION = '0.4.0' unless defined? ::SshForever::VERSION
+
+  class SecureShellForever
+    def initialize(login, options = {})
+      @login   = login
+      @options = options
+      @username, @hostname = @login.split("@")
+      cleanup_ssh_config
+      cleanup_identity_files
+      local_config_path
+      initialization_run
+      local_key_path
+    end
+
+    def initialization_run
+      unless File.exists?(config_file_path)
+        puts "You do not appear to have a config file. Expected one at #{config_file_path}"  unless @options[:quiet]
+        existing_ssh_config? ? append_ssh_config : write_ssh_config
+        puts "Next: check for a public key."  unless @options[:quiet]
+        unless File.exists?(public_key_path)
+          puts "You do not appear to have an identity file/public key. Expected one at #{public_key_path}"  unless @options[:quiet]
+          confirm_keygen
+          generate_public_key
+        end
+      end
+      args = ssh_args()
+
+      copy_public_key(args)
+
+      puts "Success. From now on you can just use plain old 'ssh'. Logging you in..."  unless @options[:quiet]
+      status = run_shell_cmd(ssh_login(args))
+      #cleanup_ssh_config
+      # exitstatus 2 is when ssh-add can't find an agent on local machine.
+      # We ought to switch to use Net::SSH libraries....
+      exit 1 unless status.exitstatus.to_i == 0 || status.exitstatus.to_i == 2
+    end
+
+    def run_interactive
+      system ssh_login_interactive(ssh_args)
+      #cleanup_ssh_config
+    end
+
+  private
+
+    def local_ssh_config_path
+      @local_ssh_config_path ||= Pathname('~/').expand_path.realpath
+    end
+
+    def authorized_keys
+      (local_ssh_config_path + '.ssh' + 'authorized_keys2').exist? ? 'authorized_keys2' : 'authorized_keys'
+    end
+
+    def append_ssh_config
+      puts "Appending host entry in local ssh config with name #{@options[:name]}" unless @options[:quiet]
+      Pathname(config_file_path).open("a") do |config|
+        config << ssh_config
+      end
+    end
+
+    def write_ssh_config
+      puts "Creating host entry in local ssh config with name #{@options[:name]||'ssh-forever'}" unless @options[:quiet]
+      Pathname(config_file_path).open("w") do |config|
+        config << ssh_config
+      end
+      @cleanup_ssh_config = config_file_path
+    end
+
+    def existing_ssh_config?
+      file = Pathname(config_file_path)
+      config = file.exist? ? file.read : ""
+      chk1 = config[/IdentityFile #{public_key_path}/] ? true : false
+      chk2 = config[/Host #{@options[:name]}/] ? true : false
+      chk1 && chk2 ? true : false
+    end
+
+    def ssh_config
+      host_config = <<-STUFF.gsub(/^ {6}/, '')
+
+      Host #{@options[:name]||'ssh-forever'}
+        HostName #{@hostname}
+        User #{@username}
+        Port #{@options[:port]||22}
+        IdentityFile #{public_key_path}
+        Protocol 2
+        PreferredAuthentications publickey
+        PubkeyAuthentication yes
+        Batchmode yes
+        ChallengeResponseAuthentication no
+        CheckHostIP yes
+        StrictHostKeyChecking #{@options[:strict] ? 'yes' : 'no'}
+        HostKeyAlias #{@options[:name] ? @options[:name] : @hostname}
+        ConnectionAttempts 3
+        ControlMaster auto
+        ControlPath #{local_ssh_config_path + '.ssh' + '%h_%p_%r'}
+        ForwardAgent no
+        ForwardX11Trusted no
+        GatewayPorts yes
+        GSSAPIAuthentication no
+        GSSAPIDelegateCredentials no
+        HashKnownHosts yes
+        HostbasedAuthentication no
+        IdentitiesOnly yes
+        LogLevel #{@options[:debug] ? 'DEBUG3' : (@options[:quiet] ? 'QUIET' : 'INFO')}
+        NoHostAuthenticationForLocalhost yes
+        PasswordAuthentication no
+        PermitLocalCommand no
+        RekeyLimit 2G
+        ServerAliveCountMax #{@options[:intense] ? '1' : '3'}
+        ServerAliveInterval #{@options[:intense] ? '1' : '15'}
+        TCPKeepAlive yes
+        Tunnel no
+
+      STUFF
+    end
+
+
+    def run_shell_cmd(cmd)
+      status = ::Open4::popen4('sh') do |pid, stdin, stdout, stderr|
+        puts "debug: #{cmd}"  if @options[:debug]
+        stdin.puts cmd
+        stdin.close
+        puts "debug: #{stderr.read.strip}" if @options[:debug]
+      end
+      status
+    end
+
+    def cleanup_ssh_config
+      file = "#{local_ssh_config_path + '.ssh' + 'ssh-forever-config'}"
+      Pathname(file).delete if Pathname(file).exist?
+    end
+
+    def cleanup_identity_files
+      file = "#{local_ssh_config_path + '.ssh' + 'ssh-forever-id'}"
+      Pathname(file).delete if Pathname(file).exist?
+      Pathname(file + '.pub').delete if Pathname(file + '.pub').exist?
+    end
+
+    def confirm_keygen
+      unless @options[:auto]
+        STDERR.print "Would you like me to generate one? [Y/n]"  unless @options[:quiet]
+        result = STDIN.gets.strip
+        unless result == '' or result == 'y' or result == 'Y'
+          cleanup_ssh_config()
+          flunk %Q{\nFair enough, I'll be off then. You can generate your own by hand using\n\n    ssh-keygen -t rsa}
+        end
+      end
+    end
+
+
+    def generate_public_key
+      status = run_shell_cmd(ssh_keygen)
+      flunk("Oh dear. I was unable to generate your public key. Run the command 'ssh-keygen -t rsa' manually to find out why.") unless status.exitstatus.to_i == 0
+    end
+
+    def copy_public_key(args)
+      puts "Copying your public key to the remote server."  unless @options[:quiet]
+      puts "Prepare to enter your password for the last time..." unless @options[:quiet]
+      status = run_shell_cmd(ssh_keycopy(args))
+      exit 1 unless status.exitstatus.to_i == 0
+    end
+
+    def ssh_args
+      [ ' ',
+        ("-F #{@options[:config_file]}" if @options[:config_file]),
+        ("-p #{@options[:port]}" if @options[:port] =~ /^\d+$/),
+        (@options[:strict] ? "-o stricthostkeychecking=yes" : "-o stricthostkeychecking=no"),
+        (@options[:debug] ? "-vvv" : "-q")
+      ].compact.join(' ')
+    end
+
+    def ssh_keygen
+      "ssh-keygen -t rsa #{@options[:debug] ? "-v" : "-q"} -C '#{local_key_path} created by ssh-forever #{Time.now.utc}' -N '' -f #{local_key_path.to_s}"
+    end
+
+    def ssh_keycopy(args)
+      "ssh-copy-id '-i #{@options[:identity_file]}.pub #{args} #{@login}'"
+    end
+
+    def ssh_login(args)
+      if @options[:name]
+        append_ssh_config unless existing_ssh_config?
+        #ssh-add #{@options[:identity_file]};
+        login_command = "ssh-add #{@options[:identity_file]}; #{ssh_keycopy(args)}; SSH_AUTH_SOCK=0 ssh #{@options[:name]}#{args} 'echo true;';"
+      else
+        login_command = "ssh-add #{@options[:identity_file]}; SSH_AUTH_SOCK=0 ssh #{@login}#{args} 'echo true;';"
+      end
+      login_command
+    end
+
+    def ssh_login_interactive(args)
+      if @options[:name]
+        append_ssh_config unless existing_ssh_config?
+        login_command = "ssh #{@options[:name]} #{args}"
+      else
+        login_command = "ssh #{@login}#{args}"
+      end
+      login_command
+    end
+
+    def flunk(message)
+      STDERR.puts message
+      exit 1
+    end
+
+    def local_key_path
+      @options[:identity_file] ||= public_key_path
+      if RUBY_VERSION =~ /^1\.8\.7/
+        @options[:identity_file] = Pathname(@options[:identity_file]).expand_path.to_s
+      else
+        @options[:identity_file] = Pathname(@options[:identity_file]).expand_path.realdirpath.to_s
+      end
+    end
+
+    def local_config_path
+      @options[:config_file] ||= config_file_path # public_key_path
+      if RUBY_VERSION =~ /^1\.8\.7/
+        @options[:config_file] = Pathname(@options[:config_file]).expand_path.to_s
+      else
+        @options[:config_file] = Pathname(@options[:config_file]).expand_path.realdirpath.to_s
+        @options[:config_file]
+      end
+    end
+
+    def public_key_path
+      File.expand_path(@options[:identity_file] || "#{local_ssh_config_path + '.ssh' + 'ssh-forever-id'}")
+    end
+
+    def config_file_path
+      File.expand_path(@options[:config_file] || "#{local_ssh_config_path + '.ssh' + 'ssh-forever-config'}")
+    end
+
+  end
+end

data/lib/cuken/chef.rb

@@ -1,3 +1,5 @@
+require 'grit'
+
 require 'cuken/common'
 require 'cuken/api/chef'
 require 'cuken/cucumber/chef'

data/lib/cuken/common.rb

@@ -1,3 +1,4 @@
+require 'fileutils'
 require 'pathname'
 require 'aruba/cucumber' unless defined? Aruba
 

data/lib/cuken/cucumber/chef/common.rb

@@ -24,11 +24,12 @@ Given /^the Chef server URI "([^"]*)"$/ do |uri|
   chef.uri = uri
 end
 
-Given /^the Chef client "([^"]*)"$/ do |name|
+Given /^the Chef client "([^"]*)" exists$/ do |name|
   chef.client_name =  name
 end
 
-Given /^the Chef admin client "([^"]*)"$/ do |name|
+Given /^the Chef admin client "([^"]*) exists"$/ do |name|
+  chef.show_client(:name => name)
   chef.admin_client_name =  name
 end
 
@@ -61,6 +62,7 @@ end
 
 Given /^I clone the remote Chef repository branch "([^"]*)" to "([^"]*)"$/ do |brnch, path|
   chef.local_chef_repo = chef_clone_repo(path, false, chef.remote_chef_repo, brnch)
+  chef.local_chef_repo.exist?.should be_true
 end
 
 Given /^a default base Chef repository in "([^"]*)"$/ do |path|

data/lib/cuken/cucumber/chef/cookbook.rb

@@ -25,13 +25,22 @@
 
 Given /^the remote Cookbook repository "([^"]*)"$/ do |ckbk_repo|
   in_current_dir do
-    chef.remote_cookbook_repo = Pathname(ckbk_repo).expand_path.realdirpath
+    repo = Dir.exist?(ckbk_repo) ? Pathname(ckbk_repo).expand_path.realdirpath : ckbk_repo
+    chef.remote_cookbook_repo = repo
+  end
+end
+
+Given /^the remote Cookbooks URI "([^"]*)"$/ do |ckbk_uri|
+  in_current_dir do
+    repo = Dir.exist?(ckbk_uri) ? Pathname(ckbk_uri).expand_path.realdirpath : ckbk_uri
+    chef.cookbooks_uri = repo
   end
 end
 
 Given /^the local Cookbook repository "([^"]*)"$/ do |ckbk_repo|
   in_current_dir do
-    chef.local_cookbook_repo = Pathname(ckbk_repo).expand_path.realdirpath
+    repo = Dir.exist?(ckbk_repo) ? Pathname(ckbk_repo).expand_path.realdirpath : ckbk_repo
+    chef.local_cookbook_repo = repo
   end
 end
 
@@ -40,42 +49,108 @@ Then /^the local Cookbook repository exists$/ do
   #TODO: check_file_presence([file], true), etc.
 end
 
-Given /^a cookbook path "([^"]*)"$/ do |path|
+Then /^the local Site\-Cookbook repository exists$/ do
+  chef.local_site_cookbook_repo.exist?.should be_true
+  #TODO: check_file_presence([file], true), etc.
+end
+
+Given /^a Cookbook path "([^"]*)"$/ do |dir_name|
+  create_dir(dir_name)
   in_current_dir do
-    update_cookbook_paths(path, true)
+    update_cookbook_paths(dir_name, true)
   end
 end
 
-Given /^a cookbooks path "([^"]*)"$/ do |path|
+Given /^a Cookbooks path "([^"]*)"$/ do |dir_name|
+  create_dir(dir_name)
   in_current_dir do
-    update_cookbook_paths(path, false)
+    update_cookbook_paths(dir_name, false)
+  end
+end
+
+Then /^the local Cookbook "([^"]*)" exists$/ do |ckbk|
+  chef.cookbook_paths.each do |pn|
+    curr_ckbk = pn.basename.to_s
+    if curr_ckbk == ckbk
+      break true if curr_ckbk.should == ckbk
+    end
   end
+  #TODO: check_file_presence([file], true), etc.
 end
 
-Then /^the local cookbook "([^"]*)" exists$/ do |ckbk|
+Then /^the local Site\-Cookbook "([^"]*)" exists$/ do |ckbk|
+  ckbk_path = 'site-cookbook'
   chef.cookbook_paths.each do |pn|
     curr_ckbk = pn.basename.to_s
-    curr_ckbk.should == ckbk if curr_ckbk == ckbk
+    if pn.to_s[/#{ckbk_path}}/] && curr_ckbk == ckbk
+      break true if curr_ckbk.should == ckbk
+    end
   end
   #TODO: check_file_presence([file], true), etc.
 end
 
+And /^these local Cookbooks exist:$/ do |table|
+  # table is a Cucumber::Ast::Table
+  table.hashes.each do |hsh|
+    chef.cookbook_paths.each do |pn|
+      curr_ckbk = pn.basename.to_s
+      if curr_ckbk == hsh['cookbook']
+        break true if curr_ckbk.should == hsh['cookbook']
+      end
+    end
+  end
+end
+
+And /^these local Site\-Cookbooks exist:$/ do |table|
+  # table is a Cucumber::Ast::Table
+  ckbk_path = 'site-cookbook'
+  table.hashes.each do |hsh|
+    chef.cookbook_paths.each do |pn|
+      curr_ckbk = pn.basename.to_s
+      if pn.to_s[/#{ckbk_path}}/] && curr_ckbk == hsh['cookbook']
+        break true if curr_ckbk.should == hsh['cookbook']
+      end
+    end
+  end
+end
+
 Given /^I clone the remote Cookbook repository branch "([^"]*)" to "([^"]*)"$/ do |brnch, ckbk_path|
-  chef.local_cookbook_repo = chef_clone_repo(ckbk_path, true, chef.remote_cookbook_repo, brnch)
+  if ckbk_path[/\/cookbooks\//]
+    chef.local_cookbook_repo = chef_clone_repo(ckbk_path, true, chef.remote_cookbook_repo, brnch)
+  elsif ckbk_path[/\/site-cookbooks\//]
+    chef.local_site_cookbook_repo = chef_clone_repo(ckbk_path, true, chef.remote_cookbook_repo, brnch)
+  end
+end
+
+Given /^I clone the Cookbook "([^"]*)" branch "([^"]*)" to "([^"]*)"$/ do |ckbk, brnch, ckbk_path|
+  if ckbk_path[/\/cookbooks\//]
+    chef.local_cookbook_repo = chef_clone_repo(ckbk_path, true, chef.cookbooks_uri + ckbk + '.git', brnch)
+  elsif ckbk_path[/\/site-cookbooks\//]
+    chef.local_site_cookbook_repo = chef_clone_repo(ckbk_path, true, chef.cookbooks_uri + ckbk + '.git', brnch)
+  end
 end
 
-When /^I successfully generate all cookbook metadata$/ do
+When /^I clone the Cookbooks:$/ do |table|
+  # table is a Cucumber::Ast::Table
+  table.hashes.each do |hsh|
+    #TODO: Accept cloning from tag and reference
+    local_repo = chef_clone_repo(hsh['destination'], true, chef.cookbooks_uri + hsh['cookbook'] + '.git', hsh['branch'])
+    Pathname(local_repo).exist?.should be_true
+  end
+end
+
+When /^I successfully generate all Cookbook metadata$/ do
   chef.cookbook_paths.each do |pn|
     curr_ckbk = pn.basename.to_s
     run_knife_command("cookbook metadata #{curr_ckbk}")
   end
 end
 
-When /^I successfully generate cookbook "([^"]*)" metadata$/ do |ckbk|
+When /^I successfully generate Cookbook "([^"]*)" metadata$/ do |ckbk|
   chef.cookbook_paths.each do |pn|
     curr_ckbk = pn.basename.to_s
     if curr_ckbk == ckbk
       run_knife_command("cookbook metadata #{curr_ckbk}")
     end
   end
-end
+end

data/lib/cuken/cucumber/chef/data_bag.rb

@@ -0,0 +1,27 @@
+###
+# Given
+###
+#Given /^I create the "([^"]*)" Data Bag$/ do |arg1|
+#  pending # express the regexp above with the code you wish you had
+#end
+
+###
+# When
+###
+#When /^I add these Cookbook Data Bag items:$/ do |table|
+#  # table is a Cucumber::Ast::Table
+#  pending # express the regexp above with the code you wish you had
+#end
+
+###
+# Then
+###
+#Then /^these Data Bags exist:$/ do |table|
+#  # table is a Cucumber::Ast::Table
+#  pending # express the regexp above with the code you wish you had
+#end
+#
+#Then /^the Data Bags contain:$/ do |table|
+#  # table is a Cucumber::Ast::Table
+#  pending # express the regexp above with the code you wish you had
+#end

data/lib/cuken/cucumber/chef/knife.rb

@@ -19,12 +19,22 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
+World(::Cuken::Api::Chef::Knife)
+
 Given /^the Knife file "([^"]*)"$/ do |path|
   in_current_dir do
     chef.client_knife_path = Pathname(path).expand_path.realdirpath
   end
 end
 
+When /^I delete the Chef admin client "([^"]*)"$/ do |client_name|
+  delete_client(client_name)
+end
+
+When /^I create the Chef admin client "([^"]*)"$/ do |client_name|
+  create_client(client_name)
+end
+
 When /^I successfully run Knife's "([^"]*)"$/ do |cmd|
   run_knife_command(cmd, false)
 end

data/lib/cuken/cucumber/chef/node.rb

@@ -19,17 +19,29 @@
 ###
 # Given
 ###
-Given /^a validated node$/ do
+Given /^a validated Node$/ do
   # client should have cached ohai assigned to it
   client.register
   client.build_node
   client.node.run_list << "integration_setup"
 end
 
-###
-# Then
-###
-Then /^the nodes are:$/ do |partial_output|
-  run_knife_command('node list')
-  all_stdout.should include(partial_output)
-end
+#Given /^a Node "([^"]*)"$/ do |arg1|
+#  pending # express the regexp above with the code you wish you had
+#end
+#
+####
+## When
+####
+#When /^I add these Node Roles to the Run Lists:$/ do |table|
+#  # table is a Cucumber::Ast::Table
+#  pending # express the regexp above with the code you wish you had
+#end
+#
+####
+## Then
+####
+#Then /^the Chef nodes are:$/ do |partial_output|
+#  run_knife_command('node list')
+#  all_stdout.should include(partial_output)
+#end

data/lib/cuken/cucumber/chef/role.rb

@@ -1,7 +1,24 @@
+###
+# Given
+###
+
+###
+# When
+###
+#When /^I load the Cookbook Roles:$/ do |table|
+#  # table is a Cucumber::Ast::Table
+#  pending # express the regexp above with the code you wish you had
+#end
+
 ###
 # Then
 ###
-Then /^the roles are:$/ do |partial_output|
+Then /^the Roles are:$/ do |partial_output|
   run_knife_command('role list')
   all_stdout.should include(partial_output)
 end
+
+#Then /^these Roles exist:$/ do |table|
+#  run_knife_command('role list')
+#  pending # express the regexp above with the code you wish you had
+#end

data/lib/cuken/cucumber/chef.rb

@@ -5,14 +5,16 @@ World(::Cuken::Api::Chef)
 
 Before do
   @aruba_timeout_seconds.nil? || @aruba_timeout_seconds < 20 ? @aruba_timeout_seconds = 20 : @aruba_timeout_seconds
+  ::Grit::Git.git_timeout = 3610
 end
 
-Before('@work_in_cwd') do
-  @dirs = [Pathname.getwd.expand_path.realpath.to_s]
-end
+#Before('@work_in_cwd') do
+#  @dirs = [Pathname.getwd.expand_path.realpath.to_s]
+#end
 
 require 'cuken/cucumber/chef/common'
 require 'cuken/cucumber/chef/knife'
 require 'cuken/cucumber/chef/cookbook'
 require 'cuken/cucumber/chef/node'
 require 'cuken/cucumber/chef/role'
+require 'cuken/cucumber/chef/data_bag'

data/lib/cuken/cucumber/cmd.rb

@@ -3,7 +3,7 @@ load 'aruba/cucumber.rb' unless defined? ::Aruba
 World(::Cuken::Api::Cmd)
 
 #
-# These have been submitted to Aruba - awaiting release...
+# These ought be submitted to Aruba...
 #
 Then /^the output from "(.*)" contains exactly:$/ do |cmd, exact_output|
   output_from(unescape(cmd)).should == exact_output

data/lib/cuken/cucumber/file.rb

@@ -4,14 +4,30 @@ World(::Cuken::Api::File)
 #
 # Refactorings, yet to be submitted to Aruba:
 #
-Given /^the empty file "([^"]*)"$/ do |file_name|
+Given /^the directory "([^"]*)"$/ do |arg1|
+  Then %Q{a directory named \"#{arg1}\"}
+end
+
+Given /^the file "([^"]*)" contains nothing$/ do |file_name|
   Given %Q{an empty file named "#{file_name}"}
 end
 
-Given /^the file "([^"]*)" with:$/ do |file_name, file_content|
+Given /^the file "([^"]*)" contains:$/ do |file_name, file_content|
   Given %Q{a file named "#{file_name}" with:}, file_content
 end
 
+When /^I write to "([^"]*)":$/ do  |file_name, file_content|
+  When %Q{I write to "#{file_name}" with:}, file_content
+end
+
+When /^I overwrite "([^"]*)":$/ do  |file_name, file_content|
+  When %Q{I overwrite "#{file_name}" with:}, file_content
+end
+
+When /^I append to "([^"]*)":$/ do  |file_name, file_content|
+  When %Q{I append to "#{file_name}" with:}, file_content
+end
+
 Then /^the file "([^"]*)" exists$/ do |arg1|
   Then %Q{a file named \"#{arg1}\" should exist}
 end

data/lib/cuken/cucumber/rvm.rb

@@ -0,0 +1,4 @@
+load 'aruba/cucumber.rb' unless defined? ::Aruba
+
+World(::Cuken::Api::Rvm)
+

data/lib/cuken/rvm.rb

@@ -0,0 +1,3 @@
+require 'cuken/common'
+require 'cuken/api/rvm'
+require 'cuken/cucumber/rvm'