hibernate 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bb37df3a1d0e0b64ae191626bfe015bca529ff81582816dea8a9198acc388f0b
-  data.tar.gz: 60eb78bd95762490b910a6dc949c666126c4e72fbd0822c5103b197642de8fbb
+  metadata.gz: 13ffe5bd9a5a78875bccaa274cc525629110332a84ce88c97928549b5857698c
+  data.tar.gz: 2543e29eb63533e946de460b89749c7853bcb419bafde2bbcd0b4b61cdc1cf6c
 SHA512:
-  metadata.gz: 3111686232c5dc02bb56a7c9f3b59ed0d9fc90a3ac1f986b32038cbbc41abeb62840228712e10ada6d7920082f1201aaa52b8cc88fbb4a33f53ec6ee843ed42a
-  data.tar.gz: 7d0389381b47eaa7be1d3938219c770b10ed5cca756cb861f4aceca5567e5abb4c460cbc56d9cddab42d548dfb768976573301b359083d045c38c453831aca72
+  metadata.gz: 4c0db9fdb1c8ebbd341cdca34dd56db61a89b975ab8a815b34991a09bb9858c325ada48c8fa5708cf99244e6fb7b8aae9b0566fadb15fff1d9ea7be8d8143d03
+  data.tar.gz: 0307afb2b28ec2d44d7901d6e60cc3c63c0ed228015a8829690d3676922aeb6c28f90d197df680c8ca77b268941b890a0c0dbbd8989862b7fdb28a635d3b6559

data/bin/hibernate

@@ -20,8 +20,8 @@ class HibernateCLI
     parser = OptionParser.new do |parser|
       parser.banner = "Usage: hibernate [command] [options]"
 
-      parser.on('--account-id=<ACCOUNT_ID>', 'Specify the AWS account ID') do |account_id|
-        ENV['AWS_ACCOUNT_ID'] = account_id
+      parser.on('--profile=<PROFILE_NAME>', 'Specify the profile name') do |profile|
+        ENV['AWS_PROFILE'] = profile
       end
 
       parser.on('--instance-name=<INSTANCE_NAME>', 'Specify the EC2 instance name') do |instance_name|

data/lib/hibernate/config_loader.rb

@@ -1,49 +1,31 @@
 require 'yaml'
-require 'fileutils'
 
 module Hibernate
   class ConfigLoader
-    CACHE_FILE_PATH = File.expand_path('~/.aws_account_cache')
-
     def initialize(config_path = 'config.yaml')
       @config_path = config_path
-      @account_id = fetch_account_id
       @config = load_config
+      @profile = ENV['AWS_PROFILE'] || default_profile
       validate_config
     end
 
     def aws_credentials
-      account_config = @config.dig('aws_accounts', @account_id)
+      account_config = @config.dig('aws_accounts', @profile)
+
       if account_config.nil?
-        raise "Account ID #{@account_id} not found in the configuration file."
+        raise "Profile #{@profile} not found in the configuration file."
       end
 
       {
-        account_id: @account_id,
+        account_id: account_config['account_id'],
         region: account_config['region'],
         access_key_id: account_config.dig('credentials', 'access_key_id'),
         secret_access_key: account_config.dig('credentials', 'secret_access_key')
       }
     end
 
-    def self.cache_account_id(account_id)
-      FileUtils.mkdir_p(File.dirname(CACHE_FILE_PATH)) # Ensure directory exists
-      File.write(CACHE_FILE_PATH, account_id)
-    end
-
     private
 
-    def fetch_account_id
-      if ENV['AWS_ACCOUNT_ID']
-        self.class.cache_account_id(ENV['AWS_ACCOUNT_ID'])
-        ENV['AWS_ACCOUNT_ID']
-      elsif File.exist?(CACHE_FILE_PATH)
-        File.read(CACHE_FILE_PATH).strip
-      else
-        raise "AWS account ID is not set. Please pass an account ID using the --account-id option or set it in the configuration."
-      end
-    end
-
     def load_config
       if File.exist?(@config_path)
         YAML.load_file(@config_path)
@@ -53,14 +35,22 @@ module Hibernate
     end
 
     def validate_config
-      unless @config.dig('aws_accounts', @account_id)
-        raise "Please set the 'account_id' in the configuration file."
+      unless @config.dig('aws_accounts', @profile)
+        raise "Profile #{@profile} is not defined in the configuration file."
       end
 
-      credentials = @config.dig('aws_accounts', @account_id, 'credentials')
+      credentials = @config.dig('aws_accounts', @profile, 'credentials')
       unless credentials && credentials['access_key_id'] && credentials['secret_access_key']
-        raise "AWS credentials for account ID #{@account_id} are missing or incomplete."
+        raise "AWS credentials for profile #{@profile} are missing or incomplete."
+      end
+    end
+
+    def default_profile
+      default_accounts = @config['aws_accounts'].select { |_, account| account['default'] }
+      if default_accounts.empty?
+        raise "No default profile found in the configuration."
       end
+      default_accounts.keys.first
     end
   end
 end

data/lib/hibernate/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Hibernate
-  VERSION = '0.1.4'
+  VERSION = '0.1.5'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: hibernate
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Manish Sharma