#!/usr/bin/python
'''
    # DESCRIPTION:
    # activator textual interface

    Copyright (C) 2007-2008 Fabio Erculiani

    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, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
'''

from entropyConstants import *
from outputTools import *
from entropy import ServerInterface
Entropy = ServerInterface()

def sync(options, justTidy = False):

    do_noask = False
    myopts = []
    for opt in options:
        if opt == "--noask":
            do_noask = True
        else:
            myopts.append(opt)
    options = myopts

    print_info(green(" * ")+red("Starting to sync data across mirrors (packages/database) ..."))

    if not justTidy:

        mirrors_tainted, mirrors_errors, successfull_mirrors, broken_mirrors, check_data = Entropy.MirrorsService.sync_packages(ask = not do_noask, pretend = etpUi['pretend'])
        if not mirrors_errors:
            if mirrors_tainted:
                if (not do_noask) and etpConst['rss-feed']:
                    etpRSSMessages['commitmessage'] = readtext(">> Please insert a commit message: ")
                elif etpConst['rss-feed']:
                    etpRSSMessages['commitmessage'] = "Autodriven Update"
            rc = database(["sync"])
            if not rc and not do_noask:
                rc = Entropy.askQuestion("Should I continue with the tidy procedure ?")
                if rc == "No":
                    sys.exit(0)
            elif rc:
                print_error(darkred(" !!! ")+red("Aborting !"))
                sys.exit(1)

    Entropy.MirrorsService.tidy_mirrors(ask = not do_noask, pretend = etpUi['pretend'])


def packages(options):

    do_pkg_check = False
    for opt in options:
        if (opt == "--do-packages-check"):
            do_pkg_check = True

    if not options:
        return

    if options[0] == "sync":
        return Entropy.MirrorsService.sync_packages(    ask = etpUi['ask'],
                                                        pretend = etpUi['pretend'],
                                                        packages_check = do_pkg_check
                                                   )


def database(options):

    cmd = options[0]

    if cmd == "lock":

        print_info(green(" * ")+green("Starting to lock mirrors' databases..."))
        rc = Entropy.MirrorsService.lock_mirrors(lock = True)
        if rc:
            print_info(green(" * ")+red("A problem occured on at least one mirror !"))
        else:
            print_info(green(" * ")+green("Databases lock complete"))
        return rc

    elif cmd == "unlock":

        print_info(green(" * ")+green("Starting to unlock mirrors' databases..."))
        rc = Entropy.MirrorsService.lock_mirrors(lock = False)
        if rc:
            print_info(green(" * ")+green("A problem occured on at least one mirror !"))
        else:
            print_info(green(" * ")+green("Databases unlock complete"))
        return rc

    elif cmd == "download-lock":

        print_info(green(" * ")+green("Starting to lock download mirrors' databases..."))
        rc = Entropy.MirrorsService.lock_mirrors_for_download(lock = True)
        if rc:
            print_info(green(" * ")+green("A problem occured on at least one mirror !"))
        else:
            print_info(green(" * ")+green("Download mirrors lock complete"))
        return rc

    elif cmd == "download-unlock":

        print_info(green(" * ")+green("Starting to unlock download mirrors' databases..."))
        rc = Entropy.MirrorsService.lock_mirrors_for_download(lock = False)
        if rc:
            print_info(green(" * ")+green("A problem occured on at least one mirror..."))
        else:
            print_info(green(" * ")+green("Download mirrors unlock complete"))
        return rc

    elif cmd == "lock-status":

        print_info(brown(" * ")+green("Mirrors status table:"))
        dbstatus = Entropy.MirrorsService.get_mirrors_lock()
        for db in dbstatus:
            if (db[1]):
                db[1] = red("Locked")
            else:
                db[1] = green("Unlocked")
            if (db[2]):
                db[2] = red("Locked")
            else:
                db[2] = green("Unlocked")
            print_info(bold("\t"+Entropy.entropyTools.extractFTPHostFromUri(db[0])+": ")+red("[")+brown("DATABASE: ")+db[1]+red("] [")+brown("DOWNLOAD: ")+db[2]+red("]"))
        return 0

    elif cmd == "sync":

        print_info(green(" * ")+red("Syncing databases ..."))
        errors, fine, broken = sync_remote_databases()
        if errors:
            print_error(darkred(" !!! ")+green("Database sync errors, cannot continue."))
            return 1
        return 0


def sync_remote_databases(noUpload = False, justStats = False):

    remoteDbsStatus = Entropy.MirrorsService.get_remote_databases_status()
    print_info(green(" * ")+red("Remote Entropy Database Repository Status:"))
    for dbstat in remoteDbsStatus:
        print_info(green("\t Host:\t")+bold(Entropy.entropyTools.extractFTPHostFromUri(dbstat[0])))
        print_info(red("\t  * Database revision: ")+blue(str(dbstat[1])))

    local_revision = Entropy.get_local_database_revision()
    print_info(red("\t  * Database local revision currently at: ")+blue(str(local_revision)))

    if justStats:
        return 0,set(),set()

    # do the rest
    errors, fine_uris, broken_uris = Entropy.MirrorsService.sync_databases(no_upload = noUpload)
    remote_status = Entropy.MirrorsService.get_remote_databases_status()
    print_info(darkgreen(" * ")+red("Remote Entropy Database Repository Status:"))
    for dbstat in remote_status:
        print_info(darkgreen("\t Host:\t")+bold(Entropy.entropyTools.extractFTPHostFromUri(dbstat[0])))
        print_info(red("\t  * Database revision: ")+blue(str(dbstat[1])))

    return errors, fine_uris, broken_uris