#include <QtGui>
#include <components/esm/esm_reader.hpp>

#include "datafilesdialog.h"
//#include "datafilesmodel.h"
//#include "datafilesitem.h"

using namespace ESM;

DataFilesDialog::DataFilesDialog()
{
    //dataFilesModel = new DataFilesModel(this);
    dataFilesModel = new QStandardItemModel();
    
   
    //dataFilesModel->setReadOnly(true); // Prevent changes to files
    //dataFilesModel->setRootPath("data");

//    sortModel = new QSortFilterProxyModel();
//    sortModel->setSourceModel(dataFilesModel);

    //selectionModel = new QItemSelectionModel(dataFilesModel);
//    selectionModel = new QItemSelectionModel(sortModel);

    /* First, show only plugin files and sort them
    QStringList acceptedfiles = (QStringList() << "*.esp");
    dataFilesModel->setNameFilters(acceptedfiles);
    dataFilesModel->setNameFilterDisables(false); // Hide all other files

    dataFilesModel->sort(3, Qt::AscendingOrder); // Sort the plugins by date
    dataFilesModel->submit(); // Force refresh of the data

    // Now show master files too, to make them appear at the top of the list
    acceptedfiles = (QStringList() << "*.esm" << "*.esp");
    dataFilesModel->setNameFilters(acceptedfiles);
    dataFilesModel->setFilter(QDir::Files);
*/
    // Column 1
    QVBoxLayout *dialogLayout = new QVBoxLayout(this);

    QGroupBox *groupFiles = new QGroupBox(tr("Morrowind Files"), this);
    groupFiles->setMinimumWidth(450);
    QVBoxLayout *groupFilesLayout = new QVBoxLayout(groupFiles);

    //QSpacerItem *vSpacer1 = new QSpacerItem(20, 2, QSizePolicy::Minimum, QSizePolicy::Fixed);

    /*QHBoxLayout *filterLayout = new QHBoxLayout();
    QLabel *labelFilter = new QLabel(tr("Filter:"), groupFiles);
    lineFilter = new LineEdit(groupFiles);

    filterLayout->addWidget(labelFilter);
    filterLayout->addWidget(lineFilter);
    */
    
    // View for the game files
    //dataFilesView = new QTreeView(groupFiles);
    dataFilesView = new QTreeView(groupFiles);
    dataFilesView->setModel(dataFilesModel);
    dataFilesView->setDragEnabled(true);
    dataFilesView->setDropIndicatorShown(true);
    dataFilesView->setAlternatingRowColors(true);
    
    
    // Put everything in the correct layouts
    //groupFilesLayout->addLayout(filterLayout);
    //groupFilesLayout->addItem(vSpacer1);
    groupFilesLayout->addWidget(dataFilesView);

    QHBoxLayout *buttonsLayout = new QHBoxLayout();
    QSpacerItem *horizontalSpacer = new QSpacerItem(40, 20, QSizePolicy::Expanding, QSizePolicy::Minimum);

    QDialogButtonBox *buttonBox = new QDialogButtonBox();
    buttonBox->setStandardButtons(QDialogButtonBox::Cancel|QDialogButtonBox::Ok|QDialogButtonBox::RestoreDefaults);
    buttonBox->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);

    buttonsLayout->addItem(horizontalSpacer);
    buttonsLayout->addWidget(buttonBox);
    
    dialogLayout->addWidget(groupFiles);
    dialogLayout->addLayout(buttonsLayout);

    setWindowTitle(tr("Data Files"));

    // Signals and slots
    /*connect(dataFilesModel, SIGNAL(directoryLoaded(const QString &)), this, SLOT(setupView()));

    connect(dataFilesView, SIGNAL(doubleClicked(QModelIndex)), this, SLOT(setCheckstate(QModelIndex)));
    connect(selectionModel, SIGNAL(currentChanged(QModelIndex, QModelIndex)), this, SLOT(changeData(QModelIndex, QModelIndex)));

    connect(lineFilter, SIGNAL(textChanged(const QString&)), this, SLOT(setFilter()));

    connect(buttonBox->button(QDialogButtonBox::RestoreDefaults), SIGNAL(clicked()), this, SLOT(restoreDefaults()));
    connect(buttonBox, SIGNAL(accepted()), this, SLOT(writeConfig()));
    connect(buttonBox, SIGNAL(rejected()), this, SLOT(close()));
*/
    //readConfig();
    //setupView();
    setupView();
}

