vagabond 0.1.0

data/lib/vagabond/server.rb

@@ -0,0 +1,158 @@
+require 'vagabond/vagabond'
+require 'mixlib/cli'
+require 'digest/md5'
+
+module Vagabond
+  
+  class Server < Vagabond
+        
+    def initialize(me, actions)
+      @name = me
+      @base_template = 'ubuntu_1204' # TODO: Make this dynamic
+      @action = actions.shift
+      setup_ui
+      load_configurations
+    end
+    
+    def create
+      if(lxc.exists?)
+        ui.warn 'Server container already exists'
+        if(lxc.frozen?)
+          ui.fatal 'Server container is currently frozen!'
+        elsif(lxc.stopped?)
+          lxc.start
+          ui.info 'Server container has been started'
+        else
+          ui.info 'Server container is currently running'
+        end
+      else
+        ui.info 'Creating Chef server container...'
+        do_create
+      end
+    end
+
+    def destroy
+      if(lxc.exists?)
+        ui.info 'Destroying Chef server container...'
+        do_destroy
+      else
+        ui.fatal 'No Chef server exists within this environment'
+      end
+    end
+
+    def do_create
+      cmd = Mixlib::ShellOut.new("#{Config[:sudo]}lxc-clone -n #{generated_name} -o #{@base_template}")
+      cmd.run_command
+      cmd.error!
+      @lxc = Lxc.new(generated_name)
+      @internal_config[:mappings][name] = generated_name
+      @internal_config.save
+      ui.info "Chef Server container created!"
+      lxc.start
+      ui.info "Bootstrapping erchef..."
+      tem_file = File.expand_path(File.join(File.dirname(__FILE__), 'bootstraps/server.erb'))
+      com = "#{Config[:sudo]}knife bootstrap #{lxc.container_ip(10, true)} --template-file #{tem_file} -i /opt/hw-lxc-config/id_rsa"
+      cmd = Mixlib::ShellOut.new(com, :live_stream => STDOUT, :timeout => 1200)
+      cmd.run_command
+      cmd.error!
+      ui.info 'Chef Server has been created!'
+      auto_upload if vagabondfile[:local_chef_server][:auto_upload]
+    end
+
+    def auto_upload
+      ui.info 'Auto uploading all assets to local Chef server...'
+      upload_roles
+      upload_databags
+      upload_environments
+      upload_cookbooks
+      ui.info 'All assets uploaded to local Chef server!'
+    end
+
+    def upload_roles
+      ui.info "Uploading roles to local Chef server..."
+      com = "knife role from file #{File.join(base_dir, 'roles/*')} #{Config[:knife_opts]}"
+      cmd = Mixlib::ShellOut.new(com)
+      cmd.run_command
+      cmd.error!
+      ui.info "Roles uploaded to local Chef server!"
+    end
+
+    def upload_databags
+      ui.info "Uploading data bags to local Chef server..."
+      Dir.glob(File.join(base_dir, "data_bags/*")).each do |b|
+        next if %w(. ..).include?(b)
+        coms = [
+          "knife data bag create #{File.basename(b)} #{Config[:knife_opts]}",
+          "knife data bag from file #{File.basename(b)} #{Config[:knife_opts]} --all"
+        ].each do |com|
+          cmd = Mixlib::ShellOut.new(com)
+          cmd.run_command
+          cmd.error!
+        end
+      end
+      ui.info "Data bags uploaded to local Chef server!"
+    end
+
+    def upload_environments
+      ui.info "Uploading environments to local Chef server..."
+      com = "knife environment from file #{File.join(base_dir, 'environments/*')} #{Config[:knife_opts]}"
+      cmd = Mixlib::ShellOut.new(com)
+      cmd.run_command
+      cmd.error!
+      ui.info "Environments uploaded to local Chef server!"
+    end
+
+    def upload_cookbooks
+      ui.info "Uploading cookbooks to local Chef server..."
+      if(vagabondfile[:local_chef_server][:berkshelf])
+        berks_upload
+      else
+        raw_upload
+      end
+    end
+
+    def berks_upload
+      write_berks_config
+      com = "berks upload -c #{File.join(vagabond_dir, 'berks.json')}"
+      cmd = Mixlib::ShellOut.new(com)
+      cmd.run_command
+      cmd.error!
+      ui.info "Berks cookbook upload complete!"
+    end
+
+    def raw_upload
+      com = "knife cookbook upload#{Config[:knife_opts]} --all"
+      cmd = Mixlib::ShellOut.new(com)
+      cmd.run_command
+      cmd.error!
+      ui.info "Cookbook upload complete!"
+    end
+
+    def write_berks_config
+      path = File.join(vagabond_dir, 'berks.json')
+      if(File.exists?(path))
+        cur = Mash.new(JSON.load(File.read(path)))
+      else
+        cur = Mash.new
+      end
+      url = "https://#{lxc.container_ip(10, true)}"
+      if(cur[:chef].nil? || cur[:chef][:chef_server_url] != url)
+        cur[:chef] = Mash.new(:chef_server_url => url)
+        cur[:ssl] = Mash.new(:verify => false)
+        File.open(path, 'w') do |file|
+          file.write(JSON.dump(cur))
+        end
+      end
+    end
+    
+    def generated_name
+      unless(@_gn)
+        s = Digest::MD5.new
+        s << vagabondfile.path
+        @_gn = "server-#{s.hexdigest}"
+      end
+      @_gn
+    end
+
+  end
+end

