debian-mirror-gitlab/lib/gitlab/git/tree.rb

104 lines
2.5 KiB
Ruby
Raw Normal View History

2019-02-15 15:39:39 +05:30
# frozen_string_literal: true
2017-08-17 22:00:37 +05:30
module Gitlab
module Git
class Tree
2017-09-10 17:25:29 +05:30
include Gitlab::EncodingHelper
2018-12-13 13:39:08 +05:30
extend Gitlab::Git::WrapsGitalyErrors
2017-08-17 22:00:37 +05:30
2020-10-24 23:57:45 +05:30
attr_accessor :id, :root_id, :type, :mode, :commit_id, :submodule_url
attr_writer :name, :path, :flat_path
2017-08-17 22:00:37 +05:30
class << self
# Get list of tree objects
# for repository based on commit sha and path
# Uses rugged for raw objects
2017-09-10 17:25:29 +05:30
#
# Gitaly migration: https://gitlab.com/gitlab-org/gitaly/issues/320
2018-03-27 19:54:05 +05:30
def where(repository, sha, path = nil, recursive = false)
2017-08-17 22:00:37 +05:30
path = nil if path == '' || path == '/'
2019-05-03 19:53:19 +05:30
tree_entries(repository, sha, path, recursive)
end
def tree_entries(repository, sha, path, recursive)
2018-12-13 13:39:08 +05:30
wrapped_gitaly_errors do
2018-11-08 19:23:39 +05:30
repository.gitaly_commit_client.tree_entries(repository, sha, path, recursive)
2017-08-17 22:00:37 +05:30
end
end
2017-09-10 17:25:29 +05:30
private
2017-08-17 22:00:37 +05:30
# Recursive search of tree id for path
#
# Ex.
# blog/ # oid: 1a
# app/ # oid: 2a
# models/ # oid: 3a
# views/ # oid: 4a
#
#
# Tree.find_id_by_path(repo, '1a', 'app/models') # => '3a'
#
def find_id_by_path(repository, root_id, path)
root_tree = repository.lookup(root_id)
path_arr = path.split('/')
entry = root_tree.find do |entry|
entry[:name] == path_arr[0] && entry[:type] == :tree
end
2019-07-07 11:18:12 +05:30
return unless entry
2017-08-17 22:00:37 +05:30
if path_arr.size > 1
path_arr.shift
find_id_by_path(repository, entry[:oid], path_arr.join('/'))
else
entry[:oid]
end
end
end
def initialize(options)
2018-03-17 18:26:18 +05:30
%w(id root_id name path flat_path type mode commit_id).each do |key|
self.send("#{key}=", options[key.to_sym]) # rubocop:disable GitlabSecurity/PublicSend
2017-08-17 22:00:37 +05:30
end
end
def name
encode! @name
end
2017-09-10 17:25:29 +05:30
def path
encode! @path
end
2018-03-17 18:26:18 +05:30
def flat_path
encode! @flat_path
end
2017-08-17 22:00:37 +05:30
def dir?
type == :tree
end
def file?
type == :blob
end
def submodule?
type == :commit
end
def readme?
name =~ /^readme/i
end
def contributing?
name =~ /^contributing/i
end
end
end
end
2019-05-03 19:53:19 +05:30
Gitlab::Git::Tree.singleton_class.prepend Gitlab::Git::RuggedImpl::Tree::ClassMethods