void DataFilesDialog::changeData(QModelIndex index, QModelIndex bottom)
{
    /*if (!index.isValid()) {
        return;
    }

    // Added for testing

    textDepends->clear(); // Clear the dependencies of previous file

    ESMReader datafile;
    QString path(dataFilesModel->filePath(index));
//    QString path(dataFilesModel->filePath(sortModel->mapToSource(index)));

    datafile.open(path.toStdString()); // Open the selected file

    // Get the author of the selected file and display it
    QString author(QString::fromStdString(datafile.getAuthor()));
    lineAuthor->setText(author);

    // Get the file desciption
    QString desc(QString::fromStdString(datafile.getDesc()));
    textDesc->setPlainText(desc);

    // Get a list of master files on which the file depends
    ESMReader::MasterList mlist = datafile.getMasters();

    for (unsigned int i = 0; i < mlist.size(); ++i) // Add each master file
        textDepends->appendPlainText(QString::fromStdString(mlist[i].name));

    /* Get the date of creation
    QDateTime dateCreated = dataFilesModel->fileInfo(index).created();
    QString created = dateCreated.toString(QString("dd.MM.yyyy"));
    labelDateCreated->setText(created);

    // Get the date last modified
    QDateTime dateModified = dataFilesModel->fileInfo(index).lastModified();
    QString modified = dateModified.toString(QString("dd.MM.yyyy"));
    labelDateModified->setText(modified);*/
}

void DataFilesDialog::setupView()
{
    QDir datadir(QString("data/"));
    QStringList acceptedfiles = (QStringList() << "*.esp");
    QStringList datafiles;
    
    datadir.setNameFilters(acceptedfiles);
    
    datafiles = datadir.entryList();
    
    QStandardItem *parentItem = dataFilesModel->invisibleRootItem();
    QStandardItem *masterFile = new QStandardItem(QString("Morrowind.esm"));
    parentItem->appendRow(masterFile);
    parentItem = masterFile;
    
    QFileIconProvider fip;
    QIcon fileIcon = fip.icon(QFileInfo("data/Morrowind.esm"));
    
    foreach (const QString &currentfile, datafiles) {
        QStandardItem *item = new QStandardItem(currentfile);
        item->setIcon(fileIcon);
        parentItem->appendRow(item);
    }
    
    
    
    /* The signal directoryLoaded is emitted after all files are in the model
    dataFilesView->setModel(dataFilesModel);
//    dataFilesView->setModel(sortModel);

    // Set the view to the data directory
    dataFilesView->setRootIndex(QModelIndex(dataFilesModel->index("data")));
//    dataFilesView->setRootIndex(sortModel->mapFromSource(QModelIndex(dataFilesModel->index("/opt/openmw/data"))));

    dataFilesView->verticalHeader()->hide();

    //dataFilesView->hideColumn(1);
    dataFilesView->hideColumn(3); // Hide Date Modified column
    dataFilesView->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
    dataFilesView->verticalHeader()->setDefaultSectionSize(25); //setHeight

    dataFilesView->setSortingEnabled(true);

    dataFilesView->setSelectionBehavior(QAbstractItemView::SelectRows);
    dataFilesView->setSelectionModel(selectionModel);


    dataFilesView->setAlternatingRowColors(true); // Fancy colors!

    dataFilesView->resizeColumnsToContents();
    dataFilesView->horizontalHeader()->setStretchLastSection(true);


    //sortModel->setSortCaseSensitivity(Qt::CaseInsensitive);*/

}

void DataFilesDialog::readConfig()
{    
/*    QString filename;
    QString path = "data/"; // TODO: Should be global
    
    QFile file("openmw.cfg"); // Specify filepath later
    if (!file.open(QIODevice::ReadOnly | QIODevice::Text)) {
        qDebug() << "error open";
        close(); // File cannot be opened or created TODO: throw error
    }
    
    QTextStream in(&file);

    QStringList datafiles;

    // Add each data file read from the config file to a QStringList
    while (!in.atEnd()) {
        QString line = in.readLine();
                
        if (line.contains("master")) {
            filename = line.remove("master=");
            filename.prepend(path);
            
            datafiles << filename << "\n";
            
        } else if (line.contains("plugin")) {
            filename = line.remove("plugin=");
            filename.prepend(path);
            
            datafiles << filename << "\n";
        }
    }

    file.close();
    
    // Check if the files are in the model, set to checked if found
    foreach(const QString &currentfile, datafiles) {
        QModelIndex index = dataFilesModel->index(currentfile, 0);
        
        if (index.isValid()) {
            // File is found in model, set it to checked
            dataFilesModel->setData(index, Qt::Checked, Qt::CheckStateRole);
        }
    }*/
}

