solutious-rudy 0.5.1 → 0.6.0

data/lib/rudy/config.rb

@@ -36,7 +36,7 @@ module Rudy
       # The "core" config file can have any or all configuration
       # but it should generally only contain the access identifiers
       # and defaults. That's why we only load one of them. 
-      core_config_paths = [adhoc_path, cwd_path, RUDY_CONFIG_FILE]
+      core_config_paths = [adhoc_path, cwd_path, Rudy::CONFIG_FILE]
       core_config_paths.each do |path|
         next unless path && File.exists?(path)
         @paths << path
@@ -44,26 +44,25 @@ module Rudy
       end
       
       # Rudy then looks for the rest of the config in these locations
-      @paths += Dir.glob(File.join('/etc', 'rudy', '*.rb')) || []
       @paths += Dir.glob(File.join(cwd, 'Rudyfile')) || []
       @paths += Dir.glob(File.join(cwd, 'config', 'rudy', '*.rb')) || []
       @paths += Dir.glob(File.join(cwd, '.rudy', '*.rb')) || []
+      @paths += Dir.glob(File.join('/etc', 'rudy', '*.rb')) || []
       @paths &&= @paths.uniq
       
       refresh
     end
 
     
-    
     def self.init_config_dir
       
-      unless File.exists?(RUDY_CONFIG_DIR)
-        puts "Creating #{RUDY_CONFIG_DIR}"
-        Dir.mkdir(RUDY_CONFIG_DIR, 0700)
+      unless File.exists?(Rudy::CONFIG_DIR)
+        puts "Creating #{Rudy::CONFIG_DIR}"
+        Dir.mkdir(Rudy::CONFIG_DIR, 0700)
       end
 
-      unless File.exists?(RUDY_CONFIG_FILE)
-        puts "Creating #{RUDY_CONFIG_FILE}"
+      unless File.exists?(Rudy::CONFIG_FILE)
+        puts "Creating #{Rudy::CONFIG_FILE}"
         rudy_config = Rudy::Utils.without_indent %Q{
           # Amazon Web Services 
           # Account access indentifiers.
@@ -77,21 +76,6 @@ module Rudy
               cert "~/path/2/cert-xxxx.pem"
             end
           end
-
-          # Machine Configuration
-          # Specify your private keys here. These can be defined globally
-          # or by environment and role like in machines.rb.
-          machines do
-            users do
-              #root :keypair => "path/2/root-private-key"
-            end
-            zone :"us-east-1b" do
-              ami 'ami-235fba4a'    # Amazon Getting Started AMI (US)
-            end
-            zone :"eu-west-1b" do
-              ami 'ami-e40f2790'    # Amazon Getting Started AMI (EU)
-            end
-          end
           
           # Global Defaults 
           # Define the values to use unless otherwise specified on the command-line. 
@@ -104,7 +88,7 @@ module Rudy
             user ENV['USER'].to_sym
           end
         }
-        Rudy::Utils.write_to_file(RUDY_CONFIG_FILE, rudy_config, 'w', 0600)
+        Rudy::Utils.write_to_file(Rudy::CONFIG_FILE, rudy_config, 'w', 0600)
       end
     end
 

data/lib/rudy/disks.rb

