class Node:
    def getDirectories(self):
        return [(sub, os.path.join(self.location, sub)) for sub in os.listdir(self.location)]

    def extractChildren(self):
        if self.isFolder == True:
            childrenLocations = self.getDirectories()
            for child in childrenLocations:
                self.children.append(Node(child))

    def __init__(self):
        self.children = []
        self.isFolder = False
        self.location = ""
        self.name = ""
        self.isOpened = True

    def __init__(self, name_location):
        self.name = name_location[0]
        self.location = name_location[1]
        self.children = []
        self.isFolder = os.path.isdir(self.location)
        self.isOpened = True


def generateHTML(node, level, closedFolders):
    code = ''
    if node.isFolder == False: # leaf/file
        if(node.isFolder == True):
            treeItemIcon = 'treeItemClosedFolder'
        else:
            treeItemIcon = 'treeItemFileIcon'

        code += "<li>\n"
        code += "<div class=\"treeItemFile treeItem\">\n"
        code += "<div class=\"treeItemNoButton\" />\n"
        code += "<div class=\"" + treeItemIcon + "\" />\n"
        code += "<div class=\"treeItemName\">" + node.name + "</div>\n"
        code += "<div class=\"file_folder_id\">" + node.location + "</div>\n"
        code += "</div>\n"
        code += "</li>\n"
        return code

    else: # folder
        if nodeOpened(node, closedFolders) == True:
            treeItemButton = "treeItemCloseButton"
        else:
            treeItemButton = "treeItemOpenButton"

        if nodeOpened(node, closedFolders) == True:
            treeItemIcon = "treeItemOpenedFolder"
        else:
            treeItemIcon = "treeItemClosedFolder"

        code += "<li>\n"
        code += "<div class=\"treeItemFolder treeItem\">\n"
        code += "<div class=\"" + treeItemButton + "\" />\n"
        code += "<div class=\""+ treeItemIcon + "\" />\n"
        code += "<div class=\"treeItemName\">" + node.name + "</div>\n"
        code += "<div class=\"file_folder_id\">" + node.location + "</div>\n"
        code += "</div>\n"

        # for sub list
        if nodeOpened(node, closedFolders) == True:
            code += "<br/><ul>\n"
            node.extractChildren()
            for sub in  node.children:
                code += generateHTML(sub, level+1, closedFolders)

            code += "</ul>\n"
        code += "</li>\n"
        return code

def nodeOpened(node, closedFolders):
    if closedFolders == None:
        return True

    for n in closedFolders:
        if node.location == n:
            return False

    return True

def backend_getProjectTree(projectPath, projectName, closedFolders):
    node = Node((projectName, projectPath))

    htmlCode = generateHTML(node, 0, closedFolders)
    return htmlCode