asteroid 0.0.1

data/lib/asteroid/provider/mock.rb

@@ -0,0 +1,51 @@
+module Asteroid
+  module Provider
+    class Mock < Abstract
+
+      def self.ip_address
+        (0..3).to_a.map{|a| rand(255)}.join(".")
+      end
+
+      def initialize(config = {})
+        @config = config || {}
+        @config[:filename] ||= File.join(File.dirname(__FILE__), '../../../world.yml')
+          
+        begin
+          @world = YAML::load_file @config[:filename]
+        rescue
+          puts "No such file #{@config[:filename]}"
+          @world = {}
+        end
+      end
+
+      def instances
+        @config[:instances] || []
+      end
+      
+      def create_instance(server)
+        instance = {
+          name: server.generate_instance_name,
+          id: SecureRandom.hex(5),
+          ip_address: self.class.ip_address,
+          provider: self,
+          type: server.type,
+          server: server
+        }
+        Instance.new instance
+      end
+
+      def destroy_instance(instance)
+
+      end
+
+      private
+
+        def save_world
+          File.open(@config[:filename], 'w') do |f|
+            f.write @world.to_yaml
+          end
+        end
+
+    end
+  end
+end

data/lib/asteroid/provider/virtual_box.rb

@@ -0,0 +1,43 @@
+# :name=>"web-1c5c2972",
+# :image_id=>3101918,
+# :size_id=>66,
+# :region_id=>4,
+# :backups_active=>false,
+# :ip_address=>"107.170.109.79",
+# :private_ip_address=>"10.128.197.52",
+# :locked=>false,
+# :status=>"active",
+# :created_at=>"2014-04-12T20:44:53Z"}
+
+module Asteroid
+  module Provider
+    class VirtualBox
+
+      def initialize(options = {})
+      end
+
+      def self.type
+        :virtual_box
+      end
+
+      def instances
+        instances = `VBoxManage list vms`.split /\n/
+        instances.map do |line|
+          matches = line.match /\"(\S+)\" {(\S+)}/
+          name, uuid = matches[0], matches[1]
+          ip_address = `VBoxManage guestproperty get "#{uuid}" "/VirtualBox/GuestInfo/Net/0/V4/IP" | awk '{ print $2 }'`
+          {
+            id: uuid,
+            name: name,
+            ip_address: ip_address,
+            provider: self
+          }
+        end
+      end
+      
+      def destroy_instance(instance)
+      end
+
+    end
+  end
+end

data/lib/asteroid/script.rb

@@ -0,0 +1,64 @@
+require 'erb'
+
+module Asteroid
+  class Script
+
+    def initialize(filename)
+      @filename = filename
+    end
+
+    def set_data(data)
+      @data = data
+    end
+
+    def data
+      @data ||= {}
+    end
+
+    def name
+      File.basename @filename
+    end
+
+    def file_type
+      @file_type ||= @filename.split('.').last
+    end
+
+    def yml?
+      file_type == "yml"
+    end
+
+    def aster?
+      file_type == "aster"
+    end
+
+    def template?
+      @filename.split('.').last == "erb"
+    end
+
+    def render(locals = {})
+      script = File.read @filename
+      if template?
+        Template.new(:erb).render(
+          script,
+          self.data.merge(locals)
+        )
+      else
+        script
+      end
+    end
+
+    def self.named(name)
+      all.select{|s| s.name == name}.first
+    end
+
+    def self.all
+      files = Dir[Asteroid::Config.script_dir + "/*"]
+      files.map do |filename|
+        new(filename)
+      end
+    end
+
+
+
+  end
+end

data/lib/asteroid/server.rb

@@ -0,0 +1,96 @@
+require "active_support/hash_with_indifferent_access"
+require "active_support/core_ext/hash"
+
+
+module Asteroid
+  class Server
+
+    attr_accessor :scripts, 
+                  :configure_steps
+
+    def self.named(name)
+      data = YAML::load_file(Asteroid::Config.server_dir + "/#{name}.yml")
+      new data
+    end
+
+    def initialize(data = {})
+      data = ActiveSupport::HashWithIndifferentAccess.new(data)
+      
+      @provider = if data[:provider]
+        Provider.named data[:provider]
+      else
+        Config.providers.first
+      end
+
+      
+      @attributes = data
+
+      @config = data["config"] || {
+        "admin_username" => "ubuntu",
+        "ssh_port" => 1983
+      }
+
+      @default_config = data["default_instance_config"] || {
+        "login.username" => "root", 
+        "ssh.port" => "22"
+      }
+    end
+
+
+    def name
+      @attributes[:name]
+    end
+
+    def attributes
+      @attributes
+    end
+
+    def commands
+      @commands
+    end
+
+    def [](k)
+      config[k.to_s]
+    end
+
+    def config
+      @config
+    end
+
+    def type
+      name.to_sym
+    end
+
+    def first_instance
+      Asteroid::Instance.all_with_type(self.type).first
+    end
+
+    def instance_config
+      @default_config
+    end
+
+    def ssh_key_filename
+      File.join Asteroid::Config.ssh_key_dir, @ssh_keys.first
+    end
+
+    def provider
+      @provider
+    end
+
+    def generate_instance_name(subname = nil)
+      "#{self.name}-#{subname || SecureRandom.hex(4)}"
+    end
+
+    def create_instance!(subname = nil)
+  
+      #key_ids = SSHKey.all.select do |k|
+      #  @ssh_keys.include?(k.name)
+      #end.map do |k|
+      #  k.id
+      #end
+
+      provider.create_instance(self)
+    end
+
+  end
+end

