rest_connection 0.0.23 → 0.1.0

data/lib/rest_connection/rightscale/rightscale_api_internal.rb

@@ -4,7 +4,7 @@ module RightScale
       def connection
         @@little_brother_connection ||= RestConnection::Connection.new
         settings = @@little_brother_connection.settings
-        settings[:common_headers]["X_API_VERSION"] = "0.1"         
+        settings[:common_headers]["X_API_VERSION"] = "0.1"
         settings[:api_href] = settings[:api_url]
         settings[:extension] = ".js"
         @@little_brother_connection
@@ -15,7 +15,7 @@ module RightScale
       def connection
         @@little_brother_connection ||= RestConnection::Connection.new
         settings = @@little_brother_connection.settings
-        settings[:common_headers]["X_API_VERSION"] = "0.1"         
+        settings[:common_headers]["X_API_VERSION"] = "0.1"
         settings[:api_href] = settings[:api_url]
         settings[:extension] = ".js"
         @@little_brother_connection
@@ -23,4 +23,4 @@ module RightScale
     end
   end
 end
- 
+

data/lib/rest_connection/rightscale/rightscale_api_mc_input.rb

@@ -0,0 +1,32 @@
+#    This file is part of RestConnection
+#
+#    RestConnection is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    RestConnection is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with RestConnection.  If not, see <http://www.gnu.org/licenses/>.
+
+#
+# You must have Beta v1.5 API access to use these internal API calls.
+#
+module RightScale
+  module Api
+    module McInput
+      def get_inputs
+        self["inputs"] = connection.get(self.href + "/inputs")
+      end
+
+      def show
+        inst_href = URI.parse(self.href)
+        @params.merge! connection.get(inst_href.path, 'view' => "inputs")
+      end
+    end
+  end
+end

data/lib/rest_connection/rightscale/rightscale_api_mc_taggable.rb

@@ -0,0 +1,57 @@
+#    This file is part of RestConnection
+#
+#    RestConnection is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    RestConnection is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with RestConnection.  If not, see <http://www.gnu.org/licenses/>.
+
+#
+# You must have Beta v1.5 API access to use these internal API calls.
+#
+module RightScale
+  module Api
+    module McTaggable
+      include RightScale::Api::Taggable
+      def add_tags(*args)
+        return false if args.empty?
+        McTag.set(self.href, args.uniq)
+        self.tags(true)
+      end
+
+      def remove_tags(*args)
+        return false if args.empty?
+        McTag.unset(self.href, args.uniq)
+        @params["tags"] -= args
+        self.tags(true)
+      end
+
+      def tags(reload=false)
+        @params["tags"] ||= []
+        @params["tags"].map! { |item| item.is_a?(Hash) ? item["name"] : item }
+        @params["tags"].deep_merge!(McTag.search_by_href(self.href).first["tags"].map { |hsh| hsh["name"] }) if reload or @params["tags"].empty?
+        @params["tags"]
+      end
+    end
+
+    module McTaggableExtend
+      def find_by_tags(*args)
+        a = Array.new
+        search = McTag.search(self.resource_plural_name, args.uniq).first
+        if search
+          search["links"].each do |hash|
+            a << self.find(hash["href"])
+          end
+        end
+        return a
+      end
+    end
+  end
+end

data/lib/rest_connection/rightscale/rightscale_api_resources.rb

@@ -1,4 +1,4 @@
-#    This file is part of RestConnection 
+#    This file is part of RestConnection
 #
 #    RestConnection is free software: you can redistribute it and/or modify
 #    it under the terms of the GNU General Public License as published by
@@ -17,6 +17,9 @@
 require 'rest_connection/rightscale/rightscale_api_base'
 require 'rest_connection/rightscale/rightscale_api_internal'
 require 'rest_connection/rightscale/rightscale_api_gateway'
+require 'rest_connection/rightscale/rightscale_api_taggable'
+require 'rest_connection/rightscale/rightscale_api_mc_taggable'
+require 'rest_connection/rightscale/rightscale_api_mc_input'
 require 'rest_connection/rightscale/executable'
 require 'rest_connection/rightscale/server'
 require 'rest_connection/rightscale/deployment'
