116 lines
3.2 KiB
Ruby
116 lines
3.2 KiB
Ruby
module API
|
|
module Helpers
|
|
module InternalHelpers
|
|
SSH_GITALY_FEATURES = {
|
|
'git-receive-pack' => [:ssh_receive_pack, Gitlab::GitalyClient::MigrationStatus::OPT_IN],
|
|
'git-upload-pack' => [:ssh_upload_pack, Gitlab::GitalyClient::MigrationStatus::OPT_OUT]
|
|
}.freeze
|
|
|
|
def wiki?
|
|
set_project unless defined?(@wiki)
|
|
@wiki
|
|
end
|
|
|
|
def project
|
|
set_project unless defined?(@project)
|
|
@project
|
|
end
|
|
|
|
def redirected_path
|
|
@redirected_path
|
|
end
|
|
|
|
def ssh_authentication_abilities
|
|
[
|
|
:read_project,
|
|
:download_code,
|
|
:push_code
|
|
]
|
|
end
|
|
|
|
def parse_env
|
|
return {} if params[:env].blank?
|
|
|
|
JSON.parse(params[:env])
|
|
rescue JSON::ParserError
|
|
{}
|
|
end
|
|
|
|
def fix_git_env_repository_paths(env, repository_path)
|
|
if obj_dir_relative = env['GIT_OBJECT_DIRECTORY_RELATIVE'].presence
|
|
env['GIT_OBJECT_DIRECTORY'] = File.join(repository_path, obj_dir_relative)
|
|
end
|
|
|
|
if alt_obj_dirs_relative = env['GIT_ALTERNATE_OBJECT_DIRECTORIES_RELATIVE'].presence
|
|
env['GIT_ALTERNATE_OBJECT_DIRECTORIES'] = alt_obj_dirs_relative.map { |dir| File.join(repository_path, dir) }
|
|
end
|
|
|
|
env
|
|
end
|
|
|
|
def log_user_activity(actor)
|
|
commands = Gitlab::GitAccess::DOWNLOAD_COMMANDS
|
|
|
|
::Users::ActivityService.new(actor, 'Git SSH').execute if commands.include?(params[:action])
|
|
end
|
|
|
|
def merge_request_urls
|
|
::MergeRequests::GetUrlsService.new(project).execute(params[:changes])
|
|
end
|
|
|
|
def redis_ping
|
|
result = Gitlab::Redis::SharedState.with { |redis| redis.ping }
|
|
|
|
result == 'PONG'
|
|
rescue => e
|
|
Rails.logger.warn("GitLab: An unexpected error occurred in pinging to Redis: #{e}")
|
|
false
|
|
end
|
|
|
|
private
|
|
|
|
def set_project
|
|
if params[:gl_repository]
|
|
@project, @wiki = Gitlab::GlRepository.parse(params[:gl_repository])
|
|
@redirected_path = nil
|
|
else
|
|
@project, @wiki, @redirected_path = Gitlab::RepoPath.parse(params[:project])
|
|
end
|
|
end
|
|
|
|
# Project id to pass between components that don't share/don't have
|
|
# access to the same filesystem mounts
|
|
def gl_repository
|
|
Gitlab::GlRepository.gl_repository(project, wiki?)
|
|
end
|
|
|
|
# Return the repository depending on whether we want the wiki or the
|
|
# regular repository
|
|
def repository
|
|
if wiki?
|
|
project.wiki.repository
|
|
else
|
|
project.repository
|
|
end
|
|
end
|
|
|
|
# Return the repository full path so that gitlab-shell has it when
|
|
# handling ssh commands
|
|
def repository_path
|
|
repository.path_to_repo
|
|
end
|
|
|
|
# Return the Gitaly Address if it is enabled
|
|
def gitaly_payload(action)
|
|
feature, status = SSH_GITALY_FEATURES[action]
|
|
return unless feature && Gitlab::GitalyClient.feature_enabled?(feature, status: status)
|
|
|
|
{
|
|
repository: repository.gitaly_repository,
|
|
address: Gitlab::GitalyClient.address(project.repository_storage),
|
|
token: Gitlab::GitalyClient.token(project.repository_storage)
|
|
}
|
|
end
|
|
end
|
|
end
|
|
end
|