v1.1.1 - Support for game-specific APIs

Author: Infinixius

plugin/messages/tf2/RegeneratePlayer.sp

@@ -0,0 +1,7 @@
+public void Message_RegeneratePlayer(const String:type[], const String:message[], const JSON_Object:jsondata, WebsocketHandle:websocket) {
+	int i = jsondata.GetInt("message")
+	
+	if (IsClientInGame(i)) {
+		TF2_RegeneratePlayer(i)
+	}
+}

plugin/sourcemod.js.sp

@@ -42,6 +42,7 @@ new Handle:g_hChilds
 #include "messages/SetPlayerRendering.sp"
 #include "messages/SlapPlayer.sp"
 #include "messages/TeleportPlayer.sp"
+#include "messages/tf2/RegeneratePlayer.sp"
 
 #include "websocket/WebsocketPing.sp"
 #include "websocket/WebsocketEvents.sp"

plugin/websocket/OnWebsocketReceive.sp

@@ -88,5 +88,7 @@ public void HandleMessage(const String:type[], const String:message[], const JSO
 		Message_SlapPlayer(type, message, jsondata, websocket)
 	} else if (StrEqual(type, "SetPlayerRendering")) {
 		Message_SetPlayerRendering(type, message, jsondata, websocket)
+	} else if (StrEqual(type, "TF2_RegeneratePlayer")) {
+		Message_RegeneratePlayer(type, message, jsondata, websocket)
 	}
 }

src/classes/Game.js

@@ -0,0 +1,23 @@
+/**
+ * @typedef Games
+ * @property {string} None
+ * @property {string} TeamFortress2
+ */
+export const Games = {
+	None: "None",
+	TeamFortress2: "TeamFortress2"
+}
+
+export class Game {
+	constructor(type) {
+		this.type = type ?? Games.None
+	}
+
+	serverModifier(server) {
+		return server
+	}
+
+	playerModifier(player) {
+		return player
+	}
+}

src/classes/Player.js

@@ -40,6 +40,7 @@ import { checkBytes } from "./Utilities.js"
 		this.partial = partial ?? false
 
 		this.update(data)
+		this.players.server.game.playerModifier(this)
 
 		this.players.on("chat", (plr, message, team) => {
 			if (plr.id != this.id) return

src/classes/Server.js

@@ -1,6 +1,7 @@
 import EventEmitter from "events"
 import { Events, Messages, Socket } from "./Socket.js"
 import { Console } from "./Console.js"
+import { Game } from "./Game.js"
 import { Players } from "./Players.js"
 
 /**
@@ -11,7 +12,6 @@ import { Players } from "./Players.js"
  * @property {object} socket - An instance of the `Socket` class used for interacting with the WebSocket. Will be null if not yet connected.
  * @property {object} console - Instance of the `Console` object for this server.
  * @property {object} players - Instance of the `Players` object for this server.
- * 
  * @property {string} currentMap - Name of the current map.  Keep in mind that this is a generic response from the engine and not the game itself.
  * @property {bool} isDedicated - Whether the server is a dedicated server or not.
  * @property {sring} authID - The SteamID of the server.
@@ -35,6 +35,7 @@ export class Server extends EventEmitter {
 		this.socket = null
 		this.console = new Console(this)
 		this.players = new Players(this)
+		this.game = null
 
 		this.currentMap = ""
 		this.nextMap = ""
@@ -55,14 +56,8 @@ export class Server extends EventEmitter {
 	 */
 	async connect() {
 		await new Socket(this, this.host)
-			.then((socket) => {
+			.then(async (socket) => {
 				this.socket = socket
-				/**
-				 * Fired when the connection to the server has been established.
-				 * 
-				 * @event Server#ready
-				 */
-				this.emit("ready", true)
 
 				this.socket.on(Events.ServerUpdate, (data) => {
 					var msg = data.message
@@ -85,6 +80,27 @@ export class Server extends EventEmitter {
 						data.message.value
 					)
 				})
+
+				await this.fetch()
+
+				switch (this.gameDescription) {
+					case "Team Fortress":
+						await import("../games/TeamFortress2.js").then(module => {
+							this.game = new module.TeamFortress2()
+							this.game.serverModifier(this)
+						})
+						break
+					default:
+						this.game = new Game()
+						this.game.serverModifier(this)
+				}
+
+				/**
+				 * Fired when the connection to the server has been established.
+				 * 
+				 * @event Server#ready
+				 */
+				this.emit("ready", true)
 			})
 			.catch(err => { throw err })
 	}

src/classes/Socket.js

@@ -202,6 +202,7 @@ export class Socket extends EventEmitter {
 	TeleportPlayer: "TeleportPlayer",
 	SlapPlayer: "SlapPlayer",
 	SetPlayerRendering: "SetPlayerRendering"
+	TF2_RegeneratePlayer: "TF2_RegeneratePlayer"
 }
 
 /**

src/games/TeamFortress2.js

@@ -0,0 +1,26 @@
+import { Game, Games } from "../classes/Game.js"
+import { Messages } from "../classes/Socket.js"
+
+export class TeamFortress2 extends Game {
+	constructor() {
+		super(Games.TeamFortress2)
+	}
+
+	serverModifier(server) {
+		return server
+	}
+
+	playerModifier(player) {
+		/**
+		 * Regenerates a player's health and ammo. Team Fortress 2 only!
+		 * [SourceMod API Reference](https://sm.alliedmods.net/new-api/tf2/TF2_RegeneratePlayer)
+		 * 
+		 * @function Player#regenerate
+		 */
+		player.regenerate = function() {
+			return player.players.server.socket.send(Messages.TF2_RegeneratePlayer, player.id)
+		}
+
+		return player
+	}
+}