Ejemplo n.º 1
0
 public static PollId GetExtendedPollId(this PollMessage message)
 {
     return(message.Poll?.GetId() ?? new PollId {
         Id = message.PollId
     });
 }
Ejemplo n.º 2
0
 public static int?GetPollId(this PollMessage message)
 {
     return(message.Poll?.Id ?? message.PollId);
 }
Ejemplo n.º 3
0
        public static async Task <InlineKeyboardMarkup> GetReplyMarkup(this PollMessage message, ChatInfo chatInfo, CancellationToken cancellationToken = default)
        {
            var pollReplyMarkup = message.Poll.GetReplyMarkup();

            var messageChat = message.Chat;

            if (messageChat == null) // inline message
            {
                return(pollReplyMarkup);
            }

            switch (messageChat.Type)
            {
            case ChatType.Channel:
                // channel, cancelled
                if (pollReplyMarkup == null)
                {
                    return(null);
                }

                var inlineKeyboardButtons = pollReplyMarkup.InlineKeyboard.Select(_ => _.ToArray()).ToArray();

                // channel, not cancelled
                // replace share button with clone button
                for (var i = 0; i < inlineKeyboardButtons.Length; i++)
                {
                    for (var j = 0; j < inlineKeyboardButtons[i].Length; j++)
                    {
                        if (inlineKeyboardButtons[i][j].Text == "🌐")
                        {
                            inlineKeyboardButtons[i][j] = InlineKeyboardButton.WithCallbackData("🌐", $"clone:{message.GetExtendedPollId()}");
                        }
                    }
                }

                return(new InlineKeyboardMarkup(inlineKeyboardButtons));

            case ChatType.Group:
            case ChatType.Supergroup:
                return(pollReplyMarkup);
            }

            if (!await chatInfo.CandEditPoll(message.Poll.Owner, message.UserId, cancellationToken))
            {
                return(pollReplyMarkup);
            }

            var pollId = message.GetExtendedPollId();

            if (pollReplyMarkup == null) // cancelled
            {
                if (!message.Poll.Cancelled)
                {
                    return(null); // can't be
                }
                return(new InlineKeyboardMarkup(new []
                {
                    InlineKeyboardButton.WithCallbackData("Resume", $"{RestoreCallbackQueryHandler.ID}:{pollId}"),
                }));
            }

            var additionalKeyboardButtons = new List <InlineKeyboardButton>();

            if (message.Poll.Time != null)
            {
                additionalKeyboardButtons.Add(InlineKeyboardButton.WithCallbackData("5' earlier", $"{AdjustCallbackQueryHandler.ID}:{pollId}:-5"));
                additionalKeyboardButtons.Add(InlineKeyboardButton.WithCallbackData("5' later", $"{AdjustCallbackQueryHandler.ID}:{pollId}:5"));
            }
            additionalKeyboardButtons.Add(InlineKeyboardButton.WithCallbackData("Cancel", $"{CancelCallbackQueryHandler.ID}:{pollId}"));
            return(new InlineKeyboardMarkup(pollReplyMarkup.InlineKeyboard.Append(additionalKeyboardButtons)));
        }
