/
update.rb
executable file
·233 lines (201 loc) · 7.08 KB
/
update.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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
# This script is designed to loop through all dependencies in a GHE or GitLab
# project, creating PRs where necessary.
#
# It is intended to be used as a stop-gap until Dependabot's hosted instance
# supports GitHub Enterprise and GitLab (coming soon!)
require "json"
require "dependabot/file_fetchers"
require "dependabot/file_parsers"
require "dependabot/update_checkers"
require "dependabot/file_updaters"
require "dependabot/pull_request_creator"
require "dependabot/pull_request_updater"
require "dependabot/omnibus"
require "gitlab"
gitlab_hostname = ENV["GITLAB_HOSTNAME"] || "gitlab.com"
credentials = [
{
"type" => "git_source",
"host" => "github.com",
"username" => "x-access-token",
"password" => ENV["KIRA_GITHUB_PERSONAL_TOKEN"] || nil
}
]
credentials << {
"type" => "git_source",
"host" => gitlab_hostname,
"username" => "x-access-token",
# A GitLab access token with API permission
"password" => ENV["KIRA_GITLAB_PERSONAL_TOKEN"]
}
json_credentials = ENV['DEPENDABOT_EXTRA_CREDENTIALS'] || ""
unless json_credentials.to_s.strip.empty?
json_credentials = JSON.parse(json_credentials)
credentials.push(*json_credentials)
end
# Full name of the repo you want to create pull requests for.
repo_name = ENV["DEPENDABOT_PROJECT_PATH"] # namespace/project
# Directory where the base dependency files are.
directory = ENV["DEPENDABOT_DIRECTORY"] || "/"
# See lists of update strategies here:
# https://github.com/wemake-services/kira-dependencies/issues/39
update_strategy = ENV['DEPENDABOT_UPDATE_STRATEGY']&.to_sym || nil
# Assignee to be set for this merge request.
# Works best with marge-bot:
# https://github.com/smarkets/marge-bot
assignees = [ENV["DEPENDABOT_ASSIGNEE_GITLAB_ID"]].compact
assignees = nil if assignees.empty?
package_manager = ENV["PACKAGE_MANAGER"] || "bundler"
# Source branch for merge requests
source_branch = ENV["DEPENDABOT_SOURCE_BRANCH"] || nil
source = Dependabot::Source.new(
provider: "gitlab",
hostname: gitlab_hostname,
api_endpoint: "https://#{gitlab_hostname}/api/v4",
repo: repo_name,
directory: directory,
branch: source_branch,
)
##############################
# Fetch the dependency files #
##############################
puts "Fetching #{package_manager} dependency files for #{repo_name}"
fetcher = Dependabot::FileFetchers.for_package_manager(package_manager).new(
source: source,
credentials: credentials,
)
files = fetcher.files
commit = fetcher.commit
##############################
# Parse the dependency files #
##############################
puts "Parsing dependencies information"
parser = Dependabot::FileParsers.for_package_manager(package_manager).new(
dependency_files: files,
source: source,
credentials: credentials,
)
dependencies = parser.parse
opened_merge_requests = 0
dependencies.select(&:top_level?).each do |dep|
if ENV["DEPENDABOT_MAX_MERGE_REQUESTS"] && opened_merge_requests >= ENV["DEPENDABOT_MAX_MERGE_REQUESTS"].to_i
puts "Opened merge request limit reached!"
break
end
#########################################
# Get update details for the dependency #
#########################################
checker = Dependabot::UpdateCheckers.for_package_manager(package_manager).new(
dependency: dep,
dependency_files: files,
credentials: credentials,
requirements_update_strategy: update_strategy
)
next if checker.up_to_date?
requirements_to_unlock =
if !checker.requirements_unlocked_or_can_be?
if checker.can_update?(requirements_to_unlock: :none) then :none
else :update_not_possible
end
elsif checker.can_update?(requirements_to_unlock: :own) then :own
elsif checker.can_update?(requirements_to_unlock: :all) then :all
else :update_not_possible
end
next if requirements_to_unlock == :update_not_possible
updated_deps = checker.updated_dependencies(
requirements_to_unlock: requirements_to_unlock
)
#####################################
# Generate updated dependency files #
#####################################
print " - Updating #{dep.name} (from #{dep.version})…"
updater = Dependabot::FileUpdaters.for_package_manager(package_manager).new(
dependencies: updated_deps,
dependency_files: files,
credentials: credentials,
)
updated_files = updater.updated_dependency_files
#####################################
# Find out if a MR already exists #
#####################################
gitlab_client = Dependabot::Clients::GitlabWithRetries.for_source(
source: source,
credentials: credentials
)
opened_merge_requests_for_this_dep = gitlab_client.merge_requests(
repo_name,
state: "opened",
search: dep.name,
in: "title",
)
conflict_merge_request_commit_id = nil
conflict_merge_request_id = nil
opened_merge_requests_for_this_dep.each do |omr|
title = omr.title
if title.include?(dep.name) && title.include?(dep.version)
if !title.include?(updated_deps[0].version)
# close old version MR
gitlab_client.update_merge_request(repo_name, omr.iid, {state_event: "close"})
gitlab_client.delete_branch(repo_name, omr.source_branch)
puts " closed merge request ##{omr.iid}"
end
if omr.merge_status != "can_be_merged"
# ignore merge request manually touched
next if gitlab_client.merge_request_commits(repo_name, omr.iid).length > 1
# keep merge request
conflict_merge_request_commit_id = omr.sha
conflict_merge_request_id = omr.iid
break
end
end
end
if conflict_merge_request_commit_id && conflict_merge_request_id
########################################
# Update merge request with conflict #
########################################
pr_updater = Dependabot::PullRequestUpdater.new(
source: source,
base_commit: commit,
old_commit: conflict_merge_request_commit_id,
files: updated_files,
credentials: credentials,
pull_request_number: conflict_merge_request_id,
)
pr_updater.update
puts " merge request ##{conflict_merge_request_id} updated"
else
########################################
# Create a pull request for the update #
########################################
pr_creator = Dependabot::PullRequestCreator.new(
source: source,
base_commit: commit,
dependencies: updated_deps,
files: updated_files,
credentials: credentials,
label_language: true,
assignees: assignees
)
pull_request = pr_creator.create
puts " submitted"
end
opened_merge_requests += 1
next unless pull_request
# Auto approve Gitlab merge request with the same user.
if ENV["DEPENDABOT_GITLAB_APPROVE_MERGE"]
g.approve_merge_request(source.repo, pull_request.iid)
puts " approved"
end
# Enable GitLab "merge when pipeline succeeds" feature.
# Merge requests created and successfully tested will be merge automatically.
if ENV["DEPENDABOT_GITLAB_AUTO_MERGE"]
g.accept_merge_request(
source.repo,
pull_request.iid,
merge_when_pipeline_succeeds: true,
should_remove_source_branch: true
)
puts " set to be accepted"
end
end
puts "Done"