tdlight/td/telegram/NotificationType.cpp
levlam e586b31a87 Add support for notifications about new calls.
GitOrigin-RevId: fc4f9c1a484832241f2059a1ec58af6e5abcb198
2018-12-18 23:59:35 +03:00

125 lines
3.3 KiB
C++

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2018
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#include "td/telegram/NotificationType.h"
#include "td/telegram/MessagesManager.h"
#include "td/telegram/Td.h"
namespace td {
class NotificationTypeMessage : public NotificationType {
bool can_be_delayed() const override {
return true;
}
MessageId get_message_id() const override {
return message_id_;
}
CallId get_call_id() const override {
return CallId();
}
td_api::object_ptr<td_api::NotificationType> get_notification_type_object(DialogId dialog_id) const override {
auto message_object = G()->td().get_actor_unsafe()->messages_manager_->get_message_object({dialog_id, message_id_});
if (message_object == nullptr) {
return nullptr;
}
return td_api::make_object<td_api::notificationTypeNewMessage>(std::move(message_object));
}
StringBuilder &to_string_builder(StringBuilder &string_builder) const override {
return string_builder << "NewMessageNotification[" << message_id_ << ']';
}
Type get_type() const override {
return Type::Message;
}
MessageId message_id_;
public:
explicit NotificationTypeMessage(MessageId message_id) : message_id_(message_id) {
}
};
class NotificationTypeSecretChat : public NotificationType {
bool can_be_delayed() const override {
return false;
}
MessageId get_message_id() const override {
return MessageId();
}
CallId get_call_id() const override {
return CallId();
}
td_api::object_ptr<td_api::NotificationType> get_notification_type_object(DialogId dialog_id) const override {
return td_api::make_object<td_api::notificationTypeNewSecretChat>();
}
StringBuilder &to_string_builder(StringBuilder &string_builder) const override {
return string_builder << "NewSecretChatNotification[]";
}
Type get_type() const override {
return Type::SecretChat;
}
public:
NotificationTypeSecretChat() {
}
};
class NotificationTypeCall : public NotificationType {
bool can_be_delayed() const override {
return false;
}
MessageId get_message_id() const override {
return MessageId::max();
}
CallId get_call_id() const override {
return call_id_;
}
td_api::object_ptr<td_api::NotificationType> get_notification_type_object(DialogId dialog_id) const override {
return td_api::make_object<td_api::notificationTypeNewCall>(call_id_.get());
}
StringBuilder &to_string_builder(StringBuilder &string_builder) const override {
return string_builder << "NewCallNotification[" << call_id_ << ']';
}
Type get_type() const override {
return Type::Call;
}
CallId call_id_;
public:
explicit NotificationTypeCall(CallId call_id) : call_id_(call_id) {
}
};
unique_ptr<NotificationType> create_new_message_notification(MessageId message_id) {
return make_unique<NotificationTypeMessage>(message_id);
}
unique_ptr<NotificationType> create_new_secret_chat_notification() {
return make_unique<NotificationTypeSecretChat>();
}
unique_ptr<NotificationType> create_new_call_notification(CallId call_id) {
return make_unique<NotificationTypeCall>(call_id);
}
} // namespace td