#include "svndeletedialog.h"

// Tnz6 includes

#include "tapp.h"

#include "tsystem.h"

#include "tfilepath.h"

#include "filebrowser.h"

#include "fileselection.h"

// TnzQt includes

#include "toonzqt/tselectionhandle.h"

#include "toonzqt/gutil.h"

// TnzLib includes

#include "toonz/txshsimplelevel.h"

// Qt includes

#include <qwidget></qwidget>

#include <qlabel></qlabel>

#include <qmovie></qmovie>

#include <qcheckbox></qcheckbox>

#include <qplaintextedit></qplaintextedit>

#include <qpushbutton></qpushbutton>

#include <qboxlayout></qboxlayout>

#include <qformlayout></qformlayout>

#include <qtreewidget></qtreewidget>

#include <qheaderview></qheaderview>

#include <qtextstream></qtextstream>

#include <qdir></qdir>

#include <qmainwindow></qmainwindow>

//=============================================================================

// SVNDeleteDialog

//-----------------------------------------------------------------------------

SVNDeleteDialog::SVNDeleteDialog(QWidget *parent, const QString &workingDir,

                                 const QStringList &files, bool isFolder,

                                 int sceneIconAdded)

    : Dialog(TApp::instance()->getMainWindow(), true, false)

    , m_deleteSceneContentsCheckBox(0)

    , m_workingDir(workingDir)

    , m_files(files)

    , m_targetTempFile(0)

    , m_isFolder(isFolder)

    , m_sceneIconAdded(sceneIconAdded) {

  setModal(false);

  setAttribute(Qt::WA_DeleteOnClose, true);

  setWindowTitle(tr("Version Control: Delete"));

  setMinimumSize(300, 180);

  QWidget *container = new QWidget;

  QVBoxLayout *mainLayout = new QVBoxLayout;

  mainLayout->setAlignment(Qt::AlignHCenter);

  mainLayout->setMargin(0);

  QHBoxLayout *hLayout = new QHBoxLayout;

  m_waitingLabel      = new QLabel;

  QMovie *waitingMove = new QMovie(":Resources/waiting.gif");

  waitingMove->setParent(this);

  m_waitingLabel->hide();

  m_waitingLabel->setMovie(waitingMove);

  waitingMove->setCacheMode(QMovie::CacheAll);

  waitingMove->start();

  m_textLabel = new QLabel;

  if (m_isFolder) {

    QDir dir(m_workingDir + "/" + m_files.at(0));

    int filesCount = dir.entryList(QDir::Files | QDir::Dirs).count();

    if (filesCount > 0)

      m_textLabel->setText(tr("Delete folder that contains %1 items.")

                               .arg(filesCount - m_sceneIconAdded));

    else

      m_textLabel->setText(tr("Delete empty folder."));

  } else

    m_textLabel->setText(

        tr("Delete %1 items.").arg(m_files.size() - m_sceneIconAdded));

  hLayout->addStretch();

  hLayout->addWidget(m_waitingLabel);

  hLayout->addWidget(m_textLabel);

  hLayout->addStretch();

  mainLayout->addLayout(hLayout);

  m_treeWidget = new QTreeWidget;

  m_treeWidget->setStyleSheet("QTreeWidget { border: 1px solid gray; }");

  m_treeWidget->header()->hide();

  m_treeWidget->hide();

  mainLayout->addWidget(m_treeWidget);

  QFormLayout *formLayout = new QFormLayout;

  formLayout->setLabelAlignment(Qt::AlignRight);

  formLayout->setFormAlignment(Qt::AlignHCenter | Qt::AlignTop);

  formLayout->setSpacing(10);

  formLayout->setMargin(0);

  m_commentTextEdit = new QPlainTextEdit;

  m_commentTextEdit->setMaximumHeight(50);

  m_commentLabel = new QLabel(tr("Comment:"));

  m_commentLabel->setFixedWidth(55);

  m_commentTextEdit->hide();

  m_commentLabel->hide();

  formLayout->addRow(m_commentLabel, m_commentTextEdit);

  if (!m_isFolder) {

    m_deleteSceneContentsCheckBox = new QCheckBox(this);

    connect(m_deleteSceneContentsCheckBox, SIGNAL(toggled(bool)), this,

            SLOT(onDeleteSceneContentsToggled(bool)));

    m_deleteSceneContentsCheckBox->setChecked(false);

    m_deleteSceneContentsCheckBox->hide();

    m_deleteSceneContentsCheckBox->setText(tr("Delete Scene Contents"));

    formLayout->addRow("", m_deleteSceneContentsCheckBox);

    int fileSize = m_files.size();

    for (int i = 0; i < fileSize; i++) {

      if (m_files.at(i).endsWith(".tnz")) {

        m_deleteSceneContentsCheckBox->show();

        break;

      }

    }

  }

  m_keepLocalCopyCheckBox = new QCheckBox(tr(" Keep Local Copy"));

  m_keepLocalCopyCheckBox->setChecked(true);

  m_keepLocalCopyCheckBox->hide();

  formLayout->addRow("", m_keepLocalCopyCheckBox);

  mainLayout->addLayout(formLayout);

  container->setLayout(mainLayout);

  beginHLayout();

  addWidget(container, false);

  endHLayout();

  m_deleteLocalButton = new QPushButton();

  m_deleteLocalButton->setText(tr("Delete Local Copy "));

  connect(m_deleteLocalButton, SIGNAL(clicked()), this,

          SLOT(onDeleteLocalButtonClicked()));

  m_deleteServerButton = new QPushButton();

  m_deleteServerButton->setText(tr("Delete on Server "));

  connect(m_deleteServerButton, SIGNAL(clicked()), this,

          SLOT(onDeleteServerButtonClicked()));

  m_cancelButton = new QPushButton(tr("Cancel"));

  connect(m_cancelButton, SIGNAL(clicked()), this, SLOT(reject()));

  addButtonBarWidget(m_deleteLocalButton, m_deleteServerButton, m_cancelButton);

  // 0. Connect for svn errors (that may occur)

  connect(&m_thread, SIGNAL(error(const QString &)), this,

          SLOT(onError(const QString &)));

}

