# 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 pytest
import urllib.parse
import mock

from rhodecode.lib.vcs.backends.base import Config
from rhodecode.tests.lib.middleware import mock_scm_app
import rhodecode.lib.middleware.simplegit as simplegit


def get_environ(url, request_method):
    """Construct a minimum WSGI environ based on the URL."""
    parsed_url = urllib.parse.urlparse(url)
    environ = {
        "PATH_INFO": parsed_url.path,
        "QUERY_STRING": parsed_url.query,
        "REQUEST_METHOD": request_method,
    }

    return environ


@pytest.mark.parametrize(
    "url, expected_action, request_method",
    [
        ("/foo/bar/info/refs?service=git-upload-pack", "pull", "GET"),
        ("/foo/bar/info/refs?service=git-receive-pack", "push", "GET"),
        ("/foo/bar/git-upload-pack", "pull", "GET"),
        ("/foo/bar/git-receive-pack", "push", "GET"),
        # Edge case: missing data for info/refs
        ("/foo/info/refs?service=", "pull", "GET"),
        ("/foo/info/refs", "pull", "GET"),
        # Edge case: git command comes with service argument
        ("/foo/git-upload-pack?service=git-receive-pack", "pull", "GET"),
        ("/foo/git-receive-pack?service=git-upload-pack", "push", "GET"),
        # Edge case: repo name conflicts with git commands
        ("/git-receive-pack/git-upload-pack", "pull", "GET"),
        ("/git-receive-pack/git-receive-pack", "push", "GET"),
        ("/git-upload-pack/git-upload-pack", "pull", "GET"),
        ("/git-upload-pack/git-receive-pack", "push", "GET"),
        ("/foo/git-receive-pack", "push", "GET"),
        # Edge case: not a smart protocol url
        ("/foo/bar", "pull", "GET"),
        # GIT LFS cases, batch
        ("/foo/bar/info/lfs/objects/batch", "push", "GET"),
        ("/foo/bar/info/lfs/objects/batch", "pull", "POST"),
        # GIT LFS oid, dl/upl
        ("/foo/bar/info/lfs/abcdeabcde", "pull", "GET"),
        ("/foo/bar/info/lfs/abcdeabcde", "push", "PUT"),
        ("/foo/bar/info/lfs/abcdeabcde", "push", "POST"),
        # Edge case: repo name conflicts with git commands
        ("/info/lfs/info/lfs/objects/batch", "push", "GET"),
        ("/info/lfs/info/lfs/objects/batch", "pull", "POST"),
    ],
)
def test_get_action(url, expected_action, request_method, baseapp, request_stub):
    app = simplegit.SimpleGit(config={"auth_ret_code": "", "base_path": ""}, registry=request_stub.registry)
    assert expected_action == app._get_action(get_environ(url, request_method))


@pytest.mark.parametrize(
    "url, expected_repo_name, request_method",
    [
        ("/foo/info/refs?service=git-upload-pack", "foo", "GET"),
        ("/foo/bar/info/refs?service=git-receive-pack", "foo/bar", "GET"),
        ("/foo/git-upload-pack", "foo", "GET"),
        ("/foo/git-receive-pack", "foo", "GET"),
        ("/foo/bar/git-upload-pack", "foo/bar", "GET"),
        ("/foo/bar/git-receive-pack", "foo/bar", "GET"),
        # GIT LFS cases, batch
        ("/foo/bar/info/lfs/objects/batch", "foo/bar", "GET"),
        ("/example-git/info/lfs/objects/batch", "example-git", "POST"),
        # GIT LFS oid, dl/upl
        ("/foo/info/lfs/abcdeabcde", "foo", "GET"),
        ("/foo/bar/info/lfs/abcdeabcde", "foo/bar", "PUT"),
        ("/my-git-repo/info/lfs/abcdeabcde", "my-git-repo", "POST"),
        # Edge case: repo name conflicts with git commands
        ("/info/lfs/info/lfs/objects/batch", "info/lfs", "GET"),
        ("/info/lfs/info/lfs/objects/batch", "info/lfs", "POST"),
    ],
)
def test_get_repository_name(url, expected_repo_name, request_method, baseapp, request_stub):
    app = simplegit.SimpleGit(config={"auth_ret_code": "", "base_path": ""}, registry=request_stub.registry)
    assert expected_repo_name == app._get_repository_name(get_environ(url, request_method))


def test_get_config(user_util, baseapp, request_stub):
    repo = user_util.create_repo(repo_type="git")
    app = simplegit.SimpleGit(config={"auth_ret_code": "", "base_path": ""}, registry=request_stub.registry)
    extras = {"foo": "FOO", "bar": "BAR"}

    # We copy the extras as the method below will change the contents.
    git_config = app._create_config(dict(extras), repo_name=repo.repo_name)

    expected_config = dict(extras)
    expected_config.update(
        {
            "git_update_server_info": False,
            "git_lfs_enabled": False,
            "git_lfs_store_path": git_config["git_lfs_store_path"],
            "git_lfs_http_scheme": "http",
        }
    )

    assert git_config == expected_config


def test_create_wsgi_app_uses_scm_app_from_simplevcs(baseapp, request_stub):
    config = {
        "auth_ret_code": "",
        "base_path": "",
        "vcs.scm_app_implementation": "rhodecode.tests.lib.middleware.mock_scm_app",
    }
    app = simplegit.SimpleGit(config=config, registry=request_stub.registry)
    wsgi_app = app._create_wsgi_app("/tmp/test", "test_repo", {})
    assert wsgi_app is mock_scm_app.mock_git_wsgi