hibernate 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 21e7dde6cd66464fa7aff8bc01d5f1338e65ba5f180d27c6d7fc176fcf9dbff5
-  data.tar.gz: 84a9f6eb6fd06373cbe68b686048107636cb2821eee5284b9a408062378e05f8
+  metadata.gz: 13ffe5bd9a5a78875bccaa274cc525629110332a84ce88c97928549b5857698c
+  data.tar.gz: 2543e29eb63533e946de460b89749c7853bcb419bafde2bbcd0b4b61cdc1cf6c
 SHA512:
-  metadata.gz: 250624e8f6984ca9cee0685c8d99a990e6fb45c66a88a16a9fd1bc68f430d252c1377d55ca31f1b95806250748e1b8be683fef73f374e1dc708bbba77b65f8e0
-  data.tar.gz: 5495ddec8a33acf9c2b4a8eeb8026f762d8f37716912bb238d6c797587403e62f0bb8023202d1755b522f6e09292146bc298e97e1bd335febed73124d54a7658
+  metadata.gz: 4c0db9fdb1c8ebbd341cdca34dd56db61a89b975ab8a815b34991a09bb9858c325ada48c8fa5708cf99244e6fb7b8aae9b0566fadb15fff1d9ea7be8d8143d03
+  data.tar.gz: 0307afb2b28ec2d44d7901d6e60cc3c63c0ed228015a8829690d3676922aeb6c28f90d197df680c8ca77b268941b890a0c0dbbd8989862b7fdb28a635d3b6559

data/bin/hibernate

@@ -20,6 +20,10 @@ class HibernateCLI
     parser = OptionParser.new do |parser|
       parser.banner = "Usage: hibernate [command] [options]"
 
+      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|
         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,56 @@
+require 'yaml'
+
+module Hibernate
+  class ConfigLoader
+    def initialize(config_path = 'config.yaml')
+      @config_path = config_path
+      @config = load_config
+      @profile = ENV['AWS_PROFILE'] || default_profile
+      validate_config
+    end
+
+    def aws_credentials
+      account_config = @config.dig('aws_accounts', @profile)
+
+      if account_config.nil?
+        raise "Profile #{@profile} not found in the configuration file."
+      end
+
+      {
+        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
+
+    private
+
+    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', @profile)
+        raise "Profile #{@profile} is not defined in the configuration file."
+      end
+
+      credentials = @config.dig('aws_accounts', @profile, 'credentials')
+      unless credentials && credentials['access_key_id'] && credentials['secret_access_key']
+        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/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.5'
 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.5
 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