/*
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 "oblivioninfo.h"
#include <tchar.h>
#include <ShlObj.h>
#include "util.h"
#include <sstream>
#include "windows_error.h"
#include "error_report.h"
#define WIN32_LEAN_AND_MEAN
#include <Windows.h>
#include <boost/assign.hpp>

namespace MOShared {


OblivionInfo::OblivionInfo(const std::wstring &moDirectory, const std::wstring &moDataDirectory, const std::wstring &gameDirectory)
  : GameInfo(moDirectory, moDataDirectory, gameDirectory)
{
  identifyMyGamesDirectory(L"oblivion");
}

bool OblivionInfo::identifyGame(const std::wstring &searchPath)
{
  return FileExists(searchPath, L"Oblivion.exe") &&
         FileExists(searchPath, L"OblivionLauncher.exe");
}

std::wstring OblivionInfo::getRegPathStatic()
{
  HKEY key;
  LONG errorcode = ::RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"Software\\Bethesda Softworks\\Oblivion",
                                  0, KEY_QUERY_VALUE, &key);

  if (errorcode != ERROR_SUCCESS) {
    return std::wstring();
  }

  WCHAR temp[MAX_PATH];
  DWORD bufferSize = MAX_PATH;

  if (::RegQueryValueExW(key, L"Installed Path", nullptr, nullptr, (LPBYTE)temp, &bufferSize) == ERROR_SUCCESS) {
    return std::wstring(temp);
  } else {
    return std::wstring();
  }
}

std::vector<std::wstring> OblivionInfo::getDLCPlugins() const
{
  return boost::assign::list_of (L"DLCShiveringIsles.esp")
                                (L"Knights.esp")
                                (L"DLCFrostcrag.esp")
                                (L"DLCSpellTomes.esp")
                                (L"DLCMehrunesRazor.esp")
                                (L"DLCOrrery.esp")
                                (L"DLCSpellTomes.esp")
                                (L"DLCThievesDen.esp")
                                (L"DLCVileLair.esp")
                                (L"DLCHorseArmor.esp")
      ;
}


std::vector<std::wstring> OblivionInfo::getSavegameAttachmentExtensions() const
{
  return boost::assign::list_of(L"obse");
}


std::vector<std::wstring> OblivionInfo::getIniFileNames() const
{
  return boost::assign::list_of(L"oblivion.ini")(L"oblivionprefs.ini");
}

std::wstring OblivionInfo::getNexusPage(bool nmmScheme) const
{
  if (nmmScheme) {
    return L"http://nmm.nexusmods.com/oblivion";
  } else {
    return L"http://www.nexusmods.com/oblivion";
  }
}


std::wstring OblivionInfo::getNexusInfoUrlStatic()
{
  return L"http://nmm.nexusmods.com/oblivion";
}


int OblivionInfo::getNexusModIDStatic()
{
  return 38277;
}

bool OblivionInfo::rerouteToProfile(const wchar_t *fileName, const wchar_t*) const
{
  static LPCWSTR profileFiles[] = { L"oblivion.ini", L"oblivionprefs.ini", L"plugins.txt", nullptr };

  for (int i = 0; profileFiles[i] != nullptr; ++i) {
    if (_wcsicmp(fileName, profileFiles[i]) == 0) {
      return true;
    }
  }
  return false;
}

std::wstring OblivionInfo::getReferenceDataFile() const
{
  return L"Oblivion - Meshes.bsa";
}

bool OblivionInfo::isValidModURL(int modID, const std::wstring &url) const
{
  return GameInfo::isValidModURL(modID, url, L"http://oblivion.nexusmods.com");
}

} // namespace MOShared