Hri7566
/
mpp-server-dev2
1
0
Fork 1
Code Issues Pull Requests Packages Projects 1 Releases Wiki Activity

Create docker build config

This commit is contained in:
Hri7566 2024-08-13 05:55:27 -04:00
parent 7b124b1d83
commit 78fc178652
29 changed files with 441 additions and 129 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
.dockerignore Normal file
View File

@ -0,0 +1,13 @@
node_modules
Dockerfile*
docker-compose*
.dockerignore
.git
.gitignore
README.md
LICENSE
.vscode
.env.template
.gitmodules
.prettierrc
.eslintrc.js

34
Dockerfile Normal file
View File

@ -0,0 +1,34 @@
FROM oven/bun:latest AS base
WORKDIR /usr/src/app
FROM base AS install
RUN mkdir -p /temp/dev
COPY package.json bun.lockb /temp/dev/
RUN cd /temp/dev && bun install --frozen-lockfile
RUN mkdir -p /temp/prod
COPY package.json bun.lockb /temp/prod
RUN cd /temp/prod && bun install --frozen-lockfile --production
FROM base AS prerelease
COPY --from=install /temp/dev/node_modules node_modules
COPY . .
ENV NODE_ENV=production
#RUN bun test
#RUN bun build
FROM base AS release
COPY --from=install /temp/prod/node_modules node_modules
COPY --from=prerelease /usr/src/app/src/ ./src
COPY --from=prerelease /usr/src/app/package.json .
COPY --from=prerelease /usr/src/app/config ./config
COPY --from=prerelease /usr/src/app/public ./public
COPY --from=prerelease /usr/src/app/mppkey ./mppkey
COPY --from=prerelease /usr/src/app/tsconfig.json .
COPY --from=prerelease /usr/src/app/prisma ./prisma
COPY --from=prerelease /usr/src/app/.env .
USER bun
EXPOSE 8443/tcp
ENTRYPOINT [ "bun", "." ]

9
README.md
View File

@ -66,9 +66,6 @@ This has always been the future intention of this project.
## TODO
- Fully implement and test tags
- Tags are sent to clients now
- Check if tags are sent to everyone
- Channel data saving
- Permission groups and permissions
- Probable permission groups: owner, admin, mod, trialmod, default
@ -104,6 +101,12 @@ This has always been the future intention of this project.
- Check for different messages?
- Check for URL?
- Notifications for server-generated XSS?
- Migrate to PostgreSQL instead of SQLite
- Likely a low priority, we use prisma anyway, but it would be nice to have a server
- Implement user caching
- Skip redis due to the infamous licensing issues
- Probably use a simple in-memory cache
- Likely store with leveldb or JSON
## How to run

BIN
bun.lockb
View File

Binary file not shown.

19
config/channels.yml
View File

@ -1,8 +1,11 @@
# Channel config file
# Which channels to keep loaded on startup
forceLoad:
- lobby
- test/awkward
# Default settings for lobbies
lobbySettings:
lobby: true
chat: true
@ -10,19 +13,35 @@ lobbySettings:
visible: true
color: "#73b3cc"
color2: "#273546"
# Default settings for regular channels
defaultSettings:
chat: true
crownsolo: false
color: "#3b5054"
color2: "#001014"
visible: true
# Regexes to match against channel names to determine whether they are lobbies or not
# This doesn't affect the `isRealLobby` function, which is used to determine "classic" lobbies
lobbyRegexes:
- ^lobby[0-9][0-9]$
- ^lobby[0-9]$
- ^lobby$
- ^lobbyNaN$
- ^test/.+$
# Backdoor channel ID for bypassing the lobby limit
lobbyBackdoor: lolwutsecretlobbybackdoor
# Channel ID for where you get sent when you join a channel that is full/you get banned/etc
fullChannel: test/awkward
# Whether to send the channel limit to the client
sendLimit: false
# Whether to give the crown to the user who had it when they rejoin
chownOnRejoin: true
# Time in milliseconds to wait before destroying an empty channel
channelDestroyTimeout: 1000

