0efd1ffc34
We have a several labels that serve as an alias for another label. Currently, we keep the specifications of each label separately, which causes a lot of duplicate data in the YAML file and makes it hard to see the relationships between them. We can store the alias information in the YAML file and automatically map the aliases to the actual labels when loading the file.
81 lines
2.4 KiB
Ruby
81 lines
2.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "octokit"
|
|
require "jwt"
|
|
require "yaml"
|
|
|
|
module Github
|
|
def self.oauth_client
|
|
Octokit::Client.new(client_id: Config.github_app_client_id, client_secret: Config.github_app_client_secret)
|
|
end
|
|
|
|
def self.app_client
|
|
current = Time.now.to_i
|
|
private_key = OpenSSL::PKey::RSA.new(Config.github_app_private_key)
|
|
key = {
|
|
iat: current,
|
|
exp: current + (10 * 60),
|
|
iss: Config.github_app_id
|
|
}
|
|
jwt = JWT.encode(key, private_key, "RS256")
|
|
|
|
Octokit::Client.new(bearer_token: jwt)
|
|
end
|
|
|
|
def self.installation_client(installation_id)
|
|
access_token = app_client.create_app_installation_access_token(installation_id)[:token]
|
|
|
|
client = Octokit::Client.new(access_token: access_token)
|
|
client.auto_paginate = true
|
|
client
|
|
end
|
|
|
|
# :nocov:
|
|
def self.freeze
|
|
runner_labels
|
|
super
|
|
end
|
|
# :nocov:
|
|
|
|
def self.runner_labels
|
|
@runner_labels ||= begin
|
|
labels = YAML.load_file("config/github_runner_labels.yml").to_h { [_1["name"], _1] }
|
|
labels.transform_values { |v| (a = v["alias_for"]) ? labels[a] : v }
|
|
end
|
|
end
|
|
|
|
def self.failed_deliveries(since, max_page = 50)
|
|
client = Github.app_client
|
|
all_deliveries = client.get("/app/hook/deliveries?per_page=100")
|
|
page = 1
|
|
while (next_url = client.last_response.rels[:next]&.href) && (since < all_deliveries.last[:delivered_at])
|
|
if page >= max_page
|
|
Clog.emit("failed deliveries page limit reached") { {deliveries: {max_page: max_page, since: since}} }
|
|
break
|
|
end
|
|
page += 1
|
|
all_deliveries += client.get(next_url)
|
|
end
|
|
|
|
Clog.emit("fetched deliveries") { {deliveries: {total: all_deliveries.count, page: page, since: since}} }
|
|
|
|
all_deliveries
|
|
.reject { _1[:delivered_at] < since }
|
|
.group_by { _1[:guid] }
|
|
.values
|
|
.reject { |group| group.any? { _1[:status] == "OK" } }
|
|
.map { |group| group.max_by { _1[:delivered_at] } }
|
|
end
|
|
|
|
def self.redeliver_failed_deliveries(*)
|
|
client = Github.app_client
|
|
failed_deliveries = Github.failed_deliveries(*)
|
|
failed_deliveries.each do |delivery|
|
|
Clog.emit("redelivering failed delivery") { {delivery: delivery.to_h} }
|
|
client.post("/app/hook/deliveries/#{delivery[:id]}/attempts")
|
|
end
|
|
Clog.emit("redelivered failed deliveries") { {deliveries: {failed: failed_deliveries.count}} }
|
|
failed_deliveries
|
|
end
|
|
end
|