From d289c2101d68080652d16437a2057d98a20425db Mon Sep 17 00:00:00 2001
From: InsanusMokrassar <ovsyannikov.alexey95@gmail.com>
Date: Sun, 20 Aug 2023 02:30:18 +0600
Subject: [PATCH] add polls sample bot

---
 PollsBot/README.md                   | 11 ++++
 PollsBot/build.gradle                | 21 +++++++
 PollsBot/src/main/kotlin/PollsBot.kt | 89 ++++++++++++++++++++++++++++
 settings.gradle                      |  2 +
 4 files changed, 123 insertions(+)
 create mode 100644 PollsBot/README.md
 create mode 100644 PollsBot/build.gradle
 create mode 100644 PollsBot/src/main/kotlin/PollsBot.kt

diff --git a/PollsBot/README.md b/PollsBot/README.md
new file mode 100644
index 0000000..d810d47
--- /dev/null
+++ b/PollsBot/README.md
@@ -0,0 +1,11 @@
+# PollsBot
+
+This bot will send test poll in the chat where commands will be received. Commands:
+
+
+
+## Launch
+
+```bash
+../gradlew run --args="BOT_TOKEN"
+```
diff --git a/PollsBot/build.gradle b/PollsBot/build.gradle
new file mode 100644
index 0000000..0188724
--- /dev/null
+++ b/PollsBot/build.gradle
@@ -0,0 +1,21 @@
+buildscript {
+    repositories {
+        mavenCentral()
+    }
+
+    dependencies {
+        classpath "org.jetbrains.kotlin:kotlin-gradle-plugin:$kotlin_version"
+    }
+}
+
+apply plugin: 'kotlin'
+apply plugin: 'application'
+
+mainClassName="HelloBotKt"
+
+
+dependencies {
+    implementation "org.jetbrains.kotlin:kotlin-stdlib:$kotlin_version"
+
+    implementation "dev.inmo:tgbotapi:$telegram_bot_api_version"
+}
diff --git a/PollsBot/src/main/kotlin/PollsBot.kt b/PollsBot/src/main/kotlin/PollsBot.kt
new file mode 100644
index 0000000..0f20259
--- /dev/null
+++ b/PollsBot/src/main/kotlin/PollsBot.kt
@@ -0,0 +1,89 @@
+import com.benasher44.uuid.uuid4
+import dev.inmo.micro_utils.coroutines.subscribeSafelyWithoutExceptions
+import dev.inmo.tgbotapi.extensions.api.bot.getMe
+import dev.inmo.tgbotapi.extensions.api.chat.get.getChat
+import dev.inmo.tgbotapi.extensions.api.send.*
+import dev.inmo.tgbotapi.extensions.api.send.polls.sendRegularPoll
+import dev.inmo.tgbotapi.extensions.behaviour_builder.telegramBotWithBehaviourAndLongPolling
+import dev.inmo.tgbotapi.extensions.behaviour_builder.triggers_handling.onCommand
+import dev.inmo.tgbotapi.extensions.behaviour_builder.triggers_handling.onContentMessage
+import dev.inmo.tgbotapi.extensions.behaviour_builder.triggers_handling.onMentionWithAnyContent
+import dev.inmo.tgbotapi.extensions.behaviour_builder.triggers_handling.onPollAnswer
+import dev.inmo.tgbotapi.extensions.utils.extensions.raw.sender_chat
+import dev.inmo.tgbotapi.extensions.utils.formatting.linkMarkdownV2
+import dev.inmo.tgbotapi.extensions.utils.formatting.textMentionMarkdownV2
+import dev.inmo.tgbotapi.extensions.utils.ifChannelChat
+import dev.inmo.tgbotapi.extensions.utils.ifFromChannelGroupContentMessage
+import dev.inmo.tgbotapi.types.ChatId
+import dev.inmo.tgbotapi.types.IdChatIdentifier
+import dev.inmo.tgbotapi.types.PollIdentifier
+import dev.inmo.tgbotapi.types.chat.*
+import dev.inmo.tgbotapi.types.chat.GroupChat
+import dev.inmo.tgbotapi.types.chat.PrivateChat
+import dev.inmo.tgbotapi.types.chat.SupergroupChat
+import dev.inmo.tgbotapi.types.message.MarkdownV2
+import dev.inmo.tgbotapi.types.polls.Poll
+import dev.inmo.tgbotapi.types.polls.PollAnswer
+import dev.inmo.tgbotapi.types.polls.PollOption
+import dev.inmo.tgbotapi.types.polls.RegularPoll
+import dev.inmo.tgbotapi.utils.PreviewFeature
+import dev.inmo.tgbotapi.utils.extensions.escapeMarkdownV2Common
+import kotlinx.coroutines.*
+import kotlinx.coroutines.sync.Mutex
+import kotlinx.coroutines.sync.withLock
+
+/**
+ * The main purpose of this bot is just to answer "Oh, hi, " and add user mention here
+ */
+@OptIn(PreviewFeature::class)
+suspend fun main(vararg args: String) {
+    val botToken = args.first()
+
+    telegramBotWithBehaviourAndLongPolling(botToken, CoroutineScope(Dispatchers.IO)) {
+        val me = getMe()
+
+        val pollToChat = mutableMapOf<PollIdentifier, IdChatIdentifier>()
+        val pollToChatMutex = Mutex()
+
+        onCommand("anonymous") {
+            val sentPoll = sendRegularPoll(
+                it.chat,
+                "Test regular anonymous poll",
+                (1 .. 10).map {
+                    it.toString()
+                },
+                isAnonymous = true,
+                replyToMessageId = it.messageId
+            )
+            pollToChatMutex.withLock {
+                pollToChat[sentPoll.content.poll.id] = sentPoll.chat.id
+            }
+        }
+
+        onCommand("public") {
+            val sentPoll = sendRegularPoll(
+                it.chat,
+                "Test regular anonymous poll",
+                (1 .. 10).map {
+                    it.toString()
+                },
+                isAnonymous = false,
+                replyToMessageId = it.messageId
+            )
+            pollToChatMutex.withLock {
+                pollToChat[sentPoll.content.poll.id] = sentPoll.chat.id
+            }
+        }
+
+        onPollAnswer {
+            val chatId = pollToChat[it.pollId] ?: return@onPollAnswer
+
+            when(it) {
+                is PollAnswer.Public -> send(chatId, "User ${it.user} have answered")
+                is PollAnswer.Anonymous -> send(chatId, "Chat ${it.voterChat} have answered")
+            }
+        }
+
+        allUpdatesFlow.subscribeSafelyWithoutExceptions(this) { println(it) }
+    }.second.join()
+}
diff --git a/settings.gradle b/settings.gradle
index a76f386..ea37daa 100644
--- a/settings.gradle
+++ b/settings.gradle
@@ -4,6 +4,8 @@ include ":RandomFileSenderBot"
 
 include ":HelloBot"
 
+include ":PollsBot"
+
 include ":GetMeBot"
 
 include ":DeepLinksBot"