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

#include "newSectionGroupDialog.h"
#include "ui_newSectionGroupDialog.h"

newSectionGroupDialog::newSectionGroupDialog(QStringList notebookColorList, QStringList notebookDirList, QStringList list, QWidget *parent)
    : QDialog(parent)
    , ui(new Ui::newSectionGroupDialog)
{
    setNotebookColorList(notebookColorList);
    setNotebookDirList(notebookDirList);
    setDirectories(list);
    ui->setupUi(this);
    init();
}

/* Destructor */
newSectionGroupDialog::~newSectionGroupDialog()
{
    delete ui;
}


void newSectionGroupDialog::on_buttonBox_accepted()
{
    if(ui->fileNameBox->text() != "" && ui->directorySelection->currentText() != "") // Cannot have Empty Filename
    {
        QDir dir(getNotebookDirList().at(ui->directorySelection->currentIndex()) + QDir::separator() + ui->fileNameBox->text());
        if(!dir.exists())
        {
            directoryCreator();
            accept();
        } else {
            QMessageBox messageBox;
            messageBox.setWindowTitle("Error");
            messageBox.setText("This Section Group already exists!");
            messageBox.exec();
        }
    }
}

void newSectionGroupDialog::on_buttonBox_rejected()
{
    reject();
}

void newSectionGroupDialog::addItemsToDirectory()
{
    int idx = 0;
    for(idx = 0; idx < getDirectories().size(); idx++)
    {
        if(!getDirectories().at(idx).startsWith("  "))
        {
            ui->directorySelection->addItem(SetNotebookIcon(notebookcolorlist.at(idx)), getDirectories().at(idx));
        } else {
            ui->directorySelection->addItem(SetSectionGroupIcon(notebookcolorlist.at(idx)), getDirectories().at(idx));
        }
    }
}

void newSectionGroupDialog::init()
{
    addItemsToDirectory();
}

void newSectionGroupDialog::directoryCreator()
{
    QString fileName, fullFilePath, fullDirPath, color;
    setSgName(ui->fileNameBox->text()); // File Name without the Extension
    int itemLocation;
    itemLocation = ui->directorySelection->currentIndex();
    fullDirPath = notebookdirList.at(itemLocation);
    fullDirPath += QDir::separator() + sgName;
    setSgDirectory(fullDirPath);
    if(!QDir(sgDirectory).exists())
    {
        QDir().mkdir(fullDirPath);
    }
    setSgColor(ui->sectionGroupColorBox->currentText());
    fileName = "sectiongroup.sectgr";
    fullFilePath = fullDirPath + QDir::separator() + fileName;
    setSgFileName(fullFilePath);
}

QIcon newSectionGroupDialog::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 newSectionGroupDialog::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;
}

/* Getter Functions */
QStringList newSectionGroupDialog::getDirectories() const
{
    return var_directoryList;
}
QStringList newSectionGroupDialog::getNotebookDirList() const
{
    return notebookdirList;
}
QStringList newSectionGroupDialog::getNotebookColorList() const
{
    return notebookcolorlist;
}
QString newSectionGroupDialog::getSgName() const
{
    return sgName;
}
QString newSectionGroupDialog::getSgDirectory() const
{
    return sgDirectory;
}
QString newSectionGroupDialog::getSgFileName() const
{
    return sgFileName;
}
QString newSectionGroupDialog::getSgColor() const
{
    return sgColor;
}


/* Setter Functions */
void newSectionGroupDialog::setDirectories(QStringList dirList)
{
    var_directoryList = dirList;
}
void newSectionGroupDialog::setNotebookDirList(QStringList nbDirList)
{
    notebookdirList = nbDirList;
}
void newSectionGroupDialog::setNotebookColorList(QStringList nbColorList)
{
    notebookcolorlist = nbColorList;
}
void newSectionGroupDialog::setSgName(QString newSgName)
{
    sgName = newSgName;
}
void newSectionGroupDialog::setSgDirectory(QString newSgDirectory)
{
    sgDirectory = newSgDirectory;
}
void newSectionGroupDialog::setSgFileName(QString newSgFileName)
{
    sgFileName = newSgFileName;
}
void newSectionGroupDialog::setSgColor(QString newSgColor)
{
    sgColor = newSgColor;
}