data/lib/vagabond/vagabond.rb

@@ -0,0 +1,109 @@
+Dir.glob(
+  File.join(
+    File.dirname(__FILE__), 'actions', '*.rb'
+  )
+).each do |action_module|
+  require action_module
+end
+
+require 'vagabond/vagabondfile'
+require 'vagabond/internal_configuration'
+require 'chef/knife/core/ui'
+require File.join(File.dirname(__FILE__), 'cookbooks/lxc/libraries/lxc.rb')
+
+module Vagabond
+  class Vagabond
+
+    class << self
+      attr_accessor :ui
+    end
+    
+    # Load available actions
+    Actions.constants.each do |const_sym|
+      const = Actions.const_get(const_sym)
+      include const if const.is_a?(Module)
+    end
+
+    attr_reader :name
+    attr_reader :lxc
+    attr_reader :vagabondfile
+    attr_reader :config
+    attr_reader :internal_config
+    attr_reader :ui
+
+    # action:: Action to perform
+    # name:: Name of vagabond
+    # config:: Hash configuration
+    #
+    # Creates an instance
+    def initialize(action, name_args)
+      setup_ui
+      @action = action
+      @name = name_args.shift
+      load_configurations
+      validate!
+    end
+
+    def load_configurations
+      @vagabondfile = Vagabondfile.new(Config[:vagabond_file])
+      Config[:sudo] = sudo
+      Config[:disable_solo] = true if @action.to_sym == :status
+      Lxc.use_sudo = @vagabondfile[:sudo].nil? ? true : @vagabondfile[:sudo]
+      @internal_config = InternalConfiguration.new(@vagabondfile, ui)
+      @config = @vagabondfile[:boxes][name]
+      @lxc = Lxc.new(@internal_config[:mappings][name] || '____nonreal____')
+      unless(Config[:disable_local_server])
+        if(@vagabondfile[:local_chef_server] && @vagabondfile[:local_chef_server][:enabled])
+          srv = Lxc.new(@internal_config[:mappings][:server])
+          Config[:knife_opts] = " -s https://#{srv.container_ip(10, true)}"
+        end
+      end
+    end
+
+    def setup_ui
+      Chef::Config[:color] = true
+      @ui = Chef::Knife::UI.new(STDOUT, STDERR, STDIN, {})
+      self.class.ui = @ui
+    end
+
+    def validate!
+      if(name.to_s == 'server')
+        raise 'Box name `server` is reserved and not allowed!'
+      end
+    end
+    
+    def execute
+      send(@action)
+    end
+
+    private
+
+    def generate_hash
+      Digest::MD5.hexdigest(@vagabondfile.path)
+    end
+
+    def check_existing!
+      if(@lxc.exists?)
+        ui.error "LXC: #{name} already exists!"
+        true
+      end
+    end
+    
+    def sudo
+      case @vagabondfile[:sudo]
+      when TrueClass
+        'sudo '
+      when String
+        "#{@vagabondfile[:sudo]} "
+      end
+    end
+
+    def base_dir
+      File.dirname(vagabondfile.path)
+    end
+
+    def vagabond_dir
+      File.join(base_dir, '.vagabond')
+    end
+  end
+end