void DataFilesDialog::writeConfig()
{
    /* Custom write method: We cannot use QSettings because it does not accept spaces

//    QList<QPersistentModelIndex> checkeditems = dataFilesModel->getCheckedItems().toList();
    QStringList checkeditems = dataFilesModel->getCheckedItems();
    //QString sectionname = "[Game Files]";
    QString filename;
    QFileInfo datafile;
    
    // Sort the items so that master files end up on top
    foreach (const QString &currentitem, checkeditems) {
        if(currentitem.endsWith(QString(".esm"), Qt::CaseInsensitive)) {
            checkeditems.move(checkeditems.indexOf(currentitem), 0);
        }
    }
    
    QFile file("openmw.cfg"); // Specify filepath later
    if (!file.open(QIODevice::ReadOnly | QIODevice::Text))
        close(); // File cannot be opened or created TODO: throw error

    //QTextStream in(&file);
    QTextStream in(&file);

    //QString buffer;
    QByteArray buffer;

    // Remove all previous master/plugin entries from config
    while (!in.atEnd()) {
        QString line = in.readLine();
        //if (!line.contains("GameFile") && line != "[Game Files]") {
        if (!line.contains("master") && !line.contains("plugin")) {
            buffer += line += "\n"; 
        }
    }

    file.close();

    // Now we write back the other config entries
    if (!file.open(QIODevice::WriteOnly | QIODevice::Text | QIODevice::Truncate))
        close(); // File cannot be opened or created TODO: throw error

    file.write(buffer);
    QTextStream out(&file);
    
    // Write the section name to the config file before we write out the data files
    //out << sectionname << endl;
    
    // Write the list of game files to the config
    for (int i = 0; i < checkeditems.size(); ++i) {
        //filename = dataFilesModel->fileName(checkeditems.at(i));
        filename = checkeditems.at(i);
        datafile = QFileInfo(filename);
        
        if (datafile.exists()) {
            if (filename.endsWith(QString(".esm"), Qt::CaseInsensitive)) {
                out << "master=" << datafile.fileName() << endl;
            } else if (filename.endsWith(QString(".esp"), Qt::CaseInsensitive)) {
                out << "plugin=" << datafile.fileName() << endl;
            }
        }
    }


    file.close();
    close(); // Exit dialog*/
}

void DataFilesDialog::restoreDefaults()
{
/*    // Uncheck all checked items
    dataFilesModel->checkedItems.clear();

    QModelIndexList indexlist; // Make a list of default master files
    indexlist.append(dataFilesModel->index("data/Morrowind.esm", 0));
    indexlist.append(dataFilesModel->index("data/Tribunal.esm", 0));
    indexlist.append(dataFilesModel->index("data/Bloodmoon.esm", 0));

    foreach (const QModelIndex &index, indexlist) {
        if (index.isValid()) {
            // Master file found, check it
            dataFilesModel->setData(index, Qt::Checked, Qt::CheckStateRole);
        }
    }
    dataFilesModel->submit(); // Force refresh of view*/
}

void DataFilesDialog::setCheckstate(QModelIndex index)
{
    /* No check if index is valid: doubleclicked() always returns
    // a valid index when emitted

    //index = QModelIndex(sortModel->mapToSource(index)); // Get a valid index
    index = index.sibling(index.row(), 0); // reset index to first column
    // because that's where te checkbox is; makes it possible to doubleclick whole row

    if (!index.isValid())
        return;

    if (dataFilesModel->data(index, Qt::CheckStateRole) == Qt::Checked) {
        // Selected row is checked, uncheck it
        dataFilesModel->setData(index, Qt::Unchecked, Qt::CheckStateRole);
    } else {
       dataFilesModel->setData(index, Qt::Checked, Qt::CheckStateRole);
    }*/
}

void DataFilesDialog::setFilter()
{
    /*
    QStringList filefilter = (QStringList() << "*.esm" << "*.esp");
    QStringList currentfilefilter;

    QString esmfilter = lineFilter->text();
    QString espfilter = lineFilter->text();

    if (lineFilter->text().isEmpty()) {
        dataFilesModel->setNameFilters(filefilter);
    //     sortModel->setFilterRegExp(QRegExp("*.esp", Qt::CaseInsensitive,
  //                                           QRegExp::FixedString));
      //   sortModel->setFilterKeyColumn(0);
        return;
    }

    esmfilter.prepend("*");
    esmfilter.append("*.esm");
    espfilter.prepend("*");
    espfilter.append("*.esp");

    currentfilefilter << esmfilter << espfilter;
//    sortModel->setFilterRegExp(QRegExp(espfilter, Qt::CaseInsensitive,
    //                                         QRegExp::FixedString));
  //  sortModel->setFilterKeyColumn(0);
    dataFilesModel->setNameFilters(currentfilefilter);

//    readConfig();
//    dataFilesModel->submit();
//    dataFilesModel->setData();
*/
}