hibernate 0.1.5 → 0.1.7

data/bin/hibernate

@@ -1,154 +1,91 @@
 #!/usr/bin/env ruby
 
-require 'optparse'
-require_relative '../lib/hibernate/lambda_setup'
-require_relative '../lib/hibernate/ec2_manager'
-
-class HibernateCLI
-  def self.run
-    command = nil
-    options = {}
-
-    if ARGV.include?('setup')
-      command = :setup
-      ARGV.delete('setup')
-    elsif ARGV.include?('rule')
-      command = :manage_ec2
-      ARGV.delete('rule')
-    end
-
-    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
-
-      parser.on('--start-expression=<START_CRON>', 'Specify the cron expression to start the instance') do |start_cron|
-        options[:start_cron] = start_cron
-      end
-
-      parser.on('--stop-expression=<STOP_CRON>', 'Specify the cron expression to stop the instance') do |stop_cron|
-        options[:stop_cron] = stop_cron
-      end
-
-      parser.on('--start-instance=<true/false>', 'Filter start rules') do |start_instance|
-        options[:start_instance] = start_instance == 'true'
-      end
-
-      parser.on('--stop-instance=<true/false>', 'Filter stop rules') do |stop_instance|
-        options[:stop_instance] = stop_instance == 'true'
-      end
-
-      parser.on('--rule-name=<RULE Name>', 'Specify the rule name to remove or update') do |rule_name|
-        options[:rule_name] = rule_name
-      end
-
-      parser.on('--update', 'Update an existing rule') do
-        options[:update] = true
-      end
-
-      parser.on('--remove', 'Remove an existing rule') do
-        options[:remove] = true
-      end
-
-      parser.on('--create', 'Create a new rule') do
-        options[:create] = true
-      end
-
-      parser.on('--list', 'List rules') do
-        options[:list] = true
-      end
-
-      parser.on('--state=<enable/disable>', 'Set rule state to either enable or disable') do |state|
-        options[:state] = state
-      end
-    end
+require 'hibernate'
+
+SUB_COMMANDS = %w[setup rule].freeze
+RULE_COMMANDS = %w[create list update remove].freeze
+
+global_opts = Optimist::options do
+  version "Hibernate #{Hibernate::VERSION} (c) #{Date.today.year} Manish Sharma"
+  banner <<-USAGE
+    Automate the shutdown and start of our EC2 instances.
+
+    Usage:
+       hibernate <command> [options]
+
+    where <command> are:
+       setup        : setup the AWS IAM role and Lambda
+       rule         : Manage the schedule for start/stop of EC2 Instance.
+        - create    : Create a new rule
+        - list      : List existing rules
+        - update    : Update an existing rule
+        - remove    : Remove an existing rule
+    and [options] are:
+       these are specific to the <command>. use hibernate <command> --help to learn about [options] for that command.
+  USAGE
+  stop_on SUB_COMMANDS
+end
 
-    begin
-      parser.parse!(ARGV)
-    rescue OptionParser::ParseError => e
-      puts e.message
-      puts parser
-      exit 1
+def parse_options_for_rule(subcmd)
+  case subcmd
+  when 'create'
+    Optimist::options do
+      opt :profile, 'Specify the profile name', short: '-p', type: :string
+      opt :instance_name, 'Specify the EC2 instance name', short: '-i', type: :string
+      opt :start, 'Specify the cron expression to start the instance', short: '-s', type: :string
+      opt :stop, 'Specify the cron expression to stop the instance', short: '-e', type: :string
     end
-
-    if command.nil?
-      puts "Please provide a command."
-      puts parser
-      exit
+  when 'list'
+    Optimist::options do
+      opt :profile, 'Specify the profile name', short: '-p', type: :string
+      opt :instance_name, 'Specify the EC2 instance name', short: '-i', type: :string
+      opt :start, 'List only the start action rules', short: '-s'
+      opt :stop, 'List only the stop action rules', short: '-e'
     end
