#
# Copyright (C) 2017  Red Hat, Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# 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 General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
import dnf
from glob import glob
import os
import shutil
import tempfile
import time
import unittest

from pylorax.sysutils import joinpaths
from pylorax.api.config import configure, make_dnf_dirs
from pylorax.api.projects import api_time, api_changelog, pkg_to_project, pkg_to_project_info, pkg_to_dep
from pylorax.api.projects import proj_to_module, projects_list, projects_info, projects_depsolve
from pylorax.api.projects import modules_list, modules_info, ProjectsError, dep_evra, dep_nevra
from pylorax.api.projects import repo_to_source, get_repo_sources, delete_repo_source, source_to_repo
from pylorax.api.projects import source_to_repodict, dnf_repo_to_file_repo
from pylorax.api.dnfbase import get_base_object

class Package(object):
    """Test class for hawkey.Package tests"""
    name = "name"
    summary = "summary"
    description = "description"
    url = "url"
    epoch = 1
    release = "release"
    arch = "arch"
    buildtime = 499222800
    license = "license"
    version = "version"

class ProjectsTest(unittest.TestCase):
    @classmethod
    def setUpClass(self):
        self.tmp_dir = tempfile.mkdtemp(prefix="lorax.test.repo.")
        self.config = configure(root_dir=self.tmp_dir, test_config=True)
        make_dnf_dirs(self.config, os.getuid(), os.getgid())
        self.dbo = get_base_object(self.config)
        os.environ["TZ"] = "UTC"
        time.tzset()

    @classmethod
    def tearDownClass(self):
        shutil.rmtree(self.tmp_dir)

    def test_api_time(self):
        self.assertEqual(api_time(499222800), "1985-10-27T01:00:00")

    def test_api_changelog(self):
        self.assertEqual(api_changelog([[0, 1, "Heavy!"], [0, 1, "Light!"]]), "Heavy!")

    def test_api_changelog_empty_list(self):
        self.assertEqual(api_changelog([]), '')

    def test_api_changelog_missing_text_entry(self):
        self.assertEqual(api_changelog([('now', 'atodorov')]), '')

    def test_pkg_to_project(self):
        result = {"name":"name",
                  "summary":"summary",
                  "description":"description",
                  "homepage":"url",
                  "upstream_vcs":"UPSTREAM_VCS"}

        pkg = Package()
        self.assertEqual(pkg_to_project(pkg), result)

    def test_pkg_to_project_info(self):
        build = {"epoch":1,
                 "release":"release",
                 "arch":"arch",
                 "build_time":"1985-10-27T01:00:00",
                 "changelog":"CHANGELOG_NEEDED",
                 "build_config_ref": "BUILD_CONFIG_REF",
                 "build_env_ref":    "BUILD_ENV_REF",
                 "metadata":    {},
                 "source":      {"license":"license",
                                 "version":"version",
                                 "source_ref": "SOURCE_REF",
                                 "metadata":   {}}}

        result = {"name":"name",
                  "summary":"summary",
                  "description":"description",
                  "homepage":"url",
                  "upstream_vcs":"UPSTREAM_VCS",
                  "builds": [build]}

        pkg = Package()
        self.assertEqual(pkg_to_project_info(pkg), result)

    def test_pkg_to_dep(self):
        result = {"name":"name",
                  "epoch":1,
                  "version":"version",
                  "release":"release",
                  "arch":"arch"}

        pkg = Package()
        self.assertEqual(pkg_to_dep(pkg), result)

    def test_proj_to_module(self):
        result = {"name":"name",
                  "group_type":"rpm"}

        proj = pkg_to_project(Package())
        self.assertEqual(proj_to_module(proj), result)

    def test_dep_evra(self):
        dep = {"arch": "noarch",
               "epoch": 0,
               "name": "basesystem",
               "release": "7.el7",
               "version": "10.0"}
        self.assertEqual(dep_evra(dep), "10.0-7.el7.noarch")

    def test_dep_evra_with_epoch_not_zero(self):
        dep = {"arch": "x86_64",
               "epoch": 2,
               "name": "tog-pegasus-libs",
               "release": "3.el7",
               "version": "2.14.1"}
        self.assertEqual(dep_evra(dep), "2:2.14.1-3.el7.x86_64")

    def test_dep_nevra(self):
        dep = {"arch": "noarch",
               "epoch": 0,
               "name": "basesystem",
               "release": "7.el7",
               "version": "10.0"}
        self.assertEqual(dep_nevra(dep), "basesystem-10.0-7.el7.noarch")

    def test_projects_list(self):
        projects = projects_list(self.dbo)
        self.assertEqual(len(projects) > 10, True)

    def test_projects_info(self):
        projects = projects_info(self.dbo, ["bash"])

        self.assertEqual(projects[0]["name"], "bash")
        self.assertEqual(projects[0]["builds"][0]["source"]["license"], "GPLv3+")

    def test_projects_depsolve(self):
        deps = projects_depsolve(self.dbo, [("bash", "*.*")], [])
        self.assertTrue(len(deps) > 3)
        self.assertTrue("basesystem" in [dep["name"] for dep in deps])

    def test_projects_depsolve_version(self):
        """Test that depsolving with a partial wildcard version works"""
        deps = projects_depsolve(self.dbo, [("bash", "5.*")], [])
        self.assertEqual(deps[1]["name"], "bash")

        deps = projects_depsolve(self.dbo, [("bash", "5.0.*")], [])
        self.assertEqual(deps[1]["name"], "bash")

    def test_projects_depsolve_oldversion(self):
        """Test that depsolving a specific non-existant version fails"""
        with self.assertRaises(ProjectsError):
            deps = projects_depsolve(self.dbo, [("bash", "1.0.0")], [])
            self.assertEqual(deps[1]["name"], "bash")

    def test_projects_depsolve_fail(self):
        with self.assertRaises(ProjectsError):
            projects_depsolve(self.dbo, [("nada-package", "*.*")], [])

    def test_modules_list_all(self):
        modules = modules_list(self.dbo, None)

        self.assertEqual(len(modules) > 10, True)
        self.assertEqual(modules[0]["group_type"], "rpm")

    def test_modules_list_glob(self):
        modules = modules_list(self.dbo, ["g*"])
        self.assertEqual(modules[0]["name"].startswith("g"), True)

    def test_modules_info(self):
        modules = modules_info(self.dbo, ["bash"])

        print(modules)
        self.assertEqual(modules[0]["name"], "bash")
        self.assertEqual(modules[0]["dependencies"][0]["name"], "basesystem")

    def test_groups_depsolve(self):
        deps = projects_depsolve(self.dbo, [], ["c-development"])
        names = [grp["name"] for grp in deps]
        self.assertTrue("autoconf" in names)            # mandatory package
        self.assertTrue("ctags" in names)               # default package
        self.assertFalse("cmake" in names)              # optional package

