blob: 557be1c75f6e9d27a8b20724801f6c0c4127f720 (
plain)
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2026
//
// 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/StarGiftId.h"
#include "td/telegram/AccessRights.h"
#include "td/telegram/DialogManager.h"
#include "td/telegram/MessageId.h"
#include "td/telegram/Td.h"
#include "td/utils/algorithm.h"
#include "td/utils/logging.h"
#include "td/utils/misc.h"
#include "td/utils/Slice.h"
#include "td/utils/SliceBuilder.h"
namespace td {
StarGiftId::StarGiftId(ServerMessageId server_message_id) {
if (server_message_id.is_valid()) {
type_ = Type::ForUser;
server_message_id_ = server_message_id;
} else if (server_message_id != ServerMessageId()) {
LOG(ERROR) << "Receive server message " << server_message_id.get();
}
}
StarGiftId::StarGiftId(DialogId dialog_id, int64 saved_id) {
if (dialog_id == DialogId()) {
return;
}
if (dialog_id.get_type() != DialogType::Channel || saved_id == 0) {
LOG(ERROR) << "Receive gift " << saved_id << " in " << dialog_id;
return;
}
type_ = Type::ForDialog;
dialog_id_ = dialog_id;
saved_id_ = saved_id;
}
StarGiftId StarGiftId::from_slug(const string &slug) {
StarGiftId result;
result.type_ = Type::Slug;
result.slug_ = slug;
return result;
}
StarGiftId::StarGiftId(const string &star_gift_id) {
if (star_gift_id.empty()) {
return;
}
if (star_gift_id[0] == '@') {
type_ = Type::Slug;
slug_ = star_gift_id.substr(1);
} else {
auto underscore_pos = star_gift_id.find('_');
if (underscore_pos == string::npos) {
type_ = Type::ForUser;
server_message_id_ = ServerMessageId(to_integer<int32>(star_gift_id));
} else {
type_ = Type::ForDialog;
dialog_id_ = DialogId(to_integer<int64>(star_gift_id));
saved_id_ = to_integer<int64>(Slice(star_gift_id).substr(underscore_pos + 1));
}
}
if (get_star_gift_id() != star_gift_id) {
*this = {};
}
}
telegram_api::object_ptr<telegram_api::InputSavedStarGift> StarGiftId::get_input_saved_star_gift(Td *td) const {
switch (type_) {
case Type::Empty:
return nullptr;
case Type::ForUser:
return telegram_api::make_object<telegram_api::inputSavedStarGiftUser>(server_message_id_.get());
case Type::ForDialog: {
auto input_peer = td->dialog_manager_->get_input_peer(dialog_id_, AccessRights::Read);
if (input_peer == nullptr) {
return nullptr;
}
return telegram_api::make_object<telegram_api::inputSavedStarGiftChat>(std::move(input_peer), saved_id_);
}
case Type::Slug:
return telegram_api::make_object<telegram_api::inputSavedStarGiftSlug>(slug_);
default:
UNREACHABLE();
return nullptr;
}
}
vector<telegram_api::object_ptr<telegram_api::InputSavedStarGift>> StarGiftId::get_input_saved_star_gifts(
Td *td, const vector<StarGiftId> &star_gift_ids) {
return transform(star_gift_ids,
[td](const StarGiftId &star_gift_id) { return star_gift_id.get_input_saved_star_gift(td); });
}
vector<StarGiftId> StarGiftId::get_star_gift_ids(const vector<string> &star_gift_ids) {
return transform(star_gift_ids, [](const string &star_gift_id) { return StarGiftId(star_gift_id); });
}
string StarGiftId::get_star_gift_id() const {
switch (type_) {
case Type::Empty:
return string();
case Type::ForUser:
return PSTRING() << server_message_id_.get();
case Type::ForDialog:
return PSTRING() << dialog_id_.get() << '_' << saved_id_;
case Type::Slug:
return PSTRING() << '@' << slug_;
default:
UNREACHABLE();
return string();
}
}
DialogId StarGiftId::get_dialog_id(const Td *td) const {
switch (type_) {
case Type::Empty:
return DialogId();
case Type::ForUser:
return td->dialog_manager_->get_my_dialog_id();
case Type::ForDialog:
return dialog_id_;
case Type::Slug:
return DialogId();
default:
UNREACHABLE();
return DialogId();
}
}
bool operator==(const StarGiftId &lhs, const StarGiftId &rhs) {
return lhs.type_ == rhs.type_ && lhs.server_message_id_ == rhs.server_message_id_ &&
lhs.dialog_id_ == rhs.dialog_id_ && lhs.saved_id_ == rhs.saved_id_ && lhs.slug_ == rhs.slug_;
}
StringBuilder &operator<<(StringBuilder &string_builder, const StarGiftId &star_gift_id) {
switch (star_gift_id.type_) {
case StarGiftId::Type::Empty:
return string_builder << "unknown gift";
case StarGiftId::Type::ForUser:
return string_builder << "user gift from " << MessageId(star_gift_id.server_message_id_);
case StarGiftId::Type::ForDialog:
return string_builder << star_gift_id.dialog_id_ << " gift " << star_gift_id.saved_id_;
case StarGiftId::Type::Slug:
return string_builder << "gift " << star_gift_id.slug_;
default:
UNREACHABLE();
return string_builder;
}
}
} // namespace td
|
