BigW Consortium Gitlab

blob_controller.rb 6.52 KB
Newer Older
1
# Controller for viewing a file's blame
2
class Projects::BlobController < Projects::ApplicationController
3
  include ExtractsPath
4
  include CreatesCommit
5
  include RendersBlob
6
  include ActionView::Helpers::SanitizeHelper
7

8
  # Raised when given an invalid file path
9
  InvalidPathError = Class.new(StandardError)
10

11 12
  prepend_before_action :authenticate_user!, only: [:edit]

13 14
  before_action :require_non_empty_project, except: [:new, :create]
  before_action :authorize_download_code!
15
  before_action :authorize_edit_tree!, only: [:new, :create, :update, :destroy]
16 17 18
  before_action :assign_blob_vars
  before_action :commit, except: [:new, :create]
  before_action :blob, except: [:new, :create]
19
  before_action :require_branch_head, only: [:edit, :update]
20
  before_action :editor_variables, except: [:show, :preview, :diff]
21
  before_action :validate_diff_params, only: :diff
22
  before_action :set_last_commit_sha, only: [:edit, :update]
23 24 25 26

  def new
    commit unless @repository.empty?
  end
27

28
  def create
29
    create_commit(Files::CreateService, success_notice: "The file has been successfully created.",
30
                                        success_path: -> { project_blob_path(@project, File.join(@branch_name, @file_path)) },
Douwe Maan committed
31
                                        failure_view: :new,
32
                                        failure_path: project_new_blob_path(@project, @ref))
33
  end
34

35
  def show
36
    conditionally_expand_blob(@blob)
37

38 39
    respond_to do |format|
      format.html do
40
        show_html
41 42 43
      end

      format.json do
44
        show_json
45 46
      end
    end
47 48
  end

49
  def edit
50
    if can_collaborate_with_project?
51
      blob.load_all_data!
52 53 54
    else
      redirect_to action: 'show'
    end
55 56 57
  end

  def update
58
    @path = params[:file_path] if params[:file_path].present?
59
    create_commit(Files::UpdateService, success_path: -> { after_edit_path },
Douwe Maan committed
60
                                        failure_view: :edit,
61
                                        failure_path: project_blob_path(@project, @id))
62 63 64 65

  rescue Files::UpdateService::FileChangedError
    @conflict = true
    render :edit
66 67 68 69
  end

  def preview
    @content = params[:content]
70
    @blob.load_all_data!
71
    diffy = Diffy::Diff.new(@blob.data, @content, diff: '-U 3', include_diff_info: true)
72 73
    diff_lines = diffy.diff.scan(/.*\n/)[2..-1]
    diff_lines = Gitlab::Diff::Parser.new.parse(diff_lines)
74
    @diff_lines = Gitlab::Diff::Highlight.new(diff_lines, repository: @repository).highlight
75 76 77 78

    render layout: false
  end

79
  def destroy
80
    create_commit(Files::DeleteService, success_notice: "The file has been successfully deleted.",
81
                                        success_path: -> { project_tree_path(@project, @branch_name) },
82
                                        failure_view: :show,
83
                                        failure_path: project_blob_path(@project, @id))
84 85
  end

skv committed
86
  def diff
87 88
    apply_diff_view_cookie!

89 90 91 92 93
    @blob.load_all_data!
    @lines = Gitlab::Highlight.highlight(@blob.path, @blob.data, repository: @repository).lines

    @form = UnfoldForm.new(params)
    @lines = @lines[@form.since - 1..@form.to - 1].map(&:html_safe)
skv committed
94 95 96 97 98 99 100 101 102 103 104 105

    if @form.bottom?
      @match_line = ''
    else
      lines_length = @lines.length - 1
      line = [@form.since, lines_length].join(',')
      @match_line = "@@ -#{line}+#{line} @@"
    end

    render layout: false
  end

106 107 108
  private

  def blob
