From 75a89dd03815f11d782a7298934f75a4a975dae9 Mon Sep 17 00:00:00 2001
From: Pierre-Yves Chibon <pingou@pingoured.fr>
Date: Apr 25 2018 09:06:03 +0000
Subject: Add support for repository templates for sources and forks


Signed-off-by: Pierre-Yves Chibon <pingou@pingoured.fr>

---

diff --git a/doc/configuration.rst b/doc/configuration.rst
index 12654e3..8f8a354 100644
--- a/doc/configuration.rst
+++ b/doc/configuration.rst
@@ -576,6 +576,28 @@ Defaults to: ``['issue_comment', 'issue_create', 'issue_change_status', 'pull_re
 Optional options
 ----------------
 
+Git repository templates
+~~~~~~~~~~~~~~~~~~~~~~~~
+
+PROJECT_TEMPLATE_PATH
+^^^^^^^^^^^^^^^^^^^^^
+
+This configuration key allows you to specify the path to a git repository
+to use as a template when creating new repository for new projects.
+This template will not be used for forks nor any of the git repository but
+the one used for the sources (ie: it will not be used for the tickets,
+requests or docs repositories).
+
+FORK_TEMPLATE_PATH
+^^^^^^^^^^^^^^^^^^
+
+This configuration key allows you to specify the path to a git repository
+to use as a template when creating new repository for new forks.
+This template will not be used for any of the git repository but
+the one used for the sources of forks (ie: it will not be used for the
+tickets, requests or docs repositories).
+
+
 SSH_KEYS
 ~~~~~~~~
 
@@ -595,6 +617,7 @@ Where `<foo>` and `<bar>` must be replaced by your values.
 
 ITEM_PER_PAGE
 ~~~~~~~~~~~~~
+
 This configuration key allows you to configure the length of a page by
 setting the number of items on the page. Items can be commits, users, groups,
 or projects for example.
diff --git a/pagure/lib/tasks.py b/pagure/lib/tasks.py
index 3e2d6fc..727e80b 100644
--- a/pagure/lib/tasks.py
+++ b/pagure/lib/tasks.py
@@ -269,13 +269,30 @@ def create_project(self, session, username, namespace, name, add_readme,
         gitrepo = os.path.join(pagure_config['GIT_FOLDER'], project.path)
 
         # Add the readme file if it was asked
-        if not add_readme:
-            _log.debug('Create git repo at: %s', gitrepo)
-            pygit2.init_repository(gitrepo, bare=True)
+        _log.debug('Create git repo at: %s', gitrepo)
+        templ = None
+        if project.is_fork:
+            templ = pagure_config.get('FORK_TEMPLATE_PATH')
         else:
+            templ = pagure_config.get('PROJECT_TEMPLATE_PATH')
+        if templ:
+            if not os.path.exists(templ):
+                _log.warning(
+                    'Invalid git template configured: %s, not found on disk',
+                    templ)
+                templ = None
+            else:
+                _log.debug('  Using template at: %s', templ)
+
+        pygit2.init_repository(
+            gitrepo, bare=True, template_path=templ)
+        if add_readme:
+            # Clone main project
             temp_gitrepo_path = tempfile.mkdtemp(prefix='pagure-')
-            temp_gitrepo = pygit2.init_repository(temp_gitrepo_path,
-                                                  bare=False)
+            temp_gitrepo = pygit2.clone_repository(
+                gitrepo, temp_gitrepo_path, bare=False)
+
+            # Add README file
             author = userobj.fullname or userobj.user
             author_email = userobj.default_email
             if six.PY2:
@@ -291,7 +308,15 @@ def create_project(self, session, username, namespace, name, add_readme,
             tree = temp_gitrepo.index.write_tree()
             temp_gitrepo.create_commit(
                 'HEAD', author, author, 'Added the README', tree, [])
-            pygit2.clone_repository(temp_gitrepo_path, gitrepo, bare=True)
+
+            # Push the README back to the main project
+            ori_remote = temp_gitrepo.remotes[0]
+            master_ref = temp_gitrepo.lookup_reference('HEAD').resolve()
+            refname = '%s:%s' % (master_ref.name, master_ref.name)
+
+            _log.info('Pushing to %s: %s', ori_remote.name, refname)
+            pagure.lib.repo.PagureRepo.push(ori_remote, refname)
+
             shutil.rmtree(temp_gitrepo_path)
 
         # Make the repo exportable via apache