awshark 1.0.0

data/lib/awshark/cloud_formation/parameters.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+module Awshark
+  module CloudFormation
+    class Parameters
+      include FileLoading
+
+      attr_reader :stage
+
+      def initialize(path:, stage: nil)
+        @filepath = Dir.glob("#{path}/parameters.*").detect do |f|
+          %w[.json .yml .yaml].include?(File.extname(f))
+        end
+        @stage = stage
+      end
+
+      def params
+        @params ||= load_parameters(@filepath)
+      end
+
+      def to_hash
+        params
+      end
+
+      def stack_parameters
+        params.each.map do |k, v|
+          {
+            parameter_key: k,
+            parameter_value: v
+          }
+        end
+      end
+
+      private
+
+      def load_parameters(filepath)
+        data = load_file(filepath) || {}
+
+        data[stage] || data
+      end
+    end
+  end
+end

data/lib/awshark/cloud_formation/stack.rb

@@ -0,0 +1,94 @@
+# frozen_string_literal: true
+
+require 'forwardable'
+
+#
+# https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/CloudFormation/Types/Stack.html
+#
+module Awshark
+  module CloudFormation
+    class Stack
+      extend Forwardable
+
+      attr_reader :name
+
+      def_delegators :@stack,
+                     :stack_id,
+                     :stack_name,
+                     :description,
+                     :parameters,
+                     :creation_time,
+                     :deletion_time,
+                     :last_updated_time,
+                     :stack_status,
+                     :stack_status_reason,
+                     :notification_arns,
+                     :capabilities,
+                     :outputs,
+                     :role_arn,
+                     :tags
+
+      def initialize(name:)
+        @name = name
+        @stack = get_stack(name)
+      end
+
+      # @return [Boolean]
+      def exists?
+        @stack.present?
+      end
+
+      def events
+        response = client.describe_stack_events(stack_name: stack_id)
+        response.stack_events
+      end
+
+      def create_or_update(params)
+        if exists?
+          client.update_stack(params)
+        else
+          client.create_stack(params)
+        end
+      end
+
+      def reload
+        @stack = get_stack(name)
+
+        self
+      end
+
+      # @return [Hash]
+      def template
+        return nil unless exists?
+        return @template if @template.present?
+
+        response = client.get_template(stack_name: stack_id)
+        @template = response.template_body
+      end
+
+      private
+
+      def client
+        return Awshark.config.cloud_formation.client if Awshark.config.cloud_formation.client
+
+        region = Aws.config[:region]
+        @client ||= Aws::CloudFormation::Client.new(region: region)
+      end
+
+      def get_stack(stack_name)
+        response = begin
+                     client.describe_stacks(stack_name: stack_name)
+                   rescue Aws::CloudFormation::Errors::ValidationError
+                     @stack = nil
+                     return
+                   end
+
+        if response.stacks.length > 1
+          raise ArgumentError, "Found too many stacks with name #{stack_name}. There should only be one."
+        end
+
+        response.stacks[0]
+      end
+    end
+  end
+end

data/lib/awshark/cloud_formation/stack_events.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+module Awshark
+  module CloudFormation
+    class StackEvents
+      attr_reader :stack
+      attr_reader :start_time, :last_event
+
+      def initialize(stack)
+        @stack = stack
+        @start_time = Time.now - 10
+      end
+
+      def done?
+        return false if last_event.nil?
+        return false if last_event.resource_type != 'AWS::CloudFormation::Stack'
+
+        last_event.resource_status.match(/IN_PROGRESS/).nil?
+      end
+
+      def new_events
+        events = stack.events
+
+        return [] if events.blank?
+
+        if last_event.nil?
+          @last_event = events.first
+          new_events = events.select { |e| e.timestamp > start_time }
+          return new_events
+        end
+
+        return [] unless new_events?(events)
+
+        last_event_index = events.index { |e| e.event_id == last_event.event_id }
+        new_events = events[0..last_event_index - 1]
+        @last_event = new_events.first
+
+        new_events
+      end
+
+      def new_events?(events)
+        events.first.event_id != last_event.event_id
+      end
+    end
+  end
+end

data/lib/awshark/cloud_formation/template.rb

@@ -0,0 +1,102 @@
+# frozen_string_literal: true
+
+module Awshark
+  module CloudFormation
+    class Template
+      include FileLoading
+
+      attr_reader :path
+      attr_reader :bucket, :name, :stage
+
+      def initialize(path, options = {})
+        @path = path
+
+        @bucket = options[:bucket]
+        @name = options[:name]
+        @stage = options[:stage]
+      end
+
+      # @returns [Hash]
+      def as_json
+        load_file(template_path, context)
+      end
+
+      # @returns [String]
+      def body
+        JSON.pretty_generate(as_json)
+      end
+
+      # @returns [Hash]
+      def context
+        @context ||= begin
+                       context = load_file(context_path) || {}
+                       context = context[stage] if context.key?(stage)
+
+                       {
+                         context: RecursiveOpenStruct.new(context),
+                         aws_account_id: Awshark.config.aws_account_id,
+                         stage: stage
+                       }
+                     end
+      end
+
+      # @returns [Integer]
+      def size
+        body.size
+      end
+
+      # @returns [Boolean]
+      def uploaded?
+        @uploaded == true
+      end
+
+      # @returns [String]
+      def url
+        upload unless uploaded?
+
+        "https://#{bucket}.s3.#{region}.amazonaws.com/#{s3_key}"
+      end
+
+      private
+
+      def region
+        Aws.config[:region] || 'eu-central-1'
+      end
+
+      def s3
+        return Awshark.config.s3.client if Awshark.config.s3.client
+
+        @s3 ||= Aws::S3::Client.new(region: region, signature_version: 'v4')
+      end
+
+      def s3_key
+        # https://apidock.com/ruby/Time/strftime
+        @s3_key ||= "awshark/#{name}/#{Time.now.strftime('%Y-%m-%d')}.json"
+      end
+
+      def upload
+        raise ArgumentError, 'Bucket for template upload to S3 is missing' if bucket.blank?
+
+        Awshark.logger.debug "[awshark] Uploading CF template to #{bucket}"
+
+        s3.put_object(bucket: bucket, key: s3_key, body: body)
+      end
+
+      def context_path
+        Dir.glob("#{path}/context.*").detect do |f|
+          %w[.json .yml .yaml].include?(File.extname(f))
+        end
+      end
+
+      def template_path
+        @template_path ||= if File.directory?(path)
+                             Dir.glob("#{path}/template.*").detect do |f|
+                               %w[.json .yml .yaml].include?(File.extname(f))
+                             end
+                           else
+                             path
+                           end
+      end
+    end
+  end
+end

