/*
Copyright 2011 Simone Tobia

This file is part of AppSet.

AppSet is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

AppSet is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with AppSet; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

#include "repoeditor.h"
#include "ui_repoeditor.h"

#include "repoconf.h"
#include "checkboxdelegate.h"
#include "optionsdelegate.h"
#include "../src/settingsmanager.h"

#include <QFileDialog>
#include <QMessageBox>
#include <QProcess>
#include <QDateTime>
#include <QCloseEvent>

RepoEditor::RepoEditor( QWidget *parent )
  : QDialog( parent )
  , ui( new Ui::RepoEditor )
{
  ui->setupUi( this );

  repoConf = new RepoConf();
  addRepoDialog = new AddRepo( this );

  ui->tableView->setModel( repoConf );
  ui->tableView->setItemDelegateForColumn( 0, new CheckBoxDelegate( this) );
  ui->tableView->setItemDelegateForColumn( 2, new OptionsDelegate( this) );

  ui->tableView->setColumnWidth( 1, 133 );

  QString dateFormat = QLocale().dateFormat(QLocale::NarrowFormat);
  dateFormat = dateFormat.remove(QStringLiteral("/"));
  QDateTime dt=QDateTime::currentDateTime();
  ui->backupFile->setText( repoConf->getConfPath() + QLatin1String(".bak.") + dt.toString(dateFormat+QLatin1String("_HHmmss")) );

  connect( ui->moveUp, SIGNAL( clicked() ),
           SLOT( moveUp() ) );

  connect( ui->moveDown, SIGNAL( clicked() ),
           SLOT( moveDown() ) );

  QItemSelectionModel *selModel = ui->tableView->selectionModel();
  connect( selModel, SIGNAL( selectionChanged( QItemSelection,QItemSelection ) ),
           SLOT( updateMovers( QItemSelection,QItemSelection ) ) );

  connect( ui->buttonBox->button( QDialogButtonBox::Reset ), SIGNAL( clicked() ),
           repoConf, SLOT( reload() ) );

  connect( ui->buttonBox->button( QDialogButtonBox::Save ), SIGNAL( clicked() ),
           SLOT( apply() ) );

  connect( ui->buttonBox->button( QDialogButtonBox::Discard ), SIGNAL( clicked() ),
           SLOT( discard() ) );

  connect( ui->remove, SIGNAL( clicked() ),
           SLOT( removeEntry() ) );

  connect( ui->add, SIGNAL( clicked() ),
           SLOT( addEntry() ) );

  connect( ui->edit, SIGNAL( clicked() ),
           SLOT( editEntry() ) );

  connect( ui->loadBackup, SIGNAL( clicked() ),
           SLOT( loadBackup() ) );

  ui->tableView->selectRow(0);
  ui->loadBackup->setVisible(false);
  restoreGeometry(SettingsManager::getRepoEditorWindowSize());
}

RepoEditor::~RepoEditor()
{
  QByteArray windowSize=saveGeometry();
  SettingsManager::setRepoEditorWindowSize(windowSize);

  delete ui;
}

void RepoEditor::closeEvent(QCloseEvent *event)
{
  if (repoConf->hasAnyChanges())
  {
    int res = QMessageBox::question(this, tr("Confirmation"),
                                    tr("There are unsaved changes.") + QLatin1Char('\n') +
                                    tr("Do you want to save them?"),
                                    QMessageBox::Yes | QMessageBox::No,
                                    QMessageBox::No);
    if (res == QMessageBox::Yes)
    {
      apply();
    }
  }

  event->accept();
  qApp->quit();
}

/*
 * Whenever user tries to reject dialog...
 */
void RepoEditor::reject()
{
  if (repoConf->hasAnyChanges())
  {
    int res = QMessageBox::question(this, tr("Confirmation"),
                                    tr("There are unsaved changes.") + QLatin1Char('\n') +
                                    tr("Do you want to save them?"),
                                    QMessageBox::Yes | QMessageBox::No,
                                    QMessageBox::No);
    if (res == QMessageBox::Yes)
    {
      apply();
    }
  }

  QDialog::reject();
}

