#!/usr/bin/python

import os
import readline
from imms import IMMSCleaner, IMMSDb
from utils import unique, copy_file, set_file_completer

class CLICleaner(IMMSCleaner):
    def check_if_uid_exist(self, uid):
        other_paths = self.db.get_library_entry(uid = uid)
        other_paths = unique(map(lambda x: x[0], other_paths))
        for opath in other_paths:
            if os.path.isfile(opath):
	        # if one, release it.
                return 1
        return 0
    def check_and_edit_path(self, path, uid, sid):
        """Must return the new file name, None to remove
        it.  If the new file name is already in the Db,
        it will be skip."""
        # if path exist, skip it.
        if os.path.isfile(path):
            return path
        # if exist another uid with a valid path, remove it.
        if self.check_if_uid_exist(uid):
            return None
	# Elsewhere, first build a list of valid candidate from sid...
        other_paths = self.db.get_library_entry(sid = sid)
        other_paths = unique(map(lambda x: x[0], other_paths))
        return self.edit_filename(path, other_paths)
    def edit_filename(self, path, other_paths):
	# Add them to the history for editing
        for opath in other_paths:
            if os.path.isfile(opath):
                readline.add_history(opath)
        # And also add the current file as a base for editing.
        readline.add_history(path)
	# Sniff!  This doesn't seems to work
        readline.insert_text('Edit')
        while 1:
            cmd = raw_input("I can't find '%s'.\n"
                            "Edit, Skip or Remove? " % path)
            if len(cmd) > 0:
                cmd = cmd.lstrip()[0].lower()
                if cmd == 'e':
                    newpath = raw_input()
		    # Already exist, so delete.
                    if newpath in other_paths:
                        return None
		    # new file so keep it.
                    if os.path.isfile(newpath):
                        return newpath
		    # Elsewhere move it.
                    print "Invalid filename!"
                elif cmd == 's':
                    return path;
                elif cmd == 'r':
                    return None

if __name__ == '__main__':
    set_file_completer()
    readline.parse_and_bind('tab: complete')
    db_file = os.environ['HOME'] + '/.imms/imms.db'
    db_backup = db_file + '.bak'
    copy_file(db_file, db_backup).close()
    try:
        CLICleaner(IMMSDb(db_file)).clean_all()
    except Exception, inst:
        print inst
        while 1:
            ans = raw_input('Do you want to get back the backup file? ')
            if len(ans) > 0:
                ans = ans.lstrip()[0].lower()
                if ans == 'y':
                    copy_file(db_backup, db_file)
                elif ans == 'n':
                    break
    print "backup file preserved: %s." % db_backup