12
config/prometheus.yml Normal file
View File

@ -0,0 +1,12 @@
global:
scrape_interval: 15s
scrape_configs:
- job_name: prometheus
scrape_interval: "5s"
static_configs:
- targets: ["localhost:9090"]
- job_name: mpp
static_configs:
- targets: ["192.168.1.24:9100"]

5
config/users.yml
View File

@ -37,14 +37,15 @@ enableAdminEval: true
# The token validation scheme. Valid values are "none", "jwt" and "uuid".
# This server will still validate existing tokens generated with other schemes if not set to "none", mimicking MPP.net's server.
# This is set to "none" by default because MPP.com does not have a token system.
tokenAuth: none
tokenAuth: jwt
# The browser challenge scheme. Valid options are "none", "obf" and "basic".
# This is to change what is sent in the "b" message.
# "none" will disable the browser challenge,
# "obf" will sent an obfuscated function to the client,
# and "basic" will just send a simple function that expects a boolean.
browserChallenge: none
# FIXME Note that "obf" is not implemented yet, and has undefined behavior.
browserChallenge: basic
# Scheme for generating user IDs.
# Valid options are "random", "sha256", "mpp" and "uuid".

1
config/util.yml Normal file
View File

@ -0,0 +1 @@
enableLogFiles: true

5
package.json
View File