-
-    case command
-    when :setup
-      create_lambda_function
-    when :manage_ec2
-      if options[:update]
-        update_ec2_rule_command(options)
-      elsif options[:remove]
-        remove_ec2_rule_command(options)
-      elsif options[:create]
-        create_ec2_rule_command(options)
-      elsif options[:list]
-        list_ec2_rule_command(options)
-      end
+  when 'update'
+    Optimist::options do
+      opt :profile, 'Specify the profile name', short: '-p', type: :string
+      opt :rule, 'Specify the rule name to update', short: '-r', type: :string
+      opt :start, 'Specify the cron expression to start the instance', short: '-s', type: :string
+      opt :stop, 'Specify the cron expression to stop the instance', short: '-e', type: :string
+      opt :state, 'Set rule state to either enable or disable', short: '-a', type: :string, default: nil
     end
-  end
-
-  def self.create_lambda_function
-    LambdaSetup.new.run
-  end
-
-  def self.create_ec2_rule_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 rule --create --instance-name=<INSTANCE_NAME> --start-expression=<START_CRON> --stop-expression=<STOP_CRON>"
-      exit
-    else
-      ec2_manager = EC2Manager.new(options[:instance_name])
-      ec2_manager.create_event_rule(options[:start_cron], options[:stop_cron])
+  when 'remove'
+    Optimist::options do
+      opt :profile, 'Specify the profile name', short: '-p', type: :string
+      opt :instance_name, 'Specify the EC2 instance name', short: '-i', type: :string
+      opt :rule, 'Specify the rule name to remove', short: '-r', type: :string
     end
+  else
+    Optimist::die "unknown subcommand #{subcmd.inspect}"
   end
+end
 
-  def self.remove_ec2_rule_command(options)
-    if options[:rule_name].nil?
-      puts "Please provide the rule name to remove."
-      puts "Usage: hibernate rule --remove --rule-name=<RULE NAME>"
-      exit
-    else
-      ec2_manager = EC2Manager.new(options[:instance_name])
-      ec2_manager.remove_event_rule(options[:rule_name])
-    end
+cmd = ARGV.shift # get the subcommand
+options = case cmd
+when 'setup'
+  Optimist::options do
+    opt :profile, 'Specify the profile name', short: '-p', type: :string
   end
-
-  def self.list_ec2_rule_command(options)
-    ec2_manager = EC2Manager.new(options[:instance_name])
-    ec2_manager.list_event_rules(options)
-  end
-
-  def self.update_ec2_rule_command(options)
-    if options[:rule_name].nil?
-      puts "Please provide the rule name to update."
-      puts "Usage: hibernate rule --update --rule-name=<RULE_NAME> [--start-expression=<START_CRON>] [--stop-expression=<STOP_CRON>] [--state=<enable/disable>]"
-      exit
-    elsif options[:start_cron].nil? && options[:stop_cron].nil? && options[:state].nil?
-      puts "Please provide atleast one attribute to update."
-      puts "Usage: hibernate rule --update --rule-name=<RULE_NAME> [--start-expression=<START_CRON>] [--stop-expression=<STOP_CRON>] [--state=<enable/disable>]"
-      exit
-    else
-      ec2_manager = EC2Manager.new(options[:instance_name])
-      ec2_manager.update_event_rule(
-        rule_name: options[:rule_name],
-        start_cron: options[:start_cron],
-        stop_cron: options[:stop_cron],
-        state: options[:state]
-      )
-    end
+when 'rule'
+  subcmd = ARGV.shift
+  unless RULE_COMMANDS.include?(subcmd)
+    Optimist::die "unknown subcommand #{subcmd.inspect}"
   end
+  parse_options_for_rule(subcmd)
+else
+  Optimist::die "unknown command #{cmd.inspect}"
 end
 
-HibernateCLI.run
+puts "Global options: #{global_opts.inspect}"
+puts "Subcommand: #{cmd.inspect}"
+puts "Subcommand options: #{options.inspect}"
+puts "Remaining arguments: #{ARGV.inspect}"
+
+HibernateCli.setup_profile(options[:profile])
+case cmd
+when 'setup'
+  HibernateCli.create_lambda_function
+when 'rule'
+  HibernateCli.handle_rule_subcommand(subcmd, options)
+end

data/config.yaml.sample

@@ -0,0 +1,13 @@
+aws_accounts:
+  profile_name:
+    account_id: "<account_id_1>"
+    region: us-east-1
+    credentials:
+      access_key_id: ACCESS_KEY_1
+      secret_access_key: SECRET_KEY_1
+  profile_other:
+    account_id:<account_id_2>
+    region: us-west-2
+    credentials:
+      access_key_id: ACCESS_KEY_2
+      secret_access_key: SECRET_KEY_2

