gitlab-org--gitlab-foss/lib/gitlab/workhorse.rb

92 lines
2.1 KiB
Ruby
Raw Normal View History

2016-02-01 05:33:22 -05:00
require 'base64'
require 'json'
module Gitlab
class Workhorse
2016-02-11 12:10:14 -05:00
SEND_DATA_HEADER = 'Gitlab-Workhorse-Send-Data'
VERSION_FILE = 'GITLAB_WORKHORSE_VERSION'
2016-02-02 08:09:55 -05:00
2016-02-11 12:10:14 -05:00
class << self
2016-04-06 11:52:12 -04:00
def git_http_ok(repository, user)
{
2016-06-15 08:59:37 -04:00
'GL_ID' => Gitlab::GlId.gl_id(user),
2016-04-06 11:52:12 -04:00
'RepoPath' => repository.path_to_repo,
}
end
2016-02-01 05:33:22 -05:00
def send_git_blob(repository, blob)
2016-02-02 08:09:55 -05:00
params = {
2016-02-01 05:33:22 -05:00
'RepoPath' => repository.path_to_repo,
'BlobId' => blob.id,
}
[
2016-02-02 08:09:55 -05:00
SEND_DATA_HEADER,
2016-06-08 08:30:15 -04:00
"git-blob:#{encode(params)}"
2016-02-01 05:33:22 -05:00
]
end
2016-02-02 08:09:55 -05:00
def send_git_archive(repository, ref:, format:)
2016-02-02 08:09:55 -05:00
format ||= 'tar.gz'
format.downcase!
params = repository.archive_metadata(ref, Gitlab.config.gitlab.repository_downloads_path, format)
2016-02-02 08:09:55 -05:00
raise "Repository or ref not found" if params.empty?
[
SEND_DATA_HEADER,
2016-06-08 08:30:15 -04:00
"git-archive:#{encode(params)}"
2016-02-02 08:09:55 -05:00
]
end
2016-05-12 14:50:49 -04:00
2016-06-08 08:30:15 -04:00
def send_git_diff(repository, diff_refs)
2016-05-12 14:50:49 -04:00
params = {
2016-06-08 08:30:15 -04:00
'RepoPath' => repository.path_to_repo,
'ShaFrom' => diff_refs.start_sha,
'ShaTo' => diff_refs.head_sha
2016-05-12 14:50:49 -04:00
}
[
SEND_DATA_HEADER,
"git-diff:#{encode(params)}"
2016-02-02 08:09:55 -05:00
]
end
2016-07-03 17:01:13 -04:00
def send_git_patch(repository, diff_refs)
params = {
2016-06-28 08:59:25 -04:00
'RepoPath' => repository.path_to_repo,
2016-07-03 17:01:13 -04:00
'ShaFrom' => diff_refs.start_sha,
'ShaTo' => diff_refs.head_sha
}
[
2016-06-28 08:59:25 -04:00
SEND_DATA_HEADER,
"git-format-patch:#{encode(params)}"
]
end
def send_artifacts_entry(build, entry)
params = {
'Archive' => build.artifacts_file.path,
'Entry' => Base64.encode64(entry.path)
}
[
SEND_DATA_HEADER,
"artifacts-entry:#{encode(params)}"
]
end
def version
path = Rails.root.join(VERSION_FILE)
path.readable? ? path.read.chomp : 'unknown'
end
2016-02-02 08:09:55 -05:00
protected
2016-02-02 08:09:55 -05:00
def encode(hash)
Base64.urlsafe_encode64(JSON.dump(hash))
end
2016-02-01 05:33:22 -05:00
end
end
2016-02-01 06:27:35 -05:00
end