data/lib/vagabond/vagabondfile.rb

@@ -0,0 +1,34 @@
+require 'chef/mash'
+
+module Vagabond
+  class Vagabondfile
+
+    attr_reader :path
+    attr_reader :config
+
+    def initialize(path=nil)
+      path = discover_path(Dir.pwd) unless path
+      @path = path
+      load_configuration!
+    end
+
+    def [](k)
+      @config[k]
+    end
+
+    def load_configuration!
+      raise 'No Vagabondfile file found!' unless @path && File.exists?(@path)
+      @config = Mash.new(self.instance_eval(IO.read(@path), @path, 1))
+    end
+
+    def discover_path(path)
+      d_path = Dir.glob(File.join(path, 'Vagabondfile')).first
+      unless(d_path)
+        cut_path = path.split(File::SEPARATOR)
+        cut_path.pop
+        d_path = discover_path(cut_path.join(File::SEPARATOR)) unless cut_path.empty?
+      end
+      d_path
+    end
+  end
+end

data/lib/vagabond/version.rb

@@ -0,0 +1,10 @@
+module Vagabond
+  class Version < Gem::Version
+    attr_reader :codename
+    def initialize(v, name)
+      @codename = name
+      super(v)
+    end
+  end
+  VERSION = Version.new('0.1.0', 'tramp')
+end

data/vagabond.gemspec

@@ -0,0 +1,18 @@
+$LOAD_PATH.unshift File.expand_path(File.dirname(__FILE__)) + '/lib/'
+require 'vagabond/version'
+
+Gem::Specification.new do |s|
+  s.name = 'vagabond'
+  s.version = Vagabond::VERSION.version
+  s.summary = 'Vagabond creator'
+  s.author = 'Chris Roberts'
+  s.email = 'chrisroberts.code@gmail.com'
+  s.homepage = 'http://github.com/chrisroberts/vagabond'
+  s.description = 'Vagabonds, lazing about'
+  s.require_path = 'lib'
+  s.has_rdoc = true
+  s.extra_rdoc_files = ['README.md']  
+  s.files = Dir['**/*']
+  s.executables = ['vagabond']
+  s.add_dependency 'chef'
+end

metadata

