hibernate 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e1e5b683d653f45c6747b92d58f5bc5258111837bec3eec25945c0e8b58de37f
-  data.tar.gz: acb1ea6c6401883e116ee44da22f5179ff3009424fc90ff3d5f5db5e650470c3
+  metadata.gz: 71a95bed5e00b8b00d0ec93a5f5f3e00faee39b1de4bc2e5f3616c6358a4c98b
+  data.tar.gz: ba6cee2ff815f36090f51e802f361079d486bce36f04ba79eee6f719c32a3437
 SHA512:
-  metadata.gz: 0b96aa6e000fbe289ea668497918d6352103e398560e112e20898ff9a2b65583340e132e369b308990eaeede6a954c951ea9bf52a1f8947a8307336abe29a18e
-  data.tar.gz: 92c1af50b90c66fe7ee91101e37ae76ca6abb63e2e5fb1671adc1fbfcf2d9f904b0dced62b999b1bbd3a09c48b43ee02d8cbd625c29ddcd2e8ca25e2935dac54
+  metadata.gz: 57e1625b9990bfd59c947a8f894495be4b0d1e5ed9e6601d3e84ee67bc1d8cb760111d917211b8f68b4fa6b4db6d491f621c168679ded265415edf4010ab97cd
+  data.tar.gz: 1cbe3c34db01b4235f31544ad87452cd722d165b47bf73a50506ccea0b0b1e083c0adde28a70bf84d1b03564ca2860b8f42a39fecab886c7d0f927635be65a89

data/bin/hibernate

@@ -4,26 +4,25 @@ require 'optparse'
 require_relative '../lib/hibernate/lambda_setup'
 require_relative '../lib/hibernate/ec2_manager'
 
-# Define the command-line interface
 class HibernateCLI
   def self.run
     command = nil
     options = {}
 
+    if ARGV.include?('setup')
+      command = :setup
+      ARGV.delete('setup')
+    elsif ARGV.include?('node')
+      command = :manage_ec2
+      ARGV.delete('node')
+    elsif ARGV.include?('remove')
+      command = :remove_ec2
+      ARGV.delete('remove')
+    end
+
     parser = OptionParser.new do |parser|
       parser.banner = "Usage: hibernate [command] [options]"
 
-      # Command for setting up the Lambda function
-      parser.on('setup', 'Create the Lambda function') do
-        command = :setup
-      end
-
-      # Command for managing EC2 instances
-      parser.on('node', 'Manage EC2 instances') do
-        command = :manage_ec2
-      end
-
-      # Options for the EC2 management command
       parser.on('--in=<INSTANCE_NAME>', 'Specify the EC2 instance name') do |instance_name|
         options[:instance_name] = instance_name
       end
@@ -37,9 +36,7 @@ class HibernateCLI
       end
     end
 
-    # Parse the command-line arguments
     begin
-      p ARGV  # Debug: Print the command-line arguments
       parser.parse!(ARGV)
     rescue OptionParser::ParseError => e
       puts e.message
@@ -47,7 +44,6 @@ class HibernateCLI
       exit 1
     end
 
-    # Ensure only one command is processed
     if command.nil?
       puts "Please provide a command."
       puts parser
@@ -58,27 +54,37 @@ class HibernateCLI
     when :setup
       create_lambda_function
     when :manage_ec2
-      # Validate EC2 management command options
-      if options[:instance_name].nil? || options[:start_cron].nil? || options[:stop_cron].nil?
-        puts "Please provide the instance name, start cron expression, and stop cron expression."
-        puts parser
-        exit
-      else
-        manage_ec2(options[:instance_name], options[:start_cron], options[:stop_cron])
-      end
+      manage_ec2_command(options)
+    when :remove_ec2
+      remove_ec2_command(options)
     end
   end
 
   def self.create_lambda_function
-    p "I am printed here in setup"
     LambdaSetup.new.run
   end
 
-  def self.manage_ec2(instance_name, start_cron, stop_cron)
-    p "I am printed here inside node"
-    ec2_manager = EC2Manager.new(instance_name, start_cron, stop_cron)
-    ec2_manager.create_event_rule
+  def self.manage_ec2_command(options)
+    if options[:instance_name].nil? || (options[:start_cron].nil? && options[:stop_cron].nil?)
+      puts "Please provide the instance name, and at least one cron expression (start or stop)."
+      puts "Usage: hibernate node --in=<INSTANCE_NAME> --start_instance=<START_CRON> --stop_instance=<STOP_CRON>"
+      exit
+    else
+      ec2_manager = EC2Manager.new(options[:instance_name], options[:start_cron], options[:stop_cron])
+      ec2_manager.create_event_rule
+    end
+  end
+
+  def self.remove_ec2_command(options)
+    if options[:instance_name].nil? || (options[:start_cron].nil? && options[:stop_cron].nil?)
+      puts "Please provide the instance name, and at least one cron expression (start or stop) to remove."
+      puts "Usage: hibernate remove --in=<INSTANCE_NAME> --start_instance=<START_CRON> --stop_instance=<STOP_CRON>"
+      exit
+    else
+      ec2_manager = EC2Manager.new(options[:instance_name], options[:start_cron], options[:stop_cron])
+      ec2_manager.remove_event_rule
+    end
   end
 end
 
