From 43756ff9f27eb37a85a69212bf16fb0eb8238c8e Mon Sep 17 00:00:00 2001
From: Farhaan Bukhsh <farhaan.bukhsh@gmail.com>
Date: Jul 28 2016 15:31:16 +0000
Subject: Add functionality to Fork and edit a file


The function added forks the project and then redirect the url to edit
the file specified. A form is added to the file UI which has a button
saying Fork and Edit.

---

diff --git a/pagure/templates/file.html b/pagure/templates/file.html
index 9478de5..5f2edca 100644
--- a/pagure/templates/file.html
+++ b/pagure/templates/file.html
@@ -98,6 +98,15 @@
                     filename=filename | unicode) }}" title="Edit file">Edit</a>
                 {% endif %}
                 {% if output_type == 'markup'  %}
+              <form class="btn btn-sm" method="POST" name="fork_project"
+                action="{{ url_for('.fork_edit_file', repo=repo.name,
+                    username=username, branchname=branchname, filename=filename) }}">
+                  <button class="btn btn-sm btn-secondary"
+                    onclick="fork_project.submit();">
+                        Fork and Edit
+                 </button>
+                {{ forkbuttonform.csrf_token }}
+                </form>
                 <a class="btn btn-sm btn-secondary" href="{{ url_for(
                     'view_file', username=username,
                     repo=repo.name, identifier=branchname,
diff --git a/pagure/ui/fork.py b/pagure/ui/fork.py
index 802cfc1..6a6f108 100644
--- a/pagure/ui/fork.py
+++ b/pagure/ui/fork.py
@@ -1161,3 +1161,62 @@ def new_remote_request_pull(repo, username=None):
         repo_admin=repo_admin,
         branch_to=orig_repo.head.shorthand,
     )
+
+
+@APP.route(
+    '/fork_edit/<repo:repo>/edit/<path:branchname>/f/<path:filename>',
+    methods=('GET', 'POST'))
+@APP.route(
+    '/fork_edit/fork/<username>/<repo:repo>/edit/<path:branchname>/f/<path:filename>',
+    methods=('GET', 'POST'))
+@login_required
+def fork_edit_file(repo, branchname, filename, username=None):
+    """ Fork the project specified and open the specific file to edit
+    """
+    repo = pagure.lib.get_project(SESSION, repo, user=username)
+
+    form = pagure.forms.ConfirmationForm()
+    if not form.validate_on_submit():
+        flask.abort(400)
+
+    if repo is None:
+        flask.abort(404)
+
+    if pagure.lib.get_project(
+            SESSION, repo.name, user=flask.g.fas_user.username):
+        flask.flash('You had already forked this project')
+        return flask.redirect(flask.url_for(
+            'edit_file',
+            username=flask.g.fas_user.username,
+            repo=repo.name,
+            branchname=branchname,
+            filename=filename
+        ))
+
+    try:
+        message = pagure.lib.fork_project(
+            session=SESSION,
+            repo=repo,
+            gitfolder=APP.config['GIT_FOLDER'],
+            docfolder=APP.config['DOCS_FOLDER'],
+            ticketfolder=APP.config['TICKETS_FOLDER'],
+            requestfolder=APP.config['REQUESTS_FOLDER'],
+            user=flask.g.fas_user.username)
+
+        SESSION.commit()
+        pagure.lib.git.generate_gitolite_acls()
+        flask.flash(message)
+        return flask.redirect(flask.url_for(
+            'edit_file',
+            username=flask.g.fas_user.username,
+            repo=repo.name,
+            branchname=branchname,
+            filename=filename
+        ))
+    except pagure.exceptions.PagureException as err:
+        flask.flash(str(err), 'error')
+    except SQLAlchemyError as err:  # pragma: no cover
+        SESSION.rollback()
+        flask.flash(str(err), 'error')
+
+    return flask.redirect(flask.url_for('view_repo', repo=repo.name))