#!/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