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
103
104
105
106
107
108
|
#pragma once
#include "types.hpp"
#include "WebPagePreview.hpp"
#include "StringView.hpp"
#include "utils.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 StaticImage;
class WebPagePreview;
struct ContentByUrlResult
{
enum class Type
{
CACHED,
DOWNLOADING,
FAILED_DOWNLOAD
};
enum class CachedType
{
NONE,
STATIC_IMAGE,
GIF,
WEB_PAGE_PREVIEW
};
ContentByUrlResult() : staticImage(nullptr), type(Type::DOWNLOADING), cachedType(CachedType::NONE) {}
ContentByUrlResult(StaticImage *_staticImage, Type _type) : staticImage(_staticImage), type(_type), cachedType(CachedType::STATIC_IMAGE) {}
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
{
StaticImage *staticImage;
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)>;
using CreateStaticImageFunc = std::function<StaticImage*(const boost::filesystem::path &filepath)>;
class Cache
{
DISABLE_COPY(Cache)
public:
// @createGifFunc can't be null
// @createStaticImageFunc can't be null
Cache(CreateGifFunc createGifFunc, CreateStaticImageFunc createStaticImageFunc);
~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;
CreateStaticImageFunc createStaticImageFunc;
};
}
|
