# -*- coding: utf-8 -*-
"""novelWriter Project Index

 novelWriter – Project Index
=============================
 Class holding the index of tags

 File History:
 Created: 2019-05-27 [0.1.4]

"""

import logging
import nw

from nw.project.document import NWDoc
from nw.enum             import nwItemType

logger = logging.getLogger(__name__)

class NWIndex():

    VALID_KEYS = [
        "todo","tag","pov","char","plot","time","location",
        "object","custom","scene","chapter","part"
    ]

    def __init__(self, theProject, theParent):

        # Internal
        self.theProject = theProject
        self.theParent  = theParent
        self.mainConf   = self.theParent.mainConf

        # Indices
        self.itemIndex  = {}
        self.tagIndex   = {}
        self.keyIndex   = {}
        for aKey in self.VALID_KEYS:
            self.keyIndex[aKey] = []

        return

    def scanFile(self, tHandle):

        theDocument = NWDoc(self.theProject, self.theParent)
        theText = theDocument.openDocument(tHandle, False)
        theItem = self.theProject.getItem(tHandle)

        if theItem is None:
            return False

        if theItem.itemType != nwItemType.FILE:
            return False

        self.itemIndex[tHandle] = {}
        for aKey in self.VALID_KEYS:
            self.itemIndex[tHandle][aKey] = []

        nLine = 0
        for aLine in theText.splitlines():
            aLine  = aLine.strip()
            nLine += 1
            nChar  = len(aLine)
            if nChar > 0 and aLine[0] == "@":
                self.indexThis(tHandle, aLine, nLine, theItem)

        for aKey in self.VALID_KEYS:
            if len(self.itemIndex[tHandle][aKey]) > 0:
                if tHandle not in self.keyIndex[aKey]:
                    self.keyIndex[aKey].append(tHandle)
            else:
                if tHandle in self.keyIndex[aKey]:
                    self.keyIndex[aKey].remove(tHandle)

        print(self.itemIndex)
        print(self.tagIndex)
        print(self.keyIndex)

        return True

    def indexThis(self, tHandle, aLine, nLine, theItem):

        nChar = len(aLine)
        nPos  = aLine.find(":")
        if nPos < 2 or nChar < nPos+2:
            return False

        aKey = aLine[1:nPos].strip().lower()
        tVal = aLine[nPos+1:].strip()
        if aKey not in self.VALID_KEYS:
            return False

        if aKey == "todo":
            self.itemIndex[tHandle]["todo"].append((nLine, tVal))
        elif aKey == "tag":
            if tVal.find(",") >- 0:
                return False
            self.itemIndex[tHandle]["tag"].append((nLine, tVal))
            self.tagIndex[tVal] = (nLine, tHandle, theItem.itemClass)
        else:
            kVal = tVal.split(",")
            cVal = []            
            for aVal in kVal:
                cVal.append(aVal.strip().lower())
            if len(cVal) > 0:
                self.itemIndex[tHandle][aKey].append((nLine, cVal))
            else:
                return False

        return True

# END Class NWIndex