class ConfigureTest(unittest.TestCase):
    @classmethod
    def setUpClass(self):
        self.tmp_dir = tempfile.mkdtemp(prefix="lorax.test.configure.")
        self.conf_file = os.path.join(self.tmp_dir, 'test.conf')
        open(self.conf_file, 'w').write("[composer]\ncache_dir = /tmp/cache-test")

    @classmethod
    def tearDownClass(self):
        shutil.rmtree(self.tmp_dir)

    def test_configure_reads_existing_file(self):
        config = configure(conf_file=self.conf_file)
        self.assertEqual(config.get('composer', 'cache_dir'), '/tmp/cache-test')

    def test_configure_reads_non_existing_file(self):
        config = configure(conf_file=self.conf_file + '.non-existing')
        self.assertEqual(config.get('composer', 'cache_dir'), '/var/tmp/composer/cache')

def fakerepo_baseurl_v0():
    return {
        "check_gpg": True,
        "check_ssl": True,
        "name": "fake-repo-baseurl",
        "system": False,
        "type": "yum-baseurl",
        "url": "https://fake-repo.base.url"
    }

def fakerepo_baseurl_v1():
    d = fakerepo_baseurl_v0()
    d["id"] = "fake-repo-baseurl"
    d["name"] = "A fake repo with a baseurl"
    return d

