1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
#pragma once
#include "types.hpp"
#include "WebPagePreview.hpp"
#include "StringView.hpp"
#include <boost/filesystem/path.hpp>
#include <string>
#include <unordered_map>
#include <thread>
#include <mutex>
#include <vector>
#include <chrono>
#include <functional>
namespace TinyProcessLib
{
class Process;
}
namespace dchat
{
class Gif;
class WebPagePreview;
struct ContentByUrlResult
{
enum class Type
{
CACHED,
DOWNLOADING,
FAILED_DOWNLOAD
};
enum class CachedType
{
NONE,
TEXTURE_FILEPATH,
GIF,
WEB_PAGE_PREVIEW
};
ContentByUrlResult() : textureFilePath(nullptr), type(Type::DOWNLOADING), cachedType(CachedType::NONE) {}
ContentByUrlResult(boost::filesystem::path *_textureFilePath, Type _type) : textureFilePath(_textureFilePath), type(_type), cachedType(CachedType::TEXTURE_FILEPATH) {}
ContentByUrlResult(Gif *_gif, Type _type) : gif(_gif), type(_type), cachedType(CachedType::GIF) {}
ContentByUrlResult(WebPagePreview *_webPagePreview, Type _type) : webPagePreview(_webPagePreview), type(_type), cachedType(CachedType::WEB_PAGE_PREVIEW) {}
// @texture is null if @type is DOWNLOADING or FAILED_DOWNLOAD
union
{
boost::filesystem::path *textureFilePath;
Gif *gif;
WebPagePreview *webPagePreview;
};
Type type;
CachedType cachedType;
i64 lastAccessed;
};
using LoadBindsCallbackFunc = std::function<void(const std::string &key, const std::string &value)>;
// @fileContent contains data allocated with new[], deallocate it with delete[] fileContent.data;
// Returned gif should be allocated with @new
using CreateGifFunc = std::function<Gif*(StringView fileContent)>;
class Cache
{
public:
// @createGifFunc can't be nullptr
Cache(CreateGifFunc createGifFunc);
~Cache();
// Creates directory if it doesn't exist (recursively). Throws boost exception on failure
static boost::filesystem::path getDchatDir();
// Creates directory if it doesn't exist (recursively). Throws boost exception on failure
static boost::filesystem::path getImagesDir();
// @callbackFunc can't be nullptr
static void loadBindsFromFile(LoadBindsCallbackFunc callbackFunc);
static void replaceBindsInFile(const std::unordered_map<std::string, std::string> &binds);
// Get cached content or download it.
// Default download file limit is 12MB
// Returns ContentByUrlResult describing texture status.
const ContentByUrlResult getContentByUrl(const std::string &url, int downloadLimitBytes = 12582912);
private:
struct ImageDownloadInfo
{
TinyProcessLib::Process *process;
std::string url;
};
std::thread downloadWaitThread;
std::thread checkContentAccessTimeThread;
std::vector<ImageDownloadInfo> imageDownloadProcesses;
std::vector<ImageDownloadInfo> imageDownloadProcessesQueue;
std::mutex imageDownloadMutex;
bool alive;
std::unordered_map<std::string, ContentByUrlResult> contentUrlCache;
CreateGifFunc createGifFunc;
};
}
|