-HibernateCLI.run
+HibernateCLI.run

data/lib/hibernate/cloud_watch_event_manager.rb

@@ -0,0 +1,111 @@
+require 'aws-sdk-cloudwatchevents'
+require 'json'
+require 'dotenv/load'
+
+class CloudWatchEventManager
+  def initialize(events_client, instance_id, instance_name, lambda_function_arn)
+    @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)
+  end
+
+  def create_start_rule(cron_expression)
+    create_rule(
+      "StartInstanceRule-#{@instance_id}",
+      cron_expression,
+      { instance_id: @instance_id, action: 'start' },
+      "start"
+    )
+  end
+
+  def create_stop_rule(cron_expression)
+    create_rule(
+      "StopInstanceRule-#{@instance_id}",
+      cron_expression,
+      { instance_id: @instance_id, action: 'stop' },
+      "stop"
+    )
+  end
+
+  def remove_start_rule
+    rule_name = "StartInstanceRule-#{@instance_id}"
+    remove_rule(rule_name)
+    remove_lambda_permission(rule_name)
+  end
+
+  def remove_stop_rule
+    rule_name = "StopInstanceRule-#{@instance_id}"
+    remove_rule(rule_name)
+    remove_lambda_permission(rule_name)
+  end
+
+  private
+
+  def create_rule(rule_name, cron_expression, input_data, action)
+    @events_client.put_rule({
+      name: rule_name,
+      schedule_expression: "cron(#{cron_expression})",
+      state: 'ENABLED',
+      description: "Rule to #{action} EC2 instance #{@instance_id} (Name: #{@instance_name}) at specified time: cron(#{cron_expression})",
+    })
+
+    add_lambda_permission(rule_name)
+
+    @events_client.put_targets({
+      rule: rule_name,
+      targets: [
+        {
+          id: '1',
+          arn: @lambda_function_arn,
+          input: input_data.to_json,
+        },
+      ],
+    })
+
+    puts "#{action.capitalize} rule created for instance '#{@instance_name}' (ID: #{@instance_id})."
+  end
+
+  def add_lambda_permission(rule_name)
+    begin
+      @lambda_client.add_permission({
+        function_name: @lambda_function_arn,
+        statement_id: "#{rule_name}-Permission",
+        action: "lambda:InvokeFunction",
+        principal: "events.amazonaws.com",
+        source_arn: "arn:aws:events:#{@aws_region}:#{@account_id}:rule/#{rule_name}"
+      })
+
+      puts "Permission added for rule #{rule_name} to invoke Lambda #{@lambda_function_arn}."
+    rescue Aws::Lambda::Errors::ResourceConflictException => e
+      puts "Permission already exists: #{e.message}"
+    end
+  end
+
+  def remove_rule(rule_name)
+    @events_client.remove_targets({
+      rule: rule_name,
+      ids: ['1']
+    })
+    @events_client.delete_rule({
+      name: rule_name
+    })
+
+    puts "Removed rule '#{rule_name}' for instance '#{@instance_name}' (ID: #{@instance_id})."
+  end
+
+  def remove_lambda_permission(rule_name)
+    begin
+      @lambda_client.remove_permission({
+        function_name: @lambda_function_arn,
+        statement_id: "#{rule_name}-Permission"
+      })
+      puts "Removed Lambda permission for rule '#{rule_name}' to invoke Lambda #{@lambda_function_arn}."
+    rescue Aws::Lambda::Errors::ResourceNotFoundException => e
+      puts "Permission not found: #{e.message}"
+    end
+  end
+end

data/lib/hibernate/ec2_manager.rb

@@ -1,7 +1,7 @@
 require 'aws-sdk-ec2'
-require 'aws-sdk-cloudwatchevents'
 require 'dotenv/load'
 require 'json'
+require_relative 'cloud_watch_event_manager' # Adjust the path to where the new class is located
 
 class EC2Manager
   def initialize(instance_name, start_cron, stop_cron)
@@ -10,12 +10,15 @@ class EC2Manager
     @stop_cron = stop_cron
     @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)
-    
+
     @lambda_function_name = "ec2_auto_shutdown_start_function"
+    @lambda_function_arn = construct_lambda_function_arn
     @instance_id = get_instance_id_by_name
