ci-bundle-update.rb

require 'circleci'
require 'httparty'

module CiBundleUpdate
  class Base
    def initialize(token, exec_days, raw_extra_build_parameters)
      @exec_days = exec_days
      @raw_extra_build_parameters = raw_extra_build_parameters
      post_initialize(token, exec_days)
    end

    def build
      raise NotImplementedError
    end

    private

    attr_reader :exec_days, :raw_extra_build_parameters

    def post_initialize(_token, _exec_days)
      nil
    end

    def extra_build_parameters
      raise NotImplementedError
    end

    def skip?
      exec_days &&
        ! exec_days.split(',').include?(Time.now.strftime('%a'))
    end
  end

  class CircleCi < Base
    def build(username, reponame, branch)
      if skip?
        puts "This build was skipped for $EXEC_DAYS (#{exec_days})"
        return
      end

      response = ::CircleCi::Project.new(username, reponame).build_branch(
        branch,
        {},
        build_parameters: { 'BUNDLE_UPDATE' => true }.merge(extra_build_parameters)
      )

      if response.success?
        puts "This build was accepted"
      else
        puts "This build was not accepted for #{response.body}"
      end
    end

    private

    def post_initialize(token, _exec_days)
      ::CircleCi.configure do |config|
        config.token = token
      end
    end

    def extra_build_parameters
      @extra_build_parameters ||=
        begin
          ary = (raw_extra_build_parameters || "")
                  .split(',')
                  .map{|e| e.split('=')}
                  .flatten
          Hash[*ary]
        end
    end
  end

  class Wercker < Base
    include HTTParty

    base_uri 'https://app.wercker.com/api/v3'

    def build(username, reponame, branch)
      if skip?
        puts "This build was skipped for $EXEC_DAYS (#{exec_days})"
        return
      end

      application = get_application(username, reponame)
      build_pipeline = get_build_pipeline(application['id'], branch)
      raise 'build pipeline not found.' unless build_pipeline
      run_build(build_pipeline['id'], branch)
    end

    private

    def post_initialize(token, _exec_days)
      @headers = {
        Authorization: "Bearer #{token}",
        'Content-type': 'application/json',
      }
    end

    def get_application(username, reponame)
      self.class.get("/applications/#{username}/#{reponame}", headers: @headers).parsed_response
    end

    def get_build_pipeline(application_id, branch)
      query = {
        applicationId: application_id,
        branch: branch,
      }

      runs = self.class.get('/runs', query: query, headers: @headers).parsed_response
      run = runs.find { |run| run['pipeline']['name'] == 'build' }
      run['pipeline'] if run
    end

    def run_build(pipeline_id, branch)
      body = {
        pipelineId: pipeline_id,
        branch: branch,
        envVars: [
          { key: 'BUNDLE_UPDATE', value: 'true' },
          *extra_build_parameters
        ],
      }

      self.class.post('/runs', body: body.to_json, headers: @headers)
    end

    def extra_build_parameters
      @extra_build_parameters ||=
        (raw_extra_build_parameters || "")
          .split(',')
          .map{|e| e.split('=')}
          .map{|k,v| {key: k, value: v}}
    end
  end
end

if $0 == __FILE__
  require 'optparse'
  options = ARGV.getopts(nil, 'ci:circle_ci')

  case options['ci']
  when 'circle_ci'
    ci_bundle_update = CiBundleUpdate::CircleCi.new(
      ENV['CIRCLECI_TOKEN'],
      ENV['EXEC_DAYS'],
      ENV['EXTRA_BUILD_PARAMETERS']
    )

  when 'wercker'
    ci_bundle_update = CiBundleUpdate::Wercker.new(
      ENV['WERCKER_TOKEN'],
      ENV['EXEC_DAYS'],
      ENV['EXTRA_BUILD_PARAMETERS']
    )
  end

  ci_bundle_update.build(ENV['GITHUB_USERNAME'], ENV['GITHUB_REPONAME'], ENV['BRANCH'])
end