extracts_path.rb 4.48 KB
Newer Older
1 2 3 4 5 6
# Module providing methods for dealing with separating a tree-ish string and a
# file path string when combined in a request parameter
module ExtractsPath
  extend ActiveSupport::Concern

  # Raised when given an invalid file path
7 8
  class InvalidPathError < StandardError; end

9 10
  included do
    if respond_to?(:before_filter)
11
      before_filter :assign_ref_vars, only: [:show]
12 13 14 15 16
    end
  end

  # Given a string containing both a Git tree-ish, such as a branch or tag, and
  # a filesystem path joined by forward slashes, attempts to separate the two.
17
  #
18 19
  # Expects a @project instance variable to contain the active project. This is
  # used to check the input against a list of valid repository refs.
20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35
  #
  # Examples
  #
  #   # No @project available
  #   extract_ref('master')
  #   # => ['', '']
  #
  #   extract_ref('master')
  #   # => ['master', '']
  #
  #   extract_ref("f4b14494ef6abf3d144c28e4af0c20143383e062/CHANGELOG")
  #   # => ['f4b14494ef6abf3d144c28e4af0c20143383e062', 'CHANGELOG']
  #
  #   extract_ref("v2.0.0/README.md")
  #   # => ['v2.0.0', 'README.md']
  #
36 37 38
  #   extract_ref('/gitlab/vagrant/tree/master/app/models/project.rb')
  #   # => ['master', 'app/models/project.rb']
  #
39 40 41 42 43 44 45 46 47
  #   extract_ref('issues/1234/app/models/project.rb')
  #   # => ['issues/1234', 'app/models/project.rb']
  #
  #   # Given an invalid branch, we fall back to just splitting on the first slash
  #   extract_ref('non/existent/branch/README.md')
  #   # => ['non', 'existent/branch/README.md']
  #
  # Returns an Array where the first value is the tree-ish and the second is the
  # path
48 49 50 51 52
  def extract_ref(input)
    pair = ['', '']

    return pair unless @project

53 54
    # Remove relative_url_root from path
    input.gsub!(/^#{Gitlab.config.gitlab.relative_url_root}/, "")
55
    # Remove project, actions and all other staff from path
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
56
    input.gsub!(/^\/#{Regexp.escape(@project.path_with_namespace)}/, "")
57
    input.gsub!(/^\/(tree|commits|blame|blob|refs|graph)\//, "") # remove actions
58 59
    input.gsub!(/\?.*$/, "") # remove stamps suffix
    input.gsub!(/.atom$/, "") # remove rss feed
60
    input.gsub!(/.json$/, "") # remove json suffix
61 62
    input.gsub!(/\/edit$/, "") # remove edit route part

63 64 65 66
    if input.match(/^([[:alnum:]]{40})(.+)/)
      # If the ref appears to be a SHA, we're done, just split the string
      pair = $~.captures
    else
67 68 69
      # Otherwise, attempt to detect the ref using a list of the project's
      # branches and tags

70 71
      # Append a trailing slash if we only get a ref and no file path
      id = input
72
      id += '/' unless id.ends_with?('/')
73

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
74
      valid_refs = @project.repository.ref_names
75 76
      valid_refs.select! { |v| id.start_with?("#{v}/") }

77 78
      if valid_refs.length != 1
        # No exact ref match, so just try our best
79
        pair = id.match(/([^\/]+)(.*)/).captures
80 81
      else
        # Partition the string into the ref and the path, ignoring the empty first value
82
        pair = id.partition(valid_refs.first)[1..-1]
83 84 85
      end
    end

86 87
    # Remove ending slashes from path
    pair[1].gsub!(/^\/|\/$/, '')
88

89 90
    pair
  end
91 92 93 94 95 96 97 98 99 100 101

  # Assigns common instance variables for views working with Git tree-ish objects
  #
  # Assignments are:
  #
  # - @id     - A string representing the joined ref and path
  # - @ref    - A string representing the ref (e.g., the branch, tag, or commit SHA)
  # - @path   - A string representing the filesystem path
  # - @commit - A CommitDecorator representing the commit from the given ref
  # - @tree   - A TreeDecorator representing the tree at the given ref/path
  #
102 103 104 105 106
  # If the :id parameter appears to be requesting a specific response format,
  # that will be handled as well.
  #
  # Automatically renders `not_found!` if a valid tree path could not be
  # resolved (e.g., when a user inserts an invalid path or ref).
107
  def assign_ref_vars
108 109 110 111 112 113
    # Handle formats embedded in the id
    if params[:id].ends_with?('.atom')
      params[:id].gsub!(/\.atom$/, '')
      request.format = :atom
    end

114
    path = CGI::unescape(request.fullpath.dup)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
115 116

    @ref, @path = extract_ref(path)
117 118 119

    @id = File.join(@ref, @path)

120 121 122 123
    # It is used "@project.repository.commits(@ref, @path, 1, 0)",
    # because "@project.repository.commit(@ref)" returns wrong commit when @ref is tag name.
    commits = @project.repository.commits(@ref, @path, 1, 0)
    @commit = CommitDecorator.decorate(commits.first)
124

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
125
    @tree = Tree.new(@commit.tree, @ref, @path)
126 127 128 129 130 131
    @tree = TreeDecorator.new(@tree)

    raise InvalidPathError if @tree.invalid?
  rescue NoMethodError, InvalidPathError
    not_found!
  end
132
end