143 lines
		
	
	
	
		
			3.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			143 lines
		
	
	
	
		
			3.4 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 "api/api_toggling_media.h"
 | |
| 
 | |
| #include "apiwrap.h"
 | |
| #include "data/data_document.h"
 | |
| #include "data/data_file_origin.h"
 | |
| #include "data/data_session.h"
 | |
| #include "data/stickers/data_stickers.h"
 | |
| #include "window/window_session_controller.h"
 | |
| #include "main/main_session.h"
 | |
| 
 | |
| namespace Api {
 | |
| namespace {
 | |
| 
 | |
| template <typename ToggleRequestCallback, typename DoneCallback>
 | |
| void ToggleExistingMedia(
 | |
| 		not_null<DocumentData*> document,
 | |
| 		Data::FileOrigin origin,
 | |
| 		ToggleRequestCallback toggleRequest,
 | |
| 		DoneCallback &&done) {
 | |
| 	const auto api = &document->owner().session().api();
 | |
| 
 | |
| 	auto performRequest = [=](const auto &repeatRequest) -> void {
 | |
| 		const auto usedFileReference = document->fileReference();
 | |
| 		api->request(
 | |
| 			toggleRequest()
 | |
| 		).done(done).fail([=](const MTP::Error &error) {
 | |
| 			if (error.code() == 400
 | |
| 				&& error.type().startsWith(u"FILE_REFERENCE_"_q)) {
 | |
| 				auto refreshed = [=](const Data::UpdatedFileReferences &d) {
 | |
| 					if (document->fileReference() != usedFileReference) {
 | |
| 						repeatRequest(repeatRequest);
 | |
| 					}
 | |
| 				};
 | |
| 				api->refreshFileReference(origin, std::move(refreshed));
 | |
| 			}
 | |
| 		}).send();
 | |
| 	};
 | |
| 	performRequest(performRequest);
 | |
| }
 | |
| 
 | |
| } // namespace
 | |
| 
 | |
| void ToggleFavedSticker(
 | |
| 		not_null<Window::SessionController*> controller,
 | |
| 		not_null<DocumentData*> document,
 | |
| 		Data::FileOrigin origin) {
 | |
| 	ToggleFavedSticker(
 | |
| 		controller,
 | |
| 		document,
 | |
| 		std::move(origin),
 | |
| 		!document->owner().stickers().isFaved(document));
 | |
| }
 | |
| 
 | |
| void ToggleFavedSticker(
 | |
| 		not_null<Window::SessionController*> controller,
 | |
| 		not_null<DocumentData*> document,
 | |
| 		Data::FileOrigin origin,
 | |
| 		bool faved) {
 | |
| 	if (faved && !document->sticker()) {
 | |
| 		return;
 | |
| 	}
 | |
| 	const auto weak = base::make_weak(controller);
 | |
| 	auto done = [=] {
 | |
| 		document->owner().stickers().setFaved(weak.get(), document, faved);
 | |
| 	};
 | |
| 	ToggleExistingMedia(
 | |
| 		document,
 | |
| 		std::move(origin),
 | |
| 		[=, d = document] {
 | |
| 			return MTPmessages_FaveSticker(d->mtpInput(), MTP_bool(!faved));
 | |
| 		},
 | |
| 		std::move(done));
 | |
| }
 | |
| 
 | |
| void ToggleRecentSticker(
 | |
| 		not_null<DocumentData*> document,
 | |
| 		Data::FileOrigin origin,
 | |
| 		bool saved) {
 | |
| 	if (!document->sticker()) {
 | |
| 		return;
 | |
| 	}
 | |
| 	auto done = [=] {
 | |
| 		if (!saved) {
 | |
| 			document->owner().stickers().removeFromRecentSet(document);
 | |
| 		}
 | |
| 	};
 | |
| 	ToggleExistingMedia(
 | |
| 		document,
 | |
| 		std::move(origin),
 | |
| 		[=] {
 | |
| 			return MTPmessages_SaveRecentSticker(
 | |
| 				MTP_flags(MTPmessages_SaveRecentSticker::Flag(0)),
 | |
| 				document->mtpInput(),
 | |
| 				MTP_bool(!saved));
 | |
| 		},
 | |
| 		std::move(done));
 | |
| }
 | |
| 
 | |
| void ToggleSavedGif(
 | |
| 		Window::SessionController *controller,
 | |
| 		not_null<DocumentData*> document,
 | |
| 		Data::FileOrigin origin,
 | |
| 		bool saved) {
 | |
| 	if (saved && !document->isGifv()) {
 | |
| 		return;
 | |
| 	}
 | |
| 	const auto weak = base::make_weak(controller);
 | |
| 	auto done = [=] {
 | |
| 		if (saved) {
 | |
| 			document->owner().stickers().addSavedGif(weak.get(), document);
 | |
| 		}
 | |
| 	};
 | |
| 	ToggleExistingMedia(
 | |
| 		document,
 | |
| 		std::move(origin),
 | |
| 		[=, d = document] {
 | |
| 			return MTPmessages_SaveGif(d->mtpInput(), MTP_bool(!saved));
 | |
| 		},
 | |
| 		std::move(done));
 | |
| }
 | |
| 
 | |
| void ToggleSavedRingtone(
 | |
| 		not_null<DocumentData*> document,
 | |
| 		Data::FileOrigin origin,
 | |
| 		Fn<void()> &&done,
 | |
| 		bool saved) {
 | |
| 	ToggleExistingMedia(
 | |
| 		document,
 | |
| 		std::move(origin),
 | |
| 		[=, d = document] {
 | |
| 			return MTPaccount_SaveRingtone(d->mtpInput(), MTP_bool(!saved));
 | |
| 		},
 | |
| 		std::move(done));
 | |
| }
 | |
| 
 | |
| } // namespace Api
 | 
