lib/awsssh.rb in awsssh-3.1.0 vs lib/awsssh.rb in awsssh-3.1.1

@@ -1,10 +1,12 @@
 require "awsssh/version"
 require "thor"
 require "inifile"
 require "aws-sdk"
 require "colorize"
+require "open-uri"
+require "json"
 
 
 module Awsssh
   class Awsssh < Thor
 
@@ -16,10 +18,11 @@
         :status => {
           :online => :green,
           :stopped => :light_red
           }
         }
+      check_version
     end
 
     desc "list_profiles", "List all your avavible profiles"
     def list_profiles()
       credentials = open_credantial_file
@@ -166,8 +169,36 @@
     end
 
     def connect_server(hostname)
       exec "ssh #{hostname}"
       exit 0
+    end
+
+    def check_version
+      temp_file = "/tmp/awsssh_version_check"
+      if File.exists?(temp_file)
+        if ((Time.now - File.mtime(temp_file)) / (60 * 60 * 2)).to_i != 0 # check all 2h
+          check = true
+        else
+          check = false
+        end
+      else
+        check = true
+      end
+      if check
+        begin
+          rubygems = JSON.parse(open("https://rubygems.org/api/v1/versions/awsssh/latest.json").read)
+          if rubygems["version"] != VERSION
+            puts "   ############################################".colorize(:red)
+            puts "   # You're using an old version of this gem! #".colorize(:red)
+            puts "   # Run `gem update awsssh`                  #".colorize(:red)
+            puts "   ############################################".colorize(:red)
+            puts
+          end
+        rescue
+        ensure
+          FileUtils.touch(temp_file)
+        end
+      end
     end
   end
 end