mirror of
https://github.com/PHIDIAS0303/ClusterChatSync.git
synced 2025-12-27 03:05:21 +09:00
158 lines
7.1 KiB
JavaScript
158 lines
7.1 KiB
JavaScript
"use strict";
|
|
const Discord = require("discord.js");
|
|
const fetch = require("node-fetch");
|
|
const {BaseControllerPlugin} = require("@clusterio/controller");
|
|
const {InstanceActionEvent} = require("./info.js");
|
|
|
|
const MAX_DISCORD_MESSAGE_LENGTH = 1950;
|
|
const MIN_CONFIDENCE_SCORE = 10.0;
|
|
|
|
class LibreTranslateAPI {
|
|
constructor(url, apiKey, logger = console) {
|
|
if (!url || !apiKey) this.logger.error('[Chat Sync] LibreTranslate API configuration is incomplete.');
|
|
try {new URL(url);} catch {this.logger.error('[Chat Sync] LibreTranslate url is invalid');}
|
|
this.url = url.endsWith('/') ? url : url + '/';
|
|
this.apiKey = apiKey;
|
|
this.logger = logger;
|
|
}
|
|
|
|
async handleResponse(response) {
|
|
if (!response.ok) this.logger.error(`[Chat Sync] API Request got HTTP ${response.status}`);
|
|
return response.json();
|
|
}
|
|
|
|
async init() {try {this.allowedLanguages = (await this.handleResponse(await fetch(`${this.url}languages?api_key=${this.apiKey}`, {method: 'GET'})))[0].targets || [];} catch (err) {this.logger.error(`[Chat Sync] failed to initialize languages:\n${err.stack}`);}}
|
|
async translateRequest(q, source, target) {try {return (await this.handleResponse(await fetch(`${this.url}translate`, {method: 'POST', headers: {'Content-Type': 'application/json'}, body: JSON.stringify({q: q, api_key: this.apiKey, source: source, target: target})}))).translatedText;} catch (err) {this.logger.error(`[Chat Sync] Translation failed:\n${err.stack}`);}}
|
|
async detectLanguage(q) {try {return (await this.handleResponse(await fetch(`${this.url}detect`, {method: 'POST', headers: {'Content-Type': 'application/json'}, body: JSON.stringify({q: q, api_key: this.apiKey})})))[0];} catch (err) {this.logger.error(`[Chat Sync] Detection failed:\n${err.stack}`);}}
|
|
|
|
async translate(query, targetLanguages = ['zh-Hant', 'en']) {
|
|
console.log(query);
|
|
const result = {action: false, passage: []};
|
|
|
|
try {
|
|
const detection = await this.detectLanguage(query);
|
|
|
|
if (!detection || typeof detection !== 'object' || !detection.confidence || !detection.language) {
|
|
this.logger.warn('[Chat Sync] Invalid language detection result:', detection);
|
|
return result;
|
|
}
|
|
|
|
if (detection.confidence > MIN_CONFIDENCE_SCORE) {
|
|
for (const targetLang of targetLanguages) {
|
|
if (!((detection.language === 'zh-Hans' || detection.language === 'zh-Hant') && (targetLang === 'zh-Hans' || targetLang === 'zh-Hant')) && detection.language !== targetLang && this.allowedLanguages.includes(detection.language) && this.allowedLanguages.includes(targetLang)) {
|
|
result.action = true;
|
|
const translated = await this.translateRequest(query, detection.language, targetLang);
|
|
result.passage.push(`[${detection.language} -> ${targetLang}] ${translated}`);
|
|
}
|
|
}
|
|
}
|
|
|
|
return result;
|
|
} catch (err) {this.logger.error(`[Chat Sync] translation failed:\n${err.stack}`);}
|
|
}
|
|
}
|
|
|
|
class ControllerPlugin extends BaseControllerPlugin {
|
|
async init() {
|
|
this.controller.config.on('fieldChanged', (field, curr, prev) => {if (field === 'ClusterChatSync.discord_bot_token') this.connect().catch(err => {this.logger.error(`[Chat Sync] Discord bot token:\n${err.stack}`);});});
|
|
this.controller.handle(InstanceActionEvent, this.handleInstanceAction.bind(this));
|
|
this.client = null;
|
|
await this.connect();
|
|
}
|
|
|
|
async clientDestroy() {
|
|
if (this.client) {
|
|
this.client.destroy();
|
|
this.client = null;
|
|
}
|
|
}
|
|
|
|
async connect() {
|
|
await this.clientDestroy()
|
|
|
|
if (!this.controller.config.get('ClusterChatSync.discord_bot_token')) {
|
|
this.logger.error('[Chat Sync] Discord bot token not configured.');
|
|
return;
|
|
}
|
|
|
|
this.client = new Discord.Client({intents: [Discord.GatewayIntentBits.Guilds, Discord.GatewayIntentBits.GuildMessages, Discord.GatewayIntentBits.MessageContent]});
|
|
this.logger.info('[Chat Sync] Logging into Discord.');
|
|
|
|
try {
|
|
await this.client.login(this.controller.config.get('ClusterChatSync.discord_bot_token'));
|
|
} catch (err) {
|
|
this.logger.error(`[Chat Sync] Discord login error:\n${err.stack}`);
|
|
await this.clientDestroy()
|
|
return;
|
|
}
|
|
|
|
this.logger.info('[Chat Sync] Logged in Discord successfully.');
|
|
|
|
if (this.controller.config.get('ClusterChatSync.use_libretranslate')) {
|
|
this.translator = new LibreTranslateAPI(this.controller.config.get('ClusterChatSync.libretranslate_url'), this.controller.config.get('ClusterChatSync.libretranslate_key'), this.logger);
|
|
await this.translator.init();
|
|
this.translator_language = this.controller.config.get('ClusterChatSync.libretranslate_language').trim().split(/\s+/);
|
|
}
|
|
}
|
|
|
|
async onShutdown() {await this.clientDestroy()}
|
|
|
|
async sendMessage(request, nrc_msg) {
|
|
const channel_id = this.controller.config.get('ClusterChatSync.discord_channel_mapping')[request.instanceName];
|
|
if (!channel_id) return;
|
|
let channel;
|
|
|
|
try {
|
|
channel = await this.client.channels.fetch(channel_id);
|
|
|
|
if (channel === null) {
|
|
this.logger.error(`[Chat Sync] Discord Channel ID ${channel_id} not found.`);
|
|
return;
|
|
}
|
|
} catch (err) {if (err.code !== 10003) this.logger.error(`[Chat Sync] Discord channel fetch error:\n${err.stack}`);}
|
|
|
|
if (this.controller.config.get("ClusterChatSync.datetime_on_message")) {
|
|
let now = new Date();
|
|
nrc_msg = `${now.getFullYear()}${String(now.getMonth() + 1).padStart(2, '0')}${String(now.getDate()).padStart(2, '0')} ${String(now.getHours()).padStart(2, '0')}${String(now.getMinutes()).padStart(2, '0')}${String(now.getSeconds()).padStart(2, '0')} ${nrc_msg}`
|
|
}
|
|
|
|
if (nrc_msg.length <= MAX_DISCORD_MESSAGE_LENGTH) {
|
|
await channel.send(nrc_msg, {allowedMentions: {parse: []}});
|
|
} else {
|
|
while (nrc_msg.length > 0) {
|
|
let nrc_cmsg = nrc_msg.slice(0, MAX_DISCORD_MESSAGE_LENGTH);
|
|
let nrc_lindex = nrc_cmsg.lastIndexOf(' ');
|
|
|
|
if (nrc_lindex !== -1) {
|
|
nrc_cmsg = nrc_cmsg.slice(0, nrc_lindex);
|
|
nrc_msg = nrc_msg.slice(nrc_lindex).trim();
|
|
} else {nrc_msg = nrc_msg.slice(MAX_DISCORD_MESSAGE_LENGTH).trim();}
|
|
|
|
await channel.send(nrc_cmsg, {allowedMentions: {parse: []}});
|
|
}
|
|
}
|
|
}
|
|
|
|
async handleInstanceAction(request, src) {
|
|
if (request.action === 'CHAT' || request.action === 'SHOUT') {
|
|
const nrc = request.content.replace(/\[special-item=.*?\]/g, '<blueprint>').replace(/<@/g, '<@\u200c>');
|
|
const nrc_index = nrc.indexOf(":");
|
|
const nrc_username = nrc.substring(0, nrc_index);
|
|
const nrc_message = nrc.substring(nrc_index + 1).trim();
|
|
await this.sendMessage(request, `**\`${nrc_username}\`**: ${nrc_message}`)
|
|
|
|
if (this.controller.config.get('ClusterChatSync.use_libretranslate')) {
|
|
this.logger.info(`[Chat Sync] Translating message from ${request.instanceName} (${nrc_username}): ${nrc_message}`);
|
|
const result = await this.translator.translate(nrc_message, this.translator_language);
|
|
|
|
if (result && result.action) {
|
|
await this.sendMessage(request, `**\`${nrc_username}\`**: ${result.passage}`)
|
|
return `[color=255,255,255]\`${nrc_username}\`: ${result}[/color]`;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
module.exports = {ControllerPlugin};
|