awful 0.0.106 → 0.0.107

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f47d50305aa745ea1d3e1ebccf4019a5834aea45
-  data.tar.gz: 68a3a50fcabc0a9afd326836fdc70181f0938844
+  metadata.gz: 51063a3a1c4c58bab96dfa743006f334a9ffbdb5
+  data.tar.gz: f6bf657fb70555adad45c6b0878bd921e83da297
 SHA512:
-  metadata.gz: df09dbe48421998771f3aa5591430ebd2cccdf2a7a1bd9061e4d9515e01289ad4dce64c376a287c3a35727a629806f6695ab4839737335ab31a122365bc81cbd
-  data.tar.gz: fc42025f14e2f16a17a5cebbbb75c9091a059fccdcb37cb2fb398e002e39030afce635f230465b8ba51e0992cfef2d45b3269828a57018d58549558fe5e29295
+  metadata.gz: 383850d370c8e8b791b0208c22749917576a37fc494b1fe849d2f59b04f0fb5abf616b959c4a73dd6819c868e8f8e519f891ae9e5029c9abb8852a3157c0edb0
+  data.tar.gz: 3eeed77f798bfb99032b9b7435f6b0c9ec87e662476f02ff89b6a288b00f344d4f3629a495bc126b68622a88b96d4a3efbcabcbe01b42ee19127041e0cf1e96c

data/lib/awful/s3.rb

@@ -11,6 +11,12 @@ module Awful
       def s3_resource
         Aws::S3::Resource.new(client: s3)
       end
+
+      def get_tags(bucket_name)
+        s3.get_bucket_tagging(bucket: bucket_name).tag_set
+      rescue Aws::S3::Errors::NoSuchTagSet # sdk throws this if no tags
+        nil
+      end
     end
 
     desc 'ls PATTERN', 'list buckets or objects'
@@ -45,6 +51,35 @@ module Awful
       end.tap { |list| puts list }
     end
 
+    desc 'tagged [NAME_PREFIX]', 'list buckets matching given tags'
+    method_option :tags,     aliases: '-t', type: :array,  default: [],  desc: 'List of tag=value to filter'
+    method_option :stack,    aliases: '-s', type: :string, default: nil, desc: 'Filter by stack name'
+    method_option :resource, aliases: '-r', type: :string, default: nil, desc: 'Filter by stack resource logical id'
+    def tagged(name = '.')
+      conditions = options[:tags].map do |tag|
+        key, value = tag.split('=')
+        ->(set) { (set[0] == key) && (set[1] == value) }
+      end
+      if options[:stack]
+        conditions << ->(set) { (set[0] == 'aws:cloudformation:stack-name') && (set[1] == options[:stack]) }
+      end
+      if options[:resource]
+        conditions << ->(set) { (set[0] == 'aws:cloudformation:logical-id') && (set[1] == options[:resource]) }
+      end
+
+      ## get all buckets and check for a match with conditions
+      s3.list_buckets.buckets.select do |b|
+        b.name.match(/^#{name}/i)
+      end.map do |bucket|
+        tags = get_tags(bucket.name) or next
+        tags.any? do |set|
+          conditions.any? { |c| c.call(set) }
+        end && bucket
+      end.select{|b| b}.tap do |buckets|
+        puts buckets.map(&:name)
+      end
+    end
+
     desc 'cat BUCKET/OBJECT', 'stream s3 object to stdout'
     def cat(path)
       bucket, key = path.split('/', 2)

data/lib/awful/version.rb

@@ -1,3 +1,3 @@
 module Awful
-  VERSION = '0.0.106'
+  VERSION = '0.0.107'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: awful
 version: !ruby/object:Gem::Version
-  version: 0.0.106
+  version: 0.0.107
 platform: ruby
 authors:
 - Ric Lister