//-----------------------------------------------------------------------------

void SVNDeleteDialog::showEvent(QShowEvent *e) {

  if (!m_isFolder) {

    initTreeWidget();

    int height = 50;

    if (m_treeWidget->isVisible()) height += (m_files.size() * 50);

    setMinimumSize(300, std::min(height, 350));

  }

  QDialog::showEvent(e);

}

//-----------------------------------------------------------------------------

void SVNDeleteDialog::onError(const QString &errorString) {

  if (m_targetTempFile) {

    QFile::remove(m_targetTempFile->fileName());

    delete m_targetTempFile;

    m_targetTempFile = 0;

  }

  m_textLabel->setText(errorString);

  switchToCloseButton();

  update();

}

//-----------------------------------------------------------------------------

void SVNDeleteDialog::switchToCloseButton() {

  if (m_deleteSceneContentsCheckBox) m_deleteSceneContentsCheckBox->hide();

  m_commentTextEdit->hide();

  m_commentLabel->hide();

  m_waitingLabel->hide();

  m_treeWidget->hide();

  m_keepLocalCopyCheckBox->hide();

  m_deleteLocalButton->disconnect();

  m_deleteLocalButton->setText("Close");

  m_deleteLocalButton->show();

  m_cancelButton->hide();

  m_deleteServerButton->hide();

  connect(m_deleteLocalButton, SIGNAL(clicked()), this, SLOT(close()));

}

//-----------------------------------------------------------------------------

void SVNDeleteDialog::onDeleteLocalButtonClicked() {

  // Delete files locally

  int fileCount = m_files.size();

  m_waitingLabel->show();

  m_textLabel->setText(

      tr("Deleting %1 items...")

          .arg(fileCount + m_sceneResources.size() - m_sceneIconAdded));

  m_textLabel->show();

  for (int i = 0; i < fileCount; i++) {

    TFilePath fp((m_workingDir + "/" + m_files.at(i)).toStdWString());

    TSystem::moveFileOrLevelToRecycleBin(fp);

  }

  int sceneResourcesCount = m_sceneResources.size();

  for (int i = 0; i < sceneResourcesCount; i++) {

    TFilePath fp(m_sceneResources.at(i).toStdWString());

    TSystem::moveFileOrLevelToRecycleBin(fp);

  }

  updateFileBrowser();

  emit done(m_files);

  m_waitingLabel->hide();

  m_textLabel->hide();

  close();

}

//-----------------------------------------------------------------------------

void SVNDeleteDialog::updateFileBrowser() {

  // Reset the selection

  TSelection *selection =

      TApp::instance()->getCurrentSelection()->getSelection();

  FileSelection *fileSelection = dynamic_cast<fileselection *="">(selection);</fileselection>

  if (fileSelection) fileSelection->selectNone();

  TFilePath fp((m_workingDir + "/" + m_files.at(0)).toStdWString());

  // Refresh FileBrowser

  FileBrowser::refreshFolder(fp.getParentDir());

}

//-----------------------------------------------------------------------------

void SVNDeleteDialog::onDeleteServerButtonClicked() {

  m_deleteLocalButton->hide();

  disconnect(m_deleteServerButton, SIGNAL(clicked()), this,

             SLOT(onDeleteServerButtonClicked()));

  m_deleteServerButton->setText(tr("Delete"));

  connect(m_deleteServerButton, SIGNAL(clicked()), this, SLOT(deleteFiles()));

  int height = 175;

  if (m_treeWidget->isVisible()) height += (m_files.size() * 50);

  setMinimumSize(300, std::min(height, 350));

  m_textLabel->setText(

      tr("You are deleting items also on repository. Are you sure ?"));

  m_commentLabel->show();

  m_commentTextEdit->show();

  m_keepLocalCopyCheckBox->show();

}