@@ -0,0 +1,248 @@
+
+
+module Rudy
+class Disk < Storable
+  include Rudy::MetaData::ObjectBase
+  
+  field :rtype
+  field :awsid
+  field :status
+  field :instid
+  
+  field :region
+  field :zone
+  field :environment
+  field :role
+  field :position
+  field :path
+  
+  field :device
+  field :size
+  #field :backups => Array
+  
+  field :mounted
+  
+  def init(path=nil, size=nil, device=nil, position=nil)
+    @path, @size, @device = path, size, device
+    @rtype = 'disk'
+    @region = @@global.region
+    @zone = @@global.zone
+    @environment = @@global.environment
+    @role = @@global.role
+    @position = position || @@global.position
+    @mounted = false
+    postprocess
+  end
+  
+  def postprocess
+    @size &&= @size.to_i
+    @mounted = true if @mounted == "true"
+  end
+  
+  def liner_note
+    info = @awsid && !@awsid.empty? ? @awsid : 'no volume'
+    "%s  %s" % [self.name.bright, info]
+  end
+  
+  def to_s(with_titles=true)
+    update
+    mtd = @mounted ? "mounted" : @status
+    "%s; %3sGB; %s; %s" % [liner_note, @size, @device, mtd]
+  end
+  
+  def inspect
+    lines = []
+    lines << liner_note
+    field_names.each do |key|
+      next unless self.respond_to?(key)
+      val = self.send(key)
+      lines << sprintf(" %22s: %s", key, (val.is_a?(Array) ? val.join(', ') : val))
+    end
+    lines.join($/)
+  end
+  
+  def name
+    sep=File::SEPARATOR
+    dirs = @path.split sep if @path && !@path.empty?
+    dirs.shift while dirs && (dirs[0].nil? || dirs[0].empty?)
+    super("disk", @zone, @environment, @role, @position, *dirs)
+  end
+  
+
+  
+  def create(snapshot=nil)
+    raise "#{name} is already running" if exists?
+    vol = @rvol.create(@size, @zone, snapshot) 
+    @awsid = vol.awsid
+    self.save
+    self
+  end
+  
+  def attach(instid)
+    raise "No volume id" unless exists?
+    vol = @rvol.attach(@awsid, instid, @device)
+  end
+  
+  def detach
+    raise "No volume id" unless exists?
+    vol = @rvol.detach(@awsid)
+  end
+  
+  def destroy(force=false)
+    if @awsid && !deleting?
+      if !force
+        raise Rudy::AWS::EC2::VolumeNotAvailable, @awsid if attached?
+      else
+        detach if exists? && attached?
+        sleep 0.1
+      end
+      raise Rudy::AWS::EC2::VolumeNotAvailable, @awsid if in_use?
+      @rvol.destroy(@awsid) if exists? && available?
+    end
+    super() # quotes, otherwise Ruby will send this method's args
+  end
+  
+  def update
+    return false unless @awsid
+    @volume = @rvol.get(@awsid) 
+    if @volume.is_a?(Rudy::AWS::EC2::Volume)
+      @status = @volume.status
+      @instid = @volume.instid
+      save
+    end
+  end
+  
+  def to_query(more=[], less=[])
+    super([:path, *more], less)  # Add path to the default fields
+  end
+  
+  def to_select(more=[], less=[])
+    super([:path, *more], less) 
+  end
+  
+  # Does this disk have enough info to be saved or used?
+  # The test is based on the same criteria for building
+  # SimpleDB queries. 
+  def valid?
+    criteria = build_criteria([:path]).flatten
+    criteria.size == criteria.compact.size
+  end
+  
+  def mounted?
+    @mounted && @mounted == true
+  end
+  
+  
+  %w[exists? deleting? available? attached? in_use?].each do |state|
+    define_method(state) do
+      return false if @awsid.nil? || @awsid.empty?
+      @rvol.send(state, @awsid) rescue false # deleting?, available?, etc...
+    end
+  end
+end
+
+class Disks
+  include Rudy::MetaData
+  
+    
+  def create(&each_mach)
+    
+  end
+
+
+  def destroy(&each_mach)
+    #raise MachineGroupNotRunning, current_machine_group unless running?
+    #raise MachineGroupNotDefined, current_machine_group unless known_machine_group?
+    list do |disk|
+      puts "Destroying #{disk.name}"
+      disk.destroy
+    end
+  end
+
+  def list(more=[], less=[], &each_disk)
+    disks = list_as_hash(&each_disk)
+    disks &&= disks.values
+    disks
+  end
+
+  def list_as_hash(more=[], less=[], &each_disk)
+    query = to_select([:rtype, 'disk'], less)
+    list = @sdb.select(query) || {}
+    disks = {}
+    list.each_pair do |n,d|
+      disks[n] = Rudy::Disk.from_hash(d)
+    end
+    disks.each_pair { |n,disk| each_disk.call(disk) } if each_disk
+    disks = nil if disks.empty?
+    disks
+  end
+
+  def get(rname=nil)
+    dhash = @sdb.get(Rudy::DOMAIN, rname)
+    return nil if dhash.nil? || dhash.empty?
+    d = Rudy::Disk.from_hash(dhash)
+    d.update if d
+    d
+  end
+
+
+  def running?
+    !list.nil?
+    # TODO: add logic that checks whether the instances are running.
+  end
+
+
+    
+  
+end
+end
+
+
+
+
+__END__
+
+def format(instance)
+  raise "No instance supplied" unless instance
+  raise "Disk not valid" unless self.valid?
+
+  begin
+    puts "Creating the filesystem (mkfs.ext3 -F #{disk.device})".bright
+    ssh_command instance.dns_public, current_user_keypairpath, @@global.user, "mkfs.ext3 -F #{disk.device}"
+    sleep 1
+  rescue => ex
+    @logger.puts ex.backtrace if debug?
+    raise "Error formatting #{disk.path}: #{ex.message}"
+  end
+  true
+end
+def mount(instance)
+  raise "No instance supplied" unless instance
+  disk = find_disk(opts[:disk] || opts[:path])
+  raise "Disk #{opts[:disk] || opts[:path]} cannot be found" unless disk
+  switch_user(:root)
+  begin
+    puts "Mounting #{disk.device} to #{disk.path}".bright
+    ssh_command instance.dns_public, current_user_keypairpath, @@global.user, "mkdir -p #{disk.path} && mount -t ext3 #{disk.device} #{disk.path}"
+  rescue => ex
+    @logger.puts ex.backtrace if debug?
+    raise "Error mounting #{disk.path}: #{ex.message}"
+  end
+  true
+end
+
+def unmount(instance)
+  raise "No instance supplied" unless instance
+  disk = find_disk(opts[:disk] || opts[:path])
+  raise "Disk #{opts[:disk] || opts[:path]} cannot be found" unless disk
+  switch_user(:root)
+  begin
+    puts "Unmounting #{disk.path}...".bright
+    ssh_command instance.dns_public, current_user_keypairpath, global.user, "umount #{disk.path}"
+    sleep 1
+  rescue => ex
+    @logger.puts ex.backtrace if debug?
+    raise "Error unmounting #{disk.path}: #{ex.message}"
+  end
+  true
+end