@@ -28,18 +31,45 @@ require 'rest_connection/rightscale/ec2_security_group'
 require 'rest_connection/rightscale/ec2_ssh_key'
 require 'rest_connection/rightscale/multi_cloud_image'
 require 'rest_connection/rightscale/tag'
+require 'rest_connection/rightscale/mc_tag'
+require 'rest_connection/rightscale/task'
 require 'rest_connection/rightscale/rs_internal'
 require 'rest_connection/rightscale/audit_entry'
 require 'rest_connection/rightscale/alert_spec'
 require 'rest_connection/rightscale/ec2_ebs_volume'
 require 'rest_connection/rightscale/ec2_ebs_snapshot'
+require 'rest_connection/rightscale/mc_volume_attachment'
+require 'rest_connection/rightscale/mc_volume'
+require 'rest_connection/rightscale/mc_volume_snapshot'
+require 'rest_connection/rightscale/mc_volume_type'
 require 'rest_connection/rightscale/server_internal'
 require 'rest_connection/rightscale/mc_server'
+require 'rest_connection/rightscale/server_interface'
+require 'rest_connection/rightscale/mc_instance'
+require 'rest_connection/rightscale/monitoring_metric'
+require 'rest_connection/rightscale/mc_multi_cloud_image_setting'
+require 'rest_connection/rightscale/mc_multi_cloud_image'
+require 'rest_connection/rightscale/mc_server_template'
 require 'rest_connection/rightscale/ec2_ssh_key_internal'
 require 'rest_connection/rightscale/server_template_internal'
 require 'rest_connection/rightscale/right_script_internal'
-require 'rest_connection/rightscale/multi_cloud_image_internal'
 require 'rest_connection/rightscale/multi_cloud_image_cloud_setting_internal'
+require 'rest_connection/rightscale/multi_cloud_image_internal'
 require 'rest_connection/rightscale/ec2_server_array'
+require 'rest_connection/rightscale/mc_server_array'
+require 'rest_connection/rightscale/mc_security_group'
+require 'rest_connection/rightscale/mc_deployment'
+require 'rest_connection/rightscale/mc_datacenter'
+require 'rest_connection/rightscale/mc_ssh_key'
 require 'rest_connection/rightscale/ec2_elastic_ip'
 require 'rest_connection/rightscale/credential'
+require 'rest_connection/rightscale/cloud'
+require 'rest_connection/rightscale/instance_type'
+require 'rest_connection/rightscale/mc_instance_type'
+require 'rest_connection/rightscale/mc_image'
+require 'rest_connection/rightscale/macro'
+require 'rest_connection/rightscale/s3_bucket'
+require 'rest_connection/rightscale/account'
+require 'rest_connection/rightscale/child_account'
+require 'rest_connection/rightscale/permission'
+require 'rest_connection/rightscale/user'

data/lib/rest_connection/rightscale/rightscale_api_taggable.rb