//-----------------------------------------------------------------------------

void SVNDeleteDialog::deleteFiles() {

  m_waitingLabel->show();

  m_textLabel->setText(

      tr("Deleting %1 items...")

          .arg(QString::number(m_files.size() + m_sceneResources.size() -

                               m_sceneIconAdded)));

  m_commentLabel->hide();

  m_commentTextEdit->hide();

  m_treeWidget->hide();

  m_keepLocalCopyCheckBox->hide();

  if (m_deleteSceneContentsCheckBox) m_deleteSceneContentsCheckBox->hide();

  m_deleteServerButton->hide();

  m_cancelButton->hide();

  QStringList args;

  args << "delete";

  if (m_keepLocalCopyCheckBox->isChecked()) args << "--keep-local";

  // Use a temporary file to store all the files list

  m_targetTempFile = new QFile(m_workingDir + "/" + "tempDeleteFile");

  if (m_targetTempFile->open(QFile::WriteOnly | QFile::Truncate)) {

    QTextStream out(m_targetTempFile);

    int filesCount = m_files.count();

    for (int i = 0; i < filesCount; i++) out << m_files.at(i) + "\n";

    int sceneResourcesCount = m_sceneResources.size();

    for (int i = 0; i < sceneResourcesCount; i++)

      out << m_sceneResources.at(i) + "\n";

  }

  m_targetTempFile->close();

  args << "--targets";

  args << "tempDeleteFile";

  m_thread.disconnect(SIGNAL(done(const QString &)));

  connect(&m_thread, SIGNAL(done(const QString &)), SLOT(commitDeletedFiles()));

  m_thread.executeCommand(m_workingDir, "svn", args, false);

}

//-----------------------------------------------------------------------------

void SVNDeleteDialog::commitDeletedFiles() {

  updateFileBrowser();

  QStringList args;

  args << "commit";

  args << "--targets";

  args << "tempDeleteFile";

  if (!m_commentTextEdit->toPlainText().isEmpty())

    args << QString("-m").append(m_commentTextEdit->toPlainText());

  else

    args << QString("-m").append(VersionControl::instance()->getUserName() +

                                 " delete files.");

  m_thread.disconnect(SIGNAL(done(const QString &)));

  connect(&m_thread, SIGNAL(done(const QString &)), SLOT(onCommitDone()));

  m_thread.executeCommand(m_workingDir, "svn", args);

}

//-----------------------------------------------------------------------------

void SVNDeleteDialog::onCommitDone() {

  if (m_targetTempFile) {

    QFile::remove(m_targetTempFile->fileName());

    delete m_targetTempFile;

    m_targetTempFile = 0;

  }

  emit done(m_files);

  close();

}

//-----------------------------------------------------------------------------

void SVNDeleteDialog::initTreeWidget() {

  int filesSize  = m_files.size();

  bool itemAdded = false;

  for (int i = 0; i < filesSize; i++) {

    QString fileName = m_files.at(i);

    TFilePath fp =

        TFilePath(m_workingDir.toStdWString()) + fileName.toStdWString();

    TFilePathSet fpset;

    TXshSimpleLevel::getFiles(fp, fpset);

    QStringList linkedFiles;

    TFilePathSet::iterator it;

    for (it = fpset.begin(); it != fpset.end(); ++it) {

      QString fn = toQString((*it).withoutParentDir());

      if (m_files.contains(fn)) linkedFiles.append(fn);

    }

    if (!linkedFiles.isEmpty()) {

      itemAdded            = true;

      QTreeWidgetItem *twi = new QTreeWidgetItem(m_treeWidget);

      twi->setText(0, fileName);

      twi->setFirstColumnSpanned(false);

      twi->setFlags(Qt::NoItemFlags);

      for (int i = 0; i < linkedFiles.size(); i++) {

        QTreeWidgetItem *child = new QTreeWidgetItem(twi);

        child->setText(0, linkedFiles.at(i));

        child->setFlags(Qt::NoItemFlags);

      }

      twi->setExpanded(true);

    }

  }

  if (itemAdded) m_treeWidget->show();

}

//-----------------------------------------------------------------------------

void SVNDeleteDialog::onDeleteSceneContentsToggled(bool checked) {

  if (!checked)

    m_sceneResources.clear();

  else {

    VersionControl *vc = VersionControl::instance();

    int fileSize = m_files.count();

    for (int i = 0; i < fileSize; i++) {

      QString fileName = m_files.at(i);

      if (fileName.endsWith(".tnz")) {

        if (m_files.contains(fileName))

          m_sceneResources.append(vc->getSceneContents(m_workingDir, fileName));

      }

    }

  }

  m_textLabel->setText(

      tr("Delete %1 items.")

          .arg(m_files.size() - m_sceneIconAdded + m_sceneResources.size()));

}