def fakesystem_repo_v0():
    return {
        "check_gpg": True,
        "check_ssl": True,
        "name": "fake-repo-baseurl",
        "system": True,
        "type": "yum-baseurl",
        "url": "https://fake-repo.base.url"
    }

def fakesystem_repo_v1():
    d = fakesystem_repo_v0()
    d["id"] = "fake-repo-baseurl"
    d["name"] = "A fake repo with a baseurl"
    return d

def fakerepo_metalink_v0():
    return {
        "check_gpg": True,
        "check_ssl": True,
        "name": "fake-repo-metalink",
        "system": False,
        "type": "yum-metalink",
        "url": "https://fake-repo.metalink"
    }

def fakerepo_metalink_v1():
    d = fakerepo_metalink_v0()
    d["id"] = "fake-repo-metalink"
    d["name"] = "A fake repo with a metalink"
    return d

def fakerepo_mirrorlist_v0():
    return {
        "check_gpg": True,
        "check_ssl": True,
        "name": "fake-repo-mirrorlist",
        "system": False,
        "type": "yum-mirrorlist",
        "url": "https://fake-repo.mirrorlist"
    }

def fakerepo_mirrorlist_v1():
    d = fakerepo_mirrorlist_v0()
    d["id"] = "fake-repo-mirrorlist"
    d["name"] = "A fake repo with a mirrorlist"
    return d

def fakerepo_proxy_v0():
    return {
        "check_gpg": True,
        "check_ssl": True,
        "name": "fake-repo-proxy",
        "proxy": "https://fake-repo.proxy",
        "system": False,
        "type": "yum-baseurl",
        "url": "https://fake-repo.base.url"
    }

def fakerepo_proxy_v1():
    d = fakerepo_proxy_v0()
    d["id"] = "fake-repo-proxy"
    d["name"] = "A fake repo with a proxy"
    return d

def fakerepo_gpgkey_v0():
    return {
        "check_gpg": True,
        "check_ssl": True,
        "gpgkey_urls": [
            "https://fake-repo.gpgkey"
        ],
        "name": "fake-repo-gpgkey",
        "system": False,
        "type": "yum-baseurl",
        "url": "https://fake-repo.base.url"
    }

def fakerepo_gpgkey_v1():
    d = fakerepo_gpgkey_v0()
    d["id"] = "fake-repo-gpgkey"
    d["name"] = "A fake repo with a gpgkey"
    return d

def singlerepo_v0():
    return {
        "check_gpg": True,
        "check_ssl": True,
        "gpgkey_urls": [
            "file:///etc/pki/rpm-gpg/RPM-GPG-KEY-fedora-32-x86_64"
        ],
        "name": "single-repo",
        "system": False,
        "type": "yum-baseurl",
        "url": "file:///tmp/lorax-empty-repo/"
    }

def singlerepo_v1():
    d = singlerepo_v0()
    d["id"] = "single-repo"
    d["name"] = "One repo in the file"
    return d

def singlerepo_vars_v0():
    return {
        "check_gpg": True,
        "check_ssl": True,
        "gpgkey_urls": [
            "file:///etc/pki/rpm-gpg/RPM-GPG-KEY-fedora-$releasever-$basearch"
        ],
        "name": "single-repo",
        "system": False,
        "type": "yum-baseurl",
        "url": "file:///tmp/lorax-empty-repo-$releasever-$basearch/"
    }

def singlerepo_vars_v1():
    d = singlerepo_v0()
    d["id"] = "single-repo"
    d["name"] = "One repo in the file"
    return d