data/docs/index.html

@@ -0,0 +1,182 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>EC2 Auto Shutdown and Start</title>
+    <link rel="stylesheet" href="style.css">
+    <script type="text/javascript" src="https://cdn.jsdelivr.net/npm/mermaid@latest/dist/mermaid.min.js"></script>
+    <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/prism/1.29.0/themes/prism.min.css" />
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/prism/1.29.0/prism.min.js"></script>
+</head>
+<body>
+  <header>
+      <h1>Hibernate</h1>
+  </header>
+  <main>
+    <section id="project-overview">
+        <h2>Project Overview</h2>
+        <p>This project automates the start and stop of AWS EC2 instances based on scheduled CloudWatch Events using an AWS Lambda function. It leverages Ruby AWS SDK for creating the Lambda function, CloudWatch Events, and EC2 instance management. You can configure the instance and schedule using the CLI.</p>
+        <div>
+            <pre class="mermaid">
+                graph TD
+                A[User - Command Line Interface]
+                B[CloudWatch Events]
+                C[Lambda Function]
+                D{Start or Stop EC2 Instance?}
+                E[Start EC2 Instance]
+                F[Stop EC2 Instance]
+                G[CloudWatch Logs]
+
+                subgraph AWS Cloud
+                    B
+                    C
+                    D
+                    E
+                    F
+                    G
+                end
+
+                A -->|Configure CloudWatch Events| B
+                B -->|Trigger Event| C
+                C -->|Execute Script| D
+                D -->|Start| E
+                D -->|Stop| F
+                C -->|Log Execution| G
+                A -.->|Monitor Logs| G
+
+                classDef default fill:#f7f7f7,stroke:#2c3e50,stroke-width:2px;
+                classDef input fill:#3498db,stroke:#2c3e50,stroke-width:2px,color:#fff;
+                classDef process fill:#1abc9c,stroke:#2c3e50,stroke-width:2px,color:#fff;
+                classDef decision fill:#f39c12,stroke:#2c3e50,stroke-width:2px;
+                classDef output fill:#2ecc71,stroke:#2c3e50,stroke-width:2px,color:#fff;
+                classDef storage fill:#e74c3c,stroke:#2c3e50,stroke-width:2px,color:#fff;
+
+                class A input;
+                class B,C process;
+                class D decision;
+                class E,F output;
+                class G storage;
+            </pre>
+        </div>
+    </section>
+
+    <section id="features">
+        <h2>Features</h2>
+        <ul>
+            <li><strong>Lambda function</strong>: Automatically manages the start and stop of EC2 instances.</li>
+            <li><strong>CloudWatch Events</strong>: Schedules based on cron expressions to run the Lambda function at specific times.</li>
+            <li><strong>CLI Interface</strong>: Easy-to-use interface for setting up the Lambda function and managing EC2 instance operations.</li>
+        </ul>
+    </section>
+
+    <section id="prerequisites">
+        <h2>Prerequisites</h2>
+        <ul>
+            <li><strong>AWS CLI</strong>: Ensure that the AWS CLI is installed and configured with access to your AWS account.</li>
+            <li><strong>Ruby 3.x</strong>: The project requires Ruby version 3.x. Install Ruby using a version manager like RVM or rbenv.</li>
+            <li><strong>AWS SDK for Ruby</strong>: The project uses <code class="language-shell">aws-sdk-ec2</code> and <code class="language-shell">aws-sdk-cloudwatchevents</code> gems.</li>
+            <li><strong>IAM Permissions</strong>: Ensure the AWS user you are using has permissions to manage CloudWatch Events, EC2, Lambda, and IAM roles.</li>
+        </ul>
+    </section>
+
+    <section id="setup-instructions">
+        <h2>Setup Instructions</h2>
+        <h3>1. Clone the Repository</h3>
+        <pre><code class="language-shell">git clone https://github.com/maniSHarma7575/hibernate.git
+cd hibernate</code></pre>
+
+        <h3>2. Environment Configuration</h3>
+        <p>Create a <code class="language-shell">config.yaml</code> file at the root of your project and configure your AWS account details as follows:</p>
+        <pre><code class="language-shell">aws_accounts:
+  profile_name:
+    account_id: "<account_id_1>"
+    region: us-east-1
+    credentials:
+      access_key_id: ACCESS_KEY_1
+      secret_access_key: SECRET_KEY_1
+  profile_other:
+    account_id: <account_id_2>
+    region: us-west-2
+    credentials:
+      access_key_id: ACCESS_KEY_2
+      secret_access_key: SECRET_KEY_2</code></pre>
+
+        <h3>3. Install Dependencies</h3>
+        <p>Make sure you have the required gems and build the gem by running:</p>
+        <pre><code class="language-shell">rake install</code></pre>
+    </section>
+
+    <section id="command-usage">
+        <h2>Command Usage</h2>
+        <p>The <code class="language-shell">hibernate</code> CLI tool provides commands for setting up the Lambda function and managing EC2 instance schedules.</p>
+
+        <h3>Setup Command</h3>
+        <pre><code class="language-shell">hibernate setup --profile &lt;PROFILE_NAME&gt;</code></pre>
+        <p><strong>Options:</strong></p>
+        <ul>
+            <li><strong>-p, --profile</strong>: Specify the AWS profile name (required).</li>
+        </ul>
+        <p><strong>Example:</strong></p>
+        <pre><code class="language-shell">hibernate setup --profile production</code></pre>
+
+        <h3>Rule Command</h3>
+        <p>Manage schedules for EC2 instance start/stop actions.</p>
+
+        <h4>Create a Schedule</h4>
+        <pre><code class="language-shell">hibernate rule create --profile &lt;PROFILE_NAME&gt; --instance-name &lt;INSTANCE_NAME&gt; --start &lt;CRON_START&gt; --stop &lt;CRON_STOP&gt;</code></pre>
+        <p><strong>Options:</strong></p>
+        <ul>
+            <li><strong>-p, --profile</strong>: Specify the AWS profile name (required).</li>
+            <li><strong>-i, --instance-name</strong>: Specify the EC2 instance name (required).</li>
+            <li><strong>-s, --start</strong>: Specify the cron expression for starting the instance (optional).</li>
+            <li><strong>-e, --stop</strong>: Specify the cron expression for stopping the instance (optional).</li>
+        </ul>
+
+        <h4>List Schedules</h4>
+        <pre><code class="language-shell">hibernate rule list --profile &lt;PROFILE_NAME&gt; --instance-name &lt;INSTANCE_NAME&gt; [--start] [--stop]</code></pre>
+        <p><strong>Options:</strong></p>
+        <ul>
+            <li><strong>-p, --profile</strong>: Specify the AWS profile name (required).</li>
+            <li><strong>-i, --instance-name</strong>: Specify the EC2 instance name (required).</li>
+            <li><strong>-s, --start</strong>: List only the start action rules (optional).</li>
+            <li><strong>-e, --stop</strong>: List only the stop action rules (optional).</li>
+        </ul>
+
+        <h4>Update a Schedule</h4>
+        <pre><code class="language-shell">hibernate rule update --profile &lt;PROFILE_NAME&gt; --rule &lt;RULE_NAME&gt; --start &lt;NEW_CRON_START&gt; --stop &lt;NEW_CRON_STOP&gt; [--state &lt;enable|disable&gt;]</code></pre>
+        <p><strong>Options:</strong></p>
+        <ul>
+            <li><strong>-p, --profile</strong>: Specify the AWS profile name (required).</li>
+            <li><strong>-r, --rule</strong>: Specify the rule name to update (required).</li>
+            <li><strong>-s, --start</strong>: Specify the new cron expression for starting the instance (optional).</li>
+            <li><strong>-e, --stop</strong>: Specify the new cron expression for stopping the instance (optional).</li>
+            <li><strong>-a, --state</strong>: Set the rule state to either enable or disable (optional).</li>
+        </ul>
+
+        <h4>Remove a Schedule</h4>
+        <pre><code class="language-shell">hibernate rule remove --profile &lt;PROFILE_NAME&gt; --instance-name &lt;INSTANCE_NAME&gt; --rule &lt;RULE_NAME&gt;</code></pre>
+        <p><strong>Options:</strong></p>
+        <ul>
+            <li><strong>-p, --profile</strong>: Specify the AWS profile name (required).</li>
+            <li><strong>-i, --instance-name</strong>: Specify the EC2 instance name (required).</li>
+            <li><strong>-r, --rule</strong>: Specify the rule name to remove (required).</li>
+        </ul>
+    </section>
+    <section id="additional-resources">
+      <h2>Additional Resources</h2>
+      <ul>
+          <li><a href="https://aws.amazon.com/documentation/lambda/">AWS Lambda Documentation</a></li>
+          <li><a href="https://aws.amazon.com/documentation/cli/">AWS CLI Documentation</a></li>
+          <li><a href="https://aws.amazon.com/documentation/ec2/">AWS EC2 Documentation</a></li>
+          <li><a href="https://rubydoc.info/gems/aws-sdk-ec2/Aws/SDK/EC2">AWS SDK for Ruby EC2 Documentation</a></li>
+          <li><a href="https://rubydoc.info/gems/aws-sdk-cloudwatchevents/Aws/SDK/CloudWatchEvents">AWS SDK for Ruby CloudWatch Events Documentation</a></li>
+      </ul>
+    </section>
+  </main>
+
+  <footer>
+    <p>&copy; 2024 Manish Sharma. All rights reserved.</p>
+  </footer>
+</body>
+</html>