@ -6,6 +6,10 @@
"keywords": [],
"author": "Hri7566",
"license": "ISC",
"scripts": {
"start": "bun run src/index.ts",
"dev": "bun run src/index.ts --watch"
},
"dependencies": {
"@prisma/client": "5.17.0",
"@t3-oss/env-core": "^0.6.1",
@ -17,6 +21,7 @@
"jsonwebtoken": "^9.0.2",
"keccak": "^2.1.0",
"nunjucks": "^3.2.4",
"prom-client": "^15.1.3",
"unique-names-generator": "^4.7.1",
"yaml": "^2.5.0",
"zod": "^3.23.8"

34
src/channel/Channel.ts
View File

@ -68,7 +68,7 @@ export class Channel extends EventEmitter {
}
private async save() {
this.logger.debug("Saving channel data");
//this.logger.debug("Saving channel data");
try {
const info = this.getInfo();
@ -78,16 +78,16 @@ export class Channel extends EventEmitter {
flags: JSON.stringify(this.flags)
};
this.logger.debug("Channel data to save:", data);
//this.logger.debug("Channel data to save:", data);
await saveChannel(this.getID(), data);
} catch (err) {
this.logger.debug("Error saving cannel:", err);
this.logger.warn("Error saving channel data:", err);
}
}
private async load() {
this.logger.debug("Loading saved data");
//this.logger.debug("Loading saved data");
try {
const data = await getSavedChannel(this.getID());
if (data) {
@ -100,9 +100,11 @@ export class Channel extends EventEmitter {
forceloadChannel(this.getID());
}
this.logger.debug("Loaded channel data:", data);
//this.logger.debug("Loaded channel data:", data);
this.emit("update", this);
} catch (err) {
this.logger.debug("Error loading channel data:", err);
this.logger.error("Error loading channel data:", err);
}
}
} catch (err) { }
@ -125,7 +127,7 @@ export class Channel extends EventEmitter {
) {
super();
this.logger = new Logger("Channel - " + _id);
this.logger = new Logger("Channel - " + _id, "logs/channel");
this.settings = {};
// Copy default settings
@ -209,7 +211,13 @@ export class Channel extends EventEmitter {
}
if (this.ppl.length == 0 && !this.stays) {
if (config.channelDestroyTimeout) {
setTimeout(() => {
this.destroy();
}, config.channelDestroyTimeout);
} else {
this.destroy();
}
}
});
@ -246,17 +254,21 @@ export class Channel extends EventEmitter {
.replace(/(\p{Mc}{5})\p{Mc}+/gu, "$1")
.trim();
const part = socket.getParticipant() as Participant;
let outgoing: ClientEvents["a"] = {
m: "a",
a: msg.message,
t: Date.now(),
p: socket.getParticipant() as Participant
p: part
};
this.sendArray([outgoing]);
this.chatHistory.push(outgoing);
await saveChatHistory(this.getID(), this.chatHistory);
this.logger.info(`${part._id} ${part.name}: ${outgoing.a}`);
if (msg.message.startsWith("/")) {
this.emit("command", msg, socket);
}
@ -373,6 +385,10 @@ export class Channel extends EventEmitter {
}
]);
});
this.on("set owner id", id => {
this.setFlag("owner_id", id);
});
}
/**
@ -450,7 +466,7 @@ export class Channel extends EventEmitter {
// Set the verified settings
for (const key of Object.keys(validatedSet)) {
this.logger.debug(`${key}: ${(validatedSet as any)[key]}`);
//this.logger.debug(`${key}: ${(validatedSet as any)[key]}`);
if ((validatedSet as any)[key] === false) continue;
(this.settings as any)[key] = (set as any)[key];
}

4
src/channel/config.ts
View File

@ -10,6 +10,7 @@ interface ChannelConfig {
fullChannel: string;
sendLimit: boolean;
chownOnRejoin: boolean;
channelDestroyTimeout: number;
}
export const config = loadConfig<ChannelConfig>("config/channels.yml", {
@ -34,5 +35,6 @@ export const config = loadConfig<ChannelConfig>("config/channels.yml", {
lobbyBackdoor: "lolwutsecretlobbybackdoor",
fullChannel: "test/awkward",
sendLimit: false,
chownOnRejoin: true
chownOnRejoin: true,
channelDestroyTimeout: 1000
});

36
src/index.ts
View File

@ -1,9 +1,15 @@
/**
* MPP Server 2
* for mpp.dev
* by Hri7566
* for https://www.multiplayerpiano.dev/
* Written by Hri7566
* This code is licensed under the GNU General Public License v3.0.
* Please see `./LICENSE` for more information.
*/
/**
* Main entry point for the server
**/
// There are a lot of unhinged bs comments in this repo
// Pay no attention to the ones that cuss you out
@ -11,15 +17,23 @@
import "./ws/server";
import { loadForcedStartupChannels } from "./channel/forceLoad";
import { Logger } from "./util/Logger";
import { startReadline } from "./util/readline";
import { startMetricsServer } from "./util/metrics";
// Let's construct an entire object just for one thing to be printed
// and then keep it in memory for the entirety of runtime
const logger = new Logger("Main");
logger.info("Forceloading startup channels...");
loadForcedStartupChannels();
// wrapper for some reason
export function startServer() {
// Let's construct an entire object just for one thing to be printed
// and then keep it in memory for the entirety of runtime
const logger = new Logger("Main");
logger.info("Forceloading startup channels...");
loadForcedStartupChannels();
// This literally breaks editors and they stick all the imports here instead of at the top
import "./util/readline";
// Break the console
startReadline();
// Nevermind we use it twice
logger.info("Ready");
// Nevermind, two things are printed
logger.info("Ready");
}
startServer();
startMetricsServer();

45
src/util/Logger.ts
View File

