132 lines
		
	
	
	
		
			3.3 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			132 lines
		
	
	
	
		
			3.3 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /*
 | |
| This file is part of Telegram Desktop,
 | |
| the official desktop application for the Telegram messaging service.
 | |
| 
 | |
| For license and copyright information please follow this link:
 | |
| https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
 | |
| */
 | |
| #include "window/notifications_utilities.h"
 | |
| 
 | |
| #include "window/main_window.h"
 | |
| #include "base/platform/base_platform_file_utilities.h"
 | |
| #include "base/random.h"
 | |
| #include "core/application.h"
 | |
| #include "data/data_peer.h"
 | |
| #include "ui/empty_userpic.h"
 | |
| #include "styles/style_window.h"
 | |
| 
 | |
| namespace Window {
 | |
| namespace Notifications {
 | |
| namespace {
 | |
| 
 | |
| // Delete notify photo file after 1 minute of not using.
 | |
| constexpr int kNotifyDeletePhotoAfterMs = 60000;
 | |
| 
 | |
| } // namespace
 | |
| 
 | |
| CachedUserpics::CachedUserpics(Type type)
 | |
| : _type(type)
 | |
| , _clearTimer([=] { clear(); }) {
 | |
| 	QDir().mkpath(cWorkingDir() + qsl("tdata/temp"));
 | |
| }
 | |
| 
 | |
| CachedUserpics::~CachedUserpics() {
 | |
| 	if (_someSavedFlag) {
 | |
| 		for (const auto &item : std::as_const(_images)) {
 | |
| 			QFile(item.path).remove();
 | |
| 		}
 | |
| 
 | |
| 		// This works about 1200ms on Windows for a folder with one image O_o
 | |
| 		//		base::Platform::DeleteDirectory(cWorkingDir() + qsl("tdata/temp"));
 | |
| 	}
 | |
| }
 | |
| 
 | |
| QString CachedUserpics::get(
 | |
| 		const InMemoryKey &key,
 | |
| 		not_null<PeerData*> peer,
 | |
| 		std::shared_ptr<Data::CloudImageView> &view) {
 | |
| 	auto ms = crl::now();
 | |
| 	auto i = _images.find(key);
 | |
| 	if (i != _images.cend()) {
 | |
| 		if (i->until) {
 | |
| 			i->until = ms + kNotifyDeletePhotoAfterMs;
 | |
| 			clearInMs(-kNotifyDeletePhotoAfterMs);
 | |
| 		}
 | |
| 	} else {
 | |
| 		Image v;
 | |
| 		if (key.first) {
 | |
| 			v.until = ms + kNotifyDeletePhotoAfterMs;
 | |
| 			clearInMs(-kNotifyDeletePhotoAfterMs);
 | |
| 		} else {
 | |
| 			v.until = 0;
 | |
| 		}
 | |
| 		v.path = u"%1tdata/temp/%2.png"_q.arg(
 | |
| 			cWorkingDir(),
 | |
| 			QString::number(base::RandomValue<uint64>(), 16));
 | |
| 		if (key.first || key.second) {
 | |
| 			if (peer->isSelf()) {
 | |
| 				const auto method = (_type == Type::Rounded)
 | |
| 					? Ui::EmptyUserpic::GenerateSavedMessagesRounded
 | |
| 					: Ui::EmptyUserpic::GenerateSavedMessages;
 | |
| 				method(st::notifyMacPhotoSize).save(v.path, "PNG");
 | |
| 			} else if (peer->isRepliesChat()) {
 | |
| 				const auto method = (_type == Type::Rounded)
 | |
| 					? Ui::EmptyUserpic::GenerateRepliesMessagesRounded
 | |
| 					: Ui::EmptyUserpic::GenerateRepliesMessages;
 | |
| 				method(st::notifyMacPhotoSize).save(v.path, "PNG");
 | |
| 			} else if (_type == Type::Rounded) {
 | |
| 				peer->saveUserpicRounded(view, v.path, st::notifyMacPhotoSize);
 | |
| 			} else {
 | |
| 				peer->saveUserpic(view, v.path, st::notifyMacPhotoSize);
 | |
| 			}
 | |
| 		} else {
 | |
| 			LogoNoMargin().save(v.path, "PNG");
 | |
| 		}
 | |
| 		i = _images.insert(key, v);
 | |
| 		_someSavedFlag = true;
 | |
| 	}
 | |
| 	return i->path;
 | |
| }
 | |
| 
 | |
| crl::time CachedUserpics::clear(crl::time ms) {
 | |
| 	crl::time result = 0;
 | |
| 	for (auto i = _images.begin(); i != _images.end();) {
 | |
| 		if (!i->until) {
 | |
| 			++i;
 | |
| 			continue;
 | |
| 		}
 | |
| 		if (i->until <= ms) {
 | |
| 			QFile(i->path).remove();
 | |
| 			i = _images.erase(i);
 | |
| 		} else {
 | |
| 			if (!result) {
 | |
| 				result = i->until;
 | |
| 			} else {
 | |
| 				accumulate_min(result, i->until);
 | |
| 			}
 | |
| 			++i;
 | |
| 		}
 | |
| 	}
 | |
| 	return result;
 | |
| }
 | |
| 
 | |
| void CachedUserpics::clearInMs(int ms) {
 | |
| 	if (ms < 0) {
 | |
| 		ms = -ms;
 | |
| 		if (_clearTimer.isActive() && _clearTimer.remainingTime() <= ms) {
 | |
| 			return;
 | |
| 		}
 | |
| 	}
 | |
| 	_clearTimer.callOnce(ms);
 | |
| }
 | |
| 
 | |
| void CachedUserpics::clear() {
 | |
| 	auto ms = crl::now();
 | |
| 	auto minuntil = clear(ms);
 | |
| 	if (minuntil) {
 | |
| 		clearInMs(int(minuntil - ms));
 | |
| 	}
 | |
| }
 | |
| 
 | |
| } // namespace Notifications
 | |
| } // namespace Window
 | 
