First commit
This commit is contained in:
Andrea Cavalli
parent
1366a5803c
commit
52ee9f8e53
2
.gitignore
vendored
2
.gitignore
vendored
|
|
@ -24,3 +24,5 @@
|
|||
hs_err_pid*
|
||||
replay_pid*
|
||||
|
||||
*.env
|
||||
.idea/
|
||||
|
|
|
|||
15
Dockerfile
Normal file
15
Dockerfile
Normal file
|
|
@ -0,0 +1,15 @@
|
|||
FROM maven:3-eclipse-temurin-22 as build
|
||||
WORKDIR /build
|
||||
COPY --link src src
|
||||
COPY --link pom.xml pom.xml
|
||||
RUN mvn package -DskipTests
|
||||
|
||||
FROM eclipse-temurin:22-jdk-alpine
|
||||
WORKDIR /build
|
||||
COPY --from=build --link /build/target/language-block-bot-*.jar language-block-bot.jar
|
||||
ENTRYPOINT ["java", "-jar", "language-block-bot.jar"]
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
7
docker-compose.yml
Normal file
7
docker-compose.yml
Normal file
|
|
@ -0,0 +1,7 @@
|
|||
services:
|
||||
language-block-bot:
|
||||
build:
|
||||
context: .
|
||||
dockerfile: Dockerfile
|
||||
environment:
|
||||
JAVA_TOOL_OPTIONS: "-Dlanguageblockbot.bot.token=$TOKEN"
|
||||
167
pom.xml
Normal file
167
pom.xml
Normal file
|
|
@ -0,0 +1,167 @@
|
|||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
|
||||
<groupId>it.cavallium</groupId>
|
||||
<artifactId>language-block-bot</artifactId>
|
||||
<version>1.0.0-SNAPSHOT</version>
|
||||
|
||||
<properties>
|
||||
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
|
||||
|
||||
<maven-compiler-plugin.version>3.12.1</maven-compiler-plugin.version>
|
||||
<maven-shade-plugin.version>3.2.4</maven-shade-plugin.version>
|
||||
<maven-surefire-plugin.version>3.0.0-M7</maven-surefire-plugin.version>
|
||||
</properties>
|
||||
|
||||
<repositories>
|
||||
<repository>
|
||||
<id>central</id>
|
||||
<url>https://repo.maven.apache.org/maven2/</url>
|
||||
</repository>
|
||||
</repositories>
|
||||
|
||||
<dependencies>
|
||||
<dependency>
|
||||
<groupId>org.jetbrains</groupId>
|
||||
<artifactId>annotations</artifactId>
|
||||
<version>24.1.0</version>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.slf4j</groupId>
|
||||
<artifactId>slf4j-api</artifactId>
|
||||
<version>2.0.12</version>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.telegram</groupId>
|
||||
<artifactId>telegrambots-abilities</artifactId>
|
||||
<version>7.2.1</version>
|
||||
<exclusions>
|
||||
<exclusion>
|
||||
<groupId>org.jetbrains</groupId>
|
||||
<artifactId>annotations</artifactId>
|
||||
</exclusion>
|
||||
<exclusion>
|
||||
<groupId>org.slf4j</groupId>
|
||||
<artifactId>slf4j-api</artifactId>
|
||||
</exclusion>
|
||||
</exclusions>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.telegram</groupId>
|
||||
<artifactId>telegrambots-longpolling</artifactId>
|
||||
<version>7.2.1</version>
|
||||
<exclusions>
|
||||
<exclusion>
|
||||
<groupId>org.jetbrains.kotlin</groupId>
|
||||
<artifactId>kotlin-stdlib-jdk8</artifactId>
|
||||
</exclusion>
|
||||
</exclusions>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.telegram</groupId>
|
||||
<artifactId>telegrambots-client</artifactId>
|
||||
<version>7.2.1</version>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>com.github.pemistahl</groupId>
|
||||
<artifactId>lingua</artifactId>
|
||||
<version>1.2.2</version>
|
||||
<exclusions>
|
||||
<exclusion>
|
||||
<groupId>com.squareup.okio</groupId>
|
||||
<artifactId>okio</artifactId>
|
||||
</exclusion>
|
||||
<exclusion>
|
||||
<groupId>org.jetbrains.kotlin</groupId>
|
||||
<artifactId>kotlin-stdlib</artifactId>
|
||||
</exclusion>
|
||||
<exclusion>
|
||||
<groupId>org.slf4j</groupId>
|
||||
<artifactId>slf4j-api</artifactId>
|
||||
</exclusion>
|
||||
<exclusion>
|
||||
<groupId>org.jetbrains.kotlin</groupId>
|
||||
<artifactId>kotlin-stdlib-jdk8</artifactId>
|
||||
</exclusion>
|
||||
</exclusions>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.slf4j</groupId>
|
||||
<artifactId>slf4j-jdk14</artifactId>
|
||||
<version>2.0.12</version>
|
||||
</dependency>
|
||||
</dependencies>
|
||||
|
||||
<build>
|
||||
<plugins>
|
||||
<plugin>
|
||||
<artifactId>maven-compiler-plugin</artifactId>
|
||||
<version>${maven-compiler-plugin.version}</version>
|
||||
<configuration>
|
||||
<release>21</release>
|
||||
</configuration>
|
||||
</plugin>
|
||||
<plugin>
|
||||
<artifactId>maven-shade-plugin</artifactId>
|
||||
<version>${maven-shade-plugin.version}</version>
|
||||
</plugin>
|
||||
<plugin>
|
||||
<artifactId>maven-surefire-plugin</artifactId>
|
||||
<version>${maven-surefire-plugin.version}</version>
|
||||
</plugin>
|
||||
<plugin>
|
||||
<groupId>org.apache.maven.plugins</groupId>
|
||||
<artifactId>maven-enforcer-plugin</artifactId>
|
||||
<version>3.0.0-M3</version>
|
||||
<executions>
|
||||
<execution>
|
||||
<id>enforce</id>
|
||||
<configuration>
|
||||
<rules>
|
||||
<dependencyConvergence/>
|
||||
</rules>
|
||||
</configuration>
|
||||
<goals>
|
||||
<goal>enforce</goal>
|
||||
</goals>
|
||||
</execution>
|
||||
</executions>
|
||||
</plugin>
|
||||
<plugin>
|
||||
<groupId>org.apache.maven.plugins</groupId>
|
||||
<artifactId>maven-shade-plugin</artifactId>
|
||||
<version>3.5.3</version>
|
||||
<executions>
|
||||
<execution>
|
||||
<phase>package</phase>
|
||||
<goals>
|
||||
<goal>shade</goal>
|
||||
</goals>
|
||||
<configuration>
|
||||
<filters>
|
||||
<filter>
|
||||
<artifact>*:*</artifact>
|
||||
<excludes>
|
||||
<exclude>META-INF/*.SF</exclude>
|
||||
<exclude>META-INF/*.DSA</exclude>
|
||||
<exclude>META-INF/*.RSA</exclude>
|
||||
</excludes>
|
||||
</filter>
|
||||
</filters>
|
||||
<transformers>
|
||||
<transformer implementation="org.apache.maven.plugins.shade.resource.ManifestResourceTransformer">
|
||||
<mainClass>it.cavallium.languageblockbot.Main</mainClass>
|
||||
</transformer>
|
||||
</transformers>
|
||||
</configuration>
|
||||
</execution>
|
||||
</executions>
|
||||
</plugin>
|
||||
</plugins>
|
||||
</build>
|
||||
|
||||
|
||||
</project>
|
||||
|
|
@ -0,0 +1,152 @@
|
|||
package it.cavallium.languageblockbot;
|
||||
|
||||
import com.github.pemistahl.lingua.api.Language;
|
||||
import com.github.pemistahl.lingua.api.LanguageDetector;
|
||||
import com.github.pemistahl.lingua.api.LanguageDetectorBuilder;
|
||||
import java.util.HashSet;
|
||||
import java.util.List;
|
||||
import java.util.Set;
|
||||
import java.util.concurrent.CompletableFuture;
|
||||
import java.util.concurrent.ConcurrentHashMap;
|
||||
import java.util.concurrent.ConcurrentMap;
|
||||
import java.util.stream.Collectors;
|
||||
import org.telegram.telegrambots.abilitybots.api.bot.AbilityBot;
|
||||
import org.telegram.telegrambots.meta.api.methods.groupadministration.GetChatAdministrators;
|
||||
import org.telegram.telegrambots.meta.api.methods.updatingmessages.DeleteMessage;
|
||||
import org.telegram.telegrambots.meta.api.objects.Update;
|
||||
import org.telegram.telegrambots.meta.api.objects.chatmember.ChatMember;
|
||||
import org.telegram.telegrambots.meta.api.objects.message.Message;
|
||||
import org.telegram.telegrambots.meta.exceptions.TelegramApiException;
|
||||
|
||||
public class LanguageBlockBot extends AbilityBot {
|
||||
|
||||
private final LanguageDetector detector;
|
||||
private final ConcurrentMap<Long, Set<Long>> chatAdmins = new ConcurrentHashMap<>();
|
||||
private final Set<Language> whitelistedLangs;
|
||||
private final Set<Language> blacklistedLangs;
|
||||
|
||||
public LanguageBlockBot(String token, Set<Language> whitelistedLangs, Set<Language> blacklistedLangs) {
|
||||
super(new org.telegram.telegrambots.client.okhttp.OkHttpTelegramClient(token), "LanguageBlockBot");
|
||||
|
||||
var allLangs = new HashSet<Language>();
|
||||
allLangs.addAll(whitelistedLangs);
|
||||
allLangs.addAll(blacklistedLangs);
|
||||
this.detector = LanguageDetectorBuilder.fromLanguages(allLangs.toArray(Language[]::new))
|
||||
.withPreloadedLanguageModels()
|
||||
.withMinimumRelativeDistance(0.8)
|
||||
.build();
|
||||
this.whitelistedLangs = whitelistedLangs;
|
||||
this.blacklistedLangs = blacklistedLangs;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void consume(Update update) {
|
||||
if (update.hasMessage()) {
|
||||
var msg = update.getMessage();
|
||||
String text;
|
||||
if (msg.hasText()) {
|
||||
text = msg.getText();
|
||||
} else if (msg.getCaption() != null) {
|
||||
text = msg.getCaption();
|
||||
} else {
|
||||
text = null;
|
||||
}
|
||||
if (text != null) {
|
||||
var detectedLangs = detector.computeLanguageConfidenceValues(text);
|
||||
double acceptableLanguageValue = whitelistedLangs
|
||||
.stream()
|
||||
.map(whitelistedLang -> detectedLangs.getOrDefault(whitelistedLang, 0.0))
|
||||
.reduce(Math::max)
|
||||
.orElse(0d);
|
||||
double blacklistedLanguageValue = blacklistedLangs
|
||||
.stream()
|
||||
.map(blacklistedLang -> detectedLangs.getOrDefault(blacklistedLang, 0.0))
|
||||
.reduce(Math::max)
|
||||
.orElse(0d);
|
||||
|
||||
var shouldBeDeletedByLanguage = !detectedLangs.isEmpty() &&
|
||||
(acceptableLanguageValue <= 0.75
|
||||
|| (acceptableLanguageValue < blacklistedLanguageValue));
|
||||
|
||||
System.out.printf("Received message: \"%s\". shouldBeDeletedByLanguage: %b. Languages: \"%s\".%n",
|
||||
text,
|
||||
shouldBeDeletedByLanguage,
|
||||
detectedLangs.entrySet().stream().map(e -> e.getKey() + "=" + e.getValue()).collect(Collectors.joining(", ", "[", "]"))
|
||||
);
|
||||
if (shouldBeDeletedByLanguage) {
|
||||
isMessageDeletable(msg).thenCompose(deletable -> {
|
||||
var deleteMessageMethod = DeleteMessage
|
||||
.builder()
|
||||
.chatId(msg.getChatId())
|
||||
.messageId(msg.getMessageId())
|
||||
.build();
|
||||
try {
|
||||
var client = this.getTelegramClient();
|
||||
return client.executeAsync(deleteMessageMethod);
|
||||
} catch (TelegramApiException e) {
|
||||
return CompletableFuture.failedFuture(e);
|
||||
}
|
||||
}).whenComplete((ok, ex) -> {
|
||||
if (ex != null) {
|
||||
ex.printStackTrace();
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private CompletableFuture<Boolean> isMessageDeletable(Message msg) {
|
||||
var chatId = msg.getChatId();
|
||||
CompletableFuture<Boolean> canDeleteBecauseItsNotAdmin;
|
||||
if (msg.getSenderChat() != null) {
|
||||
canDeleteBecauseItsNotAdmin = getChatAdmins(chatId)
|
||||
.thenApply(admins -> !admins.contains(msg.getSenderChat().getId()));
|
||||
} else {
|
||||
canDeleteBecauseItsNotAdmin = CompletableFuture.completedFuture(true);
|
||||
}
|
||||
boolean canDeleteBecauseItsNormalMessage = !(msg.getIsAutomaticForward() != null && msg.getIsAutomaticForward())
|
||||
&& (msg.getSenderChat() != null && msg.getSenderChat().isUserChat()) && !msg.hasViaBot();
|
||||
|
||||
return CompletableFuture
|
||||
.completedFuture(canDeleteBecauseItsNormalMessage)
|
||||
.thenCombine(canDeleteBecauseItsNotAdmin, (a, b) -> a && b);
|
||||
}
|
||||
|
||||
private CompletableFuture<Set<Long>> getChatAdmins(Long chatId) {
|
||||
var client = this.getTelegramClient();
|
||||
if (chatId >= 0) {
|
||||
return CompletableFuture.completedFuture(Set.of());
|
||||
} else {
|
||||
try {
|
||||
var cachedAdmins = this.chatAdmins.get(chatId);
|
||||
if (cachedAdmins != null) {
|
||||
return CompletableFuture.completedFuture(cachedAdmins);
|
||||
} else {
|
||||
return client
|
||||
.executeAsync(GetChatAdministrators.builder().chatId(chatId).build())
|
||||
.<List<ChatMember>>handle((result, err) -> {
|
||||
if (err != null) {
|
||||
err.printStackTrace();
|
||||
return List.of();
|
||||
} else {
|
||||
return result;
|
||||
}
|
||||
})
|
||||
.thenApply(admins -> {
|
||||
var newAdmins = admins.stream().map(n -> n.getUser().getId()).collect(Collectors.toCollection(HashSet<Long>::new));
|
||||
chatAdmins.put(chatId, newAdmins);
|
||||
return newAdmins;
|
||||
});
|
||||
}
|
||||
} catch (TelegramApiException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public long creatorId() {
|
||||
return 40951146;
|
||||
}
|
||||
}
|
||||
39
src/main/java/it/cavallium/languageblockbot/Main.java
Normal file
39
src/main/java/it/cavallium/languageblockbot/Main.java
Normal file
|
|
@ -0,0 +1,39 @@
|
|||
package it.cavallium.languageblockbot;
|
||||
|
||||
import com.github.pemistahl.lingua.api.Language;
|
||||
import java.util.Arrays;
|
||||
import java.util.Set;
|
||||
import java.util.stream.Collectors;
|
||||
import java.util.stream.Stream;
|
||||
import org.telegram.telegrambots.longpolling.TelegramBotsLongPollingApplication;
|
||||
|
||||
public class Main {
|
||||
|
||||
public static void main(String[] args) throws Exception {
|
||||
String token = System.getProperty("languageblockbot.bot.token", "invalid-token");
|
||||
Set<Language> whitelistedLangs;
|
||||
Set<Language> blacklistedLangs;
|
||||
try {
|
||||
whitelistedLangs = Stream
|
||||
.of(System.getProperty("languageblockbot.langs.whitelisted", "ENGLISH,FRENCH,ITALIAN").split(","))
|
||||
.map(Language::valueOf)
|
||||
.collect(Collectors.toUnmodifiableSet());
|
||||
blacklistedLangs = Stream
|
||||
.of(System.getProperty("languageblockbot.langs.blacklisted", "SOMALI,ARABIC,RUSSIAN,UKRAINIAN,CHINESE,AZERBAIJANI,SPANISH,PORTUGUESE,HINDI,POLISH,VIETNAMESE").split(","))
|
||||
.map(Language::valueOf)
|
||||
.collect(Collectors.toUnmodifiableSet());
|
||||
} catch (IllegalArgumentException ex) {
|
||||
System.err.println("Invalid language value. Allowed values: " + Arrays.toString(Language.values()));
|
||||
ex.printStackTrace();
|
||||
System.exit(1);
|
||||
return;
|
||||
}
|
||||
try (var app = new TelegramBotsLongPollingApplication()) {
|
||||
var bot = new LanguageBlockBot(token, whitelistedLangs, blacklistedLangs);
|
||||
bot.onRegister();
|
||||
app.registerBot(token, bot);
|
||||
Thread.currentThread().join();
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user