# Copyright (C) 2010-2024 RhodeCode GmbH
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# 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 Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/


import os
import pytest

from rhodecode.tests import TEST_USER_ADMIN_LOGIN
from rhodecode.tests.vcs_operations import (
    Command, _check_proper_hg_push, _check_proper_git_push, _add_files_and_push)


@pytest.mark.usefixtures(
    "init_pyramid_app",
    "repo_group_repos",
    "disable_anonymous_user",
)
class TestVCSOperationsNewBranchPush(object):

    def test_push_new_branch_hg(self, rcstack, tmpdir, user_util):
        repo = user_util.create_repo(repo_type='hg')
        clone_url = rcstack.repo_clone_url(repo.repo_name)
        Command(os.path.dirname(tmpdir.strpath)).execute(
            'hg clone', clone_url, tmpdir.strpath)

        stdout, stderr = _add_files_and_push(
            'hg', tmpdir.strpath, clone_url=clone_url)
        _check_proper_hg_push(stdout, stderr)

        # start new branch, and push file into it
        Command(tmpdir.strpath).execute(
            'hg branch dev && hg commit -m "starting dev branch"')
        stdout, stderr = _add_files_and_push(
            'hg', tmpdir.strpath, clone_url=clone_url, target_branch='dev',
            new_branch=True)

        _check_proper_hg_push(stdout, stderr)

    def test_push_new_branch_git(self, rcstack, tmpdir, user_util):
        repo = user_util.create_repo(repo_type='git')
        clone_url = rcstack.repo_clone_url(repo.repo_name)
        Command(os.path.dirname(tmpdir.strpath)).execute(
            'git clone', clone_url, tmpdir.strpath)

        stdout, stderr = _add_files_and_push(
            'git', tmpdir.strpath, clone_url=clone_url)
        _check_proper_git_push(stdout, stderr)

        # start new branch, and push file into it
        Command(tmpdir.strpath).execute('git checkout -b dev')
        stdout, stderr = _add_files_and_push(
            'git', tmpdir.strpath, clone_url=clone_url, target_branch='dev',
            new_branch=True)

        _check_proper_git_push(stdout, stderr, branch='dev')

    def test_push_new_branch_hg_with_branch_permissions_no_force_push(
            self, rcstack, tmpdir, user_util, branch_permission_setter):
        repo = user_util.create_repo(repo_type='hg')
        repo_name = repo.repo_name
        username = TEST_USER_ADMIN_LOGIN
        branch_permission_setter(repo_name, username, permission='branch.push')

        clone_url = rcstack.repo_clone_url(repo.repo_name)
        Command(os.path.dirname(tmpdir.strpath)).execute(
            'hg clone', clone_url, tmpdir.strpath)

        stdout, stderr = _add_files_and_push(
            'hg', tmpdir.strpath, clone_url=clone_url)
        _check_proper_hg_push(stdout, stderr)

        # start new branch, and push file into it
        Command(tmpdir.strpath).execute(
            'hg branch dev && hg commit -m "starting dev branch"')
        stdout, stderr = _add_files_and_push(
            'hg', tmpdir.strpath, clone_url=clone_url, target_branch='dev',
            new_branch=True)

        _check_proper_hg_push(stdout, stderr)

    def test_push_new_branch_git_with_branch_permissions_no_force_push(
            self, rcstack, tmpdir, user_util, branch_permission_setter):
        repo = user_util.create_repo(repo_type='git')
        repo_name = repo.repo_name
        username = TEST_USER_ADMIN_LOGIN
        branch_permission_setter(repo_name, username, permission='branch.push')

        clone_url = rcstack.repo_clone_url(repo.repo_name)
        Command(os.path.dirname(tmpdir.strpath)).execute(
            'git clone', clone_url, tmpdir.strpath)

        stdout, stderr = _add_files_and_push(
            'git', tmpdir.strpath, clone_url=clone_url)
        _check_proper_git_push(stdout, stderr)

        # start new branch, and push file into it
        Command(tmpdir.strpath).execute('git checkout -b dev')
        stdout, stderr = _add_files_and_push(
            'git', tmpdir.strpath, clone_url=clone_url, target_branch='dev',
            new_branch=True)

        _check_proper_git_push(stdout, stderr, branch='dev')