data/docs/style.css

@@ -0,0 +1,79 @@
+/* Reset some basic elements */
+body, h1, h2, h3, p, ul {
+  margin: 0;
+  padding: 0;
+}
+
+/* Apply a modern font */
+body {
+  font-family: 'Roboto', sans-serif;
+  line-height: 1.6;
+  background-color: #F6F6F6; /* Light background for better contrast */
+  color: #4E4E4E; /* Charcoal text */
+}
+
+/* Header styling */
+header {
+  background: black; /* Brown */
+  color: white;
+  padding: 1rem 0;
+  text-align: center;
+}
+
+/* Main styling */
+main {
+  max-width: 800px;
+  margin: 2rem auto;
+  padding: 1rem;
+  background: white;
+  box-shadow: 0 2px 10px rgba(0, 0, 0, 0.1);
+  border-radius: 8px;
+}
+
+/* Section headings */
+h2 {
+  color: #8B5A2B; /* Brown */
+  margin-bottom: 0.5rem;
+}
+
+/* Links */
+a {
+  color: #FF5733; /* Bright Orange */
+  text-decoration: none;
+}
+
+a:hover {
+  text-decoration: underline;
+}
+
+/* List styling */
+ul {
+  list-style: none;
+}
+
+ul li {
+  padding: 0.5rem 0;
+}
+
+/* Button styling */
+button {
+  background: #D2B48C; /* Tan */
+  color: white;
+  padding: 0.5rem 1rem;
+  border: none;
+  border-radius: 4px;
+  cursor: pointer;
+  transition: background 0.3s;
+}
+
+button:hover {
+  background: #FFD700; /* Gold */
+}
+
+/* Footer styling */
+footer {
+  text-align: center;
+  padding: 1rem 0;
+  background: #4E4E4E; /* Charcoal */
+  color: white;
+}