@ -1,8 +1,19 @@
import EventEmitter from "events";
import { padNum, unimportant } from "./helpers";
import { join } from "path";
import { existsSync, mkdirSync, appendFile, writeFile } from "fs";
import { config } from "./utilConfig";
export const logEvents = new EventEmitter();
const logFolder = "./logs";
// https://stackoverflow.com/questions/14693701/how-can-i-remove-the-ansi-escape-sequences-from-a-string-in-python
const logRegex = /\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])/g;
if (config.enableLogFiles) {
if (!existsSync(logFolder)) mkdirSync(logFolder);
}
/**
* A logger that doesn't fuck with the readline prompt
* timestamps are likely wrong because of js timezones
@ -14,7 +25,7 @@ export class Logger {
* @param method The method from `console` to use
* @param args The data to print
**/
private static log(method: string, ...args: any[]) {
private static log(method: string, logPath: string, ...args: any[]) {
// Un-fuck the readline prompt
process.stdout.write("\x1b[2K\r");
@ -33,6 +44,23 @@ export class Logger {
// Emit the log event for remote consoles
logEvents.emit("log", method, unimportant(this.getDate()), unimportant(this.getHHMMSSMS()), args);
if (config.enableLogFiles) {
// Write to file
(async () => {
const orig = unimportant(this.getDate()) + " " + unimportant(this.getHHMMSSMS()) + " " + args.join(" ") + "\n"
const text = orig.replace(logRegex, "");
if (!existsSync(logPath)) {
writeFile(logPath, text, (err) => {
if (err) console.error(err);
});
} else {
appendFile(logPath, text, (err) => {
if (err) console.error(err);
});
}
})();
}
}
/**
@ -62,14 +90,19 @@ export class Logger {
return new Date().toISOString().split("T")[0];
}
constructor(public id: string) { }
public logPath: string;
constructor(public id: string, logdir: string = logFolder) {
if (!existsSync(logdir)) mkdirSync(logdir);
this.logPath = join(logdir, `${encodeURIComponent(this.id)}.log`);
}
/**
* Print an info message
* @param args The data to print
**/
public info(...args: any[]) {
Logger.log("log", `[${this.id}]`, `\x1b[34m[info]\x1b[0m`, ...args);
Logger.log("log", this.logPath, `[${this.id}]`, `\x1b[34m[info]\x1b[0m`, ...args);
}
/**
@ -77,7 +110,7 @@ export class Logger {
* @param args The data to print
**/
public error(...args: any[]) {
Logger.log("error", `[${this.id}]`, `\x1b[31m[error]\x1b[0m`, ...args);
Logger.log("error", this.logPath, `[${this.id}]`, `\x1b[31m[error]\x1b[0m`, ...args);
}
/**
@ -85,7 +118,7 @@ export class Logger {
* @param args The data to print
**/
public warn(...args: any[]) {
Logger.log("warn", `[${this.id}]`, `\x1b[33m[warn]\x1b[0m`, ...args);
Logger.log("warn", this.logPath, `[${this.id}]`, `\x1b[33m[warn]\x1b[0m`, ...args);
}
/**
@ -93,6 +126,6 @@ export class Logger {
* @param args The data to print
**/
public debug(...args: any[]) {
Logger.log("debug", `[${this.id}]`, `\x1b[32m[debug]\x1b[0m`, ...args);
Logger.log("debug", this.logPath, `[${this.id}]`, `\x1b[32m[debug]\x1b[0m`, ...args);
}
}

2
src/util/config.ts
View File

@ -1,6 +1,5 @@
import { existsSync, readFileSync, writeFileSync } from "fs";
import { parse, stringify } from "yaml";
import { z } from "zod";
/**
* This file uses the synchronous functions from the fs
@ -63,6 +62,7 @@ export function loadConfig<T>(configPath: string, defaultConfig: T): T {
return config as T;
} else {
// Write default config to disk and use that
//logger.warn(`Config file "${configPath}" not found, writing default config to disk`);
writeConfig(configPath, defaultConfig);
return defaultConfig as T;
}

3
src/util/id.ts
View File

@ -35,6 +35,9 @@ export function createUserID(ip: string) {
.update("::ffff:" + ip + env.SALT)
.digest("hex")
.substring(0, 24);
} else {
// Fallback if someone typed random garbage in the config
return createID();
}
}

37
src/util/metrics.ts Normal file
View File

@ -0,0 +1,37 @@
import client, { Registry } from "prom-client";
import { Logger } from "./Logger";
const logger = new Logger("Metrics Server");
export function startMetricsServer() {
client.collectDefaultMetrics();
logger.info("Starting Prometheus metrics server...");
const server = Bun.serve({
port: 9100,
async fetch(req) {
const res = new Response(await client.register.metrics());
res.headers.set("Content-Type", client.register.contentType);
return res;
}
});
enableMetrics();
}
export const metrics = {
concurrentUsers: new client.Histogram({
name: "concurrent_users",
help: "Number of concurrent users",
}),
callbacks: [],
addCallback(callback: (...args: any[]) => void | Promise<void>) {
(this.callbacks as ((...args: any[]) => void)[]).push(callback);
}
}
function enableMetrics() {
setInterval(() => {
(metrics.callbacks as ((...args: any[]) => void)[]).forEach(callback => callback());
}, 5000);
}

29
src/util/readline/index.ts
View File

@ -3,23 +3,32 @@ import logger from "./logger";
import Command from "./Command";
import "./commands";
export const rl = readline.createInterface({
export let rl: readline.Interface;
// Turned into a function so the import isn't in a weird spot
export function startReadline() {
rl = readline.createInterface({
input: process.stdin,
output: process.stdout
});
});
rl.setPrompt("mpps> ");
rl.prompt();
rl.setPrompt("mpps> ");
rl.prompt();
rl.on("line", async line => {
rl.on("line", async line => {
const out = await Command.handleCommand(line);
logger.info(out);
rl.prompt();
});
});
rl.on("SIGINT", () => {
rl.on("SIGINT", () => {
process.exit();
});
});
// Fucking cringe but it works
(globalThis as unknown as any).rl = rl;
// Fucking cringe but it works
(globalThis as unknown as any).rl = rl;
}
export function stopReadline() {
rl.close();
}

5
src/util/utilConfig.ts Normal file
View File

@ -0,0 +1,5 @@
import { loadConfig } from "./config";
export const config = loadConfig("config/util.yml", {
enableLogFiles: true
});

56
src/ws/Gateway.ts
View File

@ -11,65 +11,71 @@
* or, you know, maybe I could log their user agent
* and IP address instead sometime in the future.
*/
import { Logger } from "../util/Logger";
const logger = new Logger("Socket Gateway");
export class Gateway {
// Whether we have correctly processed this socket's hi message
public hasProcessedHi = false;
public hasProcessedHi = false; // implemented
// Whether they have sent the MIDI devices message
public hasSentDevices = false;
public hasSentDevices = false; // implemented
// Whether they have sent a token
public hasSentToken = false;
public hasSentToken = false; // implemented
// Whether their token is valid
public isTokenValid = false;
public isTokenValid = false; // implemented
// Their user agent, if sent
public userAgent = "";
public userAgent = ""; // TODO
// Whether they have moved their cursor
public hasCursorMoved = false;
public hasCursorMoved = false; // implemented
// Whether they sent a cursor message that contained numbers instead of stringified numbers
public isCursorNotString = false;
public isCursorNotString = false; // implemented
// The last time they sent a ping message
public lastPing = Date.now();
public lastPing = Date.now(); // implemented
// Whether they have joined any channel
public hasJoinedAnyChannel = false;
public hasJoinedAnyChannel = false; // implemented
// Whether they have joined the lobby
public hasJoinedLobby = false;
// Whether they have joined a lobby
public hasJoinedLobby = false; // implemented
// Whether they have made a regular non-websocket request to the HTTP server
// probably useful for checking if they are actually on the site
// Maybe not useful if cloudflare is being used
// In that scenario, templating wouldn't work, either
public hasConnectedToHTTPServer = false;
public hasConnectedToHTTPServer = false; // implemented
// Various chat message flags
public hasSentChatMessage = false;
public hasSentChatMessageWithCapitalLettersOnly = false;
public hasSentChatMessageWithInvisibleCharacters = false;
public hasSentChatMessageWithEmoji = false;
public hasSentChatMessage = false; // implemented
public hasSentChatMessageWithCapitalLettersOnly = false; // implemented
public hasSentChatMessageWithInvisibleCharacters = false; // implemented
public hasSentChatMessageWithEmoji = false; // implemented
// Whehter or not the user has played the piano in this session
public hasPlayedPianoBefore = false;
public hasPlayedPianoBefore = false; // implemented
// Whether the user has sent a channel list subscription request, a.k.a. opened the channel list
public hasOpenedChannelList = false;
public hasOpenedChannelList = false; // implemented
// Whether the user has changed their name/color this session (not just changed from default)
public hasChangedName = false;
public hasChangedColor = false;
// Whether the user has sent
public hasSentCustomNoteData = false;
public hasChangedName = false; // implemented
public hasChangedColor = false; // implemented
// Whether they sent an admin message that was invalid (wrong password, etc)
public hasSentInvalidAdminMessage = false;
public hasSentInvalidAdminMessage = false; // implemented
// Whether or not they have passed the b message
public hasCompletedBrowserChallenge = false;
public hasCompletedBrowserChallenge = false; // implemented
public dump() {
return JSON.stringify(this, undefined, 4);
}
}

15
src/ws/Socket.ts
View File

@ -146,7 +146,7 @@ export class Socket extends EventEmitter {
// Basic function
this.sendArray([{
m: "b",
code: `~return true;`
code: `~return btoa(JSON.stringify([true, navigator.userAgent]));`
}]);
} else if (config.browserChallenge == "obf") {
// Obfuscated challenge building
@ -183,7 +183,7 @@ export class Socket extends EventEmitter {
}
/**
* Move this participant to a channel
* Move this socket to a channel
* @param _id Target channel ID
* @param set Channel settings, if the channel is instantiated
* @param force Whether to make this socket join regardless of channel properties
@ -239,6 +239,17 @@ export class Socket extends EventEmitter {
// Make them join the new channel
channel.join(this, force);
}
// Gateway stuff
this.gateway.hasJoinedAnyChannel = true;
const ch = this.getCurrentChannel();
if (ch) {
if (ch.isLobby()) {
this.gateway.hasJoinedLobby = true;
}
}
}
public admin = new EventEmitter();

2
src/ws/events/user/handlers/+ls.ts
View File

@ -16,6 +16,8 @@ export const plus_ls: ServerEventListener<"+ls"> = {
if (!socket.rateLimits.normal["+ls"].attempt()) return;
}
socket.gateway.hasOpenedChannelList = true;
socket.subscribeToChannelList();
}
};

31
src/ws/events/user/handlers/a.ts
View File

@ -1,4 +1,28 @@
import { ServerEventListener } from "../../../../util/types";
import { Socket } from "../../../Socket";
import { ServerEventListener, ServerEvents } from "../../../../util/types";
// https://stackoverflow.com/questions/64509631/is-there-a-regex-to-match-all-unicode-emojis
const emojiRegex = /(\u00a9|\u00ae|[\u2000-\u3300]|\ud83c[\ud000-\udfff]|\ud83d[\ud000-\udfff]|\ud83e[\ud000-\udfff])/g;
function populateSocketChatGatewayFlags(msg: ServerEvents["a"], socket: Socket) {
socket.gateway.hasSentChatMessage = true;
if (msg.message.toUpperCase() == msg.message) {
socket.gateway.hasSentChatMessageWithCapitalLettersOnly = true;
}
if (msg.message.includes("\u034f") || msg.message.includes("\u200b")) {
socket.gateway.hasSentChatMessageWithInvisibleCharacters = true;
}
if (msg.message.match(/[^\x00-\x7f]/gm)) {
socket.gateway.hasSentChatMessageWithInvisibleCharacters = true;
}
if (msg.message.match(emojiRegex)) {
socket.gateway.hasSentChatMessageWithEmoji = true;
}
}
export const a: ServerEventListener<"a"> = {
id: "a",
@ -7,9 +31,14 @@ export const a: ServerEventListener<"a"> = {
const flags = socket.getUserFlags();
if (!flags) return;
if (typeof msg.message !== "string") return;
// Why did I write this statement so weird
if (!flags["no chat rate limit"] || flags["no chat rate limit"] == 0)
if (!socket.rateLimits?.normal.a.attempt()) return;
populateSocketChatGatewayFlags(msg, socket);
const ch = socket.getCurrentChannel();
if (!ch) return;

11
src/ws/events/user/handlers/admin_message.ts
View File

@ -9,8 +9,15 @@ export const admin_message: ServerEventListener<"admin message"> = {
if (socket.rateLimits)
if (!socket.rateLimits.normal["admin message"].attempt()) return;
if (typeof msg.password !== "string") return;
if (msg.password !== env.ADMIN_PASS) return;
if (typeof msg.password !== "string") {
socket.gateway.hasSentInvalidAdminMessage = true;
return;
}
if (msg.password !== env.ADMIN_PASS) {
socket.gateway.hasSentInvalidAdminMessage = true;
return;
}
// Probably shouldn't be using password auth in 2024
// Maybe I'll setup a dashboard instead some day

18
src/ws/events/user/handlers/hi.ts
View File

@ -17,10 +17,20 @@ export const hi: ServerEventListener<"hi"> = {
// Browser challenge
if (config.browserChallenge == "basic") {
if (typeof msg.code !== "boolean") return;
try {
if (typeof msg.code !== "string") return;
const code = atob(msg.code);
const arr = JSON.parse(code);
if (msg.code === true) {
if (arr[0] === true) {
socket.gateway.hasCompletedBrowserChallenge = true;
if (typeof arr[1] === "string") {
socket.gateway.userAgent = arr[1];
}
}
} catch (err) {
logger.warn("Unable to parse basic browser challenge code:", err);
}
} else if (config.browserChallenge == "obf") {
// TODO
@ -34,11 +44,14 @@ export const hi: ServerEventListener<"hi"> = {
if (config.tokenAuth !== "none") {
if (typeof msg.token !== "string") {
socket.gateway.hasSentToken = true;
// Get a saved token
token = await getToken(socket.getUserID());
if (typeof token !== "string") {
// Generate a new one
token = await createToken(socket.getUserID(), socket.gateway);
socket.gateway.isTokenValid = true;
if (typeof token !== "string") {
logger.warn(`Unable to generate token for user ${socket.getUserID()}`);
@ -54,6 +67,7 @@ export const hi: ServerEventListener<"hi"> = {
//return;
} else {
token = msg.token;
socket.gateway.isTokenValid = true;
}
}
}

18
src/ws/events/user/handlers/m.ts
View File

@ -13,11 +13,21 @@ export const m: ServerEventListener<"m"> = {
let x = msg.x;
let y = msg.y;
// Make it numbers
if (typeof msg.x == "string") x = parseFloat(msg.x);
if (typeof msg.y == "string") y = parseFloat(msg.y);
// Parse cursor position if it's strings
if (typeof msg.x == "string") {
x = parseFloat(msg.x);
} else {
socket.gateway.isCursorNotString = true;
}
// Move the laggy piece of shit
if (typeof msg.y == "string") {
y = parseFloat(msg.y);
} else {
socket.gateway.isCursorNotString = true;
}
// Relocate the laggy microscopic speck
socket.setCursorPos(x, y);
socket.gateway.hasCursorMoved = true;
}
};

2
src/ws/events/user/handlers/n.ts
View File

@ -11,6 +11,8 @@ export const n: ServerEventListener<"n"> = {
if (!Array.isArray(msg.n)) return;
if (typeof msg.t !== "number") return;
socket.gateway.hasPlayedPianoBefore = true;
// This should've been here months ago
const channel = socket.getCurrentChannel();
if (!channel) return;

2
src/ws/events/user/handlers/t.ts
View File

@ -12,6 +12,8 @@ export const t: ServerEventListener<"t"> = {
if (typeof msg.e !== "number") return;
}
socket.gateway.lastPing = Date.now();
// Pong
socket.sendArray([
{

21
src/ws/events/user/handlers/userset.ts
View File

@ -1,20 +1,21 @@
import { ServerEventListener } from "../../../../util/types";
import { config } from "../../../usersConfig";
export const userset: ServerEventListener<"userset"> = {
id: "userset",
callback: async (msg, socket) => {
// Change username/color
if (!socket.rateLimits?.chains.userset.attempt()) return;
// You can disable color in the config because
// Brandon's/jacored's server doesn't allow color changes,
// and that's the OG server, but folks over at MPP.net
// said otherwise because they're dumb roleplayers
// or something and don't understand the unique value
// of the fishing bot and how it allows you to change colors
// without much control, giving it the feeling of value...
// Kinda reminds me of crypto.
// Also, Brandon's server had color changing on before.
if (!msg.set.name && !msg.set.color) return;
if (typeof msg.set.name !== "string" && typeof msg.set.color !== "string") return;
if (typeof msg.set.name == "string") {
socket.gateway.hasChangedName = true;
}
if (typeof msg.set.color == "string" && config.enableColorChanging) {
socket.gateway.hasChangedColor = true;
}
socket.userset(msg.set.name, msg.set.color);
}
};

45
src/ws/server.ts
View File

@ -7,9 +7,14 @@ import { Socket, socketsBySocketID } from "./Socket";
import env from "../util/env";
import { getMOTD } from "../util/motd";
import nunjucks from "nunjucks";
import { metrics } from "../util/metrics";
const logger = new Logger("WebSocket Server");
// ip -> timestamp
// for checking if they visited the site and are also connected to the websocket
const httpIpCache = new Map<string, number>();
/**
* Get a rendered version of the index file
* @returns Response with html in it
@ -39,17 +44,15 @@ export const app = Bun.serve<{ ip: string }>({
fetch: (req, server) => {
const reqip = server.requestIP(req);
if (!reqip) return;
const ip = req.headers.get("x-forwarded-for") || reqip.address;
if (
server.upgrade(req, {
data: {
ip
}
})
) {
// Upgrade websocket connections
if (server.upgrade(req, { data: { ip } })) {
return;
} else {
}
httpIpCache.set(ip, Date.now());
const url = new URL(req.url).pathname;
// lol
@ -67,37 +70,55 @@ export const app = Bun.serve<{ ip: string }>({
// Time for unreadable blocks of confusion
try {
// Is it a file?
if (fs.lstatSync(file).isFile()) {
// Read the file
const data = Bun.file(file);
// Return the file
if (data) {
return new Response(data);
} else {
return getIndex();
}
} else {
// Return the index file, since it's a channel name or something
return getIndex();
}
} catch (err) {
// Return the index file as a coverup of our extreme failure
return getIndex();
}
}
},
websocket: {
open: ws => {
// We got one!
// swimming in the pool
const socket = new Socket(ws, createSocketID());
// Reel 'em in...
(ws as unknown as any).socket = socket;
// logger.debug("Connection at " + socket.getIP());
// Let's put it in the dinner bucket.
if (socket.socketID == undefined) {
socket.socketID = createSocketID();
}
socketsBySocketID.set(socket.socketID, socket);
const ip = socket.getIP();
if (httpIpCache.has(ip)) {
const date = httpIpCache.get(ip);
if (date) {
if (Date.now() - date < 1000 * 60) {
// They got the page and we were connected in under a minute
socket.gateway.hasConnectedToHTTPServer = true;
} else {
// They got the page and a long time has passed
httpIpCache.delete(ip);
}
}
}
},
message: (ws, message) => {