hibernate 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: e1e5b683d653f45c6747b92d58f5bc5258111837bec3eec25945c0e8b58de37f
+  data.tar.gz: acb1ea6c6401883e116ee44da22f5179ff3009424fc90ff3d5f5db5e650470c3
+SHA512:
+  metadata.gz: 0b96aa6e000fbe289ea668497918d6352103e398560e112e20898ff9a2b65583340e132e369b308990eaeede6a954c951ea9bf52a1f8947a8307336abe29a18e
+  data.tar.gz: 92c1af50b90c66fe7ee91101e37ae76ca6abb63e2e5fb1671adc1fbfcf2d9f904b0dced62b999b1bbd3a09c48b43ee02d8cbd625c29ddcd2e8ca25e2935dac54

data/bin/hibernate

@@ -0,0 +1,84 @@
+#!/usr/bin/env ruby
+
+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 = {}
+
+    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
+
+      parser.on('--start_instance=<START_CRON>', 'Specify the cron expression to start the instance') do |start_cron|
+        options[:start_cron] = start_cron
+      end
+
+      parser.on('--stop_instance=<STOP_CRON>', 'Specify the cron expression to stop the instance') do |stop_cron|
+        options[:stop_cron] = stop_cron
+      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
+      puts parser
+      exit 1
+    end
+
+    # Ensure only one command is processed
+    if command.nil?
+      puts "Please provide a command."
+      puts parser
+      exit
+    end
+
+    case command
+    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
+    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
+  end
+end
+
+HibernateCLI.run

data/lib/hibernate/ec2_manager.rb

@@ -0,0 +1,94 @@
+require 'aws-sdk-ec2'
+require 'aws-sdk-cloudwatchevents'
+require 'dotenv/load'
+require 'json'
+
+class EC2Manager
+  def initialize(instance_name, start_cron, stop_cron)
+    @instance_name = instance_name
+    @start_cron = start_cron
+    @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"
+    @instance_id = get_instance_id_by_name
+  end
+
+  def get_instance_id_by_name
+    response = @ec2_client.describe_instances({
+      filters: [
+        { name: "tag:Name", values: [@instance_name] },
+        { name: "instance-state-name", values: ["running", "stopped"] }
+      ]
+    })
+
+    if response.reservations.empty?
+      puts "No EC2 instance found with the name '#{@instance_name}'."
+      exit 1
+    end
+
+    instance_id = response.reservations[0].instances[0].instance_id
+    puts "Found EC2 instance ID: #{instance_id} for instance name: #{@instance_name}"
+    instance_id
+  end
+
+  def create_event_rule
+    create_start_rule
+    create_stop_rule
+    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,
+        },
+      ],
+    })
+  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
+
+  def construct_lambda_function_arn
+    "arn:aws:lambda:#{@aws_region}:#{@account_id}:function:#{@lambda_function_name}"
+  end
+end

data/lib/hibernate/lambda_function/function.rb

@@ -0,0 +1,24 @@
+require 'aws-sdk-ec2'
+require 'json'
+
+def lambda_handler(event:, context:)
+  ec2 = Aws::EC2::Client.new
+
+  instance_id = event['instance_id']
+  action = event['action']
+
+  begin
+    case action
+    when 'start'
+      ec2.start_instances(instance_ids: [instance_id])
+      puts "Started EC2 instance: #{instance_id}"
+    when 'stop'
+      ec2.stop_instances(instance_ids: [instance_id])
+      puts "Stopped EC2 instance: #{instance_id}"
+    else
+      raise "Invalid action: #{action}. Must be 'start' or 'stop'."
+    end
+  rescue Aws::EC2::Errors::ServiceError => e
+    puts "Error during EC2 action: #{e.message}"
+  end
+end

data/lib/hibernate/lambda_setup.rb

@@ -0,0 +1,151 @@
+require 'aws-sdk-lambda'
+require 'aws-sdk-iam'   # For creating the IAM role and policy
+require 'zip'
+require 'dotenv/load'   # Automatically loads environment variables from .env
+require 'fileutils'
+
+class LambdaSetup
+  def initialize
+    @lambda_role_name = "ec2-auto-shutdown-start"  # Define a unique role name
+    @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)
+    @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']
+    )
+  end
+
+  # Function to zip the Lambda function code
+  def create_zip_file(dir)
+    FileUtils.rm_f(@lambda_zip)  # Remove any existing zip file
+
+    Zip::File.open(@lambda_zip, Zip::File::CREATE) do |zip|
+      Dir[File.join(dir, '**', '**')].each do |file|
+        zip.add(file.sub(dir + '/', ''), file)
+      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
+    rescue Aws::IAM::Errors::NoSuchEntity
+      false  # Role does not exist
+    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: [
+          {
+            Effect: "Allow",
+            Principal: {
+              Service: "lambda.amazonaws.com"
+            },
+            Action: "sts:AssumeRole"
+          }
+        ]
+      }.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: [
+          {
+            Effect: "Allow",
+            Action: [
+              "ec2:DescribeInstances",
+              "ec2:StartInstances",
+              "ec2:StopInstances"
+            ],
+            Resource: "*"
+          }
+        ]
+      }.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,
+        policy_name: "EC2ControlPolicy",
+        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'
+      })
+
+      puts "IAM role created and policies attached."
+    else
+      puts "IAM role '#{@lambda_role_name}' already exists. Skipping role creation."
+    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
+    rescue Aws::Lambda::Errors::ResourceNotFoundException
+      false  # Lambda function does not exist
+    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
+          handler: "#{@lambda_handler}.lambda_handler",
+          code: {
+            zip_file: zip_content,
+          },
+          description: 'Lambda function to start and stop EC2 instances',
+          timeout: 30,
+        })
+        puts "Lambda function created."
+      rescue Aws::Lambda::Errors::ServiceError => e
+        puts "Error creating Lambda function: #{e.message}"
+      end
+    end
+  end
+
+  # Main method to execute the setup
+  def run
+    create_zip_file('lambda_function')  # Pass the directory containing your Lambda function code
+    create_lambda_role
+    create_lambda_function
+  end
+end

data/lib/hibernate/version.rb

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

data/lib/hibernate.rb

@@ -0,0 +1,6 @@
+require 'aws-sdk-ec2'
+require 'aws-sdk-cloudwatch'
+require 'dotenv'
+require 'json'
+
+Dotenv.load

metadata

@@ -0,0 +1,149 @@
+--- !ruby/object:Gem::Specification
+name: hibernate
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Your Name
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2024-09-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-ec2
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-cloudwatch
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: dotenv
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubyzip
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A Ruby gem to automate the management of EC2 instances using AWS Lambda.
+email:
+- youremail@example.com
+executables:
+- hibernate
+extensions: []
+extra_rdoc_files: []
+files:
+- Readme.md
+- bin/hibernate
+- lib/hibernate.rb
+- lib/hibernate/ec2_manager.rb
+- lib/hibernate/lambda_function/function.rb
+- lib/hibernate/lambda_setup.rb
+- lib/hibernate/version.rb
+homepage: https://github.com/yourusername/ec2_manager
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.1
+signing_key:
+specification_version: 4
+summary: Manage EC2 instances with Lambda functions
+test_files: []