data/hibernate.gemspec

@@ -0,0 +1,43 @@
+# my_gem.gemspec
+require_relative 'lib/hibernate/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "hibernate"
+  spec.version       = Hibernate::VERSION
+  spec.summary       = "Automating the shutdown and start of our EC2 instances"
+  spec.description   = "A Ruby gem to automate the shutdown and start of our EC2 instances"
+  spec.authors       = ["Manish Sharma"]
+  spec.email         = ["sharma.manish7575@gmail.com"]
+  spec.homepage      = "https://github.com/maniSHarma7575/hibernate"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($\)
+  spec.executables   = spec.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  spec.bindir        = 'bin'
+  spec.require_paths  = ["lib"]
+
+  spec.required_ruby_version = '>= 3.2.2'
+
+  spec.add_dependency "aws-sdk-ec2", "~> 1.478"
+  spec.add_dependency "aws-sdk-cloudwatch", "~> 1.103"
+  spec.add_dependency "aws-sdk-lambda", "~> 1.136"
+  spec.add_dependency "aws-sdk-cloudwatchevents", "~> 1.83"
+  spec.add_dependency "aws-sdk-iam", "~> 1.111"
+
+  spec.add_dependency "dotenv", "~> 3.1"
+  spec.add_dependency "json", "~> 2.7"
+  spec.add_dependency "rubyzip", "~> 2.3"
+  spec.add_dependency "optimist", "~> 3.1"
+
+  # Specify development dependencies
+  spec.add_development_dependency "rspec", "~> 3.13"
+  spec.add_development_dependency "rubocop", "~> 1.66"
+
+  spec.metadata = {
+    "source_code_uri" => "https://github.com/maniSHarma7575/hibernate",
+    "homepage_uri"    => "https://manisharma7575.github.io/hibernate/",
+    "bug_tracker_uri" => "https://github.com/maniSHarma7575/hibernate/issues"
+  }
+
+  spec.requirements << 'Git'
+end

