kytoon 1.0.0

data/.document

@@ -0,0 +1,5 @@
+lib/**/*.rb
+bin/*
+- 
+features/**/*.feature
+LICENSE.txt

data/Gemfile

@@ -0,0 +1,15 @@
+source "http://rubygems.org"
+# Add dependencies required to use your gem here.
+# Example:
+#   gem "activesupport", ">= 2.3.5"
+
+# Add dependencies to develop your gem here.
+# Include everything needed to run rake, tests, features, etc.
+group :development do
+  gem "rdoc", "~> 3.12"
+  gem "bundler", "~> 1.1.4"
+  gem "builder", "~> 3.0.0"
+  gem "jeweler", "~> 1.8.3"
+  gem "uuidtools", "~> 2.1.2"
+  gem "mocha", "~> 0.12.1"
+end

data/Gemfile.lock

@@ -0,0 +1,29 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    builder (3.0.0)
+    git (1.2.5)
+    jeweler (1.8.4)
+      bundler (~> 1.0)
+      git (>= 1.2.5)
+      rake
+      rdoc
+    json (1.7.3)
+    metaclass (0.0.1)
+    mocha (0.12.1)
+      metaclass (~> 0.0.1)
+    rake (0.9.2.2)
+    rdoc (3.12)
+      json (~> 1.4)
+    uuidtools (2.1.2)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  builder (~> 3.0.0)
+  bundler (~> 1.1.4)
+  jeweler (~> 1.8.3)
+  mocha (~> 0.12.1)
+  rdoc (~> 3.12)
+  uuidtools (~> 2.1.2)

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2012 Red Hat Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,81 @@
+= Kytoon
+
+Create & configure ephemeral virtual private clouds
+
+== Description
+
+A set of Rake tasks that provide a framework to help automate the creation and configuration of “identical” VPC server groups. Kytoon provides the ability to create projects that can be used by team members and continuous integration systems to spin up groups of servers for development and/or testing. Configuration information is stored in JSON and YAML formats which can be easily parsed, edited, and version controlled.
+
+Inspired by and based on the Chef VPC Toolkit.
+
+== Supports
+
+- Cloud Servers VPC (Rackspace and OpenStack)
+- XenServer
+
+== Installation
+
+1) Gem install kytoon.
+
+2) Add Kytoon tasks to your projects Rakefile:
+
+	KYTOON_PROJECT = "#{File.dirname(__FILE__)}" unless defined?(KYTOON_PROJECT)
+	require 'rubygems'
+	require 'kytoon'
+	include Kytoon
+
+3) Create a .kytoon.conf file in your HOME directory that contains the following:
+
+	# Set one of the following group_types
+	group_type: cloud_server_vpc
+	#group_type: xenserver
+
+	# Cloud Servers VPC credentials
+	cloud_servers_vpc_url: https://your.vpc.url/
+	cloud_servers_vpc_username: <username>
+	cloud_servers_vpc_password: <password>
+
+== Examples
+
+Example commands:
+
+        - Create a new server group.
+
+                $ rake group:create
+
+        - List your currently running server groups.
+
+                $ rake group:list
+
+        - SSH into the current (most recently created) server group
+
+                $ rake ssh
+
+        - SSH into a server group with an ID of 3
+
+                $ rake ssh GROUP_ID=3
+
+        - Delete the server group with an ID of 3
+
+                $ rake group:delete GROUP_ID=3
+
+
+== Bash Automation Script
+
+The following is an example bash script to spin up a group and run commands via SSH.
+
+        #!/bin/bash
+        trap "rake group:delete" INT TERM EXIT # cleanup the group on exit
+
+        # create a server group
+        rake group:create
+
+        # Run some scripts on the login server
+        rake ssh bash <<-EOF_BASH
+                echo 'It works!'
+        EOF_BASH
+
+
+== Copyright
+
+Copyright (c) 2012 Red Hat Inc. See LICENSE.txt for further details.

data/Rakefile

