From 3c54c9a017a8daee508a90ac1234958ceca5c71e Mon Sep 17 00:00:00 2001
From: Stawberri <aideen@aideen.pw>
Date: Sat, 6 Dec 2014 21:57:51 -0800
Subject: [PATCH] meta.media_uris

Adds media_uris to tweet meta, which returns an array of strings
containing the URIs of media inside your tweet.
---
 lib/twitter_ebooks/bot.rb | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/lib/twitter_ebooks/bot.rb b/lib/twitter_ebooks/bot.rb
index b92938a..1455d5e 100644
--- a/lib/twitter_ebooks/bot.rb
+++ b/lib/twitter_ebooks/bot.rb
@@ -105,6 +105,38 @@ module Ebooks
       end
       @mentionless = mless
     end
+
+    # Get an array of media uris in tweet.
+    # @param size [String] A twitter image size to return. Supported sizes are thumb, small, medium (default), large
+    # @return [Array<String>] image URIs included in tweet
+    def media_uris(*args)
+      # When there's no size given, the default is medium.
+      size = ''
+      # Was one given?
+      unless args.empty?
+        case args[0]
+        when 'thumb'
+          size = ':thumb'
+        when 'small'
+          size = ':small'
+        when 'medium'
+          size = ':medium'
+        when 'large'
+          size = ':large'
+        end
+      end
+
+      # Start collecting uris.
+      uris = []
+      if @tweet.media?
+        @tweet.media.each do |each_media|
+          uris << each_media.media_url.to_s + size
+        end
+      end
+
+      # and that's pretty much it!
+      uris
+    end
   end
 
   class Bot