data/lib/hibernate/cloud_watch_event_manager.rb

@@ -221,9 +221,10 @@ class CloudWatchEventManager
 
   def matches_criteria?(rule_instance_id, action, options)
     instance_id_match = options[:instance_id].nil? || options[:instance_id] == rule_instance_id
-    action_match = (options[:start_instance] && action == 'start') ||
-                   (options[:stop_instance] && action == 'stop') ||
-                   (options[:start_instance].nil? && options[:stop_instance].nil?)
+    action_match = (options[:start] && action == 'start') ||
+                   (options[:stop] && action == 'stop') ||
+                   (options[:start].nil? && options[:stop].nil?) ||
+                   ((!options[:start].nil? && !options[:stop].nil?) && (!options[:start] && !options[:stop]))
 
     instance_id_match && action_match
   end

data/lib/hibernate/hibernate_cli.rb

@@ -0,0 +1,62 @@
+class HibernateCli
+  def self.setup_profile(profile)
+    ENV['AWS_PROFILE'] = profile
+  end
+
+  def self.create_lambda_function
+    LambdaSetup.new.run
+  end
+
+  def self.handle_rule_subcommand(subcmd, options)
+    case subcmd
+    when 'create'
+      create_ec2_rule_command(options)
+    when 'list'
+      list_ec2_rule_command(options)
+    when 'update'
+      update_ec2_rule_command(options)
+    when 'remove'
+      remove_ec2_rule_command(options)
+    end
+  end
+
+  def self.create_ec2_rule_command(options)
+    validate_options(options, %i[instance_name], 'create')
+    ec2_manager = EC2Manager.new(options[:instance_name])
+    ec2_manager.create_event_rule(options[:start], options[:stop])
+  end
+
+  def self.list_ec2_rule_command(options)
+    ec2_manager = EC2Manager.new(options[:instance_name])
+    ec2_manager.list_event_rules(options)
+  end
+
+  def self.update_ec2_rule_command(options)
+    validate_options(options, %i[rule], 'update')
+    ec2_manager = EC2Manager.new(options[:instance_name])
+    ec2_manager.update_event_rule(
+      rule_name: options[:rule],
+      start_cron: options[:start],
+      stop_cron: options[:stop],
+      state: options[:state]
+    )
+  end
+
+  def self.remove_ec2_rule_command(options)
+    validate_options(options, [:rule], 'remove')
+    ec2_manager = EC2Manager.new(options[:instance_name])
+    ec2_manager.remove_event_rule(options[:rule])
+  end
+
+  private
+
+  def self.validate_options(options, required_fields, command)
+    missing_fields = required_fields.select { |field| options[field].nil? }
+    return if missing_fields.empty?
+
+    puts "Please provide #{missing_fields.join(', ')}."
+    puts "Usage: hibernate rule --#{command} #{required_fields.map { |field| "--#{field.to_s.tr('_', '-')}" }.join(' ')}"
+    exit
+  end
+end
+

data/lib/hibernate/version.rb

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

data/lib/hibernate.rb

@@ -1,7 +1,13 @@
+# frozen_string_literal: true
+
 require 'aws-sdk-ec2'
 require 'aws-sdk-cloudwatch'
 require 'json'
+require 'optimist'
+require 'hibernate/lambda_setup'
+require 'hibernate/ec2_manager'
+require 'hibernate/version'
+require 'hibernate/hibernate_cli'
 
 module Hibernate
-  require_relative 'hibernate/version'
 end

data/spec/hibernate_spec.rb

@@ -0,0 +1,8 @@
+# Placeholder for your test cases
+require 'hibernate'
+
+RSpec.describe Hibernate do
+  it 'has a version number' do
+    expect(Hibernate::VERSION).not_to be nil
+  end
+end