jenkins_poll.rb

#!/usr/bin/env ruby
# frozen_string_literal: true
#
# Copyright (C) 2015-2016 Harald Sitter <sitter@kde.org>
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) version 3, or any
# later version accepted by the membership of KDE e.V. (or its
# successor approved by the membership of KDE e.V.), which shall
# act as a proxy defined in Section 6 of version 3 of the license.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library.  If not, see <http://www.gnu.org/licenses/>.

require 'date'
require 'logger'
require 'logger/colors'
require 'optparse'

require_relative 'lib/jenkins'
require_relative 'lib/thread_pool'
require_relative 'lib/retry'

QUALIFIER_STATES = %w[success unstable].freeze

OptionParser.new do |opts|
  opts.banner = <<-EOS
Usage: jenkins_poll.rb 'regex'

Tells jenkins to poll for any changes and start the job if they are found.

regex must be a valid Ruby regular expression matching the jobs you wish to
retry.

e.g.
  • All build jobs for vivid and utopic:
    '^(vivid|utopic)_.*_.*src'

  • All unstable builds:
    '^.*_unstable_.*src'

  • All jobs:
    '.*src'
  EOS
end.parse!

@log = Logger.new(STDOUT).tap do |l|
  l.progname = 'poll'
  l.level = Logger::INFO
end

raise 'Need ruby pattern as argv0' if ARGV.empty?

pattern = Regexp.new(ARGV[0])
@log.info pattern

job_name_queue = Queue.new
job_names = Jenkins.job.list_all
job_names.each do |name|
  next unless pattern.match(name)

  job_name_queue << name
end

module Jenkins
  # A Jenkins job that actually can be used as an object...
  class Job
    # Build Details helper class to expand builddetails from Jenkins
    class BuildDetails < OpenStruct
      def date
        @date ||= Date.parse(Time.at(timestamp / 1000).to_s)
      end
    end

    attr_reader :name

    def initialize(name)
      @name = name
    end

    def build
      BuildDetails.new(Jenkins.job.get_build_details(@name, 0))
    end

    def queued?
      Jenkins.client.queue.list.include?(@name)
    end

    private

    def method_missing(name, *args, &block)
      args.unshift(@name)
      Jenkins.job.send(name.to_sym, *args, &block)
    end
  end
end

BlockingThreadPool.run do
  until job_name_queue.empty?
    name = job_name_queue.pop(true)
    job = Jenkins::Job.new(name)
    next if job.queued?

    Retry.retry_it(times: 5, name: name) do
      @log.warn "  #{name} --> poll"
      job.poll
    end
  end
end