-
Notifications
You must be signed in to change notification settings - Fork 87
/
github.rb
72 lines (60 loc) · 2.21 KB
/
github.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
# 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
@@runner_labels = nil
def self.runner_labels
@@runner_labels ||= YAML.load_file("config/github_runner_labels.yml").to_h { [_1["name"], _1] }
end
def self.failed_deliveries(since, max_page = 20)
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