Fix diffs with edit-forking needs

Fix https://gitlab.com/gitlab-org/gitlab-ce/issues/31276

app/assets/javascripts/merge_request_tabs.js

@@ -5,6 +5,7 @@
 import Cookies from 'js-cookie';
 import './breakpoints';
 import './flash';
+import BlobForkSuggestion from './blob/blob_fork_suggestion';
 /* eslint-disable max-len */
 // MergeRequestTabs
@@ -266,6 +267,14 @@ import './flash';
           new gl.Diff();
           this.scrollToElement('#diffs');
+
+          new BlobForkSuggestion({
+            openButtons: document.querySelectorAll('.js-edit-blob-link-fork-toggler'),
+            forkButtons: document.querySelectorAll('.js-fork-suggestion-button'),
+            cancelButtons: document.querySelectorAll('.js-cancel-fork-suggestion-button'),
+            suggestionSections: document.querySelectorAll('.js-file-fork-suggestion-section'),
+            actionTextPieces: document.querySelectorAll('.js-file-fork-suggestion-section-action'),
+          });
         },
       });
     }

app/helpers/blob_helper.rb

@@ -29,7 +29,7 @@ module BlobHelper
       link_to 'Edit', edit_path(project, ref, path, options), class: "#{common_classes} btn-sm"
     elsif current_user && can?(current_user, :fork_project, project)
       continue_params = {
-        to: edit_path,
+        to: edit_path(project, ref, path, options),
         notice: edit_in_new_fork_notice,
         notice_now: edit_in_new_fork_notice_now
       }

app/views/projects/_fork_suggestion.html.haml

+- if current_user
+  .js-file-fork-suggestion-section.file-fork-suggestion.hidden
+    %span.file-fork-suggestion-note
+      You're not allowed to
+      %span.js-file-fork-suggestion-section-action
+        edit
+      files in this project directly. Please fork this project,
+      make your changes there, and submit a merge request.
+    = link_to 'Fork', nil, method: :post, class: 'js-fork-suggestion-button btn btn-grouped btn-inverted btn-new'
+    %button.js-cancel-fork-suggestion-button.btn.btn-grouped{ type: 'button' }
+      Cancel

app/views/projects/blob/_header.html.haml

@@ -39,14 +39,4 @@
         = replace_blob_link
         = delete_blob_link
-- if current_user
-  .js-file-fork-suggestion-section.file-fork-suggestion.hidden
-    %span.file-fork-suggestion-note
-      You're not allowed to
-      %span.js-file-fork-suggestion-section-action
-        edit
-      files in this project directly. Please fork this project,
-      make your changes there, and submit a merge request.
-    = link_to 'Fork', nil, method: :post, class: 'js-fork-suggestion-button btn btn-grouped btn-inverted btn-new'
-    %button.js-cancel-fork-suggestion-button.btn.btn-grouped{ type: 'button' }
-      Cancel
+= render 'projects/fork_suggestion'

app/views/projects/diffs/_file.html.haml

@@ -18,4 +18,6 @@
         = view_file_button(diff_commit.id, diff_file.new_path, project)
         = view_on_environment_button(diff_commit.id, diff_file.new_path, environment) if environment
+  = render 'projects/fork_suggestion'
+
   = render 'projects/diffs/content', diff_file: diff_file, diff_commit: diff_commit, blob: blob, project: project

spec/features/merge_requests/diffs_spec.rb

@@ -36,4 +36,9 @@ feature 'Diffs URL', js: true, feature: true do
       end
     end
   end
+
+
+  context 'when editing file' do
+
+  end
 end