hibernate 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 21e7dde6cd66464fa7aff8bc01d5f1338e65ba5f180d27c6d7fc176fcf9dbff5
-  data.tar.gz: 84a9f6eb6fd06373cbe68b686048107636cb2821eee5284b9a408062378e05f8
+  metadata.gz: bb37df3a1d0e0b64ae191626bfe015bca529ff81582816dea8a9198acc388f0b
+  data.tar.gz: 60eb78bd95762490b910a6dc949c666126c4e72fbd0822c5103b197642de8fbb
 SHA512:
-  metadata.gz: 250624e8f6984ca9cee0685c8d99a990e6fb45c66a88a16a9fd1bc68f430d252c1377d55ca31f1b95806250748e1b8be683fef73f374e1dc708bbba77b65f8e0
-  data.tar.gz: 5495ddec8a33acf9c2b4a8eeb8026f762d8f37716912bb238d6c797587403e62f0bb8023202d1755b522f6e09292146bc298e97e1bd335febed73124d54a7658
+  metadata.gz: 3111686232c5dc02bb56a7c9f3b59ed0d9fc90a3ac1f986b32038cbbc41abeb62840228712e10ada6d7920082f1201aaa52b8cc88fbb4a33f53ec6ee843ed42a
+  data.tar.gz: 7d0389381b47eaa7be1d3938219c770b10ed5cca756cb861f4aceca5567e5abb4c460cbc56d9cddab42d548dfb768976573301b359083d045c38c453831aca72

data/bin/hibernate

@@ -20,6 +20,10 @@ 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
+      end
+
       parser.on('--instance-name=<INSTANCE_NAME>', 'Specify the EC2 instance name') do |instance_name|
         options[:instance_name] = instance_name
       end

data/lib/hibernate/cloud_watch_event_manager.rb

@@ -2,16 +2,22 @@ require 'aws-sdk-cloudwatchevents'
 require 'json'
 require 'dotenv/load'
 require 'digest'
+require_relative 'config_loader'
 
 class CloudWatchEventManager
   def initialize(events_client, instance_id = nil, instance_name = nil, lambda_function_arn)
+    config_loader = Hibernate::ConfigLoader.new
     @events_client = events_client
     @instance_id = instance_id
     @instance_name = instance_name
     @lambda_function_arn = lambda_function_arn
-    @aws_region = ENV['AWS_REGION']
-    @account_id = ENV['ACCOUNT_ID']
-    @lambda_client = Aws::Lambda::Client.new(region: @aws_region)
+    @aws_region = config_loader.aws_credentials[:region]
+    @account_id = config_loader.aws_credentials[:account_id]
+    @lambda_client = Aws::Lambda::Client.new(
+      region: @aws_region,
+      access_key_id: config_loader.aws_credentials[:access_key_id],
+      secret_access_key: config_loader.aws_credentials[:secret_access_key]
+    )
   end
 
   attr_writer :instance_id, :instance_name

data/lib/hibernate/config_loader.rb

@@ -0,0 +1,66 @@
+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
+      validate_config
+    end
+
+    def aws_credentials
+      account_config = @config.dig('aws_accounts', @account_id)
+      if account_config.nil?
+        raise "Account ID #{@account_id} not found in the configuration file."
+      end
+
+      {
+        account_id: @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)
+      else
+        raise "Configuration file not found: #{@config_path}"
+      end
+    end
+
+    def validate_config
+      unless @config.dig('aws_accounts', @account_id)
+        raise "Please set the 'account_id' in the configuration file."
+      end
+
+      credentials = @config.dig('aws_accounts', @account_id, 'credentials')
+      unless credentials && credentials['access_key_id'] && credentials['secret_access_key']
+        raise "AWS credentials for account ID #{@account_id} are missing or incomplete."
+      end
+    end
+  end
+end

data/lib/hibernate/ec2_manager.rb

@@ -2,15 +2,30 @@ require 'aws-sdk-ec2'
 require 'dotenv/load'
 require 'json'
 require_relative 'cloud_watch_event_manager' # Adjust the path to where the new class is located
+require_relative 'config_loader'
 
 class EC2Manager
   def initialize(instance_name = nil)
     @instance_name = instance_name
-    @aws_region = ENV['AWS_REGION']
-    @account_id = ENV['ACCOUNT_ID']
-
-    @ec2_client = Aws::EC2::Client.new(region: @aws_region)
-    @events_client = Aws::CloudWatchEvents::Client.new(region: @aws_region)
+    config_loader = Hibernate::ConfigLoader.new
+    aws_credentials = config_loader.aws_credentials
+
+    @aws_region = aws_credentials[:region]
+    @account_id = aws_credentials[:account_id]
+    access_key_id = aws_credentials[:access_key_id]
+    secret_access_key = aws_credentials[:secret_access_key]
+
+    @ec2_client = Aws::EC2::Client.new(
+      region: @aws_region,
+      access_key_id: access_key_id,
+      secret_access_key: secret_access_key
+    )
+
+    @events_client = Aws::CloudWatchEvents::Client.new(
+      region: @aws_region,
+      access_key_id: access_key_id,
+      secret_access_key: secret_access_key
+    )
 
     @lambda_function_name = "ec2_auto_shutdown_start_function"
     @lambda_function_arn = construct_lambda_function_arn

data/lib/hibernate/lambda_setup.rb

@@ -1,21 +1,28 @@
 require 'aws-sdk-lambda'
 require 'aws-sdk-iam'
 require 'zip'
-require 'dotenv/load'
 require 'fileutils'
 require 'pry'
+require_relative 'config_loader'
 
 class LambdaSetup
   def initialize
+    config_loader = Hibernate::ConfigLoader.new
+    @aws_region = config_loader.aws_credentials[:region]
+
     @lambda_role_name = "ec2-auto-shutdown-start"
     @lambda_handler = "ec2_auto_shutdown_start_function"
     @lambda_zip = "lambda_function.zip"
-    @aws_region = ENV['AWS_REGION']
-    @iam_client = Aws::IAM::Client.new(region: @aws_region)
+    @iam_client = Aws::IAM::Client.new(
+      region: @aws_region,
+      access_key_id: config_loader.aws_credentials[:access_key_id],
+      secret_access_key: config_loader.aws_credentials[:secret_access_key]
+    )
+
     @lambda_client = Aws::Lambda::Client.new(
       region: @aws_region,
-      access_key_id: ENV['AWS_ACCESS_KEY_ID'],
-      secret_access_key: ENV['AWS_SECRET_ACCESS_KEY']
+      access_key_id: config_loader.aws_credentials[:access_key_id],
+      secret_access_key: config_loader.aws_credentials[:secret_access_key]
     )
   end
 

data/lib/hibernate/version.rb

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

data/lib/hibernate.rb

@@ -1,10 +1,7 @@
 require 'aws-sdk-ec2'
 require 'aws-sdk-cloudwatch'
-require 'dotenv'
 require 'json'
 
-Dotenv.load
-
 module Hibernate
   require_relative 'hibernate/version'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: hibernate
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Manish Sharma
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-05 00:00:00.000000000 Z
+date: 2024-10-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-ec2
@@ -119,6 +119,7 @@ files:
 - bin/hibernate
 - lib/hibernate.rb
 - lib/hibernate/cloud_watch_event_manager.rb
+- lib/hibernate/config_loader.rb
 - lib/hibernate/ec2_manager.rb
 - lib/hibernate/lambda_function/ec2_auto_shutdown_start_function.rb
 - lib/hibernate/lambda_setup.rb