cubetiq/messaging-client
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Completed telegram messaging service and provider for send message via bot

Browse Source
This commit is contained in:
Sambo Chea
2021-05-20 18:43:52 +07:00
commit 2a6dec2922
23 changed files with 791 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/main/kotlin/com/cubetiqs/messaging/client/provider/MessageProvider.kt Normal file
View File

@@ -0,0 +1,6 @@
package com.cubetiqs.messaging.client.provider
@FunctionalInterface
fun interface MessageProvider {
fun send(): Any?
}

6
src/main/kotlin/com/cubetiqs/messaging/client/sms/SmsProvider.kt Normal file
View File

@@ -0,0 +1,6 @@
package com.cubetiqs.messaging.client.sms
import com.cubetiqs.messaging.client.provider.MessageProvider
interface SmsProvider : MessageProvider {
}

90
src/main/kotlin/com/cubetiqs/messaging/client/telegram/TelegramBotUtils.kt Normal file
View File

@@ -0,0 +1,90 @@
package com.cubetiqs.messaging.client.telegram
import com.cubetiqs.messaging.client.webclient.WebClientUtils
import okhttp3.*
import okhttp3.MediaType.Companion.toMediaTypeOrNull
import okhttp3.RequestBody.Companion.asRequestBody
import java.nio.file.Files
object TelegramBotUtils {
private fun getBotUrl(endpoint: String, token: String = ""): String {
return "${TelegramConfig.TELEGRAM_API}/bot${token.removePrefix("bot")}/${endpoint.removePrefix("/")}"
}
private fun makeRequest(
request: Request,
): Response? {
println(javaClass.canonicalName + " => Start send message via telegram bot...")
return try {
WebClientUtils.makeRequest(request)
} catch (ex: Exception) {
ex.printStackTrace()
println("Make request error @${ex.message}")
null
}
}
@JvmStatic
fun sendMessage(
// usually chat id from chat group
chatId: String,
// message to send
text: String,
// config prefix for custom token
token: String = "",
): Any? {
val requestBody = MultipartBody.Builder()
.addFormDataPart("text", text)
.addFormDataPart("chat_id", chatId)
.setType(MultipartBody.FORM)
.build()
val url = getBotUrl(TelegramConfig.SEND_MESSAGE, token)
val request = Request.Builder()
.url(url)
.post(requestBody)
.build()
val result = makeRequest(request)
println(javaClass.canonicalName + " => Complete sent message to $chatId...")
return result
}
@JvmStatic
fun sendDocument(
// usually chat id from chat group
chatId: String,
// message to send
text: String,
// document extension (example: my_file.pdf, my_file.docx, etc)
filename: String,
// document in byte array
document: ByteArray,
// config prefix for custom token
token: String = "",
): Any? {
val ext = filename.split(".").lastOrNull() ?: "dat"
val tempFile = Files.createTempFile(filename.removeSuffix(ext), ".$ext")
Files.write(tempFile, document)
val requestBody = MultipartBody.Builder()
.addFormDataPart(
"document", filename,
tempFile.toFile().asRequestBody("application/octet-stream".toMediaTypeOrNull())
)
.addFormDataPart("caption", text)
.addFormDataPart("chat_id", chatId)
.setType(MultipartBody.FORM)
.build()
val url = getBotUrl(TelegramConfig.SEND_DOCUMENT, token)
val request = Request.Builder()
.url(url)
.post(requestBody)
.build()
val result = makeRequest(request)
println(javaClass.canonicalName + " => Complete sent document to $chatId...")
return result
}
}

15
src/main/kotlin/com/cubetiqs/messaging/client/telegram/TelegramChatMessage.kt Normal file
View File

@@ -0,0 +1,15 @@
package com.cubetiqs.messaging.client.telegram
/**
* Telegram Chat Message
*
* @author sombochea
* @since 1.0
*/
open class TelegramChatMessage(
private val text: String? = null,
) : TelegramMessage {
override fun getText(): String {
return text ?: ""
}
}

22
src/main/kotlin/com/cubetiqs/messaging/client/telegram/TelegramConfig.kt Normal file
View File