@@ -0,0 +1,116 @@
+#    This file is part of RestConnection
+#
+#    RestConnection is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    RestConnection is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with RestConnection.  If not, see <http://www.gnu.org/licenses/>.
+
+module RightScale
+  module Api
+    module Taggable
+      def add_tags(*args)
+        return false if args.empty?
+        Tag.set(self.href, args.uniq)
+        self.tags(true)
+      end
+
+      def remove_tags(*args)
+        return false if args.empty?
+        Tag.unset(self.href, args.uniq)
+        @params["tags"] -= args
+        self.tags(true)
+      end
+
+      def tags(reload=false)
+        @params["tags"] ||= []
+        @params["tags"].map! { |item| item.is_a?(Hash) ? item["name"] : item }
+        @params["tags"].deep_merge!(Tag.search_by_href(self.href).map { |hsh| hsh["name"] }) if reload or @params["tags"].empty?
+        @params["tags"]
+      end
+
+      def remove_info_tags(*tag_keys)
+        remove_tags_by_namespace("info", *tag_keys)
+      end
+
+      def set_info_tags(hsh={})
+        set_tags_by_namespace("info", hsh)
+      end
+
+      def get_info_tags(*tag_keys)
+        tags = get_tags_by_namespace("info")
+        tags.each { |resource,hsh|
+          hsh.reject! { |key,value|
+            rej = false
+            rej = !tag_keys.include?(key) unless tag_keys.empty?
+            rej
+          }
+        }
+        return tags
+      end
+
+      def remove_tags_by_namespace(namespace, *tag_keys)
+        tags_to_unset = []
+        tags = get_tags_by_namespace(namespace)
+        tags.each { |res,hsh|
+          hsh.each { |k,v|
+            tags_to_unset << "#{namespace}:#{k}=#{v}" if tag_keys.include?(k)
+          }
+        }
+        self.remove_tags(*tags_to_unset)
+      end
+
+      def set_tags_by_namespace(namespace, hsh={})
+        keys_to_change = []
+        tags_to_set = []
+        hsh.each { |k,v| keys_to_change << k; tags_to_set << "#{namespace}:#{k}=#{v}" }
+        self.remove_tags_by_namespace(namespace, *keys_to_change)
+        self.add_tags(*tags_to_set)
+      end
+
+      def get_tags_by_namespace(namespace)
+        ret = {}
+        tags = {"self" => self.tags(true)}
+        tags.each { |res,ary|
+          ret[res] ||= {}
+          ary.each { |tag|
+            next unless tag.start_with?("#{namespace}:")
+            key = tag.split("=").first.split(":")[1..-1].join(":")
+            value = tag.split(":")[1..-1].join(":").split("=")[1..-1].join("=")
+            ret[res][key] = value
+          }
+        }
+        return ret
+      end
+
+      def set_tags_to(*args)
+        STDERR.puts "set_tags_to(...) is deprecated"
+        self.clear_tags("info")
+        self.add_tags(*(args.uniq))
+      end
+
+      def clear_tags(namespace = nil)
+        tag_ary = self.tags(true)
+        tag_ary = tag_ary.select { |tag| tag.start_with?("#{namespace}:") } if namespace
+        self.remove_tags(*tag_ary)
+      end
+    end
+
+    module TaggableExtend
+      def find_by_tags(*args)
+        a = Array.new
+        Tag.search(self.resource_singular_name, args.uniq).each do |object|
+          a << self.new(object)
+        end
+        return a
+      end
+    end
+  end
+end

data/lib/rest_connection/rightscale/rs_internal.rb

@@ -1,4 +1,4 @@
-#    This file is part of RestConnection 
+#    This file is part of RestConnection
 #
 #    RestConnection is free software: you can redistribute it and/or modify
 #    it under the terms of the GNU General Public License as published by
@@ -13,17 +13,17 @@
 #    You should have received a copy of the GNU General Public License
 #    along with RestConnection.  If not, see <http://www.gnu.org/licenses/>.
 
-#    
+#
 # You must have special API access to use these internal API calls.
-# 
-class RsInternal 
+#
+class RsInternal
   include RightScale::Api::Base
   extend RightScale::Api::BaseExtend
 
   def connection
     @@little_brother_connection ||= RestConnection::Connection.new
     settings = @@little_brother_connection.settings
-    settings[:common_headers]["X_API_VERSION"] = "0.1"         
+    settings[:common_headers]["X_API_VERSION"] = "0.1"
     settings[:api_href] = settings[:api_url]
     settings[:extension] = ".js"
     @@little_brother_connection
@@ -32,7 +32,7 @@ class RsInternal
   def self.connection
     @@little_brother_connection ||= RestConnection::Connection.new
     settings = @@little_brother_connection.settings
-    settings[:common_headers]["X_API_VERSION"] = "0.1"         
+    settings[:common_headers]["X_API_VERSION"] = "0.1"
     settings[:api_href] = settings[:api_url]
     settings[:extension] = ".js"
     @@little_brother_connection

data/lib/rest_connection/rightscale/s3_bucket.rb

@@ -0,0 +1,36 @@
+#    This file is part of RestConnection
+#
+#    RestConnection is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    RestConnection is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with RestConnection.  If not, see <http://www.gnu.org/licenses/>.
+#
+class S3Bucket
+  include RightScale::Api::Base
+  extend RightScale::Api::BaseExtend
+
+  def self.resource_singular_name
+    "s3_bucket"
+  end
+
+  def self.resource_plural_name
+    "s3_buckets"
+  end
+
+  def resource_singular_name
+    "s3_bucket"
+  end
+
+  def resource_plural_name
+    "s3_buckets"
+  end
+
+end

data/lib/rest_connection/rightscale/server.rb

