#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# === This file is part of Calamares - <http://github.com/calamares> ===
#
#   Copyright 2014, Pier Luigi Fiorini <pierluigi.fiorini@gmail.com>
#   Copyright 2015-2017, Teo Mrnjavac <teo@kde.org>
#   Copyright 2016-2017, Kyle Robbertze <kyle@aims.ac.za>
#
#   Calamares 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 3 of the License, or
#   (at your option) any later version.
#
#   Calamares 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 Calamares. If not, see <http://www.gnu.org/licenses/>.

import subprocess
import libcalamares
from libcalamares.utils import check_target_env_call, target_env_call
from string import Template

class PackageManager:
    """ Package manager class.

    :param backend:
    """
    def __init__(self, backend):
        self.backend = backend

    def install(self, pkgs, from_local=False):
        """ Installs packages.

        :param pkgs:
        :param from_local:
        """
        if self.backend == "packagekit":
            for pkg in pkgs:
                check_target_env_call(["pkcon", "-py", "install", pkg])
        elif self.backend == "zypp":
            check_target_env_call(["zypper", "--non-interactive", "--quiet-install", "install",
                               "--auto-agree-with-licenses", "install"] + pkgs)
        elif self.backend == "yum":
            check_target_env_call(["yum", "install", "-y"] + pkgs)
        elif self.backend == "dnf":
            check_target_env_call(["dnf", "install", "-y"] + pkgs)
        elif self.backend == "urpmi":
            check_target_env_call(["urpmi", "--download-all", "--no-suggests", "--no-verify-rpm",
                               "--fastunsafe", "--ignoresize", "--nolock", "--auto"] + pkgs)
        elif self.backend == "apt":
            check_target_env_call(["apt-get", "-q", "-y", "install"] + pkgs)
        elif self.backend == "pacman":
            if from_local:
                pacman_flags = "-U"
            else:
                pacman_flags = "-Sy"

            check_target_env_call(["pacman", pacman_flags, "--noconfirm"] + pkgs)
        elif self.backend == "portage":
            check_target_env_call(["emerge", "-v"] + pkgs)
        elif self.backend == "entropy":
            check_target_env_call(["equo", "i"] + pkgs)

    def remove(self, pkgs):
        """ Removes packages.

        :param pkgs:
        """
        if self.backend == "packagekit":
            for pkg in pkgs:
                check_target_env_call(["pkcon", "-py", "remove", pkg])
        elif self.backend == "zypp":
            check_target_env_call(["zypper", "--non-interactive", "remove"] + pkgs)
        elif self.backend == "yum":
            check_target_env_call(["yum", "--disablerepo=*", "-C", "-y", "remove"] + pkgs)
        elif self.backend == "dnf":
            # ignore the error code for now because dnf thinks removing a nonexistent package is an error
            target_env_call(["dnf", "--disablerepo=*", "-C", "-y", "remove"] + pkgs)
        elif self.backend == "urpmi":
            check_target_env_call(["urpme", "--auto"] + pkgs)
        elif self.backend == "apt":
            check_target_env_call(["apt-get", "--purge", "-q", "-y", "remove"] + pkgs)
            check_target_env_call(["apt-get", "--purge", "-q", "-y", "autoremove"])
        elif self.backend == "pacman":
            check_target_env_call(["pacman", "-Rs", "--noconfirm"] + pkgs)
        elif self.backend == "portage":
            check_target_env_call(["emerge", "-C"] + pkgs)
            check_target_env_call(["emerge", "--depclean", "-q"])
        elif self.backend == "entropy":
            check_target_env_call(["equo", "rm"] + pkgs)

    def update_db(self):
        if self.backend == "packagekit":
            check_target_env_call(["pkcon", "refresh"])
        elif self.backend == "zypp":
            check_target_env_call(["zypper", "--non-interactive", "update"])
        elif self.backend == "urpmi":
            check_target_env_call(["urpmi.update", "-a"])
        elif self.backend == "apt":
            check_target_env_call(["apt-get", "update"])
        elif self.backend == "pacman":
            check_target_env_call(["pacman", "-Sy"])
        elif self.backend == "portage":
            check_target_env_call(["emerge", "--sync"])
        elif self.backend == "entropy":
            check_target_env_call(["equo", "update"])

    def run(self, script):
        if script != "":
            check_target_env_call(script.split(" "))


def subst_locale(list):
    ret = []
    locale = libcalamares.globalstorage.value("locale")
    if locale:
        for e in list:
            if  locale != "en":
                entry = Template(e)
                ret.append(entry.safe_substitute(LOCALE=locale))
            elif 'LOCALE' not in e:
                ret.append(e)
    else:
        ret = list
    return ret

def run_operations(pkgman, entry):
    """ Call package manager with given parameters.

    :param pkgman:
    :param entry:
    """
    for key in entry.keys():
        entry[key] = subst_locale(entry[key])
        if key == "install":
            if isinstance(entry[key], list):
                for package in entry[key]:
                    pkgman.run(package["pre-script"])
                    pkgman.install([package["package"]])
                    pkgman.run(package["post-script"])
            else:
                pkgman.install(entry[key])
        elif key == "try_install":
            # we make a separate package manager call for each package so a single
            # failing package won't stop all of them
            for package in entry[key]:
                if isinstance(package, str):
                    try:
                        pkgman.install([package])
                    except subprocess.CalledProcessError:
                        libcalamares.utils.debug("WARNING: could not install package {}".format(package))
                else:
                    try:
                        pkgman.run(package["pre-script"])
                        pkgman.install([package["package"]])
                        pkgman.run(package["post-script"])
                    except subprocess.CalledProcessError:
                        libcalamares.utils.debug("WARNING: could not install packages {}".format(package["package"]))
        elif key == "remove":
            pkgman.remove(entry[key])
        elif key == "try_remove":
            for package in entry[key]:
                try:
                    pkgman.remove([package])
                except subprocess.CalledProcessError:
                    libcalamares.utils.debug("WARNING: could not remove package {}".format(package))
        elif key == "localInstall":
            pkgman.install(entry[key], from_local=True)


def run():
    """ Calls routine with detected package manager to install locale packages
    or remove drivers not needed on the installed system.

    :return:
    """
    backend = libcalamares.job.configuration.get("backend")

    if backend not in ("packagekit", "zypp", "yum", "dnf", "urpmi", "apt", "pacman", "portage", "entropy"):
        return "Bad backend", "backend=\"{}\"".format(backend)

    pkgman = PackageManager(backend)
    operations = libcalamares.job.configuration.get("operations", [])

    update_db = libcalamares.job.configuration.get("update_db", False)
    if update_db and libcalamares.globalstorage.value("hasInternet"):
        pkgman.update_db()

    for entry in operations:
        run_operations(pkgman, entry)

    if libcalamares.globalstorage.contains("packageOperations"):
        run_operations(pkgman, libcalamares.globalstorage.value("packageOperations"))

    return None
