/***

  Olive - Non-Linear Video Editor
  Copyright (C) 2019 Olive Team

  This program 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 3 of the License, or
  (at your option) any later version.

  This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.

***/

#include "save.h"

#include <QDir>
#include <QFile>
#include <QXmlStreamWriter>

#include "common/filefunctions.h"

OLIVE_NAMESPACE_ENTER

ProjectSaveTask::ProjectSaveTask(ProjectPtr project) :
  project_(project)
{
  SetTitle(tr("Saving '%1'").arg(project->filename()));
}

bool ProjectSaveTask::Run()
{
  // File to temporarily save to (ensures we can't half-write the user's main file and crash)
  QString temp_save = QDir(FileFunctions::GetTempFilePath()).filePath(QStringLiteral("tempsv"));

  QFile project_file(temp_save);

  if (project_file.open(QFile::WriteOnly | QFile::Text)) {
    QXmlStreamWriter writer(&project_file);
    writer.setAutoFormatting(true);

    writer.writeStartDocument();

    writer.writeStartElement("olive");

    writer.writeTextElement("version", "0.2.0");

    writer.writeTextElement("url", project_->filename());

    writer.writeStartElement(QStringLiteral("project"));

    project_->Save(&writer);

    writer.writeEndElement(); // project

    writer.writeEndElement(); // olive

    writer.writeEndDocument();

    project_file.close();

    if (writer.hasError()) {
      SetError(tr("Failed to write XML data"));
      return false;
    }

    // Save was successful, we can now rewrite the original file
    QFile original(project_->filename());
    if ((!original.exists() || original.remove())
        && QFile::copy(temp_save, project_->filename())) {
      return true;
    } else {
      SetError(tr("Failed to write to \"%1\".").arg(project_->filename()));
      return false;
    }
  } else {
    SetError(tr("Failed to open file \"%1\" for writing.").arg(project_->filename()));
    return false;
  }
}

OLIVE_NAMESPACE_EXIT