/* 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 "inline_bots/bot_attach_web_view.h" #include "api/api_blocked_peers.h" #include "api/api_common.h" #include "core/click_handler_types.h" #include "data/data_bot_app.h" #include "data/data_changes.h" #include "data/data_user.h" #include "data/data_file_origin.h" #include "data/data_document.h" #include "data/data_document_media.h" #include "data/data_session.h" #include "main/main_session.h" #include "main/main_domain.h" #include "storage/storage_domain.h" #include "info/profile/info_profile_values.h" #include "ui/boxes/confirm_box.h" #include "ui/chat/attach/attach_bot_webview.h" #include "ui/widgets/checkbox.h" #include "ui/widgets/dropdown_menu.h" #include "ui/widgets/popup_menu.h" #include "ui/widgets/menu/menu_item_base.h" #include "ui/text/text_utilities.h" #include "ui/effects/ripple_animation.h" #include "ui/painter.h" #include "window/themes/window_theme.h" #include "window/window_controller.h" #include "window/window_peer_menu.h" #include "window/window_session_controller.h" #include "webview/webview_interface.h" #include "core/application.h" #include "core/local_url_handlers.h" #include "ui/basic_click_handlers.h" #include "history/history.h" #include "history/history_item.h" #include "payments/payments_checkout_process.h" #include "storage/storage_account.h" #include "boxes/peer_list_controllers.h" #include "lang/lang_keys.h" #include "base/random.h" #include "base/timer_rpl.h" #include "apiwrap.h" #include "mainwidget.h" #include "styles/style_boxes.h" #include "styles/style_layers.h" #include "styles/style_menu_icons.h" #include namespace InlineBots { namespace { constexpr auto kProlongTimeout = 60 * crl::time(1000); constexpr auto kRefreshBotsTimeout = 60 * 60 * crl::time(1000); [[nodiscard]] DocumentData *ResolveIcon( not_null session, const MTPDattachMenuBot &data) { for (const auto &icon : data.vicons().v) { const auto document = icon.match([&]( const MTPDattachMenuBotIcon &data ) -> DocumentData* { if (data.vname().v == "default_static") { return session->data().processDocument(data.vicon()).get(); } return nullptr; }); if (document) { return document; } } return nullptr; } [[nodiscard]] PeerTypes ResolvePeerTypes( const QVector &types) { auto result = PeerTypes(); for (const auto &type : types) { result |= type.match([&](const MTPDattachMenuPeerTypeSameBotPM &) { return PeerType::SameBot; }, [&](const MTPDattachMenuPeerTypeBotPM &) { return PeerType::Bot; }, [&](const MTPDattachMenuPeerTypePM &) { return PeerType::User; }, [&](const MTPDattachMenuPeerTypeChat &) { return PeerType::Group; }, [&](const MTPDattachMenuPeerTypeBroadcast &) { return PeerType::Broadcast; }); } return result; } [[nodiscard]] std::optional ParseAttachBot( not_null session, const MTPAttachMenuBot &bot) { auto result = bot.match([&](const MTPDattachMenuBot &data) { const auto user = session->data().userLoaded(UserId(data.vbot_id())); const auto good = user && user->isBot() && user->botInfo->supportsAttachMenu; return good ? AttachWebViewBot{ .user = user, .icon = ResolveIcon(session, data), .name = qs(data.vshort_name()), .types = (data.vpeer_types() ? ResolvePeerTypes(data.vpeer_types()->v) : PeerTypes()), .inactive = data.is_inactive(), .inMainMenu = data.is_show_in_side_menu(), .inAttachMenu = data.is_show_in_attach_menu(), .disclaimerRequired = data.is_side_menu_disclaimer_needed(), .requestWriteAccess = data.is_request_write_access(), } : std::optional(); }); if (result && result->icon) { result->icon->forceToCache(true); } if (const auto icon = result->icon) { result->media = icon->createMediaView(); icon->save(Data::FileOrigin(), {}); } return result; } [[nodiscard]] PeerTypes PeerTypesFromNames( const std::vector &names) { auto result = PeerTypes(); for (const auto &name : names) { //, bots, groups, channels result |= (name == u"users"_q) ? PeerType::User : name == u"bots"_q ? PeerType::Bot : name == u"groups"_q ? PeerType::Group : name == u"channels"_q ? PeerType::Broadcast : PeerType(0); } return result; } void ShowChooseBox( not_null controller, PeerTypes types, Fn)> callback, rpl::producer titleOverride = nullptr) { const auto weak = std::make_shared>(); auto done = [=](not_null thread) mutable { if (const auto strong = *weak) { strong->closeBox(); } callback(thread); }; auto filter = [=](not_null thread) -> bool { const auto peer = thread->peer(); if (!Data::CanSend(thread, ChatRestriction::SendInline, false)) { return false; } else if (const auto user = peer->asUser()) { if (user->isBot()) { return (types & PeerType::Bot); } else { return (types & PeerType::User); } } else if (peer->isBroadcast()) { return (types & PeerType::Broadcast); } else { return (types & PeerType::Group); } }; auto initBox = [=](not_null box) { if (titleOverride) { box->setTitle(std::move(titleOverride)); } box->addButton(tr::lng_cancel(), [box] { box->closeBox(); }); }; *weak = controller->show(Box( std::make_unique( &controller->session(), std::move(done), std::move(filter)), std::move(initBox))); } [[nodiscard]] base::flat_set> &ActiveWebViews() { static auto result = base::flat_set>(); return result; } void FillDisclaimerBox(not_null box, Fn done) { const auto updateCheck = std::make_shared>(); const auto validateCheck = std::make_shared>(); const auto callback = [=](Fn close) { if (validateCheck && (*validateCheck)()) { done(); close(); } }; const auto padding = st::boxRowPadding; Ui::ConfirmBox(box, { .text = tr::lng_mini_apps_disclaimer_text( tr::now, Ui::Text::RichLangValue), .confirmed = callback, .confirmText = tr::lng_box_ok(), .labelPadding = QMargins(padding.left(), 0, padding.right(), 0), .title = tr::lng_mini_apps_disclaimer_title(), }); auto checkView = std::make_unique( st::defaultCheck, false, [=] { if (*updateCheck) { (*updateCheck)(); } }); const auto check = checkView.get(); const auto row = box->addRow( object_ptr( box.get(), tr::lng_mini_apps_disclaimer_button( lt_link, rpl::single(Ui::Text::Link( tr::lng_mini_apps_disclaimer_link(tr::now), tr::lng_mini_apps_tos_url(tr::now))), Ui::Text::WithEntities), st::urlAuthCheckbox, std::move(checkView)), { st::boxRowPadding.left(), st::boxRowPadding.left(), st::boxRowPadding.right(), 0, }); row->setAllowTextLines(); row->setClickHandlerFilter([=]( const ClickHandlerPtr &link, Qt::MouseButton button) { ActivateClickHandler(row, link, ClickContext{ .button = button, .other = QVariant::fromValue(ClickHandlerContext{ .show = box->uiShow(), }) }); return false; }); (*updateCheck) = [=] { row->update(); }; const auto showError = Ui::CheckView::PrepareNonToggledError( check, box->lifetime()); (*validateCheck) = [=] { if (check->checked()) { return true; } showError(); return false; }; } class BotAction final : public Ui::Menu::ItemBase { public: BotAction( not_null parent, const style::Menu &st, const AttachWebViewBot &bot, Fn callback); bool isEnabled() const override; not_null action() const override; [[nodiscard]] rpl::producer forceShown() const; void handleKeyPress(not_null e) override; private: void contextMenuEvent(QContextMenuEvent *e) override; QPoint prepareRippleStartPosition() const override; QImage prepareRippleMask() const override; int contentHeight() const override; void prepare(); void paint(Painter &p); const not_null _dummyAction; const style::Menu &_st; const AttachWebViewBot _bot; MenuBotIcon _icon; base::unique_qptr _menu; rpl::event_stream _forceShown; Ui::Text::String _text; int _textWidth = 0; const int _height; }; BotAction::BotAction( not_null parent, const style::Menu &st, const AttachWebViewBot &bot, Fn callback) : ItemBase(parent, st) , _dummyAction(new QAction(parent)) , _st(st) , _bot(bot) , _icon(this, _bot.media) , _height(_st.itemPadding.top() + _st.itemStyle.font->height + _st.itemPadding.bottom()) { setAcceptBoth(false); initResizeHook(parent->sizeValue()); setClickedCallback(std::move(callback)); _icon.move(_st.itemIconPosition); paintRequest( ) | rpl::start_with_next([=] { Painter p(this); paint(p); }, lifetime()); enableMouseSelecting(); prepare(); } void BotAction::paint(Painter &p) { const auto selected = isSelected(); if (selected && _st.itemBgOver->c.alpha() < 255) { p.fillRect(0, 0, width(), _height, _st.itemBg); } p.fillRect(0, 0, width(), _height, selected ? _st.itemBgOver : _st.itemBg); if (isEnabled()) { paintRipple(p, 0, 0); } p.setPen(selected ? _st.itemFgOver : _st.itemFg); _text.drawLeftElided( p, _st.itemPadding.left(), _st.itemPadding.top(), _textWidth, width()); } void BotAction::prepare() { _text.setMarkedText(_st.itemStyle, { _bot.name }); const auto textWidth = _text.maxWidth(); const auto &padding = _st.itemPadding; const auto goodWidth = padding.left() + textWidth + padding.right(); const auto w = std::clamp(goodWidth, _st.widthMin, _st.widthMax); _textWidth = w - (goodWidth - textWidth); setMinWidth(w); update(); } bool BotAction::isEnabled() const { return true; } not_null BotAction::action() const { return _dummyAction; } void BotAction::contextMenuEvent(QContextMenuEvent *e) { _menu = nullptr; _menu = base::make_unique_q( this, st::popupMenuWithIcons); _menu->addAction(tr::lng_bot_remove_from_menu(tr::now), [=] { _bot.user->session().attachWebView().removeFromMenu(_bot.user); }, &st::menuIconDelete); QObject::connect(_menu, &QObject::destroyed, [=] { _forceShown.fire(false); }); _forceShown.fire(true); _menu->popup(e->globalPos()); e->accept(); } QPoint BotAction::prepareRippleStartPosition() const { return mapFromGlobal(QCursor::pos()); } QImage BotAction::prepareRippleMask() const { return Ui::RippleAnimation::RectMask(size()); } int BotAction::contentHeight() const { return _height; } rpl::producer BotAction::forceShown() const { return _forceShown.events(); } void BotAction::handleKeyPress(not_null e) { if (!isSelected()) { return; } const auto key = e->key(); if (key == Qt::Key_Enter || key == Qt::Key_Return) { setClicked(Ui::Menu::TriggeredSource::Keyboard); } } } // namespace MenuBotIcon::MenuBotIcon( QWidget *parent, std::shared_ptr media) : RpWidget(parent) , _media(std::move(media)) { style::PaletteChanged( ) | rpl::start_with_next([=] { _image = QImage(); update(); }, lifetime()); setAttribute(Qt::WA_TransparentForMouseEvents); resize(st::menuIconAdmin.size()); show(); } void MenuBotIcon::paintEvent(QPaintEvent *e) { validate(); if (!_image.isNull()) { QPainter(this).drawImage(0, 0, _image); } } void MenuBotIcon::validate() { const auto ratio = style::DevicePixelRatio(); const auto wanted = size() * ratio; if (_mask.size() != wanted) { if (!_media || !_media->loaded()) { return; } auto icon = QSvgRenderer(_media->bytes()); _mask = QImage(wanted, QImage::Format_ARGB32_Premultiplied); _mask.setDevicePixelRatio(style::DevicePixelRatio()); _mask.fill(Qt::transparent); if (icon.isValid()) { auto p = QPainter(&_mask); icon.render(&p, rect()); p.end(); _mask = Images::Colored(std::move(_mask), Qt::white); } } if (_image.isNull()) { _image = style::colorizeImage(_mask, st::menuIconColor); } } bool PeerMatchesTypes( not_null peer, not_null bot, PeerTypes types) { if (const auto user = peer->asUser()) { return (user == bot) ? (types & PeerType::SameBot) : user->isBot() ? (types & PeerType::Bot) : (types & PeerType::User); } else if (peer->isBroadcast()) { return (types & PeerType::Broadcast); } return (types & PeerType::Group); } PeerTypes ParseChooseTypes(QStringView choose) { auto result = PeerTypes(); for (const auto &entry : choose.split(QChar(' '))) { if (entry == u"users"_q) { result |= PeerType::User; } else if (entry == u"bots"_q) { result |= PeerType::Bot; } else if (entry == u"groups"_q) { result |= PeerType::Group; } else if (entry == u"channels"_q) { result |= PeerType::Broadcast; } } return result; } struct AttachWebView::Context { base::weak_ptr controller; Dialogs::EntryState dialogsEntryState; Api::SendAction action; bool fromSwitch = false; bool fromMainMenu = false; bool fromBotApp = false; }; AttachWebView::AttachWebView(not_null session) : _session(session) , _refreshTimer([=] { requestBots(); }) { _refreshTimer.callEach(kRefreshBotsTimeout); } AttachWebView::~AttachWebView() { ActiveWebViews().remove(this); } void AttachWebView::request( not_null controller, const Api::SendAction &action, const QString &botUsername, const QString &startCommand) { if (botUsername.isEmpty()) { return; } const auto username = _bot ? _bot->username() : _botUsername; const auto context = LookupContext(controller, action); if (IsSame(_context, context) && username.toLower() == botUsername.toLower() && _startCommand == startCommand) { if (_panel) { _panel->requestActivate(); } return; } cancel(); _context = std::make_unique(context); _botUsername = botUsername; _startCommand = startCommand; resolve(); } Webview::ThemeParams AttachWebView::botThemeParams() { return Window::Theme::WebViewParams(); } bool AttachWebView::botHandleLocalUri(QString uri, bool keepOpen) { const auto local = Core::TryConvertUrlToLocal(uri); if (uri == local || Core::InternalPassportLink(local)) { return local.startsWith(u"tg://"_q); } else if (!local.startsWith(u"tg://"_q, Qt::CaseInsensitive)) { return false; } if (!keepOpen) { botClose(); } crl::on_main([=, shownUrl = _lastShownUrl] { const auto variant = QVariant::fromValue(ClickHandlerContext{ .attachBotWebviewUrl = shownUrl, }); UrlClickHandler::Open(local, variant); }); return true; } void AttachWebView::botHandleInvoice(QString slug) { Expects(_panel != nullptr); using Result = Payments::CheckoutResult; const auto weak = base::make_weak(_panel.get()); const auto reactivate = [=](Result result) { if (const auto strong = weak.get()) { strong->invoiceClosed(slug, [&] { switch (result) { case Result::Paid: return "paid"; case Result::Failed: return "failed"; case Result::Pending: return "pending"; case Result::Cancelled: return "cancelled"; } Unexpected("Payments::CheckoutResult value."); }()); } }; _panel->hideForPayment(); Payments::CheckoutProcess::Start(&_bot->session(), slug, reactivate); } void AttachWebView::botHandleMenuButton(Ui::BotWebView::MenuButton button) { Expects(_bot != nullptr); Expects(_panel != nullptr); using Button = Ui::BotWebView::MenuButton; const auto bot = _bot; switch (button) { case Button::OpenBot: botClose(); if (bot->session().windows().empty()) { Core::App().domain().activate(&bot->session().account()); } if (!bot->session().windows().empty()) { const auto window = bot->session().windows().front(); window->showPeerHistory(bot); window->window().activate(); } break; case Button::RemoveFromMenu: case Button::RemoveFromMainMenu: const auto attached = ranges::find( _attachBots, not_null{ _bot }, &AttachWebViewBot::user); const auto name = (attached != end(_attachBots)) ? attached->name : _bot->name(); const auto done = crl::guard(this, [=] { removeFromMenu(bot); botClose(); if (const auto active = Core::App().activeWindow()) { active->activate(); } }); const auto main = (button == Button::RemoveFromMainMenu); _panel->showBox(Ui::MakeConfirmBox({ (main ? tr::lng_bot_remove_from_side_menu_sure : tr::lng_bot_remove_from_menu_sure)( tr::now, lt_bot, Ui::Text::Bold(name), Ui::Text::WithEntities), done, })); break; } } void AttachWebView::botSendData(QByteArray data) { if (!_context || _context->fromSwitch || _context->fromBotApp || _context->fromMainMenu || _context->action.history->peer != _bot || _lastShownQueryId) { return; } const auto randomId = base::RandomValue(); _session->api().request(MTPmessages_SendWebViewData( _bot->inputUser, MTP_long(randomId), MTP_string(_lastShownButtonText), MTP_bytes(data) )).done([=](const MTPUpdates &result) { _session->api().applyUpdates(result); }).send(); crl::on_main(this, [=] { cancel(); }); } void AttachWebView::botSwitchInlineQuery( std::vector chatTypes, QString query) { const auto controller = _context ? _context->controller.get() : nullptr; const auto types = PeerTypesFromNames(chatTypes); if (!_bot || !_bot->isBot() || _bot->botInfo->inlinePlaceholder.isEmpty() || !controller) { return; } else if (!types) { if (_context->dialogsEntryState.key.owningHistory()) { controller->switchInlineQuery( _context->dialogsEntryState, _bot, query); } } else { const auto bot = _bot; const auto done = [=](not_null thread) { controller->switchInlineQuery(thread, bot, query); }; ShowChooseBox( controller, types, done, tr::lng_inline_switch_choose()); } crl::on_main(this, [=] { cancel(); }); } void AttachWebView::botCheckWriteAccess(Fn callback) { _session->api().request(MTPbots_CanSendMessage( _bot->inputUser )).done([=](const MTPBool &result) { callback(mtpIsTrue(result)); }).fail([=] { callback(false); }).send(); } void AttachWebView::botAllowWriteAccess(Fn callback) { _session->api().request(MTPbots_AllowSendMessage( _bot->inputUser )).done([=](const MTPUpdates &result) { _session->api().applyUpdates(result); callback(true); }).fail([=] { callback(false); }).send(); } void AttachWebView::botSharePhone(Fn callback) { const auto bot = _bot; const auto history = _bot->owner().history(_bot); if (_bot->isBlocked()) { const auto done = [=](bool success) { if (success && _bot == bot) { Assert(!_bot->isBlocked()); botSharePhone(callback); } else { callback(false); } }; _bot->session().api().blockedPeers().unblock( _bot, crl::guard(this, done)); return; } auto action = Api::SendAction(history); action.clearDraft = false; history->session().api().shareContact( _bot->session().user(), action, std::move(callback)); } void AttachWebView::botInvokeCustomMethod( Ui::BotWebView::CustomMethodRequest request) { const auto callback = request.callback; _bot->session().api().request(MTPbots_InvokeWebViewCustomMethod( _bot->inputUser, MTP_string(request.method), MTP_dataJSON(MTP_bytes(request.params)) )).done([=](const MTPDataJSON &result) { callback(result.data().vdata().v); }).fail([=](const MTP::Error &error) { callback(base::make_unexpected(error.type())); }).send(); } void AttachWebView::botClose() { crl::on_main(this, [=] { cancel(); }); } AttachWebView::Context AttachWebView::LookupContext( not_null controller, const Api::SendAction &action) { return { .controller = controller, .dialogsEntryState = controller->currentDialogsEntryState(), .action = action, }; } bool AttachWebView::IsSame( const std::unique_ptr &a, const Context &b) { // Check fields that are sent to API in bot attach webview requests. return a && (a->controller == b.controller) && (a->dialogsEntryState == b.dialogsEntryState) && (a->fromSwitch == b.fromSwitch) && (a->fromMainMenu == b.fromMainMenu) && (a->action.history == b.action.history) && (a->action.replyTo == b.action.replyTo) && (a->action.options.sendAs == b.action.options.sendAs) && (a->action.options.silent == b.action.options.silent); } void AttachWebView::request( not_null controller, const Api::SendAction &action, not_null bot, const WebViewButton &button) { requestWithOptionalConfirm( bot, button, LookupContext(controller, action), button.fromAttachMenu ? nullptr : controller.get()); } void AttachWebView::requestWithOptionalConfirm( not_null bot, const WebViewButton &button, const Context &context, Window::SessionController *controllerForConfirm) { if (IsSame(_context, context) && _bot == bot) { if (_panel) { _panel->requestActivate(); } else if (_requestId) { return; } } cancel(); _bot = bot; _context = std::make_unique(context); if (controllerForConfirm) { confirmOpen(controllerForConfirm, [=] { request(button); }); } else { request(button); } } void AttachWebView::request(const WebViewButton &button) { Expects(_context != nullptr && _bot != nullptr); if (button.fromAttachMenu) { const auto bot = ranges::find( _attachBots, not_null{ _bot }, &AttachWebViewBot::user); if (bot == end(_attachBots) || bot->inactive) { requestAddToMenu(_bot, AddToMenuOpenAttach{ .startCommand = button.startCommand, }); return; } } _startCommand = button.startCommand; const auto &action = _context->action; using Flag = MTPmessages_RequestWebView::Flag; const auto flags = Flag::f_theme_params | (button.url.isEmpty() ? Flag(0) : Flag::f_url) | (_startCommand.isEmpty() ? Flag(0) : Flag::f_start_param) | (action.replyTo ? Flag::f_reply_to : Flag(0)) | (action.options.sendAs ? Flag::f_send_as : Flag(0)) | (action.options.silent ? Flag::f_silent : Flag(0)); _requestId = _session->api().request(MTPmessages_RequestWebView( MTP_flags(flags), action.history->peer->input, _bot->inputUser, MTP_bytes(button.url), MTP_string(_startCommand), MTP_dataJSON(MTP_bytes(Window::Theme::WebViewParams().json)), MTP_string("tdesktop"), action.mtpReplyTo(), (action.options.sendAs ? action.options.sendAs->input : MTP_inputPeerEmpty()) )).done([=](const MTPWebViewResult &result) { _requestId = 0; const auto &data = result.data(); show( data.vquery_id().v, qs(data.vurl()), button.text, button.fromAttachMenu || button.url.isEmpty()); }).fail([=](const MTP::Error &error) { _requestId = 0; if (error.type() == u"BOT_INVALID"_q) { requestBots(); } }).send(); } void AttachWebView::cancel() { Expects(!_catchingCancelInShowCall); ActiveWebViews().remove(this); _session->api().request(base::take(_requestId)).cancel(); _session->api().request(base::take(_prolongId)).cancel(); base::take(_panel); _lastShownContext = base::take(_context); _bot = nullptr; _app = nullptr; _botUsername = QString(); _botAppName = QString(); _startCommand = QString(); } void AttachWebView::requestBots(Fn callback) { if (callback) { _botsRequestCallbacks.push_back(std::move(callback)); } if (_botsRequestId) { return; } _botsRequestId = _session->api().request(MTPmessages_GetAttachMenuBots( MTP_long(_botsHash) )).done([=](const MTPAttachMenuBots &result) { _botsRequestId = 0; result.match([&](const MTPDattachMenuBotsNotModified &) { }, [&](const MTPDattachMenuBots &data) { _session->data().processUsers(data.vusers()); _botsHash = data.vhash().v; _attachBots.clear(); _attachBots.reserve(data.vbots().v.size()); for (const auto &bot : data.vbots().v) { if (auto parsed = ParseAttachBot(_session, bot)) { _attachBots.push_back(std::move(*parsed)); } } _attachBotsUpdates.fire({}); }); for (const auto &callback : base::take(_botsRequestCallbacks)) { callback(); } }).fail([=] { _botsRequestId = 0; for (const auto &callback : base::take(_botsRequestCallbacks)) { callback(); } }).send(); } bool AttachWebView::disclaimerAccepted(const AttachWebViewBot &bot) const { return _disclaimerAccepted.contains(bot.user); } bool AttachWebView::showMainMenuNewBadge( const AttachWebViewBot &bot) const { return bot.inMainMenu && bot.disclaimerRequired && !disclaimerAccepted(bot); } void AttachWebView::requestAddToMenu( not_null bot, AddToMenuOpen open) { requestAddToMenu(bot, open, nullptr, std::nullopt); } void AttachWebView::requestAddToMenu( not_null bot, AddToMenuOpen open, Window::SessionController *controller, std::optional action) { Expects(controller != nullptr || _context != nullptr); const auto wasController = (controller != nullptr); _addToMenuChooseController = base::make_weak(controller); _addToMenuOpen = open; if (!controller) { _addToMenuContext = base::take(_context); } else if (action) { _addToMenuContext = std::make_unique( LookupContext(controller, *action)); } const auto unsupported = [=] { auto context = base::take(_addToMenuContext); const auto open = base::take(_addToMenuOpen); if (const auto openApp = std::get_if(&open)) { _app = openApp->app; _startCommand = openApp->startCommand; _context = std::move(context); if (_appConfirmationRequired) { confirmAppOpen(_appRequestWriteAccess); } else { requestAppView(false); } } else { showToast( tr::lng_bot_menu_not_supported(tr::now), _addToMenuChooseController.get()); } }; if (!bot->isBot() || !bot->botInfo->supportsAttachMenu) { unsupported(); return; } if (_addToMenuId) { if (_addToMenuBot == bot) { return; } _session->api().request(base::take(_addToMenuId)).cancel(); } _addToMenuBot = bot; _addToMenuId = _session->api().request(MTPmessages_GetAttachMenuBot( bot->inputUser )).done([=](const MTPAttachMenuBotsBot &result) { _addToMenuId = 0; const auto bot = base::take(_addToMenuBot); const auto context = std::shared_ptr(base::take(_addToMenuContext)); const auto open = base::take(_addToMenuOpen); const auto chooseController = base::take(_addToMenuChooseController); const auto launch = [=](PeerTypes types) { const auto openAttach = v::is(open) ? v::get(open) : AddToMenuOpenAttach(); const auto chooseTypes = openAttach.chooseTypes; const auto strong = chooseController.get(); if (v::is(open)) { if (!context) { return false; } const auto &openApp = v::get(open); _app = openApp.app; _startCommand = openApp.startCommand; _context = std::make_unique(*context); requestAppView(true); return true; } else if (!strong) { if (wasController || !v::is(open)) { // Just ignore the click if controller was destroyed. return true; } } else if (v::is(open)) { const auto &openMenu = v::get(open); _bot = bot; requestSimple(strong, bot, { .startCommand = openMenu.startCommand, .fromMainMenu = true, }); return true; } else if (const auto useTypes = chooseTypes & types) { const auto done = [=](not_null thread) { strong->showThread(thread); requestWithOptionalConfirm( bot, { .startCommand = openAttach.startCommand }, LookupContext(strong, Api::SendAction(thread))); }; ShowChooseBox(strong, useTypes, done); return true; } if (!context) { return false; } requestWithOptionalConfirm( bot, { .startCommand = openAttach.startCommand }, *context); return true; }; result.match([&](const MTPDattachMenuBotsBot &data) { _session->data().processUsers(data.vusers()); if (const auto parsed = ParseAttachBot(_session, data.vbot())) { if (bot == parsed->user) { const auto i = ranges::find( _attachBots, not_null(bot), &AttachWebViewBot::user); if (i != end(_attachBots)) { // Save flags in our list, like 'inactive'. *i = *parsed; } const auto types = parsed->types; if (parsed->inactive) { confirmAddToMenu(*parsed, [=] { launch(types); }); } else { requestBots(); if (!launch(types)) { showToast( tr::lng_bot_menu_already_added(tr::now)); } } } } }); }).fail([=] { _addToMenuId = 0; _addToMenuBot = nullptr; unsupported(); }).send(); } void AttachWebView::removeFromMenu(not_null bot) { toggleInMenu(bot, ToggledState::Removed, [=] { showToast(tr::lng_bot_remove_from_menu_done(tr::now)); }); } std::optional AttachWebView::lookupLastAction( const QString &url) const { if (_lastShownUrl == url && _lastShownContext) { return _lastShownContext->action; } return std::nullopt; } void AttachWebView::resolve() { Expects(!_panel); resolveUsername(_botUsername, [=](not_null bot) { if (!_context) { return; } _bot = bot->asUser(); if (!_bot) { showToast(tr::lng_bot_menu_not_supported(tr::now)); return; } requestAddToMenu(_bot, AddToMenuOpenAttach{ .startCommand = _startCommand, }); }); } void AttachWebView::resolveUsername( const QString &username, Fn)> done) { if (const auto peer = _session->data().peerByUsername(username)) { done(peer); return; } _session->api().request(base::take(_requestId)).cancel(); _requestId = _session->api().request(MTPcontacts_ResolveUsername( MTP_string(username) )).done([=](const MTPcontacts_ResolvedPeer &result) { _requestId = 0; result.match([&](const MTPDcontacts_resolvedPeer &data) { _session->data().processUsers(data.vusers()); _session->data().processChats(data.vchats()); if (const auto peerId = peerFromMTP(data.vpeer())) { done(_session->data().peer(peerId)); } }); }).fail([=](const MTP::Error &error) { _requestId = 0; if (error.code() == 400) { showToast( tr::lng_username_not_found(tr::now, lt_user, username)); } }).send(); } void AttachWebView::requestSimple( not_null controller, not_null bot, const WebViewButton &button) { cancel(); _bot = bot; _context = std::make_unique(LookupContext( controller, Api::SendAction(bot->owner().history(bot)))); _context->fromSwitch = button.fromSwitch; _context->fromMainMenu = button.fromMainMenu; if (button.fromMainMenu) { acceptMainMenuDisclaimer(controller, button); } else { confirmOpen(controller, [=] { requestSimple(button); }); } } void AttachWebView::requestSimple(const WebViewButton &button) { using Flag = MTPmessages_RequestSimpleWebView::Flag; _requestId = _session->api().request(MTPmessages_RequestSimpleWebView( MTP_flags(Flag::f_theme_params | (button.fromMainMenu ? (Flag::f_from_side_menu | (button.startCommand.isEmpty() ? Flag() : Flag::f_start_param)) : Flag::f_url) | (button.fromSwitch ? Flag::f_from_switch_webview : Flag())), _bot->inputUser, MTP_bytes(button.url), MTP_string(button.startCommand), MTP_dataJSON(MTP_bytes(Window::Theme::WebViewParams().json)), MTP_string("tdesktop") )).done([=](const MTPSimpleWebViewResult &result) { _requestId = 0; result.match([&](const MTPDsimpleWebViewResultUrl &data) { show( uint64(), qs(data.vurl()), button.text, false, nullptr, button.fromMainMenu); }); }).fail([=](const MTP::Error &error) { _requestId = 0; }).send(); } void AttachWebView::requestMenu( not_null controller, not_null bot) { cancel(); _bot = bot; _context = std::make_unique(LookupContext( controller, Api::SendAction(bot->owner().history(bot)))); const auto url = bot->botInfo->botMenuButtonUrl; const auto text = bot->botInfo->botMenuButtonText; confirmOpen(controller, [=] { const auto &action = _context->action; using Flag = MTPmessages_RequestWebView::Flag; _requestId = _session->api().request(MTPmessages_RequestWebView( MTP_flags(Flag::f_theme_params | Flag::f_url | Flag::f_from_bot_menu | (action.replyTo? Flag::f_reply_to : Flag(0)) | (action.options.sendAs ? Flag::f_send_as : Flag(0)) | (action.options.silent ? Flag::f_silent : Flag(0))), action.history->peer->input, _bot->inputUser, MTP_string(url), MTPstring(), // start_param MTP_dataJSON(MTP_bytes(Window::Theme::WebViewParams().json)), MTP_string("tdesktop"), action.mtpReplyTo(), (action.options.sendAs ? action.options.sendAs->input : MTP_inputPeerEmpty()) )).done([=](const MTPWebViewResult &result) { _requestId = 0; const auto &data = result.data(); show(data.vquery_id().v, qs(data.vurl()), text); }).fail([=](const MTP::Error &error) { _requestId = 0; if (error.type() == u"BOT_INVALID"_q) { requestBots(); } }).send(); }); } void AttachWebView::requestApp( not_null controller, const Api::SendAction &action, not_null bot, const QString &appName, const QString &startParam, bool forceConfirmation) { const auto context = LookupContext(controller, action); if (_requestId && _bot == bot && _startCommand == startParam && _botAppName == appName && IsSame(_context, context)) { return; } cancel(); _bot = bot; _startCommand = startParam; _botAppName = appName; _context = std::make_unique(context); _context->fromBotApp = true; const auto already = _session->data().findBotApp(_bot->id, appName); _requestId = _session->api().request(MTPmessages_GetBotApp( MTP_inputBotAppShortName( bot->inputUser, MTP_string(appName)), MTP_long(already ? already->hash : 0) )).done([=](const MTPmessages_BotApp &result) { _requestId = 0; if (!_bot || !_context) { return; } const auto &data = result.data(); const auto firstTime = data.is_inactive(); const auto received = _session->data().processBotApp( _bot->id, data.vapp()); _app = received ? received : already; if (!_app) { cancel(); showToast(tr::lng_username_app_not_found(tr::now)); return; } // Check if this app can be added to main menu. // On fail it'll still be opened. _appConfirmationRequired = firstTime || forceConfirmation; _appRequestWriteAccess = result.data().is_request_write_access(); requestAddToMenu(_bot, AddToMenuOpenApp{ .app = _app, .startCommand = _startCommand, }); }).fail([=] { showToast(tr::lng_username_app_not_found(tr::now)); cancel(); }).send(); } void AttachWebView::confirmAppOpen(bool requestWriteAccess) { const auto controller = _context ? _context->controller.get() : nullptr; if (!controller || !_bot) { return; } controller->show(Box([=](not_null box) { const auto allowed = std::make_shared(); const auto done = [=](Fn close) { requestAppView((*allowed) && (*allowed)->checked()); close(); }; Ui::ConfirmBox(box, { tr::lng_allow_bot_webview( tr::now, lt_bot_name, Ui::Text::Bold(_bot->name()), Ui::Text::RichLangValue), done, }); if (requestWriteAccess) { (*allowed) = box->addRow( object_ptr( box, tr::lng_url_auth_allow_messages( tr::now, lt_bot, Ui::Text::Bold(_bot->name()), Ui::Text::WithEntities), true, st::urlAuthCheckbox), style::margins( st::boxRowPadding.left(), st::boxPhotoCaptionSkip, st::boxRowPadding.right(), st::boxPhotoCaptionSkip)); (*allowed)->setAllowTextLines(); } })); } void AttachWebView::requestAppView(bool allowWrite) { if (!_context || !_app) { return; } using Flag = MTPmessages_RequestAppWebView::Flag; const auto app = _app; const auto flags = Flag::f_theme_params | (_startCommand.isEmpty() ? Flag(0) : Flag::f_start_param) | (allowWrite ? Flag::f_write_allowed : Flag(0)); _requestId = _session->api().request(MTPmessages_RequestAppWebView( MTP_flags(flags), _context->action.history->peer->input, MTP_inputBotAppID(MTP_long(app->id), MTP_long(app->accessHash)), MTP_string(_startCommand), MTP_dataJSON(MTP_bytes(Window::Theme::WebViewParams().json)), MTP_string("tdesktop") )).done([=](const MTPAppWebViewResult &result) { _requestId = 0; const auto &data = result.data(); const auto queryId = uint64(); show(queryId, qs(data.vurl()), QString(), false, app); }).fail([=](const MTP::Error &error) { _requestId = 0; if (error.type() == u"BOT_INVALID"_q) { requestBots(); } }).send(); } void AttachWebView::confirmOpen( not_null controller, Fn done) { if (!_bot) { return; } else if (_bot->isVerified() || _bot->session().local().isBotTrustedOpenWebView(_bot->id)) { done(); return; } const auto callback = [=] { _bot->session().local().markBotTrustedOpenWebView(_bot->id); controller->hideLayer(); done(); }; controller->show(Ui::MakeConfirmBox({ .text = tr::lng_allow_bot_webview( tr::now, lt_bot_name, Ui::Text::Bold(_bot->name()), Ui::Text::RichLangValue), .confirmed = callback, .confirmText = tr::lng_box_ok(), })); } void AttachWebView::acceptMainMenuDisclaimer( not_null controller, const WebViewButton &button) { Expects(button.fromMainMenu); const auto local = _bot ? &_bot->session().local() : nullptr; if (!local) { return; } const auto i = ranges::find( _attachBots, not_null(_bot), &AttachWebViewBot::user); if (i == end(_attachBots)) { _attachBotsUpdates.fire({}); return; } else if (i->inactive) { requestAddToMenu(_bot, AddToMenuOpenMenu{ .startCommand = button.startCommand, }, controller, {}); return; } else if (!i->disclaimerRequired || disclaimerAccepted(*i)) { requestSimple(button); return; } const auto weak = base::make_weak(this); controller->show(Box(FillDisclaimerBox, crl::guard(this, [=] { _disclaimerAccepted.emplace(_bot); _attachBotsUpdates.fire({}); requestSimple(button); }))); } void AttachWebView::ClearAll() { while (!ActiveWebViews().empty()) { ActiveWebViews().front()->cancel(); } } void AttachWebView::show( uint64 queryId, const QString &url, const QString &buttonText, bool allowClipboardRead, const BotAppData *app, bool fromMainMenu) { Expects(_bot != nullptr && _context != nullptr); auto title = Info::Profile::NameValue(_bot); ActiveWebViews().emplace(this); using Button = Ui::BotWebView::MenuButton; const auto attached = ranges::find( _attachBots, not_null{ _bot }, &AttachWebViewBot::user); const auto hasOpenBot = !_context || (_bot != _context->action.history->peer) || fromMainMenu; const auto hasRemoveFromMenu = !app && (attached != end(_attachBots)) && (!attached->inactive || attached->inMainMenu); const auto buttons = (hasOpenBot ? Button::OpenBot : Button::None) | (!hasRemoveFromMenu ? Button::None : attached->inMainMenu ? Button::RemoveFromMainMenu : Button::RemoveFromMenu); if (attached != end(_attachBots) && (attached->inAttachMenu || attached->inMainMenu)) { allowClipboardRead = true; } _lastShownUrl = url; _lastShownQueryId = queryId; _lastShownButtonText = buttonText; base::take(_panel); _catchingCancelInShowCall = true; _panel = Ui::BotWebView::Show({ .url = url, .userDataPath = _session->domain().local().webviewDataPath(), .title = std::move(title), .bottom = rpl::single('@' + _bot->username()), .delegate = static_cast(this), .menuButtons = buttons, .allowClipboardRead = allowClipboardRead, }); _catchingCancelInShowCall = false; started(queryId); } void AttachWebView::started(uint64 queryId) { Expects(_bot != nullptr); Expects(_context != nullptr); if (_context->fromSwitch || !queryId) { return; } _session->data().webViewResultSent( ) | rpl::filter([=](const Data::Session::WebViewResultSent &sent) { return (sent.queryId == queryId); }) | rpl::start_with_next([=] { cancel(); }, _panel->lifetime()); const auto action = _context->action; base::timer_each( kProlongTimeout ) | rpl::start_with_next([=] { using Flag = MTPmessages_ProlongWebView::Flag; _session->api().request(base::take(_prolongId)).cancel(); _prolongId = _session->api().request(MTPmessages_ProlongWebView( MTP_flags(Flag(0) | (action.replyTo ? Flag::f_reply_to : Flag(0)) | (action.options.sendAs ? Flag::f_send_as : Flag(0)) | (action.options.silent ? Flag::f_silent : Flag(0))), action.history->peer->input, _bot->inputUser, MTP_long(queryId), action.mtpReplyTo(), (action.options.sendAs ? action.options.sendAs->input : MTP_inputPeerEmpty()) )).done([=] { _prolongId = 0; }).send(); }, _panel->lifetime()); } void AttachWebView::showToast( const QString &text, Window::SessionController *controller) { const auto strong = controller ? controller : _context ? _context->controller.get() : _addToMenuContext ? _addToMenuContext->controller.get() : nullptr; if (strong) { strong->showToast(text); } } void AttachWebView::confirmAddToMenu( AttachWebViewBot bot, Fn callback) { const auto active = Core::App().activeWindow(); if (!active) { return; } _confirmAddBox = active->show(Box([=](not_null box) { const auto allowed = std::make_shared(); const auto disclaimer = !disclaimerAccepted(bot); const auto done = [=](Fn close) { const auto state = (disclaimer || ((*allowed) && (*allowed)->checked())) ? ToggledState::AllowedToWrite : ToggledState::Added; toggleInMenu(bot.user, state, [=] { if (callback) { callback(); } showToast((bot.inMainMenu ? tr::lng_bot_add_to_side_menu_done : tr::lng_bot_add_to_menu_done)(tr::now)); }); close(); }; if (disclaimer) { FillDisclaimerBox(box, [=] { _disclaimerAccepted.emplace(bot.user); _attachBotsUpdates.fire({}); done([] {}); }); box->addRow(object_ptr( box, st::boxRowPadding.left())); box->addRow(object_ptr( box, tr::lng_bot_will_be_added( lt_bot, rpl::single(Ui::Text::Bold(bot.name)), Ui::Text::WithEntities), st::boxLabel)); } else { Ui::ConfirmBox(box, { (bot.inMainMenu ? tr::lng_bot_add_to_side_menu : tr::lng_bot_add_to_menu)( tr::now, lt_bot, Ui::Text::Bold(bot.name), Ui::Text::WithEntities), done, }); if (bot.requestWriteAccess) { (*allowed) = box->addRow( object_ptr( box, tr::lng_url_auth_allow_messages( tr::now, lt_bot, Ui::Text::Bold(bot.name), Ui::Text::WithEntities), true, st::urlAuthCheckbox), style::margins( st::boxRowPadding.left(), (disclaimer ? st::boxPhotoCaptionSkip : st::boxRowPadding.left()), st::boxRowPadding.right(), st::boxRowPadding.left())); (*allowed)->setAllowTextLines(); } } })); } void AttachWebView::toggleInMenu( not_null bot, ToggledState state, Fn callback) { using Flag = MTPmessages_ToggleBotInAttachMenu::Flag; _session->api().request(MTPmessages_ToggleBotInAttachMenu( MTP_flags((state == ToggledState::AllowedToWrite) ? Flag::f_write_allowed : Flag()), bot->inputUser, MTP_bool(state != ToggledState::Removed) )).done([=] { _requestId = 0; _session->api().request(base::take(_botsRequestId)).cancel(); requestBots(std::move(callback)); }).fail([=] { cancel(); }).send(); } std::unique_ptr MakeAttachBotsMenu( not_null parent, not_null controller, not_null peer, Fn actionFactory, Fn attach) { if (!Data::CanSend(peer, ChatRestriction::SendInline)) { return nullptr; } auto result = std::make_unique( parent, st::dropdownMenuWithIcons); const auto bots = &peer->session().attachWebView(); const auto raw = result.get(); auto minimal = 0; if (Data::CanSend(peer, ChatRestriction::SendPhotos, false)) { ++minimal; raw->addAction(tr::lng_attach_photo_or_video(tr::now), [=] { attach(true); }, &st::menuIconPhoto); } const auto fileTypes = ChatRestriction::SendVideos | ChatRestriction::SendGifs | ChatRestriction::SendStickers | ChatRestriction::SendMusic | ChatRestriction::SendFiles; if (Data::CanSendAnyOf(peer, fileTypes)) { ++minimal; raw->addAction(tr::lng_attach_document(tr::now), [=] { attach(false); }, &st::menuIconFile); } if (peer->canCreatePolls()) { ++minimal; raw->addAction(tr::lng_polls_create(tr::now), [=] { const auto action = actionFactory(); const auto source = action.options.scheduled ? Api::SendType::Scheduled : Api::SendType::Normal; const auto sendMenuType = action.replyTo.topicRootId ? SendMenu::Type::SilentOnly : SendMenu::Type::Scheduled; const auto flag = PollData::Flags(); const auto replyTo = action.replyTo; Window::PeerMenuCreatePoll( controller, peer, replyTo, flag, flag, source, sendMenuType); }, &st::menuIconCreatePoll); } for (const auto &bot : bots->attachBots()) { if (!bot.inAttachMenu || !PeerMatchesTypes(peer, bot.user, bot.types)) { continue; } const auto callback = [=] { bots->request( controller, actionFactory(), bot.user, { .fromAttachMenu = true }); }; auto action = base::make_unique_q( raw, raw->menu()->st(), bot, callback); action->forceShown( ) | rpl::start_with_next([=](bool shown) { if (shown) { raw->setAutoHiding(false); } else { raw->hideAnimated(); raw->setAutoHiding(true); } }, action->lifetime()); raw->addAction(std::move(action)); } if (raw->actions().size() <= minimal) { return nullptr; } return result; } } // namespace InlineBots