this bot has to be rewritten sometime soon
This commit is contained in:
parent
c7c3c06170
commit
a44ce9d23e
4 changed files with 75 additions and 274 deletions
BIN
bun.lockb
BIN
bun.lockb
Binary file not shown.
104
index.ts
104
index.ts
|
@ -1,9 +1,14 @@
|
||||||
import { Client, Events, GatewayIntentBits, TextChannel } from "discord.js";
|
import { Client, Events, GatewayIntentBits, Message, TextChannel } from "discord.js";
|
||||||
import { Rcon } from "./rcon";
|
import { Rcon } from "rconts";
|
||||||
import config from "./config.json" assert { type: "json" };
|
import config from "./config.json" assert { type: "json" };
|
||||||
|
|
||||||
const rcon = new Rcon(config.rcon.host, config.rcon.port, config.rcon.password);
|
const rcon = new Rcon({
|
||||||
let rconReconnectTimeout: Timer | null | undefined;
|
host: config.rcon.host,
|
||||||
|
port: config.rcon.port,
|
||||||
|
password: config.rcon.password
|
||||||
|
});
|
||||||
|
let ignoreDeletion: Set<string> = new Set();
|
||||||
|
|
||||||
|
|
||||||
const client = new Client({
|
const client = new Client({
|
||||||
intents: [
|
intents: [
|
||||||
|
@ -13,33 +18,56 @@ const client = new Client({
|
||||||
],
|
],
|
||||||
});
|
});
|
||||||
|
|
||||||
rcon.on("connect", () => {
|
rcon.on("connected", () => {
|
||||||
console.log("RCON connected.");
|
console.log("RCON connected.");
|
||||||
});
|
});
|
||||||
|
|
||||||
rcon.on("auth", () => {
|
async function whitelist(message: Message, username: string) {
|
||||||
console.log("RCON authenicated.");
|
const response = await rcon.exec("whitelist add " + username);
|
||||||
});
|
let deleteThisMessage;
|
||||||
|
|
||||||
|
if (response == "That player does not exist") {
|
||||||
|
try {
|
||||||
|
ignoreDeletion.add(message.id);
|
||||||
|
await message.delete();
|
||||||
|
|
||||||
|
} catch {
|
||||||
|
|
||||||
rcon.on("end", () => {
|
|
||||||
console.log("RCON ended. Reconnecting in 2000ms.")
|
|
||||||
if(rconReconnectTimeout) {
|
|
||||||
clearTimeout(rconReconnectTimeout);
|
|
||||||
}
|
}
|
||||||
rconReconnectTimeout = setTimeout(() => {
|
deleteThisMessage = await message.channel.send(
|
||||||
rcon.connect()
|
`<@${message.author.id}>, you have not been whitelisted. Please join the server at least once, and then resend your IGN.`
|
||||||
}, 2000)
|
);
|
||||||
})
|
} else if (response.includes("Incorrect argument for")) {
|
||||||
|
try {
|
||||||
|
ignoreDeletion.add(message.id);
|
||||||
|
await message.delete();
|
||||||
|
|
||||||
|
} catch {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
deleteThisMessage = await message.channel.send(
|
||||||
|
`<@${message.author.id}>, your message was invalid. Include ONLY your username, nothing else.`
|
||||||
|
);
|
||||||
|
}
|
||||||
|
if(deleteThisMessage) {
|
||||||
|
setTimeout(async () => {
|
||||||
|
await deleteThisMessage.delete();
|
||||||
|
}, 5000);
|
||||||
|
}
|
||||||
|
console.log("[whitelist] response for", username, "is", response);
|
||||||
|
}
|
||||||
|
|
||||||
client.once(Events.ClientReady, async (readyClient) => {
|
client.once(Events.ClientReady, async (readyClient) => {
|
||||||
console.log(`Ready! Logged in as ${readyClient.user.tag}`);
|
console.log(`Ready! Logged in as ${readyClient.user.tag}`);
|
||||||
let playerUsernames: string[] = [];
|
let playerUsernames: Map<Message, string[]> = new Map();
|
||||||
|
|
||||||
const channel = (await client.channels.fetch(config.channel)) as TextChannel;
|
const channel = (await client.channels.fetch(config.channel)) as TextChannel;
|
||||||
|
|
||||||
const messages = await channel.messages.fetch();
|
const messages = await channel.messages.fetch();
|
||||||
|
|
||||||
for await (const f of messages) {
|
for await (const f of messages) {
|
||||||
playerUsernames = playerUsernames.concat(f[1].content.split("\n"));
|
playerUsernames.set(f[1], f[1].content.split("\n"));
|
||||||
if (!f[1].reactions.resolve("✅")) await f[1].react("✅");
|
if (!f[1].reactions.resolve("✅")) await f[1].react("✅");
|
||||||
let member = f[1].member;
|
let member = f[1].member;
|
||||||
try {
|
try {
|
||||||
|
@ -51,21 +79,22 @@ client.once(Events.ClientReady, async (readyClient) => {
|
||||||
if (member.nickname !== f[1].content.split("\n")[0]) {
|
if (member.nickname !== f[1].content.split("\n")[0]) {
|
||||||
try {
|
try {
|
||||||
await member.setNickname(f[1].content.split("\n")[0], "whitelisted");
|
await member.setNickname(f[1].content.split("\n")[0], "whitelisted");
|
||||||
} catch {}
|
} catch { }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for (let i = 0; i < playerUsernames.length; i++) {
|
for await (let u of playerUsernames) {
|
||||||
setTimeout(() => {
|
for await(let name of u[1]) {
|
||||||
rcon.send("whitelist add " + playerUsernames[i]);
|
await whitelist(u[0], name);
|
||||||
}, i * 20);
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
console.log("Mass-whitelisted.");
|
console.log("Mass-whitelisted.");
|
||||||
});
|
});
|
||||||
|
|
||||||
client.on(Events.MessageCreate, async (message) => {
|
client.on(Events.MessageCreate, async (message) => {
|
||||||
|
|
||||||
if (
|
if (
|
||||||
message.channelId == config.channel &&
|
message.channelId == config.channel &&
|
||||||
message.author!.id != client.user!.id
|
message.author!.id != client.user!.id
|
||||||
|
@ -96,19 +125,24 @@ client.on(Events.MessageCreate, async (message) => {
|
||||||
if (config.changeNicknames) {
|
if (config.changeNicknames) {
|
||||||
try {
|
try {
|
||||||
await member.setNickname(playerUsernames[0]);
|
await member.setNickname(playerUsernames[0]);
|
||||||
} catch {}
|
} catch { }
|
||||||
}
|
}
|
||||||
for (let i = 0; i < playerUsernames.length; i++) {
|
for await (let name of playerUsernames) {
|
||||||
setTimeout(() => {
|
await whitelist(message, name)
|
||||||
rcon.send("whitelist add " + playerUsernames[i]);
|
|
||||||
}, i * 20);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
await message.react("✅");
|
await message.react("✅");
|
||||||
|
} catch {
|
||||||
|
}
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
client.on(Events.MessageDelete, async (message) => {
|
client.on(Events.MessageDelete, async (message) => {
|
||||||
|
if (ignoreDeletion.has(message.id)) {
|
||||||
|
ignoreDeletion.delete(message.id);
|
||||||
|
return;
|
||||||
|
}
|
||||||
if (
|
if (
|
||||||
message.channelId == config.channel &&
|
message.channelId == config.channel &&
|
||||||
message.author!.id != client.user!.id
|
message.author!.id != client.user!.id
|
||||||
|
@ -122,17 +156,17 @@ client.on(Events.MessageDelete, async (message) => {
|
||||||
if (config.changeNicknames) {
|
if (config.changeNicknames) {
|
||||||
try {
|
try {
|
||||||
await member.setNickname(null, "unwhitelisted");
|
await member.setNickname(null, "unwhitelisted");
|
||||||
} catch {}
|
} catch { }
|
||||||
|
|
||||||
|
}
|
||||||
setTimeout(async () => {
|
setTimeout(async () => {
|
||||||
await deleteThisMessage.delete();
|
await deleteThisMessage.delete();
|
||||||
}, 5000);
|
}, 5000);
|
||||||
}
|
const playerUsernames = message.content?.split("\n") || [];
|
||||||
const usernames = message.content?.split("\n") || [];
|
|
||||||
|
|
||||||
for (let i = 0; i < usernames.length; i++) {
|
for await (let name of playerUsernames) {
|
||||||
setTimeout(() => {
|
const response = await rcon.exec("whitelist remove " + name);
|
||||||
rcon.send("whitelist remove " + usernames[i]);
|
console.log("[delete-whitelist] response for", name, "is", response);
|
||||||
}, i * 20);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -12,6 +12,8 @@
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"bufferutil": "^4.0.8",
|
"bufferutil": "^4.0.8",
|
||||||
"discord.js": "^14.15.3",
|
"discord.js": "^14.15.3",
|
||||||
|
"node-gyp": "^11.1.0",
|
||||||
|
"rconts": "git+https://git.sad.ovh/sophie/rconts.git#8f6d5a48330a35f7e4e288b5a3f0104e249aa575",
|
||||||
"utf-8-validate": "^6.0.4",
|
"utf-8-validate": "^6.0.4",
|
||||||
"zlib-sync": "^0.1.9"
|
"zlib-sync": "^0.1.9"
|
||||||
}
|
}
|
||||||
|
|
235
rcon.ts
235
rcon.ts
|
@ -1,235 +0,0 @@
|
||||||
import EventEmitter from "events";
|
|
||||||
import * as net from "net";
|
|
||||||
import * as dgram from "dgram";
|
|
||||||
import { Buffer } from "buffer";
|
|
||||||
import type TypedEmitter from "typed-emitter";
|
|
||||||
|
|
||||||
type Events = {
|
|
||||||
error: (error: Error) => void;
|
|
||||||
auth: () => void;
|
|
||||||
response: (response: string) => void;
|
|
||||||
connect: () => void;
|
|
||||||
end: () => void;
|
|
||||||
done: () => void;
|
|
||||||
};
|
|
||||||
|
|
||||||
export const PacketType = {
|
|
||||||
COMMAND: 0x02,
|
|
||||||
AUTH: 0x03,
|
|
||||||
RESPONSE_VALUE: 0x00,
|
|
||||||
RESPONSE_AUTH: 0x02,
|
|
||||||
};
|
|
||||||
|
|
||||||
interface Options {
|
|
||||||
tcp?: boolean;
|
|
||||||
challenge?: boolean;
|
|
||||||
id?: number;
|
|
||||||
}
|
|
||||||
export class Rcon extends (EventEmitter as new () => TypedEmitter<Events>) {
|
|
||||||
private host: string;
|
|
||||||
private port: number;
|
|
||||||
private password: string;
|
|
||||||
private rconId: number;
|
|
||||||
private hasAuthed: boolean;
|
|
||||||
private outstandingData: Uint8Array | null;
|
|
||||||
private tcp: boolean;
|
|
||||||
private challenge: boolean;
|
|
||||||
private _challengeToken: string;
|
|
||||||
private _tcpSocket!: net.Socket;
|
|
||||||
private _udpSocket!: dgram.Socket;
|
|
||||||
|
|
||||||
constructor(host: string, port: number, password: string, options?: Options) {
|
|
||||||
super();
|
|
||||||
options = options || {};
|
|
||||||
this.host = host;
|
|
||||||
this.port = port;
|
|
||||||
this.password = password;
|
|
||||||
this.rconId = options.id || 0x0012d4a6; // This is arbitrary in most cases
|
|
||||||
this.hasAuthed = false;
|
|
||||||
this.outstandingData = null;
|
|
||||||
this.tcp = options.tcp ? options.tcp : true;
|
|
||||||
this.challenge = options.challenge ? options.challenge : true;
|
|
||||||
this._challengeToken = "";
|
|
||||||
}
|
|
||||||
|
|
||||||
public send = (data: string, cmd?: number, id?: number): void => {
|
|
||||||
let sendBuf: Buffer;
|
|
||||||
if (this.tcp) {
|
|
||||||
cmd = cmd || PacketType.COMMAND;
|
|
||||||
id = id || this.rconId;
|
|
||||||
|
|
||||||
const length = Buffer.byteLength(data);
|
|
||||||
sendBuf = Buffer.alloc(length + 14);
|
|
||||||
sendBuf.writeInt32LE(length + 10, 0);
|
|
||||||
sendBuf.writeInt32LE(id, 4);
|
|
||||||
sendBuf.writeInt32LE(cmd, 8);
|
|
||||||
sendBuf.write(data, 12);
|
|
||||||
sendBuf.writeInt16LE(0, length + 12);
|
|
||||||
} else {
|
|
||||||
if (this.challenge && !this._challengeToken) {
|
|
||||||
this.emit("error", new Error("Not authenticated"));
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
let str = "rcon ";
|
|
||||||
if (this._challengeToken) str += this._challengeToken + " ";
|
|
||||||
if (this.password) str += this.password + " ";
|
|
||||||
str += data + "\n";
|
|
||||||
sendBuf = Buffer.alloc(4 + Buffer.byteLength(str));
|
|
||||||
sendBuf.writeInt32LE(-1, 0);
|
|
||||||
sendBuf.write(str, 4);
|
|
||||||
}
|
|
||||||
this._sendSocket(sendBuf);
|
|
||||||
};
|
|
||||||
|
|
||||||
private _sendSocket = (buf: Buffer) => {
|
|
||||||
if (this._tcpSocket) {
|
|
||||||
this._tcpSocket.write(buf.toString("binary"), "binary");
|
|
||||||
} else if (this._udpSocket) {
|
|
||||||
this._udpSocket.send(buf, 0, buf.length, this.port, this.host);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
public connect = (): void => {
|
|
||||||
if (this.tcp) {
|
|
||||||
this._tcpSocket = net.createConnection(this.port, this.host);
|
|
||||||
this._tcpSocket.on("data", (data) => {
|
|
||||||
this._tcpSocketOnData(data);
|
|
||||||
});
|
|
||||||
this._tcpSocket.on("connect", () => {
|
|
||||||
this.socketOnConnect();
|
|
||||||
});
|
|
||||||
this._tcpSocket.on("error", (err) => {
|
|
||||||
//this.emit("error", err);
|
|
||||||
this.socketOnEnd()
|
|
||||||
});
|
|
||||||
this._tcpSocket.on("end", () => {
|
|
||||||
this.socketOnEnd();
|
|
||||||
});
|
|
||||||
} else {
|
|
||||||
this._udpSocket = dgram.createSocket("udp4");
|
|
||||||
this._udpSocket
|
|
||||||
.on("message", (data) => {
|
|
||||||
this._udpSocketOnData(data);
|
|
||||||
})
|
|
||||||
.on("listening", () => {
|
|
||||||
this.socketOnConnect();
|
|
||||||
})
|
|
||||||
.on("error", (err) => {
|
|
||||||
this.emit("error", err);
|
|
||||||
})
|
|
||||||
.on("close", () => {
|
|
||||||
this.socketOnEnd();
|
|
||||||
});
|
|
||||||
this._udpSocket.bind(0);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
public disconnect = (): void => {
|
|
||||||
if (this._tcpSocket) this._tcpSocket.end();
|
|
||||||
if (this._udpSocket) this._udpSocket.close();
|
|
||||||
};
|
|
||||||
|
|
||||||
public setTimeout = (timeout: number, callback: () => void): void => {
|
|
||||||
if (!this._tcpSocket) return;
|
|
||||||
this._tcpSocket.setTimeout(timeout, () => {
|
|
||||||
this._tcpSocket.end();
|
|
||||||
if (callback) callback();
|
|
||||||
});
|
|
||||||
};
|
|
||||||
|
|
||||||
private _udpSocketOnData = (data: Buffer) => {
|
|
||||||
const a = data.readUInt32LE(0);
|
|
||||||
if (a === 0xffffffff) {
|
|
||||||
const str = data.toString("utf-8", 4);
|
|
||||||
const tokens = str.split(" ");
|
|
||||||
if (
|
|
||||||
tokens.length === 3 &&
|
|
||||||
tokens[0] === "challenge" &&
|
|
||||||
tokens[1] === "rcon"
|
|
||||||
) {
|
|
||||||
this._challengeToken = tokens[2]
|
|
||||||
.substring(0, tokens[2].length - 1)
|
|
||||||
.trim();
|
|
||||||
this.hasAuthed = true;
|
|
||||||
this.emit("auth");
|
|
||||||
} else {
|
|
||||||
this.emit("response", str.substring(1, str.length - 2));
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
this.emit("error", new Error("Received malformed packet"));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
private _tcpSocketOnData = (data: Buffer) => {
|
|
||||||
if (this.outstandingData != null) {
|
|
||||||
data = Buffer.concat(
|
|
||||||
[this.outstandingData, data],
|
|
||||||
this.outstandingData.length + data.length
|
|
||||||
);
|
|
||||||
this.outstandingData = null;
|
|
||||||
}
|
|
||||||
|
|
||||||
while (data.length) {
|
|
||||||
const len = data.readInt32LE(0);
|
|
||||||
if (!len) return;
|
|
||||||
|
|
||||||
const id = data.readInt32LE(4);
|
|
||||||
const type = data.readInt32LE(8);
|
|
||||||
|
|
||||||
if (len >= 10 && data.length >= len + 4) {
|
|
||||||
if (id === this.rconId) {
|
|
||||||
if (!this.hasAuthed && type === PacketType.RESPONSE_AUTH) {
|
|
||||||
this.hasAuthed = true;
|
|
||||||
this.emit("auth");
|
|
||||||
} else if (type === PacketType.RESPONSE_VALUE) {
|
|
||||||
// Read just the body of the packet (truncate the last null byte)
|
|
||||||
// See https://developer.valvesoftware.com/wiki/Source_RCON_Protocol for details
|
|
||||||
let str = data.toString("utf8", 12, 12 + len - 10);
|
|
||||||
|
|
||||||
if (str.charAt(str.length - 1) === "\n") {
|
|
||||||
// Emit the response without the newline.
|
|
||||||
str = str.substring(0, str.length - 1);
|
|
||||||
}
|
|
||||||
|
|
||||||
this.emit("response", str);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
this.emit("error", new Error("Authentication failed"));
|
|
||||||
}
|
|
||||||
|
|
||||||
data = data.slice(12 + len - 8);
|
|
||||||
} else {
|
|
||||||
// Keep a reference to the chunk if it doesn't represent a full packet
|
|
||||||
this.outstandingData = data;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
public socketOnConnect = (): void => {
|
|
||||||
this.emit("connect");
|
|
||||||
|
|
||||||
if (this.tcp) {
|
|
||||||
this.send(this.password, PacketType.AUTH);
|
|
||||||
} else if (this.challenge) {
|
|
||||||
const str = "challenge rcon\n";
|
|
||||||
const sendBuf = Buffer.alloc(str.length + 4);
|
|
||||||
sendBuf.writeInt32LE(-1, 0);
|
|
||||||
sendBuf.write(str, 4);
|
|
||||||
this._sendSocket(sendBuf);
|
|
||||||
} else {
|
|
||||||
const sendBuf = Buffer.alloc(5);
|
|
||||||
sendBuf.writeInt32LE(-1, 0);
|
|
||||||
sendBuf.writeUInt8(0, 4);
|
|
||||||
this._sendSocket(sendBuf);
|
|
||||||
|
|
||||||
this.hasAuthed = true;
|
|
||||||
this.emit("auth");
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
public socketOnEnd = (): void => {
|
|
||||||
this.emit("end");
|
|
||||||
this.hasAuthed = false;
|
|
||||||
};
|
|
||||||
}
|
|
Loading…
Reference in a new issue