@@ -1,4 +1,4 @@
-#    This file is part of RestConnection 
+#    This file is part of RestConnection
 #
 #    RestConnection is free software: you can redistribute it and/or modify
 #    it under the terms of the GNU General Public License as published by
@@ -15,23 +15,27 @@
 
 require 'rest_connection/ssh_hax'
 
-class Server 
+class Server
   include RightScale::Api::Base
   extend RightScale::Api::BaseExtend
   include SshHax
+  include RightScale::Api::Taggable
+  extend RightScale::Api::TaggableExtend
 
   def self.create(opts)
     create_options = Hash.new
     create_options[self.resource_singular_name.to_sym] = opts
     create_options["cloud_id"] = opts[:cloud_id] if opts[:cloud_id]
-    location = connection.post(self.resource_plural_name,create_options)    
+    create_options[self.resource_singular_name.to_sym][:mci_href] = nil
+    create_options[self.resource_singular_name.to_sym][:inputs] = nil
+    location = connection.post(self.resource_plural_name,create_options)
     newrecord = self.new('href' => location)
     newrecord.reload
     newrecord.parameters #transform the parameters!
     newrecord
   end
 
-  # The RightScale api returns the server parameters as a hash with "name" and "value".  
+  # The RightScale api returns the server parameters as a hash with "name" and "value".
   # This must be transformed into a hash in case we want to PUT this back to the API.
   def transform_parameters(parameters)
     new_params_hash = {}
@@ -48,12 +52,13 @@ class Server
     if @params['parameters'].is_a?(Array)
       @params['parameters'] = transform_parameters(@params['parameters'])
     end
-    @params['parameters']
+    @params['parameters'] ||= {}
   end
 
   # This is overriding the default save with one that can massage the parameters
   def save
     self.parameters #transform the parameters, if they're not already!!
+    self.tags #transform the tags, if they're not already!!
     uri = URI.parse(self.href)
     connection.put(uri.path, resource_singular_name.to_sym => @params)
   end
@@ -64,36 +69,47 @@ class Server
   def wait_for_state(st,timeout=1200)
     reload
     connection.logger("#{nickname} is #{self.state}")
+    step = 15
+    catch_early_terminated = 60 / step
     while(timeout > 0)
       return true if state =~ /#{st}/
       raise "FATAL error, this server is stranded and needs to be #{st}: #{nickname}, see audit: #{self.audit_link}" if state.include?('stranded') && !st.include?('stranded')
-      sleep 30
-      timeout -= 30
+      raise "FATAL error, this server went to error state and needs to be #{st}: #{nickname}, see audit: #{self.audit_link}" if state.include?('error') && st !~ /error|terminated|stopped/
       connection.logger("waiting for server #{nickname} to go #{st}, state is #{state}")
+      if state =~ /terminated|stopped|inactive|error/ and st !~ /terminated|stopped|inactive|error/
+        if catch_early_terminated <= 0
+          raise "FATAL error, this server terminated when waiting for #{st}: #{nickname}"
+        end
+        catch_early_terminated -= 1
+      end
+      sleep step
+      timeout -= step
       reload
     end
     raise "FATAL, this server #{self.audit_link} timed out waiting for the state to be #{st}" if timeout <= 0
   end
 
   # waits until the server is operational and dns_name is available
-  def wait_for_operational_with_dns
+  def wait_for_operational_with_dns(state_wait_timeout=1200)
     timeout = 600
-    wait_for_state("operational")
+    wait_for_state("operational", state_wait_timeout)
+    step = 15
     while(timeout > 0)
       self.settings
-      break if self['dns-name'] && !self['dns-name'].empty? && self['private-dns-name'] && !self['private-dns-name'].empty? 
-      connection.logger "waiting for dns-name for #{self.nickname}"
-      sleep 30
-      timeout -= 30
+      break if self.reachable_ip
+      connection.logger "waiting for IP for #{self.nickname}"
+      sleep step
+      timeout -= step
     end
-    connection.logger "got DNS: #{self['dns-name']}"
-    raise "FATAL, this server #{self.audit_link} timed out waiting for DNS" if timeout <= 0
+    connection.logger "got IP: #{self.reachable_ip}"
+    raise "FATAL, this server #{self.audit_link} timed out waiting for IP" if timeout <= 0
   end
 
   def audit_link
     # proof of concept for now
