UpdateFetcher.h

Go to the documentation of this file.

/*
 * This file is part of the TrinityCore Project. See AUTHORS file for Copyright information
 *
 * 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 2 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/>.
 */
 
#ifndef UpdateFetcher_h__
#define UpdateFetcher_h__
 
#include "Define.h"
#include "DatabaseEnvFwd.h"
#include <set>
#include <string>
#include <unordered_map>
#include <vector>
 
namespace boost
{
 namespace filesystem
 {
 class path;
 }
}
 
struct TC_DATABASE_API UpdateResult
{
 UpdateResult()
 : updated(0), recent(0), archived(0) { }
 
 UpdateResult(size_t const updated_, size_t const recent_, size_t const archived_)
 : updated(updated_), recent(recent_), archived(archived_) { }
 
 size_t updated;
 size_t recent;
 size_t archived;
};
 
class TC_DATABASE_API UpdateFetcher
{
 typedef boost::filesystem::path Path;
 
public:
 UpdateFetcher(Path const& updateDirectory,
 std::function<void(std::string const&)> const& apply,
 std::function<void(Path const& path)> const& applyFile,
 std::function<QueryResult(std::string const&)> const& retrieve);
 ~UpdateFetcher();
 
 UpdateResult Update(bool const redundancyChecks, bool const allowRehash,
 bool const archivedRedundancy, int32 const cleanDeadReferencesMaxCount) const;
 
private:
 enum UpdateMode
 {
 MODE_APPLY,
 MODE_REHASH
 };
 
 enum State
 {
 RELEASED,
 ARCHIVED
 };
 
 struct AppliedFileEntry
 {
 AppliedFileEntry(std::string const& name_, std::string const& hash_, State state_, uint64 timestamp_)
 : name(name_), hash(hash_), state(state_), timestamp(timestamp_) { }
 
 std::string const name;
 
 std::string const hash;
 
 State const state;
 
 uint64 const timestamp;
 
 static inline State StateConvert(std::string const& state)
 {
 return (state == "RELEASED") ? RELEASED : ARCHIVED;
 }
 
 static inline std::string StateConvert(State const state)
 {
 return (state == RELEASED) ? "RELEASED" : "ARCHIVED";
 }
 
 std::string GetStateAsString() const
 {
 return StateConvert(state);
 }
 };
 
 struct DirectoryEntry;
 
 typedef std::pair<Path, State> LocaleFileEntry;
 
 struct PathCompare
 {
 bool operator()(LocaleFileEntry const& left, LocaleFileEntry const& right) const;
 };
 
 typedef std::set<LocaleFileEntry, PathCompare> LocaleFileStorage;
 typedef std::unordered_map<std::string, std::string> HashToFileNameStorage;
 typedef std::unordered_map<std::string, AppliedFileEntry> AppliedFileStorage;
 typedef std::vector<UpdateFetcher::DirectoryEntry> DirectoryStorage;
 
 LocaleFileStorage GetFileList() const;
 void FillFileListRecursively(Path const& path, LocaleFileStorage& storage,
 State const state, uint32 const depth) const;
 
 DirectoryStorage ReceiveIncludedDirectories() const;
 AppliedFileStorage ReceiveAppliedFiles() const;
 
 std::string ReadSQLUpdate(Path const& file) const;
 
 uint32 Apply(Path const& path) const;
 
 void UpdateEntry(AppliedFileEntry const& entry, uint32 const speed = 0) const;
 void RenameEntry(std::string const& from, std::string const& to) const;
 void CleanUp(AppliedFileStorage const& storage) const;
 
 void UpdateState(std::string const& name, State const state) const;
 
 std::unique_ptr<Path> const _sourceDirectory;
 
 std::function<void(std::string const&)> const _apply;
 std::function<void(Path const& path)> const _applyFile;
 std::function<QueryResult(std::string const&)> const _retrieve;
};
 
#endif // UpdateFetcher_h__