void RepoEditor::loadBackup()
{
  RepoConf conf;
  QFileInfo fi;
  QString file;

  if (!ui->backupFile->text().isEmpty())
  {
    fi.setFile(ui->backupFile->text());
    file = QFileDialog::getOpenFileName( this, QStringLiteral("Open file"), fi.path() );
  }
  else
    file = QFileDialog::getOpenFileName( this );

  if( file.isEmpty() )
    return;

  conf.loadConf( file );

  if( !conf.count() ) {
    QMessageBox mb(QMessageBox::Critical,
                   tr( "Can't load backup file" ),
                   tr( "Selected file is not valid" ),
                   QMessageBox::Ok,
                   this);
    mb.exec();
  } else {
    repoConf->loadConf( file );
    ui->backupFile->setText( repoConf->getConfPath() + QLatin1String(".bak") );
  }
}

void RepoEditor::addEntry()
{
  addRepoDialog->setRepoName(QLatin1String(""));
  addRepoDialog->setRepoLocation(QLatin1String(""));
  if( addRepoDialog->exec() == QDialog::Accepted ) {
    if( !repoConf->exists( addRepoDialog->entry.getName() ) )
      repoConf->addEntry( addRepoDialog->entry );
  }
}

void RepoEditor::editEntry()
{
  QModelIndex repoMI = ui->tableView->model()->index(ui->tableView->currentIndex().row(), 1, QModelIndex());
  QModelIndex locationMI = ui->tableView->model()->index(ui->tableView->currentIndex().row(), 2, QModelIndex());
  addRepoDialog->setRepoName(ui->tableView->model()->data(repoMI).toString());

  // take the location
  QString location = ui->tableView->model()->data(locationMI).toString();

  // take the type and address
  QRegExp locationMatch(QLatin1String("^(Server|Include)\\s*=\\s*(.+)"));
  locationMatch.indexIn(location);
  location = locationMatch.cap(2);

  // fill remaining fields
  addRepoDialog->setRepoLocation(location);
  addRepoDialog->setLocationType(locationMatch.cap(1) == QLatin1String("Server") ? 1 : 0);

  if( addRepoDialog->exec() == QDialog::Accepted ) {
    ui->tableView->model()->setData( repoMI, addRepoDialog->getRepoName() );
    const QLatin1String tmp = addRepoDialog->getLocationType() == 0 ? QLatin1String("Include = ") : QLatin1String("Server = ");
    ui->tableView->model()->setData( locationMI, QVariant(tmp + addRepoDialog->getRepoLocation()) );
  }
}

void RepoEditor::apply()
{
  if( repoConf->saveChanges( ui->checkBox->isChecked() ? ui->backupFile->text() : QLatin1String("") ) ) {
    QMessageBox::information( this,
                              tr( "Success" ),
                              tr( "Repositories configuration successfully saved." ),
                              QMessageBox::Ok );
    qApp->quit();
  } else {
    QMessageBox::critical( this,
                           tr( "Error" ),
                           tr( "Repositories configuration not saved." ),
                           QMessageBox::Ok );
  }
}

void RepoEditor::discard()
{
  done( QDialog::Rejected );
}

void RepoEditor::updateMovers( const QItemSelection &cur, const QItemSelection &)
{
  QModelIndexList list = cur.indexes();

  if( !list.count() )
    return;

  int row = list.at( 0 ).row();

  if( row > 0 )
    ui->moveUp->setEnabled( true );
  else
    ui->moveUp->setEnabled( false );

  if( row < ui->tableView->model()->rowCount() - 1)
    ui->moveDown->setEnabled( true );
  else
    ui->moveDown->setEnabled( false );
}

void RepoEditor::moveUp()
{
  QModelIndexList list = ui->tableView->selectionModel()->selectedRows();

  if( !list.count() )
    return;

  int row = list.at( 0 ).row();
  repoConf->moveUp( row );
  ui->tableView->selectRow( row - 1 );
}

void RepoEditor::moveDown()
{
  QModelIndexList list = ui->tableView->selectionModel()->selectedRows();

  if( !list.count() )
    return;

  int row = list.at( 0 ).row();
  repoConf->moveDown( row );
  ui->tableView->selectRow( row + 1 );
}

void RepoEditor::removeEntry()
{
  QModelIndexList list = ui->tableView->selectionModel()->selectedRows();

  if( !list.count() )
    return;

  int row = list.at( 0 ).row();
  repoConf->removeRow( row );
}