data/lib/asteroid/ssh_key.rb

@@ -0,0 +1,41 @@
+require 'open-uri'
+
+module Asteroid
+  class SSHKey
+
+    attr_accessor :id, :name, :ssh_pub_key
+
+    def initialize(options = {})
+      @id = options[:id]
+      @name = options[:name]
+    end
+
+    def exists?
+      names = self.class.all.map{|k| k.name }
+      names.include? @name
+    end
+
+    def self.find(name)
+      name = name.to_s
+      all.select{|k| k.name == name}.first
+    end
+
+    def self.all
+      keys = Provider.all.ssh_keys
+      keys.map do |key|
+        new id: key[:id], name: key[:name]
+      end
+    end
+
+    def save
+      if exists?
+        puts "Key exists"
+      else 
+        Digitalocean::SshKey.create(
+          name: @name, 
+          ssh_pub_key: URI::escape(@ssh_pub_key)
+        )
+      end
+    end
+  end
+end

data/lib/asteroid/template.rb

@@ -0,0 +1,25 @@
+module Asteroid
+  class Template
+
+    def initialize(type = :erb)
+      @type = type
+    end
+
+    def render(text, locals = {})
+      if @type == :erb
+        render_erb(text, locals)
+      else
+        raise "Unknown teplate: #{@type}"
+      end
+    end
+
+    private
+
+      def render_erb(text, locals = {})
+        ERB.new(text).result(
+          data = OpenStruct.new(locals).instance_eval{ binding }
+        )
+      end
+
+  end
+end

data/lib/asteroid/version.rb

@@ -0,0 +1,3 @@
+module Asteroid
+  VERSION = "0.0.1"
+end

data/test/helper.rb

@@ -0,0 +1,8 @@
+require File.join(File.dirname(__FILE__), *%w(.. lib asteroid))
+
+require "minitest/autorun"
+
+
+Asteroid::Config.providers << Asteroid::Provider::Mock.new
+
+

data/test/unit/test_instance.rb

@@ -0,0 +1,18 @@
+require_relative '../helper'
+
+class TestInstance < Minitest::Test
+  include Asteroid
+
+  def test_get_all_instances
+    instances = Instance.all
+    assert_kind_of(Array, instances)
+  end
+
+  def initialize_instance
+    instance = Instance.new
+  end
+
+
+
+
+end

data/test/unit/test_server.rb

@@ -0,0 +1,19 @@
+require_relative '../helper'
+
+class TestServer < Minitest::Test
+  include Asteroid
+
+  def test_create_instance_server
+    s = Server.new name: "web"
+    instance = s.create_instance!
+    assert_kind_of(Instance, instance)
+    assert_equal(instance.server, s)
+  end
+
+  def test_server_type
+    s = Server.new name: "web"
+    assert_equal :web, s.type
+  end
+
+
+end

metadata

@@ -0,0 +1,241 @@
+--- !ruby/object:Gem::Specification
+name: asteroid
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Sam Murphy
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-04-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-scp
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: aster
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: digitalocean
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: autotest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email:
+- sam.murphy@gmail.com
+executables:
+- asteroid
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- asteroid.gemspec
+- bin/asteroid
+- generator/README.md
+- generator/Rakefile
+- generator/asteroid/servers/default.yml
+- generator/config/asteroid.rb
+- generator/gitignore
+- generator/secrets/.keep
+- generator/secrets/config/providers.rb
+- generator/secrets/config/secrets.yml
+- lib/asteroid.rb
+- lib/asteroid/application.rb
+- lib/asteroid/config.rb
+- lib/asteroid/config_file.rb
+- lib/asteroid/generator.rb
+- lib/asteroid/instance.rb
+- lib/asteroid/instance/aster.rb
+- lib/asteroid/instance/command.rb
+- lib/asteroid/instance/scp.rb
+- lib/asteroid/instance/ssh.rb
+- lib/asteroid/instance/vars.rb
+- lib/asteroid/provider.rb
+- lib/asteroid/provider/abstract.rb
+- lib/asteroid/provider/digital_ocean.rb
+- lib/asteroid/provider/mock.rb
+- lib/asteroid/provider/virtual_box.rb
+- lib/asteroid/script.rb
+- lib/asteroid/server.rb
+- lib/asteroid/ssh_key.rb
+- lib/asteroid/template.rb
+- lib/asteroid/version.rb
+- test/helper.rb
+- test/unit/test_instance.rb
+- test/unit/test_server.rb
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: A server configuration framework
+test_files:
+- test/helper.rb
+- test/unit/test_instance.rb
+- test/unit/test_server.rb