+
+    @cloudwatch_event_manager = CloudWatchEventManager.new(@events_client, @instance_id, @instance_name, @lambda_function_arn)
   end
 
   def get_instance_id_by_name
@@ -37,58 +40,20 @@ class EC2Manager
   end
 
   def create_event_rule
-    create_start_rule
-    create_stop_rule
+    @cloudwatch_event_manager.create_start_rule(@start_cron) unless @start_cron.nil?
+    @cloudwatch_event_manager.create_stop_rule(@stop_cron) unless @stop_cron.nil?
     puts "CloudWatch Events created for instance '#{@instance_name}' (ID: #{@instance_id})."
   end
 
-  private
-
-  def create_start_rule
-    lambda_function_arn = construct_lambda_function_arn
-
-    @events_client.put_rule({
-      name: "StartInstanceRule-#{@instance_id}",
-      schedule_expression: "cron(#{@start_cron})",
-      state: 'ENABLED',
-      description: "Rule to start EC2 instance #{@instance_id} (Name: #{@instance_name}) at specified time",
-    })
-
-    @events_client.put_targets({
-      rule: "StartInstanceRule-#{@instance_id}",
-      targets: [
-        {
-          id: '1',
-          arn: lambda_function_arn,
-          input: { instance_id: @instance_id, action: 'start' }.to_json,
-        },
-      ],
-    })
+  def remove_event_rule
+    @cloudwatch_event_manager.remove_start_rule unless @start_cron.nil?
+    @cloudwatch_event_manager.remove_stop_rule unless @stop_cron.nil?
+    puts "CloudWatch Events removed for instance '#{@instance_name}' (ID: #{@instance_id})."
   end
 
-  def create_stop_rule
-    lambda_function_arn = construct_lambda_function_arn
-
-    @events_client.put_rule({
-      name: "StopInstanceRule-#{@instance_id}",
-      schedule_expression: "cron(#{@stop_cron})",
-      state: 'ENABLED',
-      description: "Rule to stop EC2 instance #{@instance_id} (Name: #{@instance_name}) at specified time",
-    })
-
-    @events_client.put_targets({
-      rule: "StopInstanceRule-#{@instance_id}",
-      targets: [
-        {
-          id: '1',
-          arn: lambda_function_arn,
-          input: { instance_id: @instance_id, action: 'stop' }.to_json,
-        },
-      ],
-    })
-  end
+  private
 
   def construct_lambda_function_arn
     "arn:aws:lambda:#{@aws_region}:#{@account_id}:function:#{@lambda_function_name}"
   end
-end
+end

data/lib/hibernate/lambda_setup.rb

@@ -1,12 +1,13 @@
 require 'aws-sdk-lambda'
-require 'aws-sdk-iam'   # For creating the IAM role and policy
+require 'aws-sdk-iam'
 require 'zip'
-require 'dotenv/load'   # Automatically loads environment variables from .env
+require 'dotenv/load'
 require 'fileutils'
+require 'pry'
 
 class LambdaSetup
   def initialize
-    @lambda_role_name = "ec2-auto-shutdown-start"  # Define a unique role name
+    @lambda_role_name = "ec2-auto-shutdown-start"
     @lambda_handler = "ec2_auto_shutdown_start_function"
     @lambda_zip = "lambda_function.zip"
     @aws_region = ENV['AWS_REGION']
@@ -18,33 +19,34 @@ class LambdaSetup
     )
   end
 
-  # Function to zip the Lambda function code
   def create_zip_file(dir)
-    FileUtils.rm_f(@lambda_zip)  # Remove any existing zip file
+    FileUtils.rm_f(@lambda_zip)
 
     Zip::File.open(@lambda_zip, Zip::File::CREATE) do |zip|
-      Dir[File.join(dir, '**', '**')].each do |file|
-        zip.add(file.sub(dir + '/', ''), file)
+      Dir.glob(File.join('lib', 'hibernate',dir, '**', '*')).each do |file|
+        next if File.directory?(file)
+        zip_path = File.basename(file)
+        puts "Adding #{file} as #{zip_path}"
+        File.open(file, 'rb') do |io|
+          zip.add(zip_path, io)
+        end
       end
     end
 
     puts "ZIP file created: #{@lambda_zip}"
   end
 
-  # Function to check if an IAM role exists
   def iam_role_exists?(role_name)
     begin
       @iam_client.get_role(role_name: role_name)
-      true  # Role exists
+      true
     rescue Aws::IAM::Errors::NoSuchEntity
-      false  # Role does not exist
+      false
     end
   end
 
-  # Function to create the IAM role and policy
   def create_lambda_role
     unless iam_role_exists?(@lambda_role_name)
-      # Trust policy for Lambda service to assume this role
       trust_policy = {
         Version: "2012-10-17",
         Statement: [
@@ -58,14 +60,12 @@ class LambdaSetup
         ]
       }.to_json
 