@@ -0,0 +1,125 @@
+--- !ruby/object:Gem::Specification
+name: vagabond
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Chris Roberts
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-03-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: chef
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Vagabonds, lazing about
+email: chrisroberts.code@gmail.com
+executables:
+- vagabond
+extensions: []
+extra_rdoc_files:
+- README.md
+files:
+- lib/vagabond/cookbooks/lxc/files/default/knife_lxc
+- lib/vagabond/cookbooks/lxc/metadata.rb
+- lib/vagabond/cookbooks/lxc/templates/default/interface.erb
+- lib/vagabond/cookbooks/lxc/templates/default/fstab.erb
+- lib/vagabond/cookbooks/lxc/templates/default/client.rb.erb
+- lib/vagabond/cookbooks/lxc/templates/default/default-lxc.erb
+- lib/vagabond/cookbooks/lxc/test/kitchen/cookbooks/lxc_test/metadata.rb
+- lib/vagabond/cookbooks/lxc/test/kitchen/cookbooks/lxc_test/recipes/lxc_files.rb
+- lib/vagabond/cookbooks/lxc/test/kitchen/cookbooks/lxc_test/recipes/centos_lxc.rb
+- lib/vagabond/cookbooks/lxc/test/kitchen/cookbooks/lxc_test/recipes/lxc_templates.rb
+- lib/vagabond/cookbooks/lxc/test/kitchen/cookbooks/lxc_test/recipes/ubuntu_lxc.rb
+- lib/vagabond/cookbooks/lxc/test/kitchen/cookbooks/lxc_test/recipes/chef-bootstrap.rb
+- lib/vagabond/cookbooks/lxc/test/kitchen/Kitchenfile
+- lib/vagabond/cookbooks/lxc/Gemfile
+- lib/vagabond/cookbooks/lxc/attributes/default.rb
+- lib/vagabond/cookbooks/lxc/README.md
+- lib/vagabond/cookbooks/lxc/providers/fstab.rb
+- lib/vagabond/cookbooks/lxc/providers/interface.rb
+- lib/vagabond/cookbooks/lxc/providers/service.rb
+- lib/vagabond/cookbooks/lxc/providers/config.rb
+- lib/vagabond/cookbooks/lxc/providers/container.rb
+- lib/vagabond/cookbooks/lxc/libraries/lxc_file_config.rb
+- lib/vagabond/cookbooks/lxc/libraries/lxc_expanded_resources.rb
+- lib/vagabond/cookbooks/lxc/libraries/lxc.rb
+- lib/vagabond/cookbooks/lxc/resources/fstab.rb
+- lib/vagabond/cookbooks/lxc/resources/interface.rb
+- lib/vagabond/cookbooks/lxc/resources/service.rb
+- lib/vagabond/cookbooks/lxc/resources/config.rb
+- lib/vagabond/cookbooks/lxc/resources/#container.rb#
+- lib/vagabond/cookbooks/lxc/resources/container.rb
+- lib/vagabond/cookbooks/lxc/CHANGELOG.md
+- lib/vagabond/cookbooks/lxc/Gemfile.lock
+- lib/vagabond/cookbooks/lxc/recipes/default.rb
+- lib/vagabond/cookbooks/lxc/recipes/install_dependencies.rb
+- lib/vagabond/cookbooks/lxc/recipes/knife.rb
+- lib/vagabond/cookbooks/lxc/recipes/containers.rb
+- lib/vagabond/cookbooks/vagabond/metadata.rb
+- lib/vagabond/cookbooks/vagabond/attributes/default.rb
+- lib/vagabond/cookbooks/vagabond/libraries/vagabond.rb
+- lib/vagabond/cookbooks/vagabond/recipes/create.rb
+- lib/vagabond/cookbooks/vagabond/recipes/default.rb
+- lib/vagabond/commands.rb
+- lib/vagabond/server.rb
+- lib/vagabond/vagabondfile.rb
+- lib/vagabond/bootstraps/server.erb
+- lib/vagabond/version.rb
+- lib/vagabond/vagabond.rb
+- lib/vagabond/config.rb
+- lib/vagabond/internal_configuration.rb
+- lib/vagabond/actions/up.rb
+- lib/vagabond/actions/create.rb
+- lib/vagabond/actions/provision.rb
+- lib/vagabond/actions/status.rb
+- lib/vagabond/actions/thaw.rb
+- lib/vagabond/actions/destroy.rb
+- lib/vagabond/actions/freeze.rb
+- lib/vagabond/actions/ssh.rb
+- lib/vagabond.rb
+- README.md
+- vagabond.gemspec
+- bin/vagabond
+- CHANGELOG.md
+homepage: http://github.com/chrisroberts/vagabond
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Vagabond creator
+test_files: []
+has_rdoc: true