109
    @blob ||= @repository.blob_at(@commit.id, @path)
110

111 112 113
    if @blob
      @blob
    else
114 115
      if tree = @repository.tree(@commit.id, @path)
        if tree.entries.any?
116
          return redirect_to project_tree_path(@project, File.join(@ref, @path))
117 118 119
        end
      end

120
      return render_404
121
    end
122
  end
123 124 125 126

  def commit
    @commit = @repository.commit(@ref)

127
    return render_404 unless @commit
128 129 130 131 132 133 134
  end

  def assign_blob_vars
    @id = params[:id]
    @ref, @path = extract_ref(@id)

  rescue InvalidPathError
135
    render_404
136 137
  end

138 139
  def after_edit_path
    from_merge_request = MergeRequestsFinder.new(current_user, project_id: @project.id).execute.find_by(iid: params[:from_merge_request_iid])
140
    if from_merge_request && @branch_name == @ref
141
      diffs_project_merge_request_path(from_merge_request.target_project, from_merge_request) +
142 143
        "##{hexdigest(@path)}"
    else
144
      project_blob_path(@project, File.join(@branch_name, @path))
145
    end
146
  end
147

148
  def editor_variables
149
    @branch_name = params[:branch_name]
150 151 152

    @file_path =
      if action_name.to_s == 'create'
153 154 155
        if params[:file].present?
          params[:file_name] = params[:file].original_filename
        end
156
        File.join(@path, params[:file_name])
157 158
      elsif params[:file_path].present?
        params[:file_path]
159 160 161 162
      else
        @path
      end

163 164 165 166 167
    if params[:file].present?
      params[:content] = Base64.encode64(params[:file].read)
      params[:encoding] = 'base64'
    end

168 169 170
    @commit_params = {
      file_path: @file_path,
      commit_message: params[:commit_message],
171
      previous_path: @path,
172
      file_content: params[:content],
173 174
      file_content_encoding: params[:encoding],
      last_commit_sha: params[:last_commit_sha]
175 176
    }
  end
177 178 179 180 181 182

  def validate_diff_params
    if [:since, :to, :offset].any? { |key| params[key].blank? }
      render nothing: true
    end
  end
183 184

  def set_last_commit_sha
185 186
    @last_commit_sha = Gitlab::Git::Commit
      .last_for_path(@repository, @ref, @path).sha
187
  end
188 189 190 191 192 193 194 195 196 197 198 199 200

  def show_html
    environment_params = @repository.branch_exists?(@ref) ? { ref: @ref } : { commit: @commit }
    @environment = EnvironmentsFinder.new(@project, current_user, environment_params).execute.last
    @last_commit = @repository.last_commit_for_path(@commit.id, @blob.path)

    render 'show'
  end

  def show_json
    json = blob_json(@blob)
    return render_404 unless json

Jacob Schatz committed
201 202 203 204
    path_segments = @path.split('/')
    path_segments.pop
    tree_path = path_segments.join('/')

205 206 207 208 209 210 211 212 213 214 215 216 217 218
    render json: json.merge(
      path: blob.path,
      name: blob.name,
      extension: blob.extension,
      size: blob.raw_size,
      mime_type: blob.mime_type,
      binary: blob.raw_binary?,
      simple_viewer: blob.simple_viewer&.class&.partial_name,
      rich_viewer: blob.rich_viewer&.class&.partial_name,
      show_viewer_switcher: !!blob.show_viewer_switcher?,
      render_error: blob.simple_viewer&.render_error || blob.rich_viewer&.render_error,
      raw_path: project_raw_path(project, @id),
      blame_path: project_blame_path(project, @id),
      commits_path: project_commits_path(project, @id),
Jacob Schatz committed
219
      tree_path: project_tree_path(project, File.join(@ref, tree_path)),
220 221 222
      permalink: project_blob_path(project, File.join(@commit.id, @path))
    )
  end
223
end