/* DrewTechs
 * Note Binder
 * @Version 1.0
 */

#include "notebooklist.h"

NotebookList::NotebookList(QObject *parent)
    : QObject{parent}
{

}

void NotebookList::addNotebookToList(Notebook* notebook)
{
    nbList.append(notebook);
}

QTreeWidgetItem* NotebookList::createSectionIcon(QFileInfo fileInfo)
{
    QTreeWidgetItem* sectionNameItem = new QTreeWidgetItem();
    sectionNameItem->setIcon(0, QIcon("../NoteBinder/Icons/SectionIcon.png"));
    sectionNameItem->setText(0, fileInfo.fileName());
    sectionNameItem->setText(1, fileInfo.absoluteFilePath());
    sectionNameItem->setText(2, getFileSize(fileInfo.size()));
    return sectionNameItem;
}
QTreeWidgetItem* NotebookList::createSectionGroupIcon(SectionGroup* sectionGroup)
{
    QTreeWidgetItem* sectionGroupNameItem = new QTreeWidgetItem();
    sectionGroupNameItem->setIcon(0, SetSectionGroupIcon(sectionGroup->color));
    sectionGroupNameItem->setText(0, sectionGroup->name);
    sectionGroupNameItem->setText(1, sectionGroup->location);
    sectionGroupNameItem->setText(2, getDirectorySize(sectionGroup->location));
    return sectionGroupNameItem;
}
QTreeWidgetItem* NotebookList::createNotebookIcon(Notebook* notebook)
{
    QTreeWidgetItem* notebookNameItem = new QTreeWidgetItem();
    notebookNameItem->setIcon(0, SetNotebookIcon(notebook->color));
    notebookNameItem->setText(0, notebook->name);
    notebookNameItem->setText(1, notebook->location);
    notebookNameItem->setText(2, getDirectorySize(notebook->location));
    return notebookNameItem;
}

QIcon NotebookList::SetNotebookIcon(QString notebookColor)
{
    QIcon icon;
    QString iconPath = "../NoteBinder/Icons";
    if(!QDir(iconPath).exists())
    {
        return icon;
    }
    if(notebookColor == "Red") {
        icon = QIcon(iconPath + "/NotebookIcon_Red.png");
    } else if(notebookColor == "Blue") {
        icon = QIcon(iconPath + "/NotebookIcon_Blue.png");
    } else if(notebookColor == "Purple") {
        icon = QIcon(iconPath + "/NotebookIcon_Purple.png");
    } else if(notebookColor == "Orange") {
        icon = QIcon(iconPath + "/NotebookIcon_Orange.png");
    } else if(notebookColor == "Yellow") {
        icon = QIcon(iconPath + "/NotebookIcon_Yellow.png");
    } else if(notebookColor == "Green") {
        icon = QIcon(iconPath + "/NotebookIcon_Green.png");
    } else if(notebookColor == "Cyan") {
        icon = QIcon(iconPath + "/NotebookIcon_Cyan.png");
    } else if(notebookColor == "Tan") {
        icon = QIcon(iconPath + "/NotebookIcon_Tan.png");
    } else if(notebookColor == "Teal") {
        icon = QIcon(iconPath + "/NotebookIcon_Teal.png");
    } else if(notebookColor == "Red Chalk") {
        icon = QIcon(iconPath + "/NotebookIcon_RedChalk.png");
    } else if(notebookColor == "Blue Mist") {
        icon = QIcon(iconPath + "/NotebookIcon_BlueMist.png");
    } else if(notebookColor == "Purple Mist") {
        icon = QIcon(iconPath + "/NotebookIcon_PurpleMist.png");
    } else if(notebookColor == "Magenta") {
        icon = QIcon(iconPath + "/NotebookIcon_Magenta.png");
    } else if(notebookColor == "Lemon Lime") {
        icon = QIcon(iconPath + "/NotebookIcon_LemonLime.png");
    } else if(notebookColor == "Apple") {
        icon = QIcon(iconPath + "/NotebookIcon_Apple.png");
    } else if(notebookColor == "Silver") {
        icon = QIcon(iconPath + "/NotebookIcon_Silver.png");
    } else if(notebookColor == "Black") {
        icon = QIcon(iconPath + "/NotebookIcon_Black.png");
    } else {
        // If it's either White/None or an invalid value, set to the white icon
        icon = QIcon(iconPath + "/NotebookIcon_White.png");
    }
    return icon;
}

QIcon NotebookList::SetSectionGroupIcon(QString sectionGroupColor)
{
    QIcon icon;
    QString iconPath = "../NoteBinder/Icons";
    if(!QDir(iconPath).exists())
    {
        return icon;
    }
    if(sectionGroupColor == "Red") {
        icon = QIcon(iconPath + "/SectionGroupIcon_Red.png");
    } else if(sectionGroupColor == "Blue") {
        icon = QIcon(iconPath + "/SectionGroupIcon_Blue.png");
    } else if(sectionGroupColor == "Purple") {
        icon = QIcon(iconPath + "/SectionGroupIcon_Purple.png");
    } else if(sectionGroupColor == "Orange") {
        icon = QIcon(iconPath + "/SectionGroupIcon_Orange.png");
    } else if(sectionGroupColor == "Yellow") {
        icon = QIcon(iconPath + "/SectionGroupIcon_Yellow.png");
    } else if(sectionGroupColor == "Green") {
        icon = QIcon(iconPath + "/SectionGroupIcon_Green.png");
    } else if(sectionGroupColor == "Cyan") {
        icon = QIcon(iconPath + "/SectionGroupIcon_Cyan.png");
    } else if(sectionGroupColor == "Tan") {
        icon = QIcon(iconPath + "/SectionGroupIcon_Tan.png");
    } else if(sectionGroupColor == "Teal") {
        icon = QIcon(iconPath + "/SectionGroupIcon_Teal.png");
    } else if(sectionGroupColor == "Red Chalk") {
        icon = QIcon(iconPath + "/SectionGroupIcon_RedChalk.png");
    } else if(sectionGroupColor == "Blue Mist") {
        icon = QIcon(iconPath + "/SectionGroupIcon_BlueMist.png");
    } else if(sectionGroupColor == "Purple Mist") {
        icon = QIcon(iconPath + "/SectionGroupIcon_PurpleMist.png");
    } else if(sectionGroupColor == "Magenta") {
        icon = QIcon(iconPath + "/SectionGroupIcon_Magenta.png");
    } else if(sectionGroupColor == "Lemon Lime") {
        icon = QIcon(iconPath + "/SectionGroupIcon_LemonLime.png");
    } else if(sectionGroupColor == "Apple") {
        icon = QIcon(iconPath + "/SectionGroupIcon_Apple.png");
    } else if(sectionGroupColor == "Silver") {
        icon = QIcon(iconPath + "/SectionGroupIcon_Silver.png");
    } else if(sectionGroupColor == "Black") {
        icon = QIcon(iconPath + "/SectionGroupIcon_Black.png");
    } else {
        // If it's either White/None or an invalid value, set to the white icon
        icon = QIcon(iconPath + "/SectionGroupIcon_White.png");
    }
    return icon;
}