-    server_id = self.href.split(/\//).last
-    audit_href = "https://my.rightscale.com/servers/#{server_id}#audit_entries"
+#    server_id = self.href.split(/\//).last
+#    audit_href = "https://my.rightscale.com/servers/#{server_id}#auditentries"
+    audit_href = self.href.gsub(/api\//,"") + "#auditentries"
     "<a href='#{audit_href}'>#{audit_href}</a>"
   end
 
@@ -116,6 +132,16 @@ class Server
     end
   end
 
+  def force_stop
+    if self.current_instance_href
+      t = URI.parse(self.href)
+      connection.post(t.path + '/stop')
+      connection.post(t.path + '/stop')
+    else
+      connection.logger("WARNING: was in #{self.state} and had a current_instance_href so skiping stop call")
+    end
+  end
+
 # Uses ServerInternal api to start and stop EBS based instances
   def start_ebs
     @server_internal = ServerInternal.new(:href => self.href)
@@ -129,7 +155,7 @@ class Server
 
   # Works on v4 and v5 images.
   # *executable can be an <~Executable> or <~RightScript>
-  def run_executable(executable, opts=nil)
+  def run_executable(executable, opts=nil, ignore_lock=false)
     script_options = Hash.new
     script_options[:server] = Hash.new
     if executable.is_a?(Executable)
@@ -143,7 +169,7 @@ class Server
     else
       raise "Invalid class passed to run_executable, needs Executable or RightScript, was:#{executable.class}"
     end
-    
+
     if not opts.nil? and opts.has_key?(:ignore_lock)
       script_options[:server][:ignore_lock] = "true"
       opts.delete(:ignore_lock)
@@ -151,6 +177,7 @@ class Server
 
     serv_href = URI.parse(self.href)
     script_options[:server][:parameters] = opts unless opts.nil?
+    script_options[:server][:ignore_lock] = "true" if ignore_lock
     location = connection.post(serv_href.path + '/run_executable', script_options)
     AuditEntry.new('href' => location)
   end
@@ -167,16 +194,32 @@ class Server
     script_options[:server][:parameters] = opts unless opts.nil?
     location = connection.post(serv_href.path + '/run_script', script_options)
     Status.new('href' => location)
-  end 
+  end
 
   def set_input(name, value)
     serv_href = URI.parse(self.href)
     connection.put(serv_href.path, :server => {:parameters => {name.to_sym => value} })
   end
 
-  def set_inputs(hash)
+  def set_inputs(hash = {})
+    set_current_inputs(hash)
+    set_next_inputs(hash)
+  end
+
+  def set_current_inputs(hash = {})
+    if self.current_instance_href and self.state != "stopped"
+      self.reload_as_current
+      serv_href = URI.parse(self.href)
+      connection.put(serv_href.path, :server => {:parameters => hash})
+      settings
+      self.reload_as_next
+    end
+  end
+
+  def set_next_inputs(hash = {})
     serv_href = URI.parse(self.href)
     connection.put(serv_href.path, :server => {:parameters => hash})
+    settings
   end
 
   def set_template(href)
@@ -191,7 +234,7 @@ class Server
 
   def attach_volume(params)
     hash = {}
-    hash[:server] = params 
+    hash[:server] = params
     serv_href = URI.parse(self.href)
     connection.post(serv_href.path + "/attach_volume", hash)
   end
@@ -211,7 +254,7 @@ class Server
     reload
     old_state = self.state
     serv_href = URI.parse(self.href)
-    connection.post(serv_href.path + "/reboot") 
+    connection.post(serv_href.path + "/reboot")
     if wait_for_state
       wait_for_state_change(old_state)
     end
@@ -220,7 +263,7 @@ class Server
   def relaunch
     self.stop
     self.wait_for_state("stopped")
-    self.start  
+    self.start
   end
 
   def wait_for_state_change(old_state = nil)
@@ -238,23 +281,129 @@ class Server
     raise("FATAL: timeout after #{timeout}s waiting for state change")
   end
 
-  # Save the servers parameters to the current server (instead of the next server)
-  def save_current
-    uri = URI.parse(self.href)
-    connection.put(uri.path + "/current", resource_singular_name.to_sym => @params)
+  # Reload the server's basic information from the current server instance
+  def reload_as_current
+    uri = URI.parse(self.href + "/current")
+    @params.merge! connection.get(uri.path)
   end
 
-  # Load server's settings from the current server (instead of the next server)
-  def settings_current
-    serv_href = URI.parse(self.href)
-    @params.merge! connection.get(serv_href.path + "/current" + "/settings")
+  # Reload the server's basic information from the next server instance
+  def reload_as_next
+    uri = URI.parse(self.href.gsub(/\/current/,""))
+    @params.merge! connection.get(uri.path)
   end
 
-  # Reload the server's basic information from the current server.
-  def reload_current
-    uri = URI.parse(self.href)
-    @params ? @params.merge!(connection.get(uri.path + "/current")) : @params = connection.get(uri.path)
+  # Complex logic for determining the cloud_id of even a stopped server
+  def cloud_id
+    self.settings
+    if self.state == "operational"
+      return self["cloud_id"]
+    end
+    cloud_ids = RestConnection::AWS_CLOUDS.map { |hsh| hsh["cloud_id"] }
+
+    api0_1 = false
+    begin
+      api0_1 = true if Ec2SshKeyInternal.find_all
+    rescue
+    end
+
+    # Try ssh keys
+    if self.ec2_ssh_key_href and api0_1
+      ref = self.ec2_ssh_key_href
+      cloud_ids.each { |cloud|
+        if Ec2SshKeyInternal.find_by_cloud_id(cloud.to_s).select { |o| o.href == ref }.first
+          return cloud
+        end
+      }
+    end
+
+    # Try security groups
+    if self.ec2_security_groups_href
+      self.ec2_security_groups_href.each { |sg|
+        cloud_ids.each { |cloud|
+          if Ec2SecurityGroup.find_by_cloud_id(cloud.to_s).select { |o| o.href == sg }.first
+            return cloud
+          end
+        }
+      }
+    end
+
+    raise "Could not determine cloud_id...try setting an ssh key or security group"
   end
 
-end
+  def run_executable_and_wait_for_completed(executable, opts=nil)
+    run_executable(executable, opts).wait_for_completed
+  end
+
+  def dns_name
+    self.settings unless self["dns_name"]
+    self["dns_name"]
+  end
+
+  def private_ip
+    self.settings unless @params["private-ip-address"]
+    @params["private-ip-address"]
+  end
+
+  def reachable_ip
+    return self.dns_name if self.dns_name
+    return self.private_ip if self.private_ip
+    nil
+  end
+
+  # Override Taggable mixin so that it sets tags on both next and current instances
+  def tags(*args)
+    self.reload_as_next if self.href =~ /current/
+    super(*args)
+  end
+
+  def current_tags(reload=true)
+    self.reload_as_next if self.href =~ /current/
+    ret = []
+    if self.current_instance_href
+      ret = Tag.search_by_href(self.current_instance_href).map { |h| h["name"] }
+    end
+    ret
+  end
+
+  def add_tags(*args)
+    self.reload_as_next if self.href =~ /current/
+    return false if args.empty?
+    args.uniq!
+    Tag.set(self.href, args)
+    Tag.set(self.current_instance_href, args) if self.current_instance_href
+    self.tags(true)
+  end
 
+  def remove_tags(*args)
+    self.reload_as_next if self.href =~ /current/
+    return false if args.empty?
+    args.uniq!
+    Tag.unset(self.href, args)
+    Tag.unset(self.current_instance_href, args) if self.current_instance_href
+    self.tags(true)
+  end
+
+  def get_tags_by_namespace(namespace)
+    ret = {}
+    tags = {"self" => self.tags(true)}
+    tags["current_instance"] = self.current_tags if self.current_instance_href
+    tags.each { |res,ary|
+      ret[res] ||= {}
+      ary.each { |tag|
+        next unless tag.start_with?("#{namespace}:")
+        key = tag.split("=").first.split(":")[1..-1].join(":")
+        value = tag.split(":")[1..-1].join(":").split("=")[1..-1].join("=")
+        ret[res][key] = value
+      }
+    }
+    return ret
+  end
+
+  def clear_tags(namespace = nil)
+    tags = self.tags(true)
+    tags.deep_merge! self.current_tags if self.current_instance_href
+    tags = tags.select { |tag| tag.start_with?("#{namespace}:") } if namespace
+    self.remove_tags(*tags)
+  end
+end