Ejemplo n.º 4
0
 public static IDictionary <string, string> GetTrackingProperties([CanBeNull] this PollMessage pollMessage)
 {
     return(new Dictionary <string, string>
     {
         { "messageId", pollMessage?.Id is int pollMessageId && pollMessageId > 0 ? pollMessageId.ToString() : null },
Ejemplo n.º 5
0
        public static async Task <InlineKeyboardMarkup> GetReplyMarkup(this PollMessage message, ChatInfo chatInfo, CancellationToken cancellationToken = default)
        {
            var pollReplyMarkup = message.Poll.GetReplyMarkup(message.PollMode);

            var messageChat = message.Chat;

            if (messageChat == null) // inline message
            {
                return(pollReplyMarkup);
            }

            if (pollReplyMarkup == null) // cancelled poll
            {
                return(null);
            }

            var inlineKeyboardButtons = pollReplyMarkup.InlineKeyboard.Select(_ => _.ToArray()).ToArray();

            bool modified = false;

            for (var i = 0; i < inlineKeyboardButtons.Length; i++)
            {
                for (var j = 0; j < inlineKeyboardButtons[i].Length; j++)
                {
                    var inlineKeyboardButton = inlineKeyboardButtons[i][j];

                    // channel, not cancelled
                    // replace share button with clone button
                    if (messageChat.Type == ChatType.Channel && (inlineKeyboardButton.CallbackData?.StartsWith(ShareInlineQueryHandler.ID) ?? false))
                    {
                        modified = true;
                        inlineKeyboardButtons[i][j] = InlineKeyboardButton.WithCallbackData(VoteEnum.Share.AsString(EnumFormat.DisplayName) !, $"{CloneCallbackQueryHandler.ID}:{message.GetExtendedPollId()}");
                    }
                }
            }

            if (modified)
            {
                pollReplyMarkup = new InlineKeyboardMarkup(inlineKeyboardButtons);
            }

            if (messageChat.Type != ChatType.Private)
            {
                return(pollReplyMarkup);
            }

            if (message.PollMode?.HasFlag(PollMode.Invitation) ?? false)
            {
                var hosters = message.Poll.Votes.Where(v => v.Team?.HasAnyFlags(VoteEnum.Hosting) ?? false).Select(_ => _.UserId).ToHashSet();
                if (message.Poll.Owner == message.ChatId || hosters.Contains(messageChat.Id))
                {
                    pollReplyMarkup = new InlineKeyboardMarkup(inlineKeyboardButtons.Concat(new []
                    {
                        new []
                        {
                            InlineKeyboardButton.WithCallbackData("Invite", $"{InviteCallbackQueryHandler.ID}:{message.GetExtendedPollId()}"),
                            InlineKeyboardButton.WithCallbackData("GO", $"{GoCallbackQueryHandler.ID}:{message.PollId}")
                        }
                    }));
                }
            }

            // TODO: Currently, no additional admin buttons
            return(pollReplyMarkup);

            if (!await chatInfo.CandEditPoll(message.Poll.Owner, message.UserId, cancellationToken))
            {
                return(pollReplyMarkup);
            }

            var pollId = message.GetExtendedPollId();

            if (pollReplyMarkup == null) // cancelled
            {
                if (!message.Poll.Cancelled)
                {
                    return(null); // can't be
                }
                return(new InlineKeyboardMarkup(new []
                {
                    InlineKeyboardButton.WithCallbackData("Resume", $"{RestoreCallbackQueryHandler.ID}:{pollId}"),
                }));
            }

            var additionalKeyboardButtons = new List <InlineKeyboardButton>();

            if (message.Poll.Time != null)
            {
                additionalKeyboardButtons.Add(InlineKeyboardButton.WithCallbackData("5' earlier", $"{AdjustCallbackQueryHandler.ID}:{pollId}:-5"));
                additionalKeyboardButtons.Add(InlineKeyboardButton.WithCallbackData("5' later", $"{AdjustCallbackQueryHandler.ID}:{pollId}:5"));
            }
            additionalKeyboardButtons.Add(InlineKeyboardButton.WithCallbackData("Cancel", $"{CancelCallbackQueryHandler.ID}:{pollId}"));
            return(new InlineKeyboardMarkup(pollReplyMarkup.InlineKeyboard.Append(additionalKeyboardButtons)));
        }
Ejemplo n.º 6
0
 public static IDictionary <string, string> GetTrackingProperties([CanBeNull] this PollMessage pollMessage, IDictionary <string, string> properties = null)
 {
     return(new Dictionary <string, string>(properties ?? EmptyProperties)
     {
         { "messageId", pollMessage?.Id is { } pollMessageId and > 0 ? pollMessageId.ToString() : null },