data/lib/awshark/configuration.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module Awshark
+  class Configuration
+    attr_accessor :cloud_formation, :s3
+
+    def initialize
+      @cloud_formation = OpenStruct.new(
+        client: nil,
+        event_polling: 3
+      )
+
+      @s3 = OpenStruct.new(
+        client: nil
+      )
+    end
+
+    def aws_account_id
+      return @aws_account_id if defined?(@aws_account_id)
+
+      response = sts_client.get_caller_identity
+
+      @aws_account_id = response.account
+    end
+
+    def sts_client
+      @sts_client ||= Aws::STS::Client.new
+    end
+
+    private
+
+    attr_writer :sts_client
+  end
+end

data/lib/awshark/ec2/instance.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+module Awshark
+  module EC2
+    class Instance
+      attr_reader :id
+      attr_reader :type
+      attr_reader :state
+      attr_reader :private_ip_address
+      attr_reader :public_ip_address
+      attr_reader :public_dns_name
+      attr_reader :vpc_id
+      attr_reader :subnet_id
+      attr_reader :tags
+
+      def initialize(instance)
+        @id = instance.instance_id
+        @type = instance.instance_type
+        @state = instance.state.name
+        @private_ip_address = instance.private_ip_address
+        @public_ip_address = instance.public_ip_address
+        @public_dns_name = instance.public_dns_name
+        @vpc_id = instance.vpc_id
+        @subnet_id = instance.subnet_id
+        @tags = instance.tags
+      end
+
+      def name
+        tag_value(tags, 'Name').split(' - ').last
+      rescue StandardError
+        'null'
+      end
+
+      private
+
+      def tag_value(tags, key)
+        return nil if tags.empty?
+
+        tag = tags.detect { |t| t.key == key }
+        return tag.value if tag
+
+        'null'
+      end
+    end
+  end
+end

data/lib/awshark/ec2/manager.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+module Awshark
+  module EC2
+    class Manager
+      def all_instances
+        return @all_instances if defined?(@all_instances)
+
+        @all_instances = []
+        response = client.describe_instances
+
+        response.each_page do |page|
+          page.reservations.each do |reservation|
+            reservation.instances.each do |instance|
+              @all_instances << Instance.new(instance)
+            end
+          end
+        end
+
+        @all_instances
+      end
+
+      %i[running stopped terminated].each do |state|
+        define_method "#{state}_instances" do
+          all_instances.select { |i| i.state == state.to_s }
+        end
+      end
+
+      private
+
+      def client
+        @client ||= Aws::EC2::Client.new(region: region)
+      end
+
+      def region
+        Aws.config[:region] || 'eu-central-1'
+      end
+    end
+  end
+end

data/lib/awshark/profile_resolver.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+module Awshark
+  class ProfileResolver
+    attr_reader :region
+
+    def initialize(options)
+      @profile = options[:profile] || ENV['AWS_PROFILE']
+      @shared_config = ::Aws::SharedConfig.new(
+        profile_name: @profile,
+        config_enabled: true
+      )
+      @region = options[:region] || @shared_config.region || 'eu-central-1'
+    end
+
+    def credentials
+      user_credentials || role_credentials
+    end
+
+    # Returns Aws credentials for configuration with
+    #  [profile]
+    #  aws_access_key_id=AWS_ACCESS_KEY_ID
+    #  aws_secret_access_key=AWS_SECRET_ACCESS_KEY
+    #
+    # Returns nil for configuration with
+    #  [profile]
+    #  role_arn = ROLE_ARN
+    #  source_profile = SOURCE_PROFILE
+    #
+    # @returns [Aws::Credentials]
+    def user_credentials
+      @shared_config.credentials
+    end
+
+    # Returns Aws credentials for configuration with
+    #  [profile]
+    #  role_arn = ROLE_ARN
+    #  source_profile = SOURCE_PROFILE
+    #
+    # @throws [Aws::STS::Errors::AccessDenied] if MultiFactorAuthentication failed
+    # @returns [Aws::Credentials]
+    def role_credentials
+      @shared_config.assume_role_credentials_from_config(
+        region: region,
+        token_code: mfa_token
+      )
+    end
+
+    def mfa_token
+      print %(Please enter MFA token for AWS account #{@profile}: )
+      $stdin.gets.strip
+    end
+  end
+end