lib/ansible_spec.rb in ansible_spec-0.0.1.4 vs lib/ansible_spec.rb in ansible_spec-0.1

@@ -1,6 +1,7 @@
 require "ansible_spec/version"
+require "ansible_spec/load_ansible"
 require "fileutils"
 
 # Reference
 # https://github.com/serverspec/serverspec/blob/master/lib/serverspec/setup.rb
 # Reference License (MIT)
@@ -14,157 +15,28 @@
     safe_create_ansiblespec
   end
 
 
   def self.safe_create_spec_helper
-    content = <<'EOF'
-require 'serverspec'
-require 'pathname'
-require 'net/ssh'
-
-include SpecInfra::Helper::Ssh
-include SpecInfra::Helper::DetectOS
-
-RSpec.configure do |c|
-  if ENV['ASK_SUDO_PASSWORD']
-    require 'highline/import'
-    c.sudo_password = ask("Enter sudo password: ") { |q| q.echo = false }
-  else
-    c.sudo_password = ENV['SUDO_PASSWORD']
-  end
-  c.before :all do
-    block = self.class.metadata[:example_group_block]
-    if RUBY_VERSION.start_with?('1.8')
-      file = block.to_s.match(/.*@(.*):[0-9]+>/)[1]
-    else
-      file = block.source_location.first
-    end
-    host  = ENV['TARGET_HOST']
-    if c.host != host
-      c.ssh.close if c.ssh
-      c.host  = host
-      options = Net::SSH::Config.for(c.host)
-      user    = ENV['TARGET_USER']
-      options[:keys] = ENV['TARGET_PRIVATE_KEY']
-      c.ssh   = Net::SSH.start(host, user, options)
-    end
-  end
-end
-
-EOF
+    content = File.open(File.dirname(__FILE__) + "/../lib/src/spec/spec_helper.rb").read
     safe_mkdir("spec")
     safe_touch("spec/spec_helper.rb")
     File.open("spec/spec_helper.rb", 'w') do |f|
       f.puts content
     end
 
   end
 
   def self.safe_create_rakefile
-    content = <<'EOF'
-require 'rake'
-require 'rspec/core/rake_task'
-require 'yaml'
-
-# param: none
-# return: @playbook, @inventoryfile
-def load_ansiblespec()
-  f = '.ansiblespec'
-  if File.exist?(f)
-    y = YAML.load_file(f)
-    @playbook = y[0]['playbook']
-    @inventoryfile = y[0]['inventory']
-  else
-    @playbook = 'site.yml'
-    @inventoryfile = 'hosts'
-  end
-  if File.exist?(@playbook) == false
-    puts 'Error: ' + @playbook + ' is not Found. create site.yml or /.ansiblespec  See https://github.com/volanja/ansible_spec'
-    exit 1
-  elsif File.exist?(@inventoryfile) == false
-    puts 'Error: ' + @inventoryfile + ' is not Found. create hosts or /.ansiblespec  See https://github.com/volanja/ansible_spec'
-    exit 1
-  end
-end
-
-# param: inventory file of Ansible
-# return: Hash {"active_group_name" => ["192.168.0.1","192.168.0.2"]}
-def load_host(file)
-  hosts = File.open(file).read
-  active_group = Hash.new
-  active_group_name = ''
-  hosts.each_line{|line|
-    line = line.chomp
-    next if line.start_with?('#')
-    if line.start_with?('[') && line.end_with?(']')
-      active_group_name = line.gsub('[','').gsub(']','')
-      active_group["#{active_group_name}"] = Array.new
-    elsif active_group_name.empty? == false
-      next if line.empty? == true
-      active_group["#{active_group_name}"] << line
-    end
-  }
-  return active_group
-end
-
-# main
-load_ansiblespec
-load_file = YAML.load_file(@playbook)
-
-# e.g. comment-out
-if load_file === false
-  puts 'Error: No data in site.yml'
-  exit
-end
-
-properties = Array.new
-load_file.each do |site|
-  if site.has_key?("include")
-    properties.push YAML.load_file(site["include"])[0]
-  else
-    properties.push site
-  end
-end
-
-
-#load inventry file
-hosts = load_host(@inventoryfile)
-properties.each do |var|
-  if hosts.has_key?("#{var["hosts"]}")
-    var["hosts"] = hosts["#{var["hosts"]}"]
-  end
-end
-
-namespace :serverspec do
-  properties.each do |var|
-    var["hosts"].each do |host|
-      desc "Run serverspec for #{var["name"]}"
-      RSpec::Core::RakeTask.new(var["name"].to_sym) do |t|
-        puts "Run serverspec for #{var["name"]} to #{host}"
-        ENV['TARGET_HOST'] = host
-        ENV['TARGET_PRIVATE_KEY'] = '~/.ssh/id_rsa'
-        ENV['TARGET_USER'] = var["user"]
-        t.pattern = 'roles/{' + var["roles"].join(',') + '}/spec/*_spec.rb'
-      end
-    end
-  end
-end
-
-EOF
+    content = File.open(File.dirname(__FILE__) + "/../lib/src/Rakefile").read
     safe_touch("Rakefile")
     File.open("Rakefile", 'w') do |f|
       f.puts content
     end
-    
   end
 
   def self.safe_create_ansiblespec
-    content = <<'EOF'
---- 
-- 
-  playbook: site.yml
-  inventory: hosts
-EOF
+    content = File.open(File.dirname(__FILE__) + "/../lib/src/.ansiblespec").read
     safe_touch(".ansiblespec")
     File.open(".ansiblespec", 'w') do |f|
       f.puts content
     end
   end