aws_role_creds 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: fd22edec6ed2ef8b88158dad1699d24ffbe9204f
-  data.tar.gz: 5b2113869f655c4f7d2d490a8592885d05c3f870
+  metadata.gz: 753860b6d50a59c85f94b10065c06927d14b2a97
+  data.tar.gz: 815dbe6bd48b3160d5d65c23b65729dce665a60d
 SHA512:
-  metadata.gz: 2524764e052f74bb59767dcca9c43d0d7caf3793d75c74eedb0a0584c29be5f6a02c7212cc41cc695722577283c9b0e39a11f9ba3dfd43886d3190457607df36
-  data.tar.gz: fa250abf2b813b45da0bb433bb5d92bb751c7a087e54684f2277f8c7382517f362a945ec1c2df719ccd81764c76298def76f28ee9610cfe4f4e0d812f06f74d6
+  metadata.gz: 6711dbe34131e618a6a3bf96de2f8b70cf7f4cea25a63eb3971a9f158ab68d56bf9d15dc10ac21fa7cf7f9985b05723843962f35d020e2b91d87befc4c92e8f4
+  data.tar.gz: c787a65c55190a64c174f5cd08ce51471a207a03e59fd0b332aefad5fd3b940f950e0a977f956d052b8e584962ff141ad79d46666ecb446967aba9b2dc988365

data/lib/aws_role_creds.rb

@@ -9,7 +9,8 @@ module AwsRoleCreds
     
     IN_FILE = "#{ENV['HOME']}/.aws/config.yaml"
     # The config file we write out
-    OUT_FILE = ["#{ENV['HOME']}/.aws/config", "#{ENV['HOME']}/.aws/credentials"]
+    CONFIG_OUT_FILE = "#{ENV['HOME']}/.aws/config"
+    CREDENTIALS_OUT_FILE = "#{ENV['HOME']}/.aws/credentials"
     SESSION_CREDS_FILE = "#{ENV['HOME']}/.aws/session.yaml"
     SESSION_DURATION = 86400
     ROLE_DURATION = 3600
@@ -36,9 +37,8 @@ module AwsRoleCreds
         region = p['region'] || REGION
         duration = p['duration'] || SESSION_DURATION
         if @session_credentials.key?(name)
-            break if @session_credentials[name]['expiration'] > Time.now 
+            next if @session_credentials[name]['expiration'] > Time.now 
         end
-        puts "#{name}"
         
         if p['id'] and p['key']
             client = Aws::STS::Client.new(
@@ -114,38 +114,49 @@ module AwsRoleCreds
 
     # Write out config file
     # first make a backup
-    OUT_FILE.each do |o|
-        FileUtils.cp( o, "#{o}.backup" )
-
-        # create a new ini file object
-        config = IniFile.new
-        config.filename = o
-
-        config['default'] = { "region" => REGION }
-
-        # set properties
-        @session_credentials.each do |k, c|
-            config["profile #{k}"] = {
-                "aws_access_key_id" => "#{c['access_key_id']}",
-                "aws_secret_access_key" => "#{c['secret_access_key']}",
-                "aws_security_token" => "#{c['session_token']}",
-                "region" => "#{c['region']}",
-            }
-            puts "Profile #{k} created"
-        end
 
-        @role_credentials.each do |k, c|
-            config["profile #{k}"] = {
-                "aws_access_key_id" => "#{c['access_key_id']}",
-                "aws_secret_access_key" => "#{c['secret_access_key']}",
-                "aws_security_token" => "#{c['session_token']}",
-                "region" => "#{c['region']}",
-            }
-            puts "Profile #{k} created for role #{c['role']}"
-        end
+    FileUtils.cp( CONFIG_OUT_FILE, "#{CONFIG_OUT_FILE}.backup" )
+    FileUtils.cp( CREDENTIALS_OUT_FILE, "#{CREDENTIALS_OUT_FILE}.backup" )
+
+    # create a new ini file object
+    config = IniFile.new
+    config.filename = CONFIG_OUT_FILE
+
+    credentials = IniFile.new
+    credentials.filename = CREDENTIALS_OUT_FILE
 
-        # save file
-        config.write()
+    config['default'] = { "region" => REGION }
+
+    # set properties
+    @session_credentials.each do |k, c|
+        profile = {
+            "aws_access_key_id" => "#{c['access_key_id']}",
+            "aws_secret_access_key" => "#{c['secret_access_key']}",
+            "aws_security_token" => "#{c['session_token']}",
+            "region" => "#{c['region']}",
+        }
+        
+        config["profile #{k}"] = profile
+        credentials["#{k}"] = profile
     end
 
+    @role_credentials.each do |k, c|
+        profile = {
+            "aws_access_key_id" => "#{c['access_key_id']}",
+            "aws_secret_access_key" => "#{c['secret_access_key']}",
+            "aws_security_token" => "#{c['session_token']}",
+            "region" => "#{c['region']}",
+        }
+        
+        config["profile #{k}"] = profile
+        credentials["#{k}"] = profile
+    end
+
+    # save file
+    config.write()
+    puts "#{CONFIG_OUT_FILE} updated"
+    credentials.write()
+    puts "#{CREDENTIALS_OUT_FILE} updated"
+
+
 end

data/lib/aws_role_creds/version.rb

@@ -1,3 +1,3 @@
 module AwsRoleCreds
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aws_role_creds
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Jack Thomas