# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import logging

import formencode

import traceback

from formencode import htmlfill

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, \

    NotAnonymous, HasRepoPermissionAny

from rhodecode.lib.base import BaseRepoController, render

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__)

class ForksController(BaseRepoController):

    @LoginRequired()

    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)\

    @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},

        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') %

        allow_extra_fields = True

        filter_extra_fields = False

        repo_name = All(v.UnicodeString(strip=True, min=1, not_empty=True),

                        v.SlugifyName())

        clone_uri = All(v.UnicodeString(strip=True, min=1, not_empty=False))

        repo_group = v.OneOf(repo_groups, hideList=True)

        repo_type = v.OneOf(supported_backends)

        description = v.UnicodeString(strip=True, min=1, not_empty=False)

        private = v.StringBoolean(if_missing=False)

        enable_statistics = v.StringBoolean(if_missing=False)

        enable_downloads = v.StringBoolean(if_missing=False)

        landing_rev = v.OneOf(landing_revs, hideList=True)

        if edit:

            #this is repo owner

            user = All(v.UnicodeString(not_empty=True), v.ValidRepoUser())

        chained_validators = [v.ValidCloneUri(),

                              v.ValidRepoName(edit, old_data),

                              v.ValidPerms()]

    return _RepoForm

def RepoForkForm(edit=False, old_data={}, supported_backends=BACKENDS.keys(),

    class _RepoForkForm(formencode.Schema):

        allow_extra_fields = True

        filter_extra_fields = False

        repo_name = All(v.UnicodeString(strip=True, min=1, not_empty=True),

                        v.SlugifyName())

        repo_group = v.OneOf(repo_groups, hideList=True)

        repo_type = All(v.ValidForkType(old_data), v.OneOf(supported_backends))

        description = v.UnicodeString(strip=True, min=1, not_empty=True)

        private = v.StringBoolean(if_missing=False)

        copy_permissions = v.StringBoolean(if_missing=False)

        update_after_clone = v.StringBoolean(if_missing=False)

        fork_parent_id = v.UnicodeString()

        chained_validators = [v.ValidForkName(edit, old_data)]

    return _RepoForkForm

def RepoSettingsForm(edit=False, old_data={},

                     supported_backends=BACKENDS.keys(), repo_groups=[],

                     landing_revs=[]):

    class _RepoForm(formencode.Schema):

        allow_extra_fields = True

        filter_extra_fields = False

        repo_name = All(v.UnicodeString(strip=True, min=1, not_empty=True),

                        v.SlugifyName())

        description = v.UnicodeString(strip=True, min=1, not_empty=True)

        repo_group = v.OneOf(repo_groups, hideList=True)

        private = v.StringBoolean(if_missing=False)

        landing_rev = v.OneOf(landing_revs, hideList=True)

        chained_validators = [v.ValidRepoName(edit, old_data), v.ValidPerms(),

                              v.ValidSettings()]

    return _RepoForm

def ApplicationSettingsForm():

    class _ApplicationSettingsForm(formencode.Schema):

        allow_extra_fields = True

                <label for="repo_name">${_('Name')}:</label>

            </div>

            <div class="input">

                ${h.text('repo_name',c.new_repo,class_="small")}

                %if not h.HasPermissionAll('hg.admin')('repo create form'):

                    ${h.hidden('user_created',True)}

                %endif

            </div>

         </div>

        <div class="field">

            <div class="label">

                <label for="clone_uri">${_('Clone from')}:</label>

            </div>

            <div class="input">

                ${h.text('clone_uri',class_="small")}

                <span class="help-block">${_('Optional http[s] url from which repository should be cloned.')}</span>

            </div>

         </div>

         <div class="field">

             <div class="label">

                 <label for="repo_group">${_('Repository group')}:</label>

             </div>

             <div class="input">

                 ${h.select('repo_group',request.GET.get('parent_group'),c.repo_groups,class_="medium")}

             </div>

         </div>

        <div class="field">

            <div class="label">

                <label for="repo_type">${_('Type')}:</label>

            </div>

            <div class="input">

                ${h.select('repo_type','hg',c.backends,class_="small")}

                <span class="help-block">${_('Type of repository to create.')}</span>

            </div>

         </div>

         <div class="field">

            <div class="label">

                <label for="landing_rev">${_('Landing revision')}:</label>

            </div>

            <div class="input">

                ${h.select('landing_rev','',c.landing_revs,class_="medium")}

                <span class="help-block">${_('Default revision for files page, downloads, whoosh and readme')}</span>

            </div>

        </div>

        <div class="field">

            <div class="label label-textarea">

                <label for="description">${_('Description')}:</label>

            </div>

<%def name="page_nav()">

	${self.menu('')}

</%def>

<%def name="main()">

<div class="box">

    <!-- box / title -->

    <div class="title">

        ${self.breadcrumbs()}

    </div>

    ${h.form(url('repo_fork_create_home',repo_name=c.repo_info.repo_name))}

    <div class="form">

        <!-- fields -->

        <div class="fields">

            <div class="field">

              <div class="label">

                  <label for="repo_name">${_('Fork name')}:</label>

              </div>

              <div class="input">

                  ${h.text('repo_name',class_="small")}

                  ${h.hidden('repo_type',c.repo_info.repo_type)}

                  ${h.hidden('fork_parent_id',c.repo_info.repo_id)}

              </div>

            </div>

            <div class="field">

                 <div class="label">

                     <label for="repo_group">${_('Repository group')}:</label>

                 </div>

                 <div class="input">

                     ${h.select('repo_group','',c.repo_groups,class_="medium")}

                 </div>

            </div>

            <div class="field">

                <div class="label label-textarea">

                    <label for="description">${_('Description')}:</label>

                </div>

                <div class="textarea text-area editor">

                    ${h.textarea('description',cols=23,rows=5)}

                </div>

             </div>

            <div class="field">

                <div class="label label-checkbox">

                    <label for="private">${_('Private')}:</label>

                </div>

                <div class="checkboxes">

                    ${h.checkbox('private',value="True")}

                </div>

            </div>

            <div class="field">

                <div class="label label-checkbox">

                    <label for="private">${_('Copy permissions')}:</label>

                </div>

                <div class="checkboxes">

                    ${h.checkbox('copy_permissions',value="True", checked="checked")}

                </div>

             </div>

            <div class="field">

                <div class="label label-checkbox">

                    <label for="private">${_('Update after clone')}:</label>

                </div>

                <div class="checkboxes">

                    ${h.checkbox('update_after_clone',value="True")}

                </div>

             </div>

	        <div class="buttons">

	          ${h.submit('',_('fork this repository'),class_="ui-button")}

	        </div>

        </div>

    </div>

    ${h.end_form()}

</div>

</%def>