@@ -0,0 +1,35 @@
+require 'rake/testtask'
+
+KYTOON_PROJECT = "#{File.dirname(__FILE__)}" unless defined?(KYTOON_PROJECT)
+
+$:.unshift File.join(File.dirname(__FILE__),'lib')
+require 'kytoon'
+include Kytoon
+
+Dir[File.join(File.dirname(__FILE__), 'rake', '*.rake')].each do  |rakefile|
+        import(rakefile)
+end
+
+Rake::TestTask.new(:test) do |t|
+        t.pattern = 'test/*_test.rb'
+        t.verbose = true
+end
+Rake::Task['test'].comment = "Unit"
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gemspec|
+    gemspec.name = "kytoon"
+    gemspec.summary = "Create & configure ephemeral virtual private clouds."
+    gemspec.description = "A set of Rake tasks that provide a framework to help automate the creation and configuration of VPC server groups."
+    gemspec.email = "dprince@redhat.com"
+    gemspec.homepage = "http://github.com/dprince/kytoon"
+    gemspec.authors = ["Dan Prince"]
+    gemspec.add_dependency 'rake'
+    gemspec.add_dependency 'builder'
+    gemspec.add_dependency 'json'
+    gemspec.add_dependency 'uuidtools'
+  end
+rescue LoadError
+  puts "Jeweler not available. Install it with: sudo gem install jeweler"
+end

data/VERSION

@@ -0,0 +1 @@
+1.0.0

data/config/server_group_vpc.json

@@ -0,0 +1,14 @@
+{
+	"name": "Fedora",
+	"domain_name": "vpc",
+	"description": "Fedora",
+	"vpn_device": "tap",
+	"vpn_proto": "udp",
+	"servers": {
+		"login": {
+			"image_id": "24344414",
+			"flavor_id": "3",
+			"openvpn_server": "true"
+		}
+	}
+}

data/config/server_group_xen.json

@@ -0,0 +1,24 @@
+{
+    "name": "Fedora",
+    "netmask": "255.255.255.0",
+    "gateway": "192.168.0.1",
+    "broadcast": "192.168.0.127",
+    "dns_nameserver": "8.8.8.8",
+    "network_type": "static",
+    "public_ip_bridge": "xenbr0",
+    "bridge": "xenbr1",
+    "servers": [
+        {
+            "hostname": "login",
+            "image_path": "/images/fedora-agent2.xva",
+            "ip_address": "192.168.0.2",
+            "mac": "e2:6d:71:67:7e:66"
+        },
+        {
+            "hostname": "nova1",
+            "image_path": "/images/fedora-agent2.xva",
+            "ip_address": "192.168.0.3",
+            "mac": "e2:ad:a1:a7:ae:67"
+        }
+    ]
+}

data/lib/kytoon.rb

@@ -0,0 +1,8 @@
+require 'kytoon/util'
+require 'kytoon/ssh_util'
+require 'kytoon/version'
+require 'kytoon/xml_util'
+require 'kytoon/server_group'
+require 'kytoon/vpn/vpn_connection'
+require 'kytoon/vpn/vpn_network_manager'
+require 'kytoon/vpn/vpn_openvpn'

data/lib/kytoon/providers/cloud_servers_vpc.rb

@@ -0,0 +1,6 @@
+require 'kytoon/providers/cloud_servers_vpc/connection'
+require 'kytoon/providers/cloud_servers_vpc/client'
+require 'kytoon/providers/cloud_servers_vpc/server'
+require 'kytoon/providers/cloud_servers_vpc/server_group'
+require 'kytoon/providers/cloud_servers_vpc/ssh_public_key'
+require 'kytoon/providers/cloud_servers_vpc/vpn_network_interface'

data/lib/kytoon/providers/cloud_servers_vpc/client.rb

