diff --git a/rhodecode/controllers/forks.py b/rhodecode/controllers/forks.py --- a/rhodecode/controllers/forks.py +++ b/rhodecode/controllers/forks.py @@ -23,13 +23,23 @@ # You should have received a copy of the GNU General Public License # along with this program. If not, see . import logging +import formencode +import traceback +from formencode import htmlfill -from pylons import tmpl_context as c, request +from pylons import tmpl_context as c, request, url +from pylons.controllers.util import redirect +from pylons.i18n.translation import _ + +import rhodecode.lib.helpers as h from rhodecode.lib.helpers import Page -from rhodecode.lib.auth import LoginRequired, HasRepoPermissionAnyDecorator +from rhodecode.lib.auth import LoginRequired, HasRepoPermissionAnyDecorator, \ + NotAnonymous from rhodecode.lib.base import BaseRepoController, render -from rhodecode.model.db import Repository, User, UserFollowing +from rhodecode.model.db import Repository, RepoGroup, UserFollowing, User +from rhodecode.model.repo import RepoModel +from rhodecode.model.forms import RepoForkForm log = logging.getLogger(__name__) @@ -37,11 +47,59 @@ log = logging.getLogger(__name__) class ForksController(BaseRepoController): @LoginRequired() - @HasRepoPermissionAnyDecorator('repository.read', 'repository.write', - 'repository.admin') def __before__(self): super(ForksController, self).__before__() + def __load_defaults(self): + c.repo_groups = RepoGroup.groups_choices() + c.repo_groups_choices = map(lambda k: unicode(k[0]), c.repo_groups) + + def __load_data(self, repo_name=None): + """ + Load defaults settings for edit, and update + + :param repo_name: + """ + self.__load_defaults() + + c.repo_info = db_repo = Repository.get_by_repo_name(repo_name) + repo = db_repo.scm_instance + + if c.repo_info is None: + h.flash(_('%s repository is not mapped to db perhaps' + ' it was created or renamed from the filesystem' + ' please run the application again' + ' in order to rescan repositories') % repo_name, + category='error') + + return redirect(url('repos')) + + c.default_user_id = User.get_by_username('default').user_id + c.in_public_journal = UserFollowing.query()\ + .filter(UserFollowing.user_id == c.default_user_id)\ + .filter(UserFollowing.follows_repository == c.repo_info).scalar() + + if c.repo_info.stats: + last_rev = c.repo_info.stats.stat_on_revision + else: + last_rev = 0 + c.stats_revision = last_rev + + c.repo_last_rev = repo.count() - 1 if repo.revisions else 0 + + if last_rev == 0 or c.repo_last_rev == 0: + c.stats_percentage = 0 + else: + c.stats_percentage = '%.2f' % ((float((last_rev)) / + c.repo_last_rev) * 100) + + defaults = RepoModel()._get_defaults(repo_name) + # add prefix to fork + defaults['repo_name'] = 'fork-' + defaults['repo_name'] + return defaults + + @HasRepoPermissionAnyDecorator('repository.read', 'repository.write', + 'repository.admin') def forks(self, repo_name): p = int(request.params.get('page', 1)) repo_id = c.rhodecode_db_repo.repo_id @@ -54,3 +112,63 @@ class ForksController(BaseRepoController return c.forks_data return render('/forks/forks.html') + + @NotAnonymous() + @HasRepoPermissionAnyDecorator('repository.read', 'repository.write', + 'repository.admin') + def fork(self, repo_name): + c.repo_info = Repository.get_by_repo_name(repo_name) + if not c.repo_info: + h.flash(_('%s repository is not mapped to db perhaps' + ' it was created or renamed from the file system' + ' please run the application again' + ' in order to rescan repositories') % repo_name, + category='error') + + return redirect(url('home')) + + defaults = self.__load_data(repo_name) + + return htmlfill.render( + render('forks/fork.html'), + defaults=defaults, + encoding="UTF-8", + force_defaults=False + ) + + + @NotAnonymous() + @HasRepoPermissionAnyDecorator('repository.read', 'repository.write', + 'repository.admin') + def fork_create(self, repo_name): + self.__load_defaults() + c.repo_info = Repository.get_by_repo_name(repo_name) + _form = RepoForkForm(old_data={'repo_type': c.repo_info.repo_type}, + repo_groups=c.repo_groups_choices,)() + form_result = {} + try: + form_result = _form.to_python(dict(request.POST)) + # add org_path of repo so we can do a clone from it later + form_result['org_path'] = c.repo_info.repo_name + + # create fork is done sometimes async on celery, db transaction + # management is handled there. + RepoModel().create_fork(form_result, self.rhodecode_user) + h.flash(_('forked %s repository as %s') \ + % (repo_name, form_result['repo_name']), + category='success') + except formencode.Invalid, errors: + c.new_repo = errors.value['repo_name'] + + return htmlfill.render( + render('forks/fork.html'), + defaults=errors.value, + errors=errors.error_dict or {}, + prefix_error=False, + encoding="UTF-8") + except Exception: + log.error(traceback.format_exc()) + h.flash(_('An error occurred during repository forking %s') % + repo_name, category='error') + + return redirect(url('home'))