-/*

- * Copyright (c) 2010 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- */

-

-#include <Swiften/Avatars/AvatarFileStorage.h>

-

-#include <iostream>

-#include <boost/filesystem/fstream.hpp>

-#include <boost/filesystem.hpp>

-

-#include <Swiften/Base/foreach.h>

-#include <Swiften/Base/String.h>

-#include <Swiften/StringCodecs/SHA1.h>

-#include <Swiften/StringCodecs/Hexify.h>

-

-namespace Swift {

-

-AvatarFileStorage::AvatarFileStorage(const boost::filesystem::path& avatarsDir, const boost::filesystem::path& avatarsFile) : avatarsDir(avatarsDir), avatarsFile(avatarsFile) {

- if (boost::filesystem::exists(avatarsFile)) {

- try {

- boost::filesystem::ifstream file(avatarsFile);

- std::string line;

- if (file.is_open()) {

- while (!file.eof()) {

- getline(file, line);

- std::pair<std::string, std::string> r = String::getSplittedAtFirst(line, ' ');

- JID jid(r.second);

- if (jid.isValid()) {

- jidAvatars.insert(std::make_pair(jid, r.first));

- }

- else if (!r.first.empty() || !r.second.empty()) {

- std::cerr << "Invalid entry in avatars file: " << r.second << std::endl;

- }

- }

- }

- }

- catch (...) {

- std::cerr << "Error reading avatars file" << std::endl;

- }

- }

-}

-

-bool AvatarFileStorage::hasAvatar(const std::string& hash) const {

- return boost::filesystem::exists(getAvatarPath(hash));

-}

-

-void AvatarFileStorage::addAvatar(const std::string& hash, const ByteArray& avatar) {

- assert(Hexify::hexify(SHA1::getHash(avatar)) == hash);

-

- boost::filesystem::path avatarPath = getAvatarPath(hash);

- if (!boost::filesystem::exists(avatarPath.parent_path())) {

- try {

- boost::filesystem::create_directories(avatarPath.parent_path());

- }

- catch (const boost::filesystem::filesystem_error& e) {

- std::cerr << "ERROR: " << e.what() << std::endl;

- }

- }

- boost::filesystem::ofstream file(avatarPath, boost::filesystem::ofstream::binary|boost::filesystem::ofstream::out);

- file.write(reinterpret_cast<const char*>(avatar.getData()), static_cast<std::streamsize>(avatar.getSize()));

- file.close();

-}

-

-boost::filesystem::path AvatarFileStorage::getAvatarPath(const std::string& hash) const {

- return avatarsDir / hash;

-}

-

-ByteArray AvatarFileStorage::getAvatar(const std::string& hash) const {

- ByteArray data;

- data.readFromFile(getAvatarPath(hash).string());

- return data;

-}

-

-void AvatarFileStorage::setAvatarForJID(const JID& jid, const std::string& hash) {

- std::pair<JIDAvatarMap::iterator, bool> r = jidAvatars.insert(std::make_pair(jid, hash));

- if (r.second) {

- saveJIDAvatars();

- }

- else if (r.first->second != hash) {

- r.first->second = hash;

- saveJIDAvatars();

- }

-}

-

-std::string AvatarFileStorage::getAvatarForJID(const JID& jid) const {

- JIDAvatarMap::const_iterator i = jidAvatars.find(jid);

- return i == jidAvatars.end() ? "" : i->second;

-}

-

-void AvatarFileStorage::saveJIDAvatars() {

- try {

- boost::filesystem::ofstream file(avatarsFile);

- for (JIDAvatarMap::const_iterator i = jidAvatars.begin(); i != jidAvatars.end(); ++i) {

- file << i->second << " " << i->first.toString() << std::endl;

- }

- file.close();

- }

- catch (...) {

- std::cerr << "Error writing avatars file" << std::endl;

- }

-}

-

-}

-/*

- * Copyright (c) 2010 Remko Tronçon

- * Licensed under the GNU General Public License v3.

- * See Documentation/Licenses/GPLv3.txt for more information.

- */

-

-#pragma once

-

-#include <map>

-#include <string>

-#include <boost/filesystem/path.hpp>

-

-#include <Swiften/JID/JID.h>

-#include "Swiften/Base/ByteArray.h"

-#include "Swiften/Avatars/AvatarStorage.h"

-

-namespace Swift {

- class AvatarFileStorage : public AvatarStorage {

- public:

- AvatarFileStorage(const boost::filesystem::path& avatarsDir, const boost::filesystem::path& avatarsFile);

-

- virtual bool hasAvatar(const std::string& hash) const;

- virtual void addAvatar(const std::string& hash, const ByteArray& avatar);

- virtual ByteArray getAvatar(const std::string& hash) const;

-

- virtual boost::filesystem::path getAvatarPath(const std::string& hash) const;

-

- virtual void setAvatarForJID(const JID& jid, const std::string& hash);

- virtual std::string getAvatarForJID(const JID& jid) const;

-

- private:

- void saveJIDAvatars();

-

- private:

- boost::filesystem::path avatarsDir;

- boost::filesystem::path avatarsFile;

- typedef std::map<JID, std::string> JIDAvatarMap;

- JIDAvatarMap jidAvatars;

- };

-

-}

- "AvatarFileStorage.cpp",