@@ -0,0 +1,197 @@
+module Kytoon
+
+module Providers
+
+module CloudServersVPC
+
+class Client
+
+  @@data_dir=File.join(KYTOON_PROJECT, "tmp", "clients")
+
+  def self.data_dir
+    @@data_dir
+  end
+
+  def self.data_dir=(dir)
+    @@data_dir=dir
+  end
+
+  attr_accessor :id
+  attr_accessor :name
+  attr_accessor :description
+  attr_accessor :status
+  attr_accessor :server_group_id
+  attr_accessor :cache_file
+
+  def initialize(options={})
+    @id=options[:id].to_i
+    @name=options[:name]
+    @description=options[:description]
+    if options[:status]
+      @status=options[:status]
+    else
+      @status = "Pending"
+    end
+    @status=options[:status] or @status = "Pending"
+    @server_group_id=options[:server_group_id]
+    if options[:cache_file] then
+      @cache_file=options[:cache_file]
+    else
+      @cache_file=options[:server_group_id]
+    end
+    @vpn_network_interfaces=[]
+  end
+
+  def vpn_network_interfaces
+    @vpn_network_interfaces
+  end
+
+    def cache_to_disk
+        FileUtils.mkdir_p(@@data_dir)
+        File.open(File.join(@@data_dir, "#{@cache_file}.xml"), 'w') do |f|
+            f.chmod(0600)
+            f.write(self.to_xml)
+        end
+    end
+
+  def delete
+    client_xml_file=File.join(@@data_dir, "#{@cache_file}.xml")
+        if File.exists?(client_xml_file) then
+            File.delete(client_xml_file)
+        end
+  end
+
+  def self.from_xml(xml)
+    client=nil
+    dom = REXML::Document.new(xml)
+    REXML::XPath.each(dom, "/client") do |cxml|
+
+      client=Client.new(
+        :id => XMLUtil.element_text(cxml,"id").to_i,
+        :name => XMLUtil.element_text(cxml, "name"),
+        :description => XMLUtil.element_text(cxml,"description"),
+        :status => XMLUtil.element_text(cxml,"status"),
+        :server_group_id => XMLUtil.element_text(cxml, "server-group-id").to_i
+      )
+      REXML::XPath.each(dom, "//vpn-network-interface") do |vni|
+        vni = VpnNetworkInterface.new(
+          :id => XMLUtil.element_text(vni, "id"),
+          :vpn_ip_addr => XMLUtil.element_text(vni, "vpn-ip-addr"),
+          :ptp_ip_addr => XMLUtil.element_text(vni, "ptp-ip-addr"),
+          :client_key => XMLUtil.element_text(vni, "client-key"),
+          :client_cert => XMLUtil.element_text(vni, "client-cert"),
+          :ca_cert => XMLUtil.element_text(vni, "ca-cert")
+        )
+        client.vpn_network_interfaces << vni
+      end
+    end
+    client
+  end 
+
+    def to_xml
+
+        xml = Builder::XmlMarkup.new
+        xml.tag! "client" do |sg|
+            sg.id(@id)
+            sg.name(@name)
+            sg.description(@description)
+            sg.status(@status)
+            sg.tag! "server-group-id", @server_group_id
+            sg.tag! "vpn-network-interfaces", {"type" => "array"} do |interfaces|
+        @vpn_network_interfaces.each do |vni|
+          interfaces.tag! "vpn-network-interface" do |xml_vni|
+            xml_vni.id(vni.id)
+            xml_vni.tag! "vpn-ip-addr", vni.vpn_ip_addr
+            xml_vni.tag! "ptp-ip-addr", vni.ptp_ip_addr
+            xml_vni.tag! "client-key", vni.client_key
+            xml_vni.tag! "client-cert", vni.client_cert
+            xml_vni.tag! "ca-cert", vni.ca_cert
+          end  
+        end
+      end
+
+    end
+        xml.target!
+
+  end
+
+
+    # Poll the server group until it is online.
+    # :timeout - max number of seconds to wait before raising an exception.
+    #            Defaults to 1500
+    def poll_until_online(options={})
+
+        timeout=options[:timeout] or timeout = ENV['VPN_CLIENT_TIMEOUT']
+        if timeout.nil? or timeout.empty? then
+            timeout=300 # defaults to 5 minutes
+        end 
+
+    online = false
+    count=0
+    until online or (count*5) >= timeout.to_i do
+      count+=1
+      begin
+        client=Client.get(:id => @id, :source => "remote")
+
+        if client.status == "Online" then
+          online = true
+        else
+          yield client if block_given?
+          sleep 5
+        end
+      rescue EOFError
+      end
+    end
+    if (count*20) >= timeout.to_i then
+      raise "Timeout waiting for client to come online."
+    end
+
+  end
+
+  def self.create(server_group, client_name, cache_to_disk=true)
+
+    xml = Builder::XmlMarkup.new
+    xml.client do |client|
+      client.name(client_name)
+      client.description("Toolkit Client: #{client_name}")
+      client.tag! "server-group-id", server_group.id
+    end
+
+    xml=Connection.post("/clients.xml", xml.target!)
+    client=Client.from_xml(xml)
+    client.cache_to_disk if cache_to_disk
+    client
+
+  end
+
+    # Get a client. The following options are available:
+    #
+    # :id - The ID of the client to get.
+    # :source - valid options are 'remote' and 'cache'
+    #
+  def self.get(options = {})
+
+        source = options[:source] or source = "remote"
+
+        if source == "remote" then
+      id=options[:id] or raise "Please specify a Client ID."
+      xml=Connection.get("/clients/#{id}.xml")
+      Client.from_xml(xml)
+    elsif source == "cache" then
+      id=options[:id] or id = ENV['GROUP_ID']
+      client_xml_file=File.join(@@data_dir, "#{id}.xml")
+      raise "No client files exist." if not File.exists?(client_xml_file)
+      Client.from_xml(IO.read(client_xml_file))
+    else
+      raise "Invalid get :source specified."
+    end
+
+  end
+
+end
+
+end
+
+end
+
+end