#! /usr/bin/python
#
# $Header$
#
# Distributed under the terms of the GNU General Public License v2
# Copyright (c) 2003 Karl Trygve Kalleberg
#
# Based on previous versions, by
#  - Brandon Low <lostlogic@gentoo.org>
#  - Jochem Kossen <j.kossen@home.nl>
#  - Leo Lipelis <aeoo@gentoo.org>

from dialog import Dialog
import portage
import time
import re
import os

__author__ = "Karl Trygve Kalleberg"
__email__ = "karltk@gentoo.org"
__version__ = "0.2.0"
__productname__ = "etc-update"
__description__ = "Interactive config file updater"

globals = portage.settings.configdict["globals"]

for i in globals["CONFIG_PROTECT"].split():
    print i

# list all files in all CONFIG_PROTECT dirs
# list them in the gui
# one-by-one:
#  - is update to header only? 
#  - is the original unmodified from the previous package? (not checkable - duh!)
#  - 

class Config:
    pass

def loadConfig():
    cfg = Config()
    globals = portage.settings.configdict["globals"]
    cfg.config_protect = globals["CONFIG_PROTECT"].split()
    return cfg

def _recurseFiles(path):
    files = []
    if os.path.exists(path):
        try:
            tmpfiles = os.listdir(path)
            for i in tmpfiles:
                fn = path + "/" + i
                if os.path.isdir(fn):
                    files += _recurseFiles(fn)
                elif os.path.isfile(fn):
                    m = re.search("\._cfg...._",fn)
                    if m:
                        files.append(fn)
                else:
                    print "What is this anyway?:", fn
        except OSError:
            pass
            # print "Access denied:", path
            
    return files
                
def findAllFiles(dlg, config):
    files = []
    gauge = dlg.gauge(0,
                      "Processing CONFIG_PROTECT directories...",
                      7,
                      60,
                      "Gauge",
                      sleep=3)
    num_dirs = len(config.config_protect)
    for i in xrange(num_dirs):
        rem = repr(num_dirs - i / num_dirs)
        gauge.update(rem, "Directories remaining: %s" % rem)
        files += _recurseFiles(config.config_protect[i])
    return files

def prettifyFiles(files):
    rx = re.compile("\._cfg...._")
    def strip_cfg(x):
        """Remove ._cfg????_ part """
        m = rx.search(x)
        if m:
            s,e = m.span(0)
            return x[:s] + x[e:]
        return x
    return map(strip_cfg, files)

def updateFile(dlg, original):

    # Find candidates

    dir = os.path.dirname(original)
    filename = os.path.basename(original)
    rx = re.compile("\._cfg...._" + filename)
    cand = filter(lambda x: rx.search(x), os.listdir(dir))

    if len(cand) > 1:
        
        # Add mtimes
        for i in xrange(len(cand)):
            stamp = time.localtime(os.path.getmtime(dir + "/" + cand[i]))
            tstr = time.strftime("%a, %d %b %Y %H:%M:%S", stamp)
            cand[i] = cand[i] + " - " + tstr

    
        # Show selection
        replacement = dlg.menu("Files that need updating",
                          Dialog.AUTO_SIZE,
                          list = cand,
                          showHelp = False,
                          title="Checklist")

    else:
        replacement = cand[0]

    
    # Display diff

    dlg.yesno("Would you like to update \n" + \
              "[" + original + "]with\n[" + dir + "/" + replacement + "] ?")
    
def displayFiles(dlg, config, files):

    pretty_files = prettifyFiles(files)

    while 1:
        result = dlg.menu("Files that need updating",
                          Dialog.AUTO_SIZE,
                          list = pretty_files,
                          showHelp = False,
                          title="Checklist")
        if result == None:
            if dlg.ERR_CANCEL:
                break

        updateFile(dlg, result)
        
    if len(pretty_files):
        print "!!! Warning: There are still files that require updating."
        
    
def main():
    dlg = Dialog("etc-update")
#    dlg = None
    
    config = loadConfig()
    files = findAllFiles(dlg, config)
    displayFiles(dlg, config,files)
    
    
if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        print "Operation aborted!"

# TODO:
# - option for automatically update untouched files
# - show coloured diff
# - proper progress bar