class SourceTest(unittest.TestCase):
    @classmethod
    def setUpClass(self):
        self.tmp_dir = tempfile.mkdtemp(prefix="lorax.test.repo.")
        for f in glob("./tests/pylorax/repos/*.repo"):
            shutil.copy2(f, self.tmp_dir)

        self.dbo = dnf.Base()

        # Load all the test repos
        self.dbo.conf.reposdir = [self.tmp_dir]
        self.dbo.read_all_repos()

    @classmethod
    def tearDownClass(self):
        shutil.rmtree(self.tmp_dir)

    def _read(self, repo_file):
        return open(joinpaths(self.tmp_dir, repo_file), "r").read()

    def test_repo_to_source_baseurl(self):
        """Test a repo with a baseurl API v0"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("fake-repo-baseurl"), False, 0), fakerepo_baseurl_v0())

    def test_repo_to_source_baseurl_v1(self):
        """Test a repo with a baseurl API v1"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("fake-repo-baseurl"), False, 1), fakerepo_baseurl_v1())

    def test_system_repo(self):
        """Test a system repo with a baseurl API v0"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("fake-repo-baseurl"), True, 0), fakesystem_repo_v0())

    def test_system_repo_v1(self):
        """Test a system repo with a baseurl API v1"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("fake-repo-baseurl"), True, 1), fakesystem_repo_v1())

    def test_repo_to_source_metalink(self):
        """Test a repo with a metalink API v0"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("fake-repo-metalink"), False, 0), fakerepo_metalink_v0())

    def test_repo_to_source_metalink_v1(self):
        """Test a repo with a metalink API v1"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("fake-repo-metalink"), False, 1), fakerepo_metalink_v1())

    def test_repo_to_source_mirrorlist(self):
        """Test a repo with a mirrorlist API v0"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("fake-repo-mirrorlist"), False, 0), fakerepo_mirrorlist_v0())

    def test_repo_to_source_mirrorlist_v1(self):
        """Test a repo with a mirrorlist API v1"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("fake-repo-mirrorlist"), False, 1), fakerepo_mirrorlist_v1())

    def test_repo_to_source_proxy(self):
        """Test a repo with a proxy API v0"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("fake-repo-proxy"), False, 0), fakerepo_proxy_v0())

    def test_repo_to_source_proxy_v1(self):
        """Test a repo with a proxy API v1"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("fake-repo-proxy"), False, 1), fakerepo_proxy_v1())

    def test_repo_to_source_gpgkey(self):
        """Test a repo with a GPG key API v0"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("fake-repo-gpgkey"), False, 0), fakerepo_gpgkey_v0())

    def test_repo_to_source_gpgkey_v1(self):
        """Test a repo with a GPG key API v1"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("fake-repo-gpgkey"), False, 1), fakerepo_gpgkey_v1())

    def test_get_repo_sources(self):
        """Test getting a list of sources from a repo directory"""
        sources = get_repo_sources(joinpaths(self.tmp_dir, "*.repo"))
        self.assertTrue("lorax-1" in sources)
        self.assertTrue("lorax-2" in sources)

    def test_delete_source_multiple(self):
        """Test deleting a source from a repo file with multiple entries"""
        delete_repo_source(joinpaths(self.tmp_dir, "*.repo"), "lorax-3")
        sources = get_repo_sources(joinpaths(self.tmp_dir, "*.repo"))
        self.assertTrue("lorax-3" not in sources)

    def test_delete_source_single(self):
        """Test deleting a source from a repo with only 1 entry"""
        delete_repo_source(joinpaths(self.tmp_dir, "*.repo"), "single-repo")
        sources = get_repo_sources(joinpaths(self.tmp_dir, "*.repo"))
        self.assertTrue("single-repo" not in sources)
        self.assertTrue(not os.path.exists(joinpaths(self.tmp_dir, "single.repo")))

    def test_delete_source_other(self):
        """Test deleting a source from a repo that doesn't match the source name"""
        with self.assertRaises(ProjectsError):
            delete_repo_source(joinpaths(self.tmp_dir, "*.repo"), "unknown-source")
        sources = get_repo_sources(joinpaths(self.tmp_dir, "*.repo"))
        self.assertTrue("lorax-1" in sources)
        self.assertTrue("lorax-2" in sources)
        self.assertTrue("lorax-4" in sources)
        self.assertTrue("other-repo" in sources)

    def test_source_to_repo_baseurl(self):
        """Test creating a dnf.Repo with a baseurl API v0"""
        repo = source_to_repo(fakerepo_baseurl_v0(), self.dbo.conf)
        self.assertEqual(repo.baseurl[0], fakerepo_baseurl_v0()["url"])

    def test_source_to_repodict_baseurl(self):
        """Test creating a repodict with a baseurl API v0"""
        repo = source_to_repodict(fakerepo_baseurl_v0())
        self.assertEqual(repo[1][0], fakerepo_baseurl_v0()["url"])

    def test_source_to_repo_baseurl_v1(self):
        """Test creating a dnf.Repo with a baseurl API v1"""
        repo = source_to_repo(fakerepo_baseurl_v1(), self.dbo.conf)
        self.assertEqual(repo.baseurl[0], fakerepo_baseurl_v1()["url"])

    def test_source_to_repodict_baseurl_v1(self):
        """Test creating a repodict with a baseurl API v1"""
        repo = source_to_repodict(fakerepo_baseurl_v1())
        self.assertEqual(repo[1][0], fakerepo_baseurl_v1()["url"])

    def test_source_to_repo_metalink(self):
        """Test creating a dnf.Repo with a metalink API v0"""
        repo = source_to_repo(fakerepo_metalink_v0(), self.dbo.conf)
        self.assertEqual(repo.metalink, fakerepo_metalink_v0()["url"])

    def test_source_to_repodict_metalink(self):
        """Test creating a repodict with a metalink API v0"""
        repo = source_to_repodict(fakerepo_metalink_v0())
        self.assertEqual(repo[2]["metalink"], fakerepo_metalink_v0()["url"])

    def test_source_to_repo_metalink_v1(self):
        """Test creating a dnf.Repo with a metalink API v1"""
        repo = source_to_repo(fakerepo_metalink_v1(), self.dbo.conf)
        self.assertEqual(repo.metalink, fakerepo_metalink_v1()["url"])

    def test_source_to_repodict_metalink_v1(self):
        """Test creating a repodict with a metalink API v1"""
        repo = source_to_repodict(fakerepo_metalink_v1())
        self.assertEqual(repo[2]["metalink"], fakerepo_metalink_v1()["url"])

    def test_source_to_repo_mirrorlist(self):
        """Test creating a dnf.Repo with a mirrorlist API v0"""
        repo = source_to_repo(fakerepo_mirrorlist_v0(), self.dbo.conf)
        self.assertEqual(repo.mirrorlist, fakerepo_mirrorlist_v0()["url"])

    def test_source_to_repodict_mirrorlist(self):
        """Test creating a repodict with a mirrorlist API v0"""
        repo = source_to_repodict(fakerepo_mirrorlist_v0())
        self.assertEqual(repo[2]["mirrorlist"], fakerepo_mirrorlist_v0()["url"])

    def test_source_to_repo_mirrorlist_v1(self):
        """Test creating a dnf.Repo with a mirrorlist"""
        repo = source_to_repo(fakerepo_mirrorlist_v1(), self.dbo.conf)
        self.assertEqual(repo.mirrorlist, fakerepo_mirrorlist_v1()["url"])

    def test_source_to_repodict_mirrorlist_v1(self):
        """Test creating a repodict with a mirrorlist"""
        repo = source_to_repodict(fakerepo_mirrorlist_v1())
        self.assertEqual(repo[2]["mirrorlist"], fakerepo_mirrorlist_v1()["url"])

    def test_source_to_repo_proxy(self):
        """Test creating a dnf.Repo with a proxy API v0"""
        repo = source_to_repo(fakerepo_proxy_v0(), self.dbo.conf)
        self.assertEqual(repo.proxy, fakerepo_proxy_v0()["proxy"])

    def test_source_to_repodict_proxy(self):
        """Test creating a repodict with a proxy API v0"""
        repo = source_to_repodict(fakerepo_proxy_v0())
        self.assertEqual(repo[2]["proxy"], fakerepo_proxy_v0()["proxy"])

    def test_source_to_repo_proxy_v1(self):
        """Test creating a dnf.Repo with a proxy API v1"""
        repo = source_to_repo(fakerepo_proxy_v1(), self.dbo.conf)
        self.assertEqual(repo.proxy, fakerepo_proxy_v1()["proxy"])

    def test_source_to_repodict_proxy_v1(self):
        """Test creating a repodict with a proxy API v1"""
        repo = source_to_repodict(fakerepo_proxy_v1())
        self.assertEqual(repo[2]["proxy"], fakerepo_proxy_v1()["proxy"])

    def test_source_to_repo_gpgkey(self):
        """Test creating a dnf.Repo with a proxy API v0"""
        repo = source_to_repo(fakerepo_gpgkey_v0(), self.dbo.conf)
        self.assertEqual(repo.gpgkey[0], fakerepo_gpgkey_v0()["gpgkey_urls"][0])

    def test_source_to_repodict_gpgkey(self):
        """Test creating a repodict with a proxy API v0"""
        repo = source_to_repodict(fakerepo_gpgkey_v0())
        self.assertEqual(repo[2]["gpgkey"][0], fakerepo_gpgkey_v0()["gpgkey_urls"][0])

    def test_source_to_repo_gpgkey_v1(self):
        """Test creating a dnf.Repo with a proxy API v1"""
        repo = source_to_repo(fakerepo_gpgkey_v1(), self.dbo.conf)
        self.assertEqual(repo.gpgkey[0], fakerepo_gpgkey_v1()["gpgkey_urls"][0])

    def test_source_to_repodict_gpgkey_v1(self):
        """Test creating a repodict with a proxy API v1"""
        repo = source_to_repodict(fakerepo_gpgkey_v1())
        self.assertEqual(repo[2]["gpgkey"][0], fakerepo_gpgkey_v1()["gpgkey_urls"][0])

    def test_drtfr_baseurl(self):
        """Test creating a dnf .repo file from a baseurl Repo object"""
        self.assertEqual(dnf_repo_to_file_repo(self.dbo.repos.get("fake-repo-baseurl")),
                         self._read("baseurl-test.repo"))

    def test_drtfr_metalink(self):
        """Test creating a dnf .repo file from a metalink Repo object"""
        self.assertEqual(dnf_repo_to_file_repo(self.dbo.repos.get("fake-repo-metalink")),
                         self._read("metalink-test.repo"))

    def test_drtfr_mirrorlist(self):
        """Test creating a dnf .repo file from a mirrorlist Repo object"""
        self.assertEqual(dnf_repo_to_file_repo(self.dbo.repos.get("fake-repo-mirrorlist")),
                         self._read("mirrorlist-test.repo"))

    def test_drtfr_proxy(self):
        """Test creating a dnf .repo file from a baseurl Repo object with proxy"""
        self.assertEqual(dnf_repo_to_file_repo(self.dbo.repos.get("fake-repo-proxy")),
                         self._read("proxy-test.repo"))

    def test_drtfr_gpgkey(self):
        """Test creating a dnf .repo file from a baseurl Repo object with gpgkey"""
        self.assertEqual(dnf_repo_to_file_repo(self.dbo.repos.get("fake-repo-gpgkey")),
                         self._read("gpgkey-test.repo"))

    def test_repo_to_source_json(self):
        """Test serializing repo_to_source results API v0"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("single-repo"), False, 0), singlerepo_v0())

    def test_repo_to_source_json_v1(self):
        """Test serializing repo_to_source results API v1"""
        self.assertEqual(repo_to_source(self.dbo.repos.get("single-repo"), False, 1), singlerepo_v1())