import os
import yaml
import shutil

class OrvaniTag:

    DB = dict()
    FILENAME = ''
    FILE = ''
    CTX = {
        'contexts' : '@',
        'projects' : '+',
        'tags'     : '#'
    }

    def __init__(self, path: str='context.yaml', contexts=CTX):
        self.FILENAME = os.path.basename(path)
        self.__cd(path)

    def __cd(self, path: str):
        """change file context to subdirectory"""
        self.FILE = os.path.join(os.path.dirname(path), self.FILENAME)
        if os.path.isfile(self.FILE):
            try:
                with open(self.FILE, 'r') as data:
                    self.DB = yaml.safe_load(data)
            except Exception:
                print('File does not exist')
        else:
            self.DB = dict()

    def __save(self):
        """save context file to current directory"""
        try:
            with open(self.FILE, 'w+') as data:
                yaml.dump(self.DB, data)
        except Exception:
            print('File write error')
            
    def __context(self, argv) -> dict:
        """create context for given filename"""
        context = dict()
        for ctx in self.CTX.keys():
            context.update( { ctx : []})
        for arg in argv:
            for ctx in self.CTX.keys():
                if str(arg)[0] in self.CTX.values():
                    if str(arg).startswith(self.CTX[ctx]):
                        context[ctx].append(str(arg)[1:])
                else:
                    context.update( { 'note' : str(arg)})
        return context

    def add(self, file: str, *argv: str):
        if os.path.isfile(file):
            if os.path.dirname(file):
                self.__cd(file)
            file = os.path.basename(file)
            context = self.__context(argv)
            if file in self.DB:
                for ctx in self.CTX.keys():
                    self.DB[file][ctx] = list(set( self.DB[file][ctx] + context[ctx]))
                if 'note' in context.keys():
                    self.DB[file]['note'] = context['note']
            else:
                self.DB.update( { file : context } )
            self.__save()

    def remove(self, file: str, *argv: str):
        if os.path.isfile(file):
            if os.path.dirname(file):
                self.__cd(file)
            file = os.path.basename(file)
            context = self.__context(argv)
            if file in self.DB:
                for ctx in self.CTX.keys():
                    for item in context[ctx]:
                        if item in self.DB[file][ctx]:
                            self.DB[file][ctx].remove(item)
            self.__save()

    def delete(self, file: str, fop=False):
        if os.path.dirname(file):
            self.__cd(file)
        if fop:
            os.remove(file)
        file = os.path.basename(file)
        if file in self.DB:
            del self.DB[file]
        self.__save()

    def move(self, source: str, dest: str, retain=False, fop=True):
        if os.path.isfile(source) and os.path.isdir(os.path.dirname(dest)):
            sourcefolder = os.path.dirname(source)
            sourcefile = os.path.basename(source)
            destfolder = os.path.dirname(dest)
            destfile = os.path.basename(dest)
            if not destfile:
                destfile = sourcefile
            sourcedata = {}
            self.__cd(source)
            if self.DB[sourcefile]:
                sourcedata = self.DB[sourcefile]
            if not retain:
                self.purge(source)
            self.__cd(dest)
            self.DB.update( { destfile : sourcedata } )
            self.__save()
            if fop:
                if retain:
                    shutil.copy2(source, dest)
                else:
                    shutil.move(source, dest, copy_function=shutil.copy2)
            
    def purge(self, path: str=''):
        if os.path.dirname(path):
            self.__cd(path)
        for file in self.DB.keys():
            if not os.path.isfile(file):
                if file in self.DB:
                    del self.DB[file]
        self.__save()
        if not self.DB:
            os.remove(self.FILE)


o = OrvaniTag()
#o.add('TODO.md', '#tagx', '+projectX')
o.purge()