codebuild-tail 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 725f46b3de3bd6b62ec3103f2f9dfb250b36a40d
+  data.tar.gz: d3c99c52a294767bfdf85bb108461a36950a213a
+SHA512:
+  metadata.gz: 220fcf23d5428dd8c95ea796f1a9eff3e36b343c368aaf3382bd5f6cafb9637a46a3c7bbfb3472172fbc045f14efb0dbfb6920662b5fd7898be7aad6d09349bf
+  data.tar.gz: 4a2c100af5a95112bcae79fa01a19e339543acd2816547eefd938134f5daf1630dd82dff7d1282422b7f4459a8d9f1285fb377ab2beacf61c363808e5c733a72

data/bin/codebuild-tail

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+require 'code_build_tail/main'
+
+STDOUT.sync = true
+main = CodeBuildTail::Main.new($stdin)
+exit(main.run)

data/lib/code_build_tail/build.rb

@@ -0,0 +1,37 @@
+
+module CodeBuildTail
+  class Build
+    attr_reader :build_id, :build, :refresh_count
+    def initialize(client, build_id)
+      @refresh_count = 0
+      @client = client
+      @build_id = build_id
+    end
+
+    def logs
+      @build.logs
+    end
+
+    def missing?
+      @build.nil?
+    end
+
+    def running?
+      @build.build_status == "IN_PROGRESS"
+    end
+
+    def successful?
+      @build.build_status == "SUCCEEDED"
+    end
+
+    def refresh
+      response = @client.batch_get_builds({ ids: [@build_id] })
+      @refresh_count += 1
+      @build = if response[:builds].empty?
+                 nil
+               else
+                 response[:builds][0]
+               end
+    end
+  end
+end

data/lib/code_build_tail/logs.rb

@@ -0,0 +1,34 @@
+
+module CodeBuildTail
+  class Logs
+    def initialize(_cloudwatch_client, poll_from_start, log_limit)
+      @log_limit = log_limit
+      @next_forward_token = nil
+      @poll_from_start = poll_from_start
+      @cloudwatch_client = Aws::CloudWatchLogs::Client.new
+    end
+
+    def get_latest_logs(group_name, stream_name)
+      params = {
+        log_group_name: group_name,
+        log_stream_name: stream_name,
+        next_token: @next_forward_token,
+        start_from_head: @poll_from_start,
+        limit: @log_limit
+      }
+      resp = @cloudwatch_client.get_log_events(params)
+      @next_forward_token = resp.next_forward_token
+      resp.events
+    end
+
+    def poll_and_show_logs(group_name, stream_name)
+      log_events = get_latest_logs(group_name, stream_name)
+      until log_events.empty?
+        log_events.each do |event|
+          print event.message
+        end
+        log_events = get_latest_logs(group_name, stream_name)
+      end
+    end
+  end
+end

data/lib/code_build_tail/main.rb

@@ -0,0 +1,64 @@
+
+require 'json'
+require 'aws-sdk-codebuild'
+require 'aws-sdk-cloudwatchlogs'
+require 'code_build_tail/logs'
+require 'code_build_tail/build'
+
+module CodeBuildTail
+  class Main
+    def initialize(source)
+      @source = source
+      @logs = CodeBuildTail::Logs.new(Aws::CloudWatchLogs::Client.new, true, 100)
+    end
+
+    def read_build
+      build_str = @source.read
+      JSON.parse(build_str)
+    rescue JSON::ParserError => e
+      puts e
+      nil
+    end
+
+    def run
+      build_hash = read_build
+      if build_hash.nil?
+        puts "Unable to parse build"
+        return 1
+      end
+      unless build_hash.key?("build") && build_hash["build"].key?("id")
+        puts "Parsed build json, but no id field"
+        return 1
+      end
+      @build = CodeBuildTail::Build.new(Aws::CodeBuild::Client.new, build_hash["build"]["id"])
+      follow_build
+    end
+
+    def follow_build
+      @build.refresh
+      if @build.missing?
+        puts "No build found for id #{@build.build_id}"
+        return 1
+      end
+
+      loop do
+        @logs.poll_and_show_logs(@build.logs.group_name, @build.logs.stream_name) unless @build.logs.nil?
+        @build.refresh
+        break unless @build.running?
+
+        # Prevent us from getting rate limitted.
+        sleep(sleep_time)
+      end
+      @logs.poll_and_show_logs(@build.logs.group_name, @build.logs.stream_name) unless @build.logs.nil?
+      if @build.successful?
+        0
+      else
+        1
+      end
+    end
+
+    def sleep_time
+      5 + @build.refresh_count
+    end
+  end
+end

data/lib/code_build_tail/version.rb

@@ -0,0 +1,5 @@
+
+
+module CodeBuildTail
+  VERSION = "0.1.0".freeze
+end

data/lib/code_build_tail.rb

@@ -0,0 +1,5 @@
+
+
+require "code_build_tail/build"
+require "code_build_tail/logs"
+require "code_build_tail/main"

metadata

@@ -0,0 +1,108 @@
+--- !ruby/object:Gem::Specification
+name: codebuild-tail
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Mark Roddy
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-08-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.8'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.8'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.43.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.43.0
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-cloudwatchlogs
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.23'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.23'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-codebuild
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.38'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.38'
+description: codebuild-tail lets you follow the progress of running AWS CodeBuild
+  jobs
+email:
+- mroddy@primary.com
+executables:
+- codebuild-tail
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/codebuild-tail
+- lib/code_build_tail.rb
+- lib/code_build_tail/build.rb
+- lib/code_build_tail/logs.rb
+- lib/code_build_tail/main.rb
+- lib/code_build_tail/version.rb
+homepage: https://github.com/PrimaryKids/codebuild-tail
+licenses:
+- Apache
+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: []
+rubyforge_project: 
+rubygems_version: 2.4.5.1
+signing_key: 
+specification_version: 4
+summary: Follow the status of CodeBuild jobs
+test_files: []