aws-cleaner 0.1.3 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d960635f67336a50cd494f8c4886cb2d288dac07
-  data.tar.gz: b2e13460c1390ff2a38484d42c19cbe9cb2b7805
+  metadata.gz: f4e9cb3c17a953b6e31c33653adfabcc8e624de4
+  data.tar.gz: 2a02b8c09de9ca83022b0da35a97fbf3a71fa55a
 SHA512:
-  metadata.gz: e17e67a8eb0ad493b54cc106f019c5a70201ad55ad032309fcef253ef76f78c29e4bc71e155a614c40f5ccb48a23e84decbb2049944a70f71f65254b79d92a10
-  data.tar.gz: 4ff7266734e55f77f702f83b90f2c32eb28c46d569786437addcacc1960db6196950264b4650dcd48999b953526cc26b080c67bae706d7e609f787346dccf3ba
+  metadata.gz: 26e021cc9cdd9a950de13bc6246cb3458b61c79308f4acbb7dbeb8ee3ff4c9ff340f3b62be4827f1d10411293f5b8d76ca3ce8c3c3689c9c2f9eefcc1e35c081
+  data.tar.gz: c5abb5b0ed1a2244b14f99734b06a5e38450e26cd0d1ba0cd6260227dee4e60337bbf3c910a6d3e0d43b91276095754599fa932ee731046c0cbf2ab59ec4466c

data/bin/aws_cleaner.rb

@@ -22,13 +22,13 @@ end
 
 def config(file)
   YAML.load(File.read(file))
-  rescue StandardError => e
-    raise "Failed to open config file: #{e}"
+rescue StandardError => e
+  raise "Failed to open config file: #{e}"
 end
 
 # get options
 opts = Trollop::options do
-  opt :config, 'Path to config file', :type => :string, :default => 'config.yml'
+  opt :config, 'Path to config file', type: 'string', default: 'config.yml'
 end
 
 @config = config(opts[:config])
@@ -46,15 +46,15 @@ def delete_message(id)
   delete = @sqs.delete_message(
     queue_url: @config[:sqs][:queue],
     receipt_handle: id
-   )
+  )
   delete ? true : false
 end
 
 # return the body of the SQS message in JSON
 def parse(body)
   JSON.parse(body)
-  rescue JSON::ParserError
-    return false
+rescue JSON::ParserError
+  return false
 end
 
 # return the instance_id of the terminated instance
@@ -81,6 +81,16 @@ def get_chef_node_name(instance_id)
   end
 end
 
+# call the Chef API to get the FQDN of the instance
+def get_chef_fqdn(instance_id)
+  results = @chef.search.query(:node, "ec2_instance_id:#{instance_id} OR chef_provisioning_reference_server_id:#{instance_id}")
+  if results.rows.size > 0
+    return results.rows.first['automatic']['fqdn']
+  else
+    return false
+  end
+end
+
 # check if the node exists in Sensu
 def in_sensu?(node_name)
   begin
@@ -131,6 +141,37 @@ def notify_hipchat(msg)
   hipchat[room].send('AWS Cleaner', msg)
 end
 
+# generate the URL for the webhook
+def generate_webhook_url(url, template_variable_method, template_variable_argument, template_variable)
+  replacement = send(template_variable_method, eval(template_variable_argument))
+  url.gsub!(/{#{template_variable}/, replacement)
+  url
+end
+
+# call an HTTP endpoint
+def fire_webhook(config)
+  # generate templated URL
+  if config[:template_variables] && config[:url] =~ /\{\S+\}/
+    url = generate_webhook_url(
+      config[:url],
+      config[:template_variables][:method],
+      config[:template_variables][:argument],
+      config[:template_variables][:variable]
+    )
+  else
+    url = config[:url]
+  end
+
+  hook = { method: config[:method].to_sym, url: url }
+  begin
+    RestClient::Request.execute(hook)
+  rescue
+    return false
+  else
+    return true
+  end
+end
+
 # main loop
 loop do
   # get messages from SQS
@@ -176,6 +217,14 @@ loop do
           delete_message(id)
         end
       end
+
+      if @config[:webhooks]
+        @config[:webbooks].each do |_hook, config|
+          fire_webhook(config)
+        end
+        delete_message(id)
+      end
+
     else
       puts 'Message not relevant, deleting'
       delete_message(id)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-cleaner
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.2.0
 platform: ruby
 authors:
 - Eric Heydrick
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-07-31 00:00:00.000000000 Z
+date: 2015-12-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core