-      # Create the IAM role
       puts "Creating IAM role..."
       @iam_client.create_role({
         role_name: @lambda_role_name,
         assume_role_policy_document: trust_policy
       })
 
-      # Define the custom policy for EC2 actions
       policy_document = {
         Version: "2012-10-17",
         Statement: [
@@ -81,7 +81,6 @@ class LambdaSetup
         ]
       }.to_json
 
-      # Attach custom policy to the IAM role
       puts "Attaching custom EC2 policy to IAM role..."
       @iam_client.put_role_policy({
         role_name: @lambda_role_name,
@@ -89,7 +88,6 @@ class LambdaSetup
         policy_document: policy_document
       })
 
-      # Attach basic execution role to allow CloudWatch logging
       @iam_client.attach_role_policy({
         role_name: @lambda_role_name,
         policy_arn: 'arn:aws:iam::aws:policy/service-role/AWSLambdaBasicExecutionRole'
@@ -101,39 +99,35 @@ class LambdaSetup
     end
   end
 
-  # Function to check if the Lambda function exists
   def lambda_function_exists?(function_name)
     begin
       @lambda_client.get_function(function_name: function_name)
-      true  # Lambda function exists
+      true
     rescue Aws::Lambda::Errors::ResourceNotFoundException
-      false  # Lambda function does not exist
+      false
     end
   end
 
-  # Function to create the Lambda function
   def create_lambda_function
     role_arn = @iam_client.get_role(role_name: @lambda_role_name).role.arn
 
     if lambda_function_exists?(@lambda_handler)
       puts "Lambda function '#{@lambda_handler}' already exists. Skipping creation."
     else
-      # Read the ZIP file as binary content
       zip_content = File.read(@lambda_zip)
 
-      # Create Lambda function
       begin
         puts "Creating Lambda function..."
         @lambda_client.create_function({
           function_name: @lambda_handler,
-          runtime: 'ruby3.2',  # Specify your desired runtime
-          role: role_arn,  # Use the ARN of the newly created role
+          runtime: 'ruby3.2',
+          role: role_arn,
           handler: "#{@lambda_handler}.lambda_handler",
           code: {
             zip_file: zip_content,
           },
           description: 'Lambda function to start and stop EC2 instances',
-          timeout: 30,
+          timeout: 30
         })
         puts "Lambda function created."
       rescue Aws::Lambda::Errors::ServiceError => e
@@ -142,10 +136,9 @@ class LambdaSetup
     end
   end
 
-  # Main method to execute the setup
   def run
-    create_zip_file('lambda_function')  # Pass the directory containing your Lambda function code
+    create_zip_file('lambda_function')
     create_lambda_role
     create_lambda_function
   end
-end
+end

data/lib/hibernate/version.rb

@@ -1,3 +1,3 @@
 module Hibernate
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/lib/hibernate.rb

@@ -3,4 +3,8 @@ require 'aws-sdk-cloudwatch'
 require 'dotenv'
 require 'json'
 
-Dotenv.load
+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.0
+  version: 0.1.1
 platform: ruby
 authors:
-- Your Name
+- Manish Sharma
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-09-29 00:00:00.000000000 Z
+date: 2024-09-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-ec2
@@ -108,25 +108,28 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: A Ruby gem to automate the management of EC2 instances using AWS Lambda.
+description: A Ruby gem to automate the shutdown and start of our EC2 instances
 email:
-- youremail@example.com
+- sharma.manish7575@gmail.com
 executables:
 - hibernate
 extensions: []
 extra_rdoc_files: []
 files:
-- Readme.md
 - bin/hibernate
 - lib/hibernate.rb
+- lib/hibernate/cloud_watch_event_manager.rb
 - lib/hibernate/ec2_manager.rb
-- lib/hibernate/lambda_function/function.rb
+- lib/hibernate/lambda_function/ec2_auto_shutdown_start_function.rb
 - lib/hibernate/lambda_setup.rb
 - lib/hibernate/version.rb
-homepage: https://github.com/yourusername/ec2_manager
+homepage: https://github.com/maniSHarma7575/hibernate
 licenses:
 - MIT
-metadata: {}
+metadata:
+  source_code_uri: https://github.com/maniSHarma7575/hibernate
+  homepage_uri: https://github.com/maniSHarma7575/hibernate
+  bug_tracker_uri: https://github.com/maniSHarma7575/hibernate/issues
 post_install_message:
 rdoc_options: []
 require_paths:
@@ -145,5 +148,5 @@ requirements: []
 rubygems_version: 3.5.1
 signing_key:
 specification_version: 4
-summary: Manage EC2 instances with Lambda functions
+summary: Automating the shutdown and start of our EC2 instances
 test_files: []