@@ -0,0 +1,22 @@
package com.cubetiqs.messaging.client.telegram
import com.cubetiqs.messaging.client.util.ConfigUtils
object TelegramConfig {
const val TELEGRAM_API = "https://api.telegram.org"
const val SEND_DOCUMENT = "/sendDocument"
const val SEND_MESSAGE = "/sendMessage"
const val POS_SYSTEM_ID = ""
private const val CUBETIQ_TELEGRAM_TOKEN = "CUBETIQ_TELEGRAM_TOKEN"
private const val CUBETIQ_TELEGRAM_RECEIVER = "CUBETIQ_TELEGRAM_RECEIVER"
@JvmStatic
fun getToken(): String {
return ConfigUtils.getEnv(CUBETIQ_TELEGRAM_TOKEN, ConfigUtils.getProperty(CUBETIQ_TELEGRAM_TOKEN)) ?: throw NullPointerException("CUBETIQ_TELEGRAM_TOKEN is required for Telegram Sender!")
}
@JvmStatic
fun getReceiver(): String {
return ConfigUtils.getEnv(CUBETIQ_TELEGRAM_RECEIVER, ConfigUtils.getProperty(CUBETIQ_TELEGRAM_RECEIVER, "-360594386")) ?: throw NullPointerException("CUBETIQ_TELEGRAM_RECEIVER is required for Telegram Receiver!")
}
}

16
src/main/kotlin/com/cubetiqs/messaging/client/telegram/TelegramDocument.kt Normal file
View File

@@ -0,0 +1,16 @@
package com.cubetiqs.messaging.client.telegram
/**
* Telegram Chat Document
*
* @author sombochea
* @since 1.0
*/
open class TelegramDocument(
val document: TelegramFileDocument? = null,
private val text: String? = null,
) : TelegramMessage {
override fun getText(): String {
return text ?: ""
}
}

16
src/main/kotlin/com/cubetiqs/messaging/client/telegram/TelegramFileDocument.kt Normal file
View File

@@ -0,0 +1,16 @@
package com.cubetiqs.messaging.client.telegram
import java.io.Serializable
/**
* File Document
*
* @author sombochea
* @since 1.0
*/
open class TelegramFileDocument (
var file: ByteArray? = null,
var name: String? = null,
var type: String? = null,
var extension: String? = null,
) : Serializable

13
src/main/kotlin/com/cubetiqs/messaging/client/telegram/TelegramMessage.kt Normal file
View File

@@ -0,0 +1,13 @@
package com.cubetiqs.messaging.client.telegram
import java.io.Serializable
/**
* Telegram Message
*
* @author sombochea
* @since 1.0
*/
interface TelegramMessage : Serializable {
fun getText(): String
}

84
src/main/kotlin/com/cubetiqs/messaging/client/telegram/TelegramProvider.kt Normal file
View File

@@ -0,0 +1,84 @@
package com.cubetiqs.messaging.client.telegram
import com.cubetiqs.messaging.client.provider.MessageProvider
import com.sun.org.slf4j.internal.LoggerFactory
import kotlin.IllegalArgumentException
/**
* Telegram Provider
*
* @author sombochea
* @since 1.0
*/
class TelegramProvider : MessageProvider {
private val log = LoggerFactory.getLogger(this::class.java)
private var _token: String = ""
private var _chatId: String = ""
private var _message: TelegramMessage? = null
fun sendToChatId(chatId: String) = apply {
this._chatId = chatId
}
fun messageToSend(message: TelegramMessage?) = apply {
this._message = message
}
fun setBotToken(token: String) = apply {
this._token = token
}
fun send(
chatId: String,
message: TelegramMessage,
): TelegramResponse? {
if (_token.isEmpty() || _token.isBlank()) {
throw IllegalArgumentException("token must be provide for send message!")
}
if (_message?.getText().isNullOrEmpty()) return null
return try {
val response = TelegramBotUtils.sendMessage(
chatId = chatId,
token = this._token,
text = this._message!!.getText(),
)
TelegramResponse(
response = response,
)
} catch (ex: Exception) {
log.error("[Telegram] Message send error: ${ex.message}!")
null
}
}
override fun send(): Any? {
if (this._message?.getText().isNullOrEmpty()) {
throw IllegalArgumentException("message must be non-null or non-empty!")
}
return send(this._chatId, this._message!!)
}
override fun toString(): String {
return "TelegramProvider(_token='$_token', _chatId='$_chatId', _message=$_message)"
}
companion object {
@JvmStatic
fun init(token: String): TelegramProvider {
return TelegramProvider()
.setBotToken(token)
}
@JvmStatic
fun sendMessage(token: String, chatId: String, text: String): Any? {
return init(token)
.sendToChatId(chatId)
.messageToSend(TelegramChatMessage(text))
.send()
}
}
}