data/lib/rudy/global.rb

@@ -0,0 +1,121 @@
+
+
+module Rudy
+  
+  # = Rudy::Global
+  #
+  # This global class is used by all Huxtable objects.
+  # When a new CLI global is added, the appropriate field must
+  # be added to this class (optional: a default value in initialize).
+  class Global < Storable
+    
+    field :region
+    field :zone
+    field :environment
+    field :role
+    field :position
+    field :user
+    field :pkey
+    
+    field :nocolor
+    field :quiet
+    field :verbose
+    field :format
+    field :print_header
+    field :yes
+    
+    field :accesskey
+    field :secretkey
+    field :accountnum
+    field :accountname  # TODO: use this. And accounttype (aws)
+    field :cert
+    field :privatekey
+    
+    field :local_user
+    field :local_hostname
+    
+    field :config => String
+    
+    attr_accessor :print_header
+    
+    def initialize
+      postprocess
+      # These attributes MUST have values. 
+      @verbose ||= 0
+      @nocolor ||= false
+      @quiet ||= false
+      @format ||= 'string' # as in, to_s
+      @print_header = true if @print_header == nil
+      @yes = false if @yes.nil?
+    end
+      
+    def apply_config(config)
+      return unless config.is_a?(Rudy::Config)
+      if config.defaults?
+        %w[region zone environment role position user nocolor quiet].each do |name|
+          val = config.defaults.send(name)
+          self.send("#{name}=", val) if val
+        end
+      end
+      
+      if config.accounts? && config.accounts.aws
+        %w[accesskey secretkey accountnum cert privatekey].each do |name|
+          val = config.accounts.aws.send(name)
+          self.send("#{name}=", val) if val
+        end
+      end
+      
+      postprocess
+    end
+    
+    
+    def update(ghash={})
+      ghash = ghash.marshal_dump if ghash.is_a?(OpenStruct)
+      
+      if ghash.is_a?(Hash)
+        ghash.each_pair { |n,v| self.send("#{n}=", v) } 
+      else
+        raise "Unexpected #{ghash.class.to_s}"
+      end
+      
+      postprocess
+    end
+    
+    
+  private 
+    
+    
+    def postprocess
+      apply_environment_variables
+      apply_system_defaults
+
+      @cert &&= File.expand_path(@cert)
+      @privatekey &&= File.expand_path(@privatekey)
+      @position &&= @position.to_s.rjust(2, '0')  
+      
+      String.disable_color if @nocolor
+      Rudy.enable_quiet if @quiet
+      Annoy.enable_skip if @yes
+    end
+    
+    def apply_environment_variables
+      @accesskey ||= ENV['AWS_ACCESS_KEY']
+      @secretkey ||= ENV['AWS_SECRET_KEY'] || ENV['AWS_SECRET_ACCESS_KEY']
+      @accountnum ||= ENV['AWS_ACCOUNT_NUMBER']
+      @cert ||= ENV['EC2_CERT']
+      @privatekey ||= ENV['EC2_PRIVATE_KEY']
+      @local_user = ENV['USER'] || :rudy
+      @local_hostname = Socket.gethostname || :localhost  
+    end
+    
+    def apply_system_defaults
+      @region ||= Rudy::DEFAULT_REGION
+      @zone ||= Rudy::DEFAULT_ZONE
+      @environment ||= Rudy::DEFAULT_ENVIRONMENT
+      @role ||= Rudy::DEFAULT_ROLE
+      @position ||= Rudy::DEFAULT_POSITION
+      @user ||= Rudy::DEFAULT_USER
+    end
+    
+  end
+end