1 files changed, 802 insertions, 799 deletions
diff --git a/src/shared/directoryentry.cpp b/src/shared/directoryentry.cpp
index aa3bce5f..eaba91a6 100644
--- a/src/shared/directoryentry.cpp
+++ b/src/shared/directoryentry.cpp
@@ -1,799 +1,802 @@
-/*
-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 "directoryentry.h"
-#define WIN32_LEAN_AND_MEAN
-#include <Windows.h>
-#include <sstream>
-#include <algorithm>
-#include <bsatk.h>
-#include "error_report.h"
-#include "util.h"
-#include "windows_error.h"
-#include <boost/bind.hpp>
-
-
-
-class OriginConnection {
-
-public:
-
-  typedef int Index;
-  static const int INVALID_INDEX = INT_MIN;
-
-public:
-
-  OriginConnection()
-    : m_NextID(0)
-  {}
-
-  FilesOrigin& createOrigin(const std::wstring &originName, const std::wstring &directory, int priority,
-                            boost::shared_ptr<FileRegister> fileRegister, boost::shared_ptr<OriginConnection> originConnection) {
-    int newID = createID();
-    m_Origins[newID] = FilesOrigin(newID, originName, directory, priority, fileRegister, originConnection);
-    m_OriginsNameMap[originName] = newID;
-    m_OriginsPriorityMap[priority] = newID;
-    return m_Origins[newID];
-  }
-
-  bool exists(const std::wstring &name) {
-    std::map<std::wstring, int>::iterator iter = m_OriginsNameMap.find(name);
-    return iter != m_OriginsNameMap.end();
-  }
-
-  FilesOrigin &getByID(Index ID) {
-    return m_Origins[ID];
-  }
-
-  FilesOrigin &getByName(const std::wstring &name) {
-    std::map<std::wstring, int>::iterator iter = m_OriginsNameMap.find(name);
-    if (iter != m_OriginsNameMap.end()) {
-      return m_Origins[iter->second];
-    } else {
-      std::ostringstream stream;
-      stream << "invalid origin name: " << ToString(name, false);
-      throw std::runtime_error(stream.str());
-    }
-  }
-
-  void changePriorityLookup(int oldPriority, int newPriority)
-  {
-    auto iter = m_OriginsPriorityMap.find(oldPriority);
-    if (iter != m_OriginsPriorityMap.end()) {
-      Index idx = iter->second;
-      m_OriginsPriorityMap.erase(iter);
-      m_OriginsPriorityMap[newPriority] = idx;
-    }
-  }
-
-  void changeNameLookup(const std::wstring &oldName, const std::wstring &newName)
-  {
-    auto iter = m_OriginsNameMap.find(oldName);
-    if (iter != m_OriginsNameMap.end()) {
-      Index idx = iter->second;
-      m_OriginsNameMap.erase(iter);
-      m_OriginsNameMap[newName] = idx;
-    } else {
-      log("failed to change name lookup from %ls to %ls", oldName.c_str(), newName.c_str());
-    }
-  }
-
-private:
-
-  Index createID() {
-    return m_NextID++;
-  }
-
-private:
-
-  Index m_NextID;
-
-  std::map<Index, FilesOrigin> m_Origins;
-  std::map<std::wstring, Index> m_OriginsNameMap;
-  std::map<int, Index> m_OriginsPriorityMap;
-
-};
-
-
-//
-// FilesOrigin
-//
-
-
-void FilesOrigin::enable(bool enabled)
-{
-  if (!enabled) {
-    std::set<FileEntry::Index> copy = m_Files;
-    for (auto iter = copy.begin(); iter != copy.end(); ++iter) {
-      m_FileRegister->removeOrigin(*iter, m_ID);
-    }
-    m_Files.clear();
-  }
-  m_Disabled = !enabled;
-}
-
-
-void FilesOrigin::removeFile(FileEntry::Index index)
-{
-  auto iter = m_Files.find(index);
-  if (iter != m_Files.end()) {
-    m_Files.erase(iter);
-  }
-}
-
-
-
-std::wstring tail(const std::wstring &source, const size_t count)
-{
-  if (count >= source.length()) {
-    return source;
-  }
-
-  return source.substr(source.length() - count);
-}
-
-
-void FilesOrigin::setPriority(int priority)
-{
-  m_OriginConnection->changePriorityLookup(m_Priority, priority);
-
-    m_Priority = priority;
-}
-
-
-void FilesOrigin::setName(const std::wstring &name)
-{
-  m_OriginConnection->changeNameLookup(m_Name, name);
-  // change path too
-  if (tail(m_Path, m_Name.length()) == m_Name) {
-    m_Path = m_Path.substr(0, m_Path.length() - m_Name.length()).append(name);
-  }
-  m_Name = name;
-}
-
-std::vector<FileEntry*> FilesOrigin::getFiles() const
-{
-  std::vector<FileEntry*> result;
-
-  for (auto iter = m_Files.begin(); iter != m_Files.end(); ++iter) {
-    result.push_back(m_FileRegister->getFile(*iter));
-  }
-
-  return result;
-}
-
-
-
-//
-// FileEntry
-//
-
-void FileEntry::addOrigin(int origin, FILETIME fileTime, const std::wstring &archive)
-{
-  if (m_Origin == -1) {
-    m_Origin = origin;
-    m_FileTime = fileTime;
-    m_Archive = archive;
-//  } else if (FilesOrigin::getByID(origin).getPriority() > FilesOrigin::getByID(m_Origin).getPriority()) {
-  } else if (m_Parent->getOriginByID(origin).getPriority() > m_Parent->getOriginByID(m_Origin).getPriority()) {
-    if (std::find(m_Alternatives.begin(), m_Alternatives.end(), m_Origin) == m_Alternatives.end()) {
-      m_Alternatives.push_back(m_Origin);
-    }
-    m_Origin = origin;
-    m_FileTime = fileTime;
-    m_Archive = archive;
-  } else {
-    bool found = false;
-    if (m_Origin == origin) {
-      // already an origin
-      return;
-    }
-    for (std::vector<int>::iterator iter = m_Alternatives.begin(); iter != m_Alternatives.end(); ++iter) {
-      if (*iter == origin) {
-        // already an origin
-        return;
-      }
-      if (m_Parent->getOriginByID(*iter).getPriority() < m_Parent->getOriginByID(origin).getPriority()) {
-        m_Alternatives.insert(iter, origin);
-        found = true;
-        break;
-      }
-    }
-    if (!found) {
-      m_Alternatives.push_back(origin);
-    }
-  }
-}
-
-bool FileEntry::removeOrigin(int origin)
-{
-  if (m_Origin == origin) {
-    if (!m_Alternatives.empty()) {
-      // find alternative with the highest priority
-      std::vector<int>::iterator currentIter = m_Alternatives.begin();
-      for (std::vector<int>::iterator iter = m_Alternatives.begin(); iter != m_Alternatives.end(); ++iter) {
-        if ((m_Parent->getOriginByID(*iter).getPriority() > m_Parent->getOriginByID(*currentIter).getPriority()) &&
-            (*iter != origin)) {
-          currentIter = iter;
-        }
-      }
-      int currentID = *currentIter;
-      m_Alternatives.erase(currentIter);
-
-      m_Origin = currentID;
-
-      // now we need to update the file time...
-      std::wstring filePath = getFullPath();
-      HANDLE file = ::CreateFile(filePath.c_str(), GENERIC_READ | GENERIC_WRITE,
-                                 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
-      if (!::GetFileTime(file, NULL, NULL, &m_FileTime)) {
-        // maybe this file is in a bsa, but there is no easy way to find out which. User should refresh
-        // the view to find out
-        m_Archive = L"bsa?";
-      } else {
-        m_Archive = L"";
-      }
-
-      ::CloseHandle(file);
-
-    } else {
-      m_Origin = -1;
-      return true;
-    }
-  } else {
-    std::vector<int>::iterator newEnd = std::remove(m_Alternatives.begin(), m_Alternatives.end(), origin);
-    m_Alternatives.erase(newEnd, m_Alternatives.end());
-  }
-  return false;
-}
-
-
-// sorted by priority descending
-static bool ByOriginPriority(DirectoryEntry *entry, int LHS, int RHS)
-{
-  return entry->getOriginByID(LHS).getPriority() < entry->getOriginByID(RHS).getPriority();
-}
-
-
-FileEntry::FileEntry(Index index, const std::wstring &name, DirectoryEntry *parent)
-  : m_Index(index), m_Name(name), m_Parent(parent), m_Origin(-1), m_Archive(L"")
-{
-}
-
-
-void FileEntry::sortOrigins()
-{
-  m_Alternatives.push_back(m_Origin);
-  std::sort(m_Alternatives.begin(), m_Alternatives.end(), boost::bind(ByOriginPriority, m_Parent, _1, _2));
-  m_Origin = m_Alternatives[m_Alternatives.size() - 1];
-  m_Alternatives.pop_back();
-}
-
-
-bool FileEntry::recurseParents(std::wstring &path, const DirectoryEntry *parent) const
-{
-  if (parent == NULL) {
-    return false;
-  } else {
-    // don't append the topmost parent because it is the virtual data-root
-    if (recurseParents(path, parent->getParent())) {
-      path.append(L"\\").append(parent->getName());
-    }
-    return true;
-  }
-}
-
-std::wstring FileEntry::getFullPath() const
-{
-  std::wstring result;
-  bool ignore = false;
-  result = m_Parent->getOriginByID(getOrigin(ignore)).getPath(); //base directory for origin
-  recurseParents(result, m_Parent); // all intermediate directories
-  result.append(L"\\").append(m_Name); // the actual filename
-  return result;
-}
-
-std::wstring FileEntry::getRelativePath() const
-{
-  std::wstring result;
-  recurseParents(result, m_Parent); // all intermediate directories
-  result.append(L"\\").append(m_Name); // the actual filename
-  return result;
-}
-
-
-
-
-
-//
-// DirectoryEntry
-//
-DirectoryEntry::DirectoryEntry(const std::wstring &name, DirectoryEntry *parent, int originID)
-  : m_OriginConnection(new OriginConnection),
-    m_Name(name), m_Parent(parent), m_Populated(false), m_Origin(originID)
-{
-  m_FileRegister.reset(new FileRegister(m_OriginConnection));
-}
-
-DirectoryEntry::DirectoryEntry(const std::wstring &name, DirectoryEntry *parent, int originID,
-               boost::shared_ptr<FileRegister> fileRegister, boost::shared_ptr<OriginConnection> originConnection)
-  : m_FileRegister(fileRegister), m_OriginConnection(originConnection),
-    m_Name(name), m_Parent(parent), m_Populated(false), m_Origin(originID)
-{}
-
-
-DirectoryEntry::~DirectoryEntry()
-{
-  clear();
-}
-
-
-const std::wstring &DirectoryEntry::getName() const
-{
-  return m_Name;
-}
-
-
-void DirectoryEntry::clear()
-{
-  m_Files.clear();
-  for (std::vector<DirectoryEntry*>::iterator iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
-    delete *iter;
-  }
-  m_SubDirectories.clear();
-}
-
-
-FilesOrigin &DirectoryEntry::createOrigin(const std::wstring &originName, const std::wstring &directory, int priority)
-{
-  if (m_OriginConnection->exists(originName)) {
-    FilesOrigin &origin = m_OriginConnection->getByName(originName);
-    origin.enable(true);
-    return origin;
-  } else {
-    return m_OriginConnection->createOrigin(originName, directory, priority, m_FileRegister, m_OriginConnection);
-  }
-}
-
-
-void DirectoryEntry::addFromOrigin(const std::wstring &originName, const std::wstring &directory, int priority)
-{
-  FilesOrigin &origin = createOrigin(originName, directory, priority);
-  wchar_t *buffer = new wchar_t[MAXPATH_UNICODE + 1];
-  memset(buffer, L'\0', MAXPATH_UNICODE + 1);
-  try {
-    int offset = _snwprintf(buffer, MAXPATH_UNICODE, L"%ls", directory.c_str());
-    addFiles(origin, buffer, offset);
-  } catch (...) {
-    delete [] buffer;
-    buffer = NULL;
-  }
-  delete [] buffer;
-  m_Populated = true;
-}
-
-
-void DirectoryEntry::addFromBSA(const std::wstring &originName, std::wstring &directory, const std::wstring &fileName, int priority)
-{
-  FilesOrigin &origin = createOrigin(originName, directory, priority);
-
-  WIN32_FILE_ATTRIBUTE_DATA fileData;
-  if (::GetFileAttributesExW(fileName.c_str(), GetFileExInfoStandard, &fileData) == 0) {
-    throw windows_error("failed to determine file time");
-  }
-
-  BSA::Archive archive;
-  BSA::EErrorCode res = archive.read(ToString(fileName, false).c_str());
-  if ((res != BSA::ERROR_NONE) && (res != BSA::ERROR_INVALIDHASHES)) {
-    std::ostringstream stream;
-    stream << "invalid bsa file: " << ToString(fileName, false) << " errorcode " << res << " - " << ::GetLastError();
-    throw std::runtime_error(stream.str());
-  }
-  size_t namePos = fileName.find_last_of(L"\\/");
-  if (namePos == std::wstring::npos) {
-    namePos = 0;
-  } else {
-    ++namePos;
-  }
-
-  addFiles(origin, archive.getRoot(), fileData.ftLastWriteTime, fileName.substr(namePos));
-  m_Populated = true;
-}
-
-
-void DirectoryEntry::addFiles(FilesOrigin &origin, wchar_t *buffer, int bufferOffset)
-{
-  WIN32_FIND_DATAW findData;
-
-  _snwprintf(buffer + bufferOffset, MAXPATH_UNICODE - bufferOffset, L"\\*");
-  HANDLE searchHandle = ::FindFirstFileExW(buffer, FindExInfoStandard, &findData, FindExSearchNameMatch, NULL, FIND_FIRST_EX_CASE_SENSITIVE);
-  if (searchHandle != INVALID_HANDLE_VALUE) {
-    BOOL result = true;
-    while (result) {
-      if (findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
-        if ((wcscmp(findData.cFileName, L".") != 0) &&
-            (wcscmp(findData.cFileName, L"..") != 0)) {
-          int offset = _snwprintf(buffer + bufferOffset, MAXPATH_UNICODE, L"\\%ls", findData.cFileName);
-          // recurse into subdirectories
-          getSubDirectory(findData.cFileName, true, origin.getID())->addFiles(origin, buffer, bufferOffset + offset);
-        }
-      } else {
-        insert(findData.cFileName, origin, findData.ftLastWriteTime, L"");
-      }
-      result = ::FindNextFileW(searchHandle, &findData);
-    }
-  }
-  ::FindClose(searchHandle);
-}
-
-
-void DirectoryEntry::addFiles(FilesOrigin &origin, BSA::Folder::Ptr archiveFolder, FILETIME &fileTime, const std::wstring &archiveName)
-{
-  // add files
-  for (unsigned int fileIdx = 0; fileIdx < archiveFolder->getNumFiles(); ++fileIdx) {
-    BSA::File::Ptr file = archiveFolder->getFile(fileIdx);
-    insert(ToWString(file->getName(), true), origin, fileTime, archiveName);
-  }
-
-  // recurse into subdirectories
-  for (unsigned int folderIdx = 0; folderIdx < archiveFolder->getNumSubFolders(); ++folderIdx) {
-    BSA::Folder::Ptr folder = archiveFolder->getSubFolder(folderIdx);
-    DirectoryEntry *folderEntry = getSubDirectoryRecursive(ToWString(folder->getName(), true), true, origin.getID());
-
-    folderEntry->addFiles(origin, folder, fileTime, archiveName);
-  }
-}
-
-
-void DirectoryEntry::removeFile(const std::wstring &filePath, int *origin)
-{
-  size_t pos = filePath.find_first_of(L"\\/");
-  if (pos == std::string::npos) {
-    this->remove(filePath, origin);
-  } else {
-    std::wstring dirName = filePath.substr(0, pos);
-    std::wstring rest = filePath.substr(pos + 1);
-    DirectoryEntry *entry = getSubDirectoryRecursive(dirName, false);
-    if (entry != NULL) {
-      entry->removeFile(rest, origin);
-    }
-  }
-}
-
-void DirectoryEntry::removeDirRecursive()
-{
-  while (!m_Files.empty()) {
-    m_FileRegister->removeFile(m_Files.begin()->second);
-  }
-
-  for (auto iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
-    (*iter)->removeDirRecursive();
-  }
-  m_SubDirectories.clear();
-}
-
-void DirectoryEntry::removeDir(const std::wstring &path)
-{
-  size_t pos = path.find_first_of(L"\\/");
-  if (pos == std::string::npos) {
-    for (auto iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
-      if (_wcsicmp((*iter)->getName().c_str(), path.c_str()) == 0) {
-        (*iter)->removeDirRecursive();
-        m_SubDirectories.erase(iter);
-        break;
-      }
-    }
-  } else {
-    std::wstring dirName = path.substr(0, pos);
-    std::wstring rest = path.substr(pos + 1);
-    DirectoryEntry *entry = getSubDirectoryRecursive(dirName, false);
-    if (entry != NULL) {
-      entry->removeDir(rest);
-    }
-  }
-}
-
-
-void DirectoryEntry::insertFile(const std::wstring &filePath, FilesOrigin &origin, FILETIME fileTime)
-{
-  size_t pos = filePath.find_first_of(L"\\/");
-  if (pos == std::string::npos) {
-    this->insert(filePath, origin, fileTime, L"");
-  } else {
-    std::wstring dirName = filePath.substr(0, pos);
-    std::wstring rest = filePath.substr(pos + 1);
-    getSubDirectoryRecursive(dirName, true, origin.getID())->insertFile(rest, origin, fileTime);
-  }
-}
-
-
-void DirectoryEntry::removeFile(FileEntry::Index index)
-{
-  if (m_Files.size() != 0) {
-    auto iter = std::find_if(m_Files.begin(), m_Files.end(),
-                             [&index](const std::pair<std::wstring, FileEntry::Index> &iter) -> bool {
-                               return iter.second == index; } );
-    if (iter != m_Files.end()) {
-      m_Files.erase(iter);
-    } else {
-      log("file \"%ls\" not in directory \"%ls\"",
-          m_FileRegister->getFile(index)->getName().c_str(),
-          this->getName().c_str());
-    }
-  } else {
-    log("file \"%ls\" not in directory \"%ls\", directory empty",
-        m_FileRegister->getFile(index)->getName().c_str(),
-        this->getName().c_str());
-  }
-}
-
-
-int DirectoryEntry::anyOrigin() const
-{
-  bool ignore;
-  for (auto iter = m_Files.begin(); iter != m_Files.end(); ++iter) {
-    FileEntry *entry = m_FileRegister->getFile(iter->second);
-    if (!entry->isFromArchive()) {
-      return entry->getOrigin(ignore);
-    }
-  }
-
-  // if we got here, no file directly within this directory is a valid indicator for a mod, thus
-  // we continue looking in subdirectories
-  for (std::vector<DirectoryEntry*>::const_iterator iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
-    int res = (*iter)->anyOrigin();
-    if (res != -1){
-      return res;
-    }
-  }
-  return m_Origin;
-}
-
-
-bool DirectoryEntry::originExists(const std::wstring &name) const
-{
-  return m_OriginConnection->exists(name);
-}
-
-
-FilesOrigin &DirectoryEntry::getOriginByID(int ID) const
-{
-  return m_OriginConnection->getByID(ID);
-}
-
-
-FilesOrigin &DirectoryEntry::getOriginByName(const std::wstring &name) const
-{
-  return m_OriginConnection->getByName(name);
-}
-
-
-int DirectoryEntry::getOrigin(const std::wstring &path, bool &archive)
-{
-  const DirectoryEntry *directory = NULL;
-  const FileEntry *file = searchFile(path, &directory);
-  if (file != NULL) {
-    return file->getOrigin(archive);
-  } else {
-    if (directory != NULL) {
-      return directory->anyOrigin();
-    } else {
-      return -1;
-    }
-  }
-}
-
-std::vector<FileEntry*> DirectoryEntry::getFiles() const
-{
-  std::vector<FileEntry*> result;
-  for (auto iter = m_Files.begin(); iter != m_Files.end(); ++iter) {
-    result.push_back(m_FileRegister->getFile(iter->second));
-  }
-  return result;
-}
-
-
-const FileEntry *DirectoryEntry::searchFile(const std::wstring &path, const DirectoryEntry **directory) const
-{
-  if (directory != NULL) {
-    *directory = NULL;
-  }
-
-  if ((path.length() == 0) ||
-      (path == L"*")) {
-    // no file name -> the path ended on a (back-)slash
-    *directory = this;
-
-    return NULL;
-  }
-
-  size_t len =  path.find_first_of(L"\\/");
-
-  if (len == std::string::npos) {
-    // no more path components
-    auto iter = m_Files.find(path);
-    if (iter != m_Files.end()) {
-      return m_FileRegister->getFile(iter->second);
-    } else if (directory != NULL) {
-      DirectoryEntry *temp = findSubDirectory(path);
-      if (temp != NULL) {
-        *directory = temp;
-      }
-    }
-  } else {
-    // file is in in a subdirectory, recurse into the matching subdirectory
-    std::wstring pathComponent = path.substr(0, len);
-    DirectoryEntry *temp = findSubDirectory(pathComponent);
-    if (temp != NULL) {
-      return temp->searchFile(path.substr(len + 1), directory);
-    }
-  }
-  return NULL;
-}
-
-/*
-void DirectoryEntry::sortOrigins()
-{
-  for (std::set<FileEntry>::iterator iter = m_Files.begin(); iter != m_Files.end(); ++iter) {
-    const_cast<FileEntry&>(*iter).sortOrigins();
-  }
-  for (std::vector<DirectoryEntry*>::iterator iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
-    (*iter)->sortOrigins();
-  }
-}
-*/
-
-DirectoryEntry *DirectoryEntry::findSubDirectory(const std::wstring &name) const
-{
-  for (std::vector<DirectoryEntry*>::const_iterator iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
-    if (_wcsicmp((*iter)->getName().c_str(), name.c_str()) == 0) {
-      return *iter;
-    }
-  }
-  return NULL;
-}
-
-
-
-const FileEntry *DirectoryEntry::findFile(const std::wstring &name)
-{
-  auto iter = m_Files.find(name);
-  if (iter != m_Files.end()) {
-    return m_FileRegister->getFile(iter->second);
-  } else {
-    return NULL;
-  }
-}
-
-DirectoryEntry *DirectoryEntry::getSubDirectory(const std::wstring &name, bool create, int originID)
-{
-  for (std::vector<DirectoryEntry*>::iterator iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
-    if (_wcsicmp((*iter)->getName().c_str(), name.c_str()) == 0) {
-      return *iter;
-    }
-  }
-  if (create) {
-    std::vector<DirectoryEntry*>::iterator iter = m_SubDirectories.insert(m_SubDirectories.end(),
-        new DirectoryEntry(name, this, originID, m_FileRegister, m_OriginConnection));
-    return *iter;
-  } else {
-    return NULL;
-  }
-}
-
-
-DirectoryEntry *DirectoryEntry::getSubDirectoryRecursive(const std::wstring &path, bool create, int originID)
-{
-  if (path.length() == 0) {
-    // path ended with a backslash?
-    return this;
-  }
-
-  size_t pos = path.find_first_of(L"\\/");
-  if (pos == std::wstring::npos) {
-    return getSubDirectory(path, create);
-  } else {
-    DirectoryEntry *nextChild = getSubDirectory(path.substr(0, pos), create, originID);
-    if (nextChild == NULL) {
-      return NULL;
-    } else {
-      return nextChild->getSubDirectoryRecursive(path.substr(pos + 1), create, originID);
-    }
-  }
-}
-
-
-
-FileRegister::FileRegister(boost::shared_ptr<OriginConnection> originConnection)
-  : m_OriginConnection(originConnection)
-{
-}
-
-FileEntry::Index FileRegister::generateIndex()
-{
-  static FileEntry::Index sIndex = 0;
-  return sIndex++;
-}
-
-bool FileRegister::indexValid(FileEntry::Index index) const
-{
-  return m_Files.find(index) != m_Files.end();
-}
-
-FileEntry &FileRegister::createFile(const std::wstring &name, DirectoryEntry *parent)
-{
-  FileEntry::Index index = generateIndex();
-  m_Files[index] = FileEntry(index, name, parent);
-  return m_Files[index];
-}
-
-
-FileEntry *FileRegister::getFile(FileEntry::Index index)
-{
-  auto iter = m_Files.find(index);
-  if (iter != m_Files.end()) {
-    return &iter->second;
-  }
-  return NULL;
-}
-
-
-void FileRegister::unregisterFile(FileEntry &file)
-{
-  bool ignore;
-  // unregister from origin
-  int originID = file.getOrigin(ignore);
-  m_OriginConnection->getByID(originID).removeFile(file.getIndex());
-  const std::vector<int> &alternatives = file.getAlternatives();
-  for (auto iter = alternatives.begin(); iter != alternatives.end(); ++iter) {
-    m_OriginConnection->getByID(*iter).removeFile(file.getIndex());
-  }
-
-  // unregister from directory
-  if (file.getParent() != NULL) {
-    file.getParent()->removeFile(file.getIndex());
-  }
-}
-
-
-void FileRegister::removeFile(FileEntry::Index index)
-{
-  auto iter = m_Files.find(index);
-  if (iter != m_Files.end()) {
-    unregisterFile(iter->second);
-    m_Files.erase(index);
-  }
-}
-
-void FileRegister::removeOrigin(FileEntry::Index index, int originID)
-{
-  auto iter = m_Files.find(index);
-  if (iter != m_Files.end()) {
-    if (iter->second.removeOrigin(originID)) {
-      unregisterFile(iter->second);
-    }
-  }
-}
-
-void FileRegister::sortOrigins()
-{
-  for (auto iter = m_Files.begin(); iter != m_Files.end(); ++iter) {
-    iter->second.sortOrigins();
-  }
-}
+/*
+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 "directoryentry.h"
+#define WIN32_LEAN_AND_MEAN
+#include <Windows.h>
+#include <sstream>
+#include <algorithm>
+#include <bsatk.h>
+#include "error_report.h"
+#include "util.h"
+#include "windows_error.h"
+#include <boost/bind.hpp>
+
+namespace MOShared {
+
+
+
+class OriginConnection {
+
+public:
+
+  typedef int Index;
+  static const int INVALID_INDEX = INT_MIN;
+
+public:
+
+  OriginConnection()
+    : m_NextID(0)
+  {}
+
+  FilesOrigin& createOrigin(const std::wstring &originName, const std::wstring &directory, int priority,
+                            boost::shared_ptr<FileRegister> fileRegister, boost::shared_ptr<OriginConnection> originConnection) {
+    int newID = createID();
+    m_Origins[newID] = FilesOrigin(newID, originName, directory, priority, fileRegister, originConnection);
+    m_OriginsNameMap[originName] = newID;
+    m_OriginsPriorityMap[priority] = newID;
+    return m_Origins[newID];
+  }
+
+  bool exists(const std::wstring &name) {
+    std::map<std::wstring, int>::iterator iter = m_OriginsNameMap.find(name);
+    return iter != m_OriginsNameMap.end();
+  }
+
+  FilesOrigin &getByID(Index ID) {
+    return m_Origins[ID];
+  }
+
+  FilesOrigin &getByName(const std::wstring &name) {
+    std::map<std::wstring, int>::iterator iter = m_OriginsNameMap.find(name);
+    if (iter != m_OriginsNameMap.end()) {
+      return m_Origins[iter->second];
+    } else {
+      std::ostringstream stream;
+      stream << "invalid origin name: " << ToString(name, false);
+      throw std::runtime_error(stream.str());
+    }
+  }
+
+  void changePriorityLookup(int oldPriority, int newPriority)
+  {
+    auto iter = m_OriginsPriorityMap.find(oldPriority);
+    if (iter != m_OriginsPriorityMap.end()) {
+      Index idx = iter->second;
+      m_OriginsPriorityMap.erase(iter);
+      m_OriginsPriorityMap[newPriority] = idx;
+    }
+  }
+
+  void changeNameLookup(const std::wstring &oldName, const std::wstring &newName)
+  {
+    auto iter = m_OriginsNameMap.find(oldName);
+    if (iter != m_OriginsNameMap.end()) {
+      Index idx = iter->second;
+      m_OriginsNameMap.erase(iter);
+      m_OriginsNameMap[newName] = idx;
+    } else {
+      log("failed to change name lookup from %ls to %ls", oldName.c_str(), newName.c_str());
+    }
+  }
+
+private:
+
+  Index createID() {
+    return m_NextID++;
+  }
+
+private:
+
+  Index m_NextID;
+
+  std::map<Index, FilesOrigin> m_Origins;
+  std::map<std::wstring, Index> m_OriginsNameMap;
+  std::map<int, Index> m_OriginsPriorityMap;
+
+};
+
+
+//
+// FilesOrigin
+//
+
+
+void FilesOrigin::enable(bool enabled)
+{
+  if (!enabled) {
+    std::set<FileEntry::Index> copy = m_Files;
+    for (auto iter = copy.begin(); iter != copy.end(); ++iter) {
+      m_FileRegister->removeOrigin(*iter, m_ID);
+    }
+    m_Files.clear();
+  }
+  m_Disabled = !enabled;
+}
+
+
+void FilesOrigin::removeFile(FileEntry::Index index)
+{
+  auto iter = m_Files.find(index);
+  if (iter != m_Files.end()) {
+    m_Files.erase(iter);
+  }
+}
+
+
+
+std::wstring tail(const std::wstring &source, const size_t count)
+{
+  if (count >= source.length()) {
+    return source;
+  }
+
+  return source.substr(source.length() - count);
+}
+
+
+void FilesOrigin::setPriority(int priority)
+{
+  m_OriginConnection->changePriorityLookup(m_Priority, priority);
+
+    m_Priority = priority;
+}
+
+
+void FilesOrigin::setName(const std::wstring &name)
+{
+  m_OriginConnection->changeNameLookup(m_Name, name);
+  // change path too
+  if (tail(m_Path, m_Name.length()) == m_Name) {
+    m_Path = m_Path.substr(0, m_Path.length() - m_Name.length()).append(name);
+  }
+  m_Name = name;
+}
+
+std::vector<FileEntry*> FilesOrigin::getFiles() const
+{
+  std::vector<FileEntry*> result;
+
+  for (auto iter = m_Files.begin(); iter != m_Files.end(); ++iter) {
+    result.push_back(m_FileRegister->getFile(*iter));
+  }
+
+  return result;
+}
+
+
+
+//
+// FileEntry
+//
+
+void FileEntry::addOrigin(int origin, FILETIME fileTime, const std::wstring &archive)
+{
+  if (m_Origin == -1) {
+    m_Origin = origin;
+    m_FileTime = fileTime;
+    m_Archive = archive;
+//  } else if (FilesOrigin::getByID(origin).getPriority() > FilesOrigin::getByID(m_Origin).getPriority()) {
+  } else if (m_Parent->getOriginByID(origin).getPriority() > m_Parent->getOriginByID(m_Origin).getPriority()) {
+    if (std::find(m_Alternatives.begin(), m_Alternatives.end(), m_Origin) == m_Alternatives.end()) {
+      m_Alternatives.push_back(m_Origin);
+    }
+    m_Origin = origin;
+    m_FileTime = fileTime;
+    m_Archive = archive;
+  } else {
+    bool found = false;
+    if (m_Origin == origin) {
+      // already an origin
+      return;
+    }
+    for (std::vector<int>::iterator iter = m_Alternatives.begin(); iter != m_Alternatives.end(); ++iter) {
+      if (*iter == origin) {
+        // already an origin
+        return;
+      }
+      if (m_Parent->getOriginByID(*iter).getPriority() < m_Parent->getOriginByID(origin).getPriority()) {
+        m_Alternatives.insert(iter, origin);
+        found = true;
+        break;
+      }
+    }
+    if (!found) {
+      m_Alternatives.push_back(origin);
+    }
+  }
+}
+
+bool FileEntry::removeOrigin(int origin)
+{
+  if (m_Origin == origin) {
+    if (!m_Alternatives.empty()) {
+      // find alternative with the highest priority
+      std::vector<int>::iterator currentIter = m_Alternatives.begin();
+      for (std::vector<int>::iterator iter = m_Alternatives.begin(); iter != m_Alternatives.end(); ++iter) {
+        if ((m_Parent->getOriginByID(*iter).getPriority() > m_Parent->getOriginByID(*currentIter).getPriority()) &&
+            (*iter != origin)) {
+          currentIter = iter;
+        }
+      }
+      int currentID = *currentIter;
+      m_Alternatives.erase(currentIter);
+
+      m_Origin = currentID;
+
+      // now we need to update the file time...
+      std::wstring filePath = getFullPath();
+      HANDLE file = ::CreateFile(filePath.c_str(), GENERIC_READ | GENERIC_WRITE,
+                                 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
+      if (!::GetFileTime(file, NULL, NULL, &m_FileTime)) {
+        // maybe this file is in a bsa, but there is no easy way to find out which. User should refresh
+        // the view to find out
+        m_Archive = L"bsa?";
+      } else {
+        m_Archive = L"";
+      }
+
+      ::CloseHandle(file);
+
+    } else {
+      m_Origin = -1;
+      return true;
+    }
+  } else {
+    std::vector<int>::iterator newEnd = std::remove(m_Alternatives.begin(), m_Alternatives.end(), origin);
+    m_Alternatives.erase(newEnd, m_Alternatives.end());
+  }
+  return false;
+}
+
+
+// sorted by priority descending
+static bool ByOriginPriority(DirectoryEntry *entry, int LHS, int RHS)
+{
+  return entry->getOriginByID(LHS).getPriority() < entry->getOriginByID(RHS).getPriority();
+}
+
+
+FileEntry::FileEntry(Index index, const std::wstring &name, DirectoryEntry *parent)
+  : m_Index(index), m_Name(name), m_Parent(parent), m_Origin(-1), m_Archive(L"")
+{
+}
+
+
+void FileEntry::sortOrigins()
+{
+  m_Alternatives.push_back(m_Origin);
+  std::sort(m_Alternatives.begin(), m_Alternatives.end(), boost::bind(ByOriginPriority, m_Parent, _1, _2));
+  m_Origin = m_Alternatives[m_Alternatives.size() - 1];
+  m_Alternatives.pop_back();
+}
+
+
+bool FileEntry::recurseParents(std::wstring &path, const DirectoryEntry *parent) const
+{
+  if (parent == NULL) {
+    return false;
+  } else {
+    // don't append the topmost parent because it is the virtual data-root
+    if (recurseParents(path, parent->getParent())) {
+      path.append(L"\\").append(parent->getName());
+    }
+    return true;
+  }
+}
+
+std::wstring FileEntry::getFullPath() const
+{
+  std::wstring result;
+  bool ignore = false;
+  result = m_Parent->getOriginByID(getOrigin(ignore)).getPath(); //base directory for origin
+  recurseParents(result, m_Parent); // all intermediate directories
+  result.append(L"\\").append(m_Name); // the actual filename
+  return result;
+}
+
+std::wstring FileEntry::getRelativePath() const
+{
+  std::wstring result;
+  recurseParents(result, m_Parent); // all intermediate directories
+  result.append(L"\\").append(m_Name); // the actual filename
+  return result;
+}
+
+
+
+
+
+//
+// DirectoryEntry
+//
+DirectoryEntry::DirectoryEntry(const std::wstring &name, DirectoryEntry *parent, int originID)
+  : m_OriginConnection(new OriginConnection),
+    m_Name(name), m_Parent(parent), m_Populated(false), m_Origin(originID)
+{
+  m_FileRegister.reset(new FileRegister(m_OriginConnection));
+}
+
+DirectoryEntry::DirectoryEntry(const std::wstring &name, DirectoryEntry *parent, int originID,
+               boost::shared_ptr<FileRegister> fileRegister, boost::shared_ptr<OriginConnection> originConnection)
+  : m_FileRegister(fileRegister), m_OriginConnection(originConnection),
+    m_Name(name), m_Parent(parent), m_Populated(false), m_Origin(originID)
+{}
+
+
+DirectoryEntry::~DirectoryEntry()
+{
+  clear();
+}
+
+
+const std::wstring &DirectoryEntry::getName() const
+{
+  return m_Name;
+}
+
+
+void DirectoryEntry::clear()
+{
+  m_Files.clear();
+  for (std::vector<DirectoryEntry*>::iterator iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
+    delete *iter;
+  }
+  m_SubDirectories.clear();
+}
+
+
+FilesOrigin &DirectoryEntry::createOrigin(const std::wstring &originName, const std::wstring &directory, int priority)
+{
+  if (m_OriginConnection->exists(originName)) {
+    FilesOrigin &origin = m_OriginConnection->getByName(originName);
+    origin.enable(true);
+    return origin;
+  } else {
+    return m_OriginConnection->createOrigin(originName, directory, priority, m_FileRegister, m_OriginConnection);
+  }
+}
+
+
+void DirectoryEntry::addFromOrigin(const std::wstring &originName, const std::wstring &directory, int priority)
+{
+  FilesOrigin &origin = createOrigin(originName, directory, priority);
+  wchar_t *buffer = new wchar_t[MAXPATH_UNICODE + 1];
+  memset(buffer, L'\0', MAXPATH_UNICODE + 1);
+  try {
+    int offset = _snwprintf(buffer, MAXPATH_UNICODE, L"%ls", directory.c_str());
+    addFiles(origin, buffer, offset);
+  } catch (...) {
+    delete [] buffer;
+    buffer = NULL;
+  }
+  delete [] buffer;
+  m_Populated = true;
+}
+
+
+void DirectoryEntry::addFromBSA(const std::wstring &originName, std::wstring &directory, const std::wstring &fileName, int priority)
+{
+  FilesOrigin &origin = createOrigin(originName, directory, priority);
+
+  WIN32_FILE_ATTRIBUTE_DATA fileData;
+  if (::GetFileAttributesExW(fileName.c_str(), GetFileExInfoStandard, &fileData) == 0) {
+    throw windows_error("failed to determine file time");
+  }
+
+  BSA::Archive archive;
+  BSA::EErrorCode res = archive.read(ToString(fileName, false).c_str());
+  if ((res != BSA::ERROR_NONE) && (res != BSA::ERROR_INVALIDHASHES)) {
+    std::ostringstream stream;
+    stream << "invalid bsa file: " << ToString(fileName, false) << " errorcode " << res << " - " << ::GetLastError();
+    throw std::runtime_error(stream.str());
+  }
+  size_t namePos = fileName.find_last_of(L"\\/");
+  if (namePos == std::wstring::npos) {
+    namePos = 0;
+  } else {
+    ++namePos;
+  }
+
+  addFiles(origin, archive.getRoot(), fileData.ftLastWriteTime, fileName.substr(namePos));
+  m_Populated = true;
+}
+
+
+void DirectoryEntry::addFiles(FilesOrigin &origin, wchar_t *buffer, int bufferOffset)
+{
+  WIN32_FIND_DATAW findData;
+
+  _snwprintf(buffer + bufferOffset, MAXPATH_UNICODE - bufferOffset, L"\\*");
+  HANDLE searchHandle = ::FindFirstFileExW(buffer, FindExInfoStandard, &findData, FindExSearchNameMatch, NULL, FIND_FIRST_EX_CASE_SENSITIVE);
+  if (searchHandle != INVALID_HANDLE_VALUE) {
+    BOOL result = true;
+    while (result) {
+      if (findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
+        if ((wcscmp(findData.cFileName, L".") != 0) &&
+            (wcscmp(findData.cFileName, L"..") != 0)) {
+          int offset = _snwprintf(buffer + bufferOffset, MAXPATH_UNICODE, L"\\%ls", findData.cFileName);
+          // recurse into subdirectories
+          getSubDirectory(findData.cFileName, true, origin.getID())->addFiles(origin, buffer, bufferOffset + offset);
+        }
+      } else {
+        insert(findData.cFileName, origin, findData.ftLastWriteTime, L"");
+      }
+      result = ::FindNextFileW(searchHandle, &findData);
+    }
+  }
+  ::FindClose(searchHandle);
+}
+
+
+void DirectoryEntry::addFiles(FilesOrigin &origin, BSA::Folder::Ptr archiveFolder, FILETIME &fileTime, const std::wstring &archiveName)
+{
+  // add files
+  for (unsigned int fileIdx = 0; fileIdx < archiveFolder->getNumFiles(); ++fileIdx) {
+    BSA::File::Ptr file = archiveFolder->getFile(fileIdx);
+    insert(ToWString(file->getName(), true), origin, fileTime, archiveName);
+  }
+
+  // recurse into subdirectories
+  for (unsigned int folderIdx = 0; folderIdx < archiveFolder->getNumSubFolders(); ++folderIdx) {
+    BSA::Folder::Ptr folder = archiveFolder->getSubFolder(folderIdx);
+    DirectoryEntry *folderEntry = getSubDirectoryRecursive(ToWString(folder->getName(), true), true, origin.getID());
+
+    folderEntry->addFiles(origin, folder, fileTime, archiveName);
+  }
+}
+
+
+void DirectoryEntry::removeFile(const std::wstring &filePath, int *origin)
+{
+  size_t pos = filePath.find_first_of(L"\\/");
+  if (pos == std::string::npos) {
+    this->remove(filePath, origin);
+  } else {
+    std::wstring dirName = filePath.substr(0, pos);
+    std::wstring rest = filePath.substr(pos + 1);
+    DirectoryEntry *entry = getSubDirectoryRecursive(dirName, false);
+    if (entry != NULL) {
+      entry->removeFile(rest, origin);
+    }
+  }
+}
+
+void DirectoryEntry::removeDirRecursive()
+{
+  while (!m_Files.empty()) {
+    m_FileRegister->removeFile(m_Files.begin()->second);
+  }
+
+  for (auto iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
+    (*iter)->removeDirRecursive();
+  }
+  m_SubDirectories.clear();
+}
+
+void DirectoryEntry::removeDir(const std::wstring &path)
+{
+  size_t pos = path.find_first_of(L"\\/");
+  if (pos == std::string::npos) {
+    for (auto iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
+      if (_wcsicmp((*iter)->getName().c_str(), path.c_str()) == 0) {
+        (*iter)->removeDirRecursive();
+        m_SubDirectories.erase(iter);
+        break;
+      }
+    }
+  } else {
+    std::wstring dirName = path.substr(0, pos);
+    std::wstring rest = path.substr(pos + 1);
+    DirectoryEntry *entry = getSubDirectoryRecursive(dirName, false);
+    if (entry != NULL) {
+      entry->removeDir(rest);
+    }
+  }
+}
+
+
+void DirectoryEntry::insertFile(const std::wstring &filePath, FilesOrigin &origin, FILETIME fileTime)
+{
+  size_t pos = filePath.find_first_of(L"\\/");
+  if (pos == std::string::npos) {
+    this->insert(filePath, origin, fileTime, L"");
+  } else {
+    std::wstring dirName = filePath.substr(0, pos);
+    std::wstring rest = filePath.substr(pos + 1);
+    getSubDirectoryRecursive(dirName, true, origin.getID())->insertFile(rest, origin, fileTime);
+  }
+}
+
+
+void DirectoryEntry::removeFile(FileEntry::Index index)
+{
+  if (m_Files.size() != 0) {
+    auto iter = std::find_if(m_Files.begin(), m_Files.end(),
+                             [&index](const std::pair<std::wstring, FileEntry::Index> &iter) -> bool {
+                               return iter.second == index; } );
+    if (iter != m_Files.end()) {
+      m_Files.erase(iter);
+    } else {
+      log("file \"%ls\" not in directory \"%ls\"",
+          m_FileRegister->getFile(index)->getName().c_str(),
+          this->getName().c_str());
+    }
+  } else {
+    log("file \"%ls\" not in directory \"%ls\", directory empty",
+        m_FileRegister->getFile(index)->getName().c_str(),
+        this->getName().c_str());
+  }
+}
+
+
+int DirectoryEntry::anyOrigin() const
+{
+  bool ignore;
+  for (auto iter = m_Files.begin(); iter != m_Files.end(); ++iter) {
+    FileEntry *entry = m_FileRegister->getFile(iter->second);
+    if (!entry->isFromArchive()) {
+      return entry->getOrigin(ignore);
+    }
+  }
+
+  // if we got here, no file directly within this directory is a valid indicator for a mod, thus
+  // we continue looking in subdirectories
+  for (std::vector<DirectoryEntry*>::const_iterator iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
+    int res = (*iter)->anyOrigin();
+    if (res != -1){
+      return res;
+    }
+  }
+  return m_Origin;
+}
+
+
+bool DirectoryEntry::originExists(const std::wstring &name) const
+{
+  return m_OriginConnection->exists(name);
+}
+
+
+FilesOrigin &DirectoryEntry::getOriginByID(int ID) const
+{
+  return m_OriginConnection->getByID(ID);
+}
+
+
+FilesOrigin &DirectoryEntry::getOriginByName(const std::wstring &name) const
+{
+  return m_OriginConnection->getByName(name);
+}
+
+
+int DirectoryEntry::getOrigin(const std::wstring &path, bool &archive)
+{
+  const DirectoryEntry *directory = NULL;
+  const FileEntry *file = searchFile(path, &directory);
+  if (file != NULL) {
+    return file->getOrigin(archive);
+  } else {
+    if (directory != NULL) {
+      return directory->anyOrigin();
+    } else {
+      return -1;
+    }
+  }
+}
+
+std::vector<FileEntry*> DirectoryEntry::getFiles() const
+{
+  std::vector<FileEntry*> result;
+  for (auto iter = m_Files.begin(); iter != m_Files.end(); ++iter) {
+    result.push_back(m_FileRegister->getFile(iter->second));
+  }
+  return result;
+}
+
+
+const FileEntry *DirectoryEntry::searchFile(const std::wstring &path, const DirectoryEntry **directory) const
+{
+  if (directory != NULL) {
+    *directory = NULL;
+  }
+
+  if ((path.length() == 0) ||
+      (path == L"*")) {
+    // no file name -> the path ended on a (back-)slash
+    *directory = this;
+
+    return NULL;
+  }
+
+  size_t len =  path.find_first_of(L"\\/");
+
+  if (len == std::string::npos) {
+    // no more path components
+    auto iter = m_Files.find(path);
+    if (iter != m_Files.end()) {
+      return m_FileRegister->getFile(iter->second);
+    } else if (directory != NULL) {
+      DirectoryEntry *temp = findSubDirectory(path);
+      if (temp != NULL) {
+        *directory = temp;
+      }
+    }
+  } else {
+    // file is in in a subdirectory, recurse into the matching subdirectory
+    std::wstring pathComponent = path.substr(0, len);
+    DirectoryEntry *temp = findSubDirectory(pathComponent);
+    if (temp != NULL) {
+      return temp->searchFile(path.substr(len + 1), directory);
+    }
+  }
+  return NULL;
+}
+
+/*
+void DirectoryEntry::sortOrigins()
+{
+  for (std::set<FileEntry>::iterator iter = m_Files.begin(); iter != m_Files.end(); ++iter) {
+    const_cast<FileEntry&>(*iter).sortOrigins();
+  }
+  for (std::vector<DirectoryEntry*>::iterator iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
+    (*iter)->sortOrigins();
+  }
+}
+*/
+
+DirectoryEntry *DirectoryEntry::findSubDirectory(const std::wstring &name) const
+{
+  for (std::vector<DirectoryEntry*>::const_iterator iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
+    if (_wcsicmp((*iter)->getName().c_str(), name.c_str()) == 0) {
+      return *iter;
+    }
+  }
+  return NULL;
+}
+
+
+
+const FileEntry *DirectoryEntry::findFile(const std::wstring &name)
+{
+  auto iter = m_Files.find(name);
+  if (iter != m_Files.end()) {
+    return m_FileRegister->getFile(iter->second);
+  } else {
+    return NULL;
+  }
+}
+
+DirectoryEntry *DirectoryEntry::getSubDirectory(const std::wstring &name, bool create, int originID)
+{
+  for (std::vector<DirectoryEntry*>::iterator iter = m_SubDirectories.begin(); iter != m_SubDirectories.end(); ++iter) {
+    if (_wcsicmp((*iter)->getName().c_str(), name.c_str()) == 0) {
+      return *iter;
+    }
+  }
+  if (create) {
+    std::vector<DirectoryEntry*>::iterator iter = m_SubDirectories.insert(m_SubDirectories.end(),
+        new DirectoryEntry(name, this, originID, m_FileRegister, m_OriginConnection));
+    return *iter;
+  } else {
+    return NULL;
+  }
+}
+
+
+DirectoryEntry *DirectoryEntry::getSubDirectoryRecursive(const std::wstring &path, bool create, int originID)
+{
+  if (path.length() == 0) {
+    // path ended with a backslash?
+    return this;
+  }
+
+  size_t pos = path.find_first_of(L"\\/");
+  if (pos == std::wstring::npos) {
+    return getSubDirectory(path, create);
+  } else {
+    DirectoryEntry *nextChild = getSubDirectory(path.substr(0, pos), create, originID);
+    if (nextChild == NULL) {
+      return NULL;
+    } else {
+      return nextChild->getSubDirectoryRecursive(path.substr(pos + 1), create, originID);
+    }
+  }
+}
+
+
+
+FileRegister::FileRegister(boost::shared_ptr<OriginConnection> originConnection)
+  : m_OriginConnection(originConnection)
+{
+}
+
+FileEntry::Index FileRegister::generateIndex()
+{
+  static FileEntry::Index sIndex = 0;
+  return sIndex++;
+}
+
+bool FileRegister::indexValid(FileEntry::Index index) const
+{
+  return m_Files.find(index) != m_Files.end();
+}
+
+FileEntry &FileRegister::createFile(const std::wstring &name, DirectoryEntry *parent)
+{
+  FileEntry::Index index = generateIndex();
+  m_Files[index] = FileEntry(index, name, parent);
+  return m_Files[index];
+}
+
+
+FileEntry *FileRegister::getFile(FileEntry::Index index)
+{
+  auto iter = m_Files.find(index);
+  if (iter != m_Files.end()) {
+    return &iter->second;
+  }
+  return NULL;
+}
+
+
+void FileRegister::unregisterFile(FileEntry &file)
+{
+  bool ignore;
+  // unregister from origin
+  int originID = file.getOrigin(ignore);
+  m_OriginConnection->getByID(originID).removeFile(file.getIndex());
+  const std::vector<int> &alternatives = file.getAlternatives();
+  for (auto iter = alternatives.begin(); iter != alternatives.end(); ++iter) {
+    m_OriginConnection->getByID(*iter).removeFile(file.getIndex());
+  }
+
+  // unregister from directory
+  if (file.getParent() != NULL) {
+    file.getParent()->removeFile(file.getIndex());
+  }
+}
+
+
+void FileRegister::removeFile(FileEntry::Index index)
+{
+  auto iter = m_Files.find(index);
+  if (iter != m_Files.end()) {
+    unregisterFile(iter->second);
+    m_Files.erase(index);
+  }
+}
+
+void FileRegister::removeOrigin(FileEntry::Index index, int originID)
+{
+  auto iter = m_Files.find(index);
+  if (iter != m_Files.end()) {
+    if (iter->second.removeOrigin(originID)) {
+      unregisterFile(iter->second);
+    }
+  }
+}
+
+void FileRegister::sortOrigins()
+{
+  for (auto iter = m_Files.begin(); iter != m_Files.end(); ++iter) {
+    iter->second.sortOrigins();
+  }
+}
+} // namespace MOShared