18
src/main/kotlin/com/cubetiqs/messaging/client/telegram/TelegramResponse.kt Normal file
View File

@@ -0,0 +1,18 @@
package com.cubetiqs.messaging.client.telegram
import java.io.Serializable
/**
* Telegram Response
*
* @author sombochea
* @since 1.0
*/
class TelegramResponse(
val request: Any? = null,
val response: Any? = null,
) : Serializable {
override fun toString(): String {
return "TelegramResponse(request=$request, response=$response)"
}
}

13
src/main/kotlin/com/cubetiqs/messaging/client/util/ConfigUtils.kt Normal file
View File

@@ -0,0 +1,13 @@
package com.cubetiqs.messaging.client.util
object ConfigUtils {
@JvmStatic
fun getEnv(key: String, defaultValue: String? = null): String? {
return System.getenv(key) ?: defaultValue
}
@JvmStatic
fun getProperty(key: String, defaultValue: String? = null): String? {
return System.getProperty(key, defaultValue)
}
}

31
src/main/kotlin/com/cubetiqs/messaging/client/webclient/WebClientUtils.kt Normal file
View File

@@ -0,0 +1,31 @@
package com.cubetiqs.messaging.client.webclient
import okhttp3.OkHttpClient
import okhttp3.Request
import okhttp3.Response
object WebClientUtils {
private var webClient: OkHttpClient? = null
private fun getClient(): OkHttpClient {
if (this.webClient == null) {
this.webClient = OkHttpClient()
}
return this.webClient!!
}
@JvmStatic
fun makeRequest(request: Request): Response {
val call = getClient().newCall(request)
var response: Response? = null
return try {
response = call.execute()
response
} catch (ex: Exception) {
throw Exception(ex)
} finally {
response?.close()
}
}
}

BIN
src/main/resources/cubetiq.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 30 KiB

48
src/test/java/com/cubetiqs/example/TelegramExampleJavaTests.java Normal file
View File

@@ -0,0 +1,48 @@
package com.cubetiqs.example;
import com.cubetiqs.messaging.client.telegram.TelegramBotUtils;
import com.cubetiqs.messaging.client.telegram.TelegramConfig;
import com.cubetiqs.messaging.client.telegram.TelegramProvider;
import org.junit.jupiter.api.Test;
import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
public final class TelegramExampleJavaTests {
private final String token = TelegramConfig.getToken();
private final String chatId = TelegramConfig.getReceiver();
@Test
public void sendMessage() {
String text = "Hello, Sambo!";
TelegramBotUtils.sendMessage(
chatId,
text,
token
);
}
@Test
public void sendDocument() throws IOException {
String text = "Hello, Sambo with my paper!";
File file = new File("src/main/resources/cubetiq.png");
TelegramBotUtils.sendDocument(
chatId,
text,
"my paper.png",
Files.readAllBytes(file.toPath()),
token
);
}
@Test
public void sendMessageProvider() {
String text = "Hello, Sambo with Provider!";
TelegramProvider.sendMessage(
chatId,
text,
token
);
}
}

44
src/test/kotlin/com/cubetiqs/example/TelegramExampleKotlinTests.kt Normal file
View File

@@ -0,0 +1,44 @@
package com.cubetiqs.example
import com.cubetiqs.messaging.client.telegram.TelegramBotUtils
import com.cubetiqs.messaging.client.telegram.TelegramConfig
import com.cubetiqs.messaging.client.telegram.TelegramProvider
import org.junit.jupiter.api.Test
import java.io.File
class TelegramExampleKotlinTests {
private val token = TelegramConfig.POS_SYSTEM_ID
private val chatId = "-360594386"
@Test
fun sendMessage() {
val text = "Hello World"
TelegramBotUtils.sendMessage(
chatId = chatId,
token = token,
text = text,
)
}
@Test
fun sendDocument() {
val text = "My document caption"
TelegramBotUtils.sendDocument(
chatId = chatId,
token = token,
text = text,
filename = "my exam paper.png",
document = File("src/main/resources/cubetiq.png").readBytes(),
)
}
@Test
fun sendMessageProvider() {
val text = "Hello World from Provider"
TelegramProvider.sendMessage(
chatId = chatId,
token = token,
text = text,
)
}
}