/*
Copyright (C) 2012 Sebastian Herbord. All rights reserved.

This file is part of Mod Organizer.

Mod Organizer 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.

Mod Organizer 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 Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <QApplication>
#include <QPushButton>
#include <QListWidget>
#include <QComboBox>
#include <QCheckBox>
#include <QDir>
#include <QFileInfo>
#include <QSettings>
#include <QWhatsThis>
#include <QToolBar>
#include <QFileDialog>
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <DbgHelp.h>
#include <cstdarg>
#include <inject.h>
#include <appconfig.h>
#include <utility.h>
#include <stdexcept>
#include "mainwindow.h"
#include "report.h"
#include "modlist.h"
#include "profile.h"
#include "gameinfo.h"
#include "fallout3info.h"
#include "falloutnvinfo.h"
#include "oblivioninfo.h"
#include "skyriminfo.h"
#include "spawn.h"
#include "executableslist.h"
#include "singleinstance.h"
#include "utility.h"
#include "helper.h"
#include "logbuffer.h"
#include "selectiondialog.h"
#include "moapplication.h"
#include "tutorialmanager.h"
#include <QLibrary>
#include <iostream>
#include <QMessageBox>
#include <QSharedMemory>
#include <QBuffer>
#include <QSplashScreen>
#include <QDirIterator>
#include <QDesktopServices>
#include <eh.h>
#include <windows_error.h>
#include <boost/scoped_array.hpp>

#pragma comment(linker, "/manifestDependency:\"name='dlls' processorArchitecture='x86' version='1.0.0.0' type='win32' \"")


using namespace MOBase;
using namespace MOShared;


void removeOldLogfiles()
{
  QFileInfoList files = QDir(ToQString(GameInfo::instance().getLogDir())).entryInfoList(QStringList("ModOrganizer*.log"),
                            QDir::Files, QDir::Name);

  if (files.count() > 5) {
    QStringList deleteFiles;
    for (int i = 0; i < files.count() - 5; ++i) {
      deleteFiles.append(files.at(i).absoluteFilePath());
    }

    if (!shellDelete(deleteFiles)) {
      qWarning("failed to remove log files: %s", qPrintable(windowsErrorString(::GetLastError())));
    }
  }
}


// set up required folders (for a first install or after an update or to fix a broken installation)
bool bootstrap()
{
  GameInfo &gameInfo = GameInfo::instance();

  // remove the temporary backup directory in case we're restarting after an update
  QString moDirectory = QDir::fromNativeSeparators(ToQString(gameInfo.getOrganizerDirectory()));
  QString backupDirectory = moDirectory.mid(0).append("/update_backup");
  if (QDir(backupDirectory).exists()) {
    shellDelete(QStringList(backupDirectory));
  }

  // cycle logfile
  removeOldLogfiles();

  // create organizer directories
  QString dirNames[] = {
      QDir::fromNativeSeparators(ToQString(gameInfo.getProfilesDir())),
      QDir::fromNativeSeparators(ToQString(gameInfo.getModsDir())),
      QDir::fromNativeSeparators(ToQString(gameInfo.getDownloadDir())),
      QDir::fromNativeSeparators(ToQString(gameInfo.getOverwriteDir())),
      QDir::fromNativeSeparators(ToQString(gameInfo.getLogDir())),
      QDir::fromNativeSeparators(ToQString(gameInfo.getTutorialDir()))
    };
  static const int NUM_DIRECTORIES = sizeof(dirNames) / sizeof(QString);

  // optimistic run: try to simply create the directories:
  for (int i = 0; i < NUM_DIRECTORIES; ++i) {
    if (!QDir(dirNames[i]).exists()) {
      QDir().mkdir(dirNames[i]);
    }
  }

  // verify all directories exist and are writable,
  // otherwise invoke the helper to create them and make them writable
  for (int i = 0; i < NUM_DIRECTORIES; ++i) {
    QFileInfo fileInfo(dirNames[i]);
    if (!fileInfo.exists() || !fileInfo.isWritable()) {
      if (QMessageBox::question(NULL, QObject::tr("Permissions required"),
          QObject::tr("The current user account doesn't have the required access rights to run "
             "Mod Organizer. The neccessary changes can be made automatically (the MO directory "
             "will be made writable for the current user account). You will be asked to run "
             "\"helper.exe\" with administrative rights)."),
             QMessageBox::Yes | QMessageBox::Cancel) == QMessageBox::Yes) {
        if (!Helper::init(GameInfo::instance().getOrganizerDirectory())) {
          return false;
        }
      } else {
        return false;
      }
      // no matter which directory didn't exist/wasn't writable, the helper
      // should have created them all so we can break the loop
      break;
    }
  }

  // verify the hook-dll exists
  QString dllName = qApp->applicationDirPath() + "/" + ToQString(AppConfig::hookDLLName());
  HMODULE dllMod = ::LoadLibraryW(ToWString(dllName).c_str());
  if (dllMod == NULL) {
    throw windows_error("hook.dll is missing or invalid");
  }
  ::FreeLibrary(dllMod);

  return true;
}


void cleanupDir()
{
  // files from previous versions of MO that are no longer
  // required (in that location)
  QString fileNames[] = {
    "ModOrganiser.exe",
    "ModOrganizer.log",
    "ModOrganizer.log.old",
    "7z.dll",
    "mo1.dll",
    "mo_archive.dll",
    "mo_helper.exe",
    "msvcp90.dll",
    "msvcr90.dll",
    "phonon4.dll",
    "QtCore4.dll",
    "QtGui4.dll",
    "QtNetwork4.dll",
    "QtXml4.dll",
    "QtWebKit4.dll",
    "qjpeg4.dll",
    "dlls/phonon4.dll",
    "dlls/QtCore4.dll",
    "dlls/QtGui4.dll",
    "dlls/QtNetwork4.dll",
    "dlls/QtXml4.dll",
    "dlls/QtXmlPatterns4.dll",
    "dlls/QtWebKit4.dll",
    "dlls/QtDeclarative4.dll",
    "dlls/QtScript4.dll",
    "dlls/QtSql4.dll"
  };

  static const int NUM_FILES = sizeof(fileNames) / sizeof(QString);

  qDebug("cleaning up unused files");

  for (int i = 0; i < NUM_FILES; ++i) {
    if (QFile::remove(QApplication::applicationDirPath().append("/").append(fileNames[i]))) {
      qDebug("%s removed in cleanup",
             QApplication::applicationDirPath().append("/").append(fileNames[i]).toUtf8().constData());
    }
  }
}


bool isNxmLink(const QString &link)
{
  return link.left(6).toLower() == "nxm://";
}


LONG WINAPI MyUnhandledExceptionFilter(struct _EXCEPTION_POINTERS *exceptionPtrs)
{
  typedef BOOL (WINAPI *FuncMiniDumpWriteDump)(HANDLE process, DWORD pid, HANDLE file, MINIDUMP_TYPE dumpType,
                                               const PMINIDUMP_EXCEPTION_INFORMATION exceptionParam,
                                               const PMINIDUMP_USER_STREAM_INFORMATION userStreamParam,
                                               const PMINIDUMP_CALLBACK_INFORMATION callbackParam);
  LONG result = EXCEPTION_CONTINUE_SEARCH;

  HMODULE dbgDLL = ::LoadLibrary(L"dbghelp.dll");

  static const int errorLen = 200;
  char errorBuffer[errorLen + 1];
  memset(errorBuffer, '\0', errorLen + 1);

  if (dbgDLL) {
    FuncMiniDumpWriteDump funcDump = (FuncMiniDumpWriteDump)::GetProcAddress(dbgDLL, "MiniDumpWriteDump");
    if (funcDump) {

      if (QMessageBox::question(NULL, QObject::tr("Woops"),
                            QObject::tr("ModOrganizer has crashed! Should a diagnostic file be created? If you send me this file "
                               "(%1) to sherb@gmx.net, the bug is a lot more likely to be fixed. "
                               "Please include a short description of what you were doing when the crash happened").arg(qApp->applicationFilePath().append(".dmp")),
                            QMessageBox::Yes | QMessageBox::No) == QMessageBox::Yes) {

        std::wstring dumpName = ToWString(qApp->applicationFilePath().append(".dmp"));

        HANDLE dumpFile = ::CreateFile(dumpName.c_str(),
                                       GENERIC_WRITE, FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
        if (dumpFile != INVALID_HANDLE_VALUE) {
          _MINIDUMP_EXCEPTION_INFORMATION exceptionInfo;
          exceptionInfo.ThreadId = ::GetCurrentThreadId();
          exceptionInfo.ExceptionPointers = exceptionPtrs;
          exceptionInfo.ClientPointers = NULL;

          BOOL success = funcDump(::GetCurrentProcess(), ::GetCurrentProcessId(), dumpFile, MiniDumpNormal, &exceptionInfo, NULL, NULL);

          ::CloseHandle(dumpFile);
          if (success) {
            return EXCEPTION_EXECUTE_HANDLER;
          }
          _snprintf(errorBuffer, errorLen, "failed to save minidump to %ls (error %lu)",
                     dumpName.c_str(), ::GetLastError());
        } else {
          _snprintf(errorBuffer, errorLen, "failed to create %ls (error %lu)",
                     dumpName.c_str(), ::GetLastError());
        }
      } else {
        return result;
      }
    } else {
      _snprintf(errorBuffer, errorLen, "dbghelp.dll outdated");
    }
  } else {
    _snprintf(errorBuffer, errorLen, "dbghelp.dll not found");
  }

  QMessageBox::critical(NULL, QObject::tr("Woops"),
                        QObject::tr("ModOrganizer has crashed! Unfortunately I was not able to write a diagnostic file: %1").arg(errorBuffer));
  return result;
}


void registerMetaTypes()
{
  registerExecutable();
}

int main(int argc, char *argv[])
{
  MOApplication application(argc, argv);

  qApp->addLibraryPath(application.applicationDirPath() + "/dlls");

  SetUnhandledExceptionFilter(MyUnhandledExceptionFilter);

  LogBuffer::init(20, QtDebugMsg, application.applicationDirPath() + "/logs/mo_interface.log");

  qDebug("Working directory: %s", qPrintable(QDir::currentPath()));
  qDebug("MO at: %s", qPrintable(application.applicationDirPath()));
  qDebug("user name: %s", getenv("USERNAME"));
  QPixmap pixmap(":/MO/gui/splash");
  QSplashScreen splash(pixmap);
  splash.show();

  { // extend path to include dll directory so plugins don't need a manifest
    static const int BUFSIZE = 4096;

    boost::scoped_array<TCHAR> oldPath(new TCHAR[BUFSIZE]);
    DWORD offset = ::GetEnvironmentVariable(TEXT("PATH"), oldPath.get(), BUFSIZE);
    if (offset > BUFSIZE) {
      oldPath.reset(new TCHAR[offset]);
      ::GetEnvironmentVariable(TEXT("PATH"), oldPath.get(), offset);
    }

    std::tstring newPath(oldPath.get());
    newPath += TEXT(";");
    newPath += ToWString(QDir::toNativeSeparators(QCoreApplication::applicationDirPath())).c_str();
    newPath += TEXT("\\dlls");

    ::SetEnvironmentVariable(TEXT("PATH"), newPath.c_str());
  }

  registerMetaTypes();

  QStringList arguments = application.arguments();

  bool forcePrimary = false;
  if (arguments.contains("update")) {
    arguments.removeAll("update");
    forcePrimary = true;
  }

  try {
    SingleInstance instance(forcePrimary);
    if (!instance.primaryInstance()) {
      if ((arguments.size() == 2) && isNxmLink(arguments.at(1))) {
        qDebug("not primary instance, sending download message");
        instance.sendMessage(arguments.at(1));
        return 0;
      } else if (arguments.size() == 1) {
        QMessageBox::information(NULL, QObject::tr("Mod Organizer"), QObject::tr("An instance of Mod Organizer is already running"));
        return 0;
      }
    } // we continue for the primary instance OR if MO has been called with parameters


    // TODO: this should be MAX_PATH_UNICODE!
    wchar_t moPath[MAX_PATH];
    memset(moPath, 0, sizeof(TCHAR) * MAX_PATH);
    ::GetModuleFileNameW(NULL, moPath, MAX_PATH);
    wchar_t *lastBSlash = wcsrchr(moPath, TEXT('\\'));
    if (lastBSlash != NULL) {
      *lastBSlash = TEXT('\0');
    }
    QSettings settings(ToQString(std::wstring(moPath).append(L"\\ModOrganizer.ini")), QSettings::IniFormat);

    QString gamePath = QString::fromUtf8(settings.value("gamePath", "").toByteArray());

    bool done = false;
    while (!done) {
      if (!GameInfo::init(moPath, ToWString(gamePath))) {
        if (!gamePath.isEmpty()) {
          reportError(QObject::tr("No game identified in \"%1\". The directory is required to contain "
                                  "the game binary and its launcher.").arg(gamePath));
        }
        SelectionDialog selection(QObject::tr("Please select the game to manage"), NULL);

        { // add options
          QString skyrimPath    = ToQString(SkyrimInfo::getRegPathStatic());
          QString falloutNVPath = ToQString(FalloutNVInfo::getRegPathStatic());
          QString fallout3Path  = ToQString(Fallout3Info::getRegPathStatic());
          QString oblivionPath  = ToQString(OblivionInfo::getRegPathStatic());
          if (skyrimPath.length() != 0) {
            selection.addChoice(QString("Skyrim"), skyrimPath, skyrimPath);
          }
          if (falloutNVPath.length() != 0) {
            selection.addChoice(QString("Fallout NV"), falloutNVPath, falloutNVPath);
          }
          if (fallout3Path.length() != 0) {
            selection.addChoice(QString("Fallout 3"), fallout3Path, fallout3Path);
          }
          if (oblivionPath.length() != 0) {
            selection.addChoice(QString("Oblivion"), oblivionPath, oblivionPath);
          }

          selection.addChoice(QString("Browse..."), QString(), QString());
        }

        if (selection.exec() == QDialog::Rejected) {
          gamePath = "";
          done = true;
        } else {
          gamePath = QDir::cleanPath(selection.getChoiceData().toString());
          if (gamePath.isEmpty()) {
            gamePath = QFileDialog::getExistingDirectory(NULL, QObject::tr("Please select the game to manage"), QString(),
                                                         QFileDialog::ShowDirsOnly);
          }
        }
      } else {
        done = true;
        gamePath = ToQString(GameInfo::instance().getGameDirectory());
      }
    }

    if (gamePath.isEmpty()) {
      // game not found and user canceled
      return -1;
    } else if (gamePath.length() != 0) {
      // user selected a folder and game was initialised with it
      settings.setValue("gamePath", gamePath.toUtf8().constData());
    }

    qDebug("managing game at %s", qPrintable(gamePath));

    ExecutablesList executablesList;

    executablesList.init();

    if (!bootstrap()) { // requires gameinfo to be initialised!
      return -1;
    }

    cleanupDir();

    qDebug("setting up configured executables");

    int numCustomExecutables = settings.beginReadArray("customExecutables");
    for (int i = 0; i < numCustomExecutables; ++i) {
      settings.setArrayIndex(i);
      CloseMOStyle closeMO = settings.value("closeOnStart").toBool() ? DEFAULT_CLOSE : DEFAULT_STAY;
      executablesList.addExecutable(settings.value("title").toString(),
                                    settings.value("binary").toString(),
                                    settings.value("arguments").toString(),
                                    settings.value("workingDirectory", "").toString(),
                                    closeMO,
                                    settings.value("steamAppID", "").toString(),
                                    settings.value("custom", true).toBool(),
                                    settings.value("toolbar", false).toBool());
    }

    settings.endArray();

    qDebug("initializing tutorials");
    TutorialManager::init(QDir::fromNativeSeparators(ToQString(GameInfo::instance().getTutorialDir())).append("/"));

    application.setStyleFile(settings.value("Settings/style", "").toString());

    // set up main window and its data structures
    MainWindow mainWindow(argv[0], settings);
    QObject::connect(&mainWindow, SIGNAL(styleChanged(QString)), &application, SLOT(setStyleFile(QString)));
    QObject::connect(&instance, SIGNAL(messageSent(QString)), &mainWindow, SLOT(externalMessage(QString)));

    mainWindow.setExecutablesList(executablesList);
    mainWindow.readSettings();

    QString selectedProfileName = QString::fromUtf8(settings.value("selected_profile", "").toByteArray());

    { // see if there is a profile on the command line
      int profileIndex = arguments.indexOf("-p", 1);
      if ((profileIndex != -1) && (profileIndex < arguments.size() - 1)) {
        qDebug("profile overwritten on command line");
        selectedProfileName = arguments.at(profileIndex + 1);
      }
      arguments.removeAt(profileIndex);
      arguments.removeAt(profileIndex);
    }
    qDebug("configured profile: %s", qPrintable(selectedProfileName));

    // if we have a command line parameter, it is either a nxm link or
    // a binary to start
    if ((arguments.size() > 1) && (!isNxmLink(arguments.at(1)))) {
      QString exeName = arguments.at(1);
      qDebug("starting %s from command line", qPrintable(exeName));
      arguments.removeFirst(); // remove application name (ModOrganizer.exe)
      arguments.removeFirst(); // remove binary name
      // pass the remaining parameters to the binary
      mainWindow.spawnProgram(exeName, arguments.join(" "), selectedProfileName, QDir());
      return 0;
    }

    mainWindow.createFirstProfile();

    if (selectedProfileName.length() != 0) {
      if (!mainWindow.setCurrentProfile(selectedProfileName)) {
        mainWindow.setCurrentProfile(1);
        qWarning("failed to set profile: %s",
                 selectedProfileName.toUtf8().constData());
      }
    } else {
      mainWindow.setCurrentProfile(1);
    }

    qDebug("displaying main window");
    mainWindow.show();

    if ((arguments.size() > 1) &&
        (isNxmLink(arguments.at(1)))) {
      qDebug("starting download from command line: %s", qPrintable(arguments.at(1)));
      mainWindow.externalMessage(arguments.at(1));
    }
    splash.finish(&mainWindow);
    return application.exec();
  } catch (const std::exception &e) {
    reportError(e.what());
    return 1;
  }
}