mirror of
https://github.com/clusterio/clusterio.git
synced 2026-11-13 07:04:53 +09:00
Hornwitser
7f8a700716
Split up the action of registring an event or request data class to be decoded by the Link class and the action of registring a handler for those requests and events. This moves the validation work from Link instantiation time to startup initialisation which makes logic errors fail on startup instead of failing instantiation of the link when a connection attempt is made. Additionally implement a partial validation logic for messages received using the newly refactored message registration. This unfortunately ended up merged with the registration refactor.
73 lines
2.4 KiB
JavaScript
73 lines
2.4 KiB
JavaScript
"use strict";
|
|
const libData = require("@clusterio/lib/data");
|
|
const libErrors = require("@clusterio/lib/errors");
|
|
const libLink = require("@clusterio/lib/link");
|
|
const { logger } = require("@clusterio/lib/logging");
|
|
const libPlugin = require("@clusterio/lib/plugin");
|
|
const ControllerRouter = require("./ControllerRouter");
|
|
|
|
|
|
/**
|
|
* Base class for controller connections
|
|
*
|
|
* @extends module:lib/link.Link
|
|
* @alias module:controller/src/BaseConnection
|
|
*/
|
|
class BaseConnection extends libLink.Link {
|
|
constructor(connector, controller) {
|
|
/** @member {module:controller/src/WsServerConnector} module:controller/src/BaseConnection#connector */
|
|
super(connector);
|
|
this.router = new ControllerRouter(controller);
|
|
this._controller = controller;
|
|
for (let [Request, handler] of controller._registeredRequests) { this.handle(Request, handler); }
|
|
for (let [Request, handler] of controller._fallbackedRequests) { this.fallbackRequest(Request, handler); }
|
|
for (let [Event, handler] of controller._registeredEvents) { this.handle(Event, handler); }
|
|
for (let [Event, handler] of controller._snoopedEvents) { this.snoopEvent(Event, handler); }
|
|
|
|
this.handle(libData.ModPackGetRequest, this.handleModPackGetRequest.bind(this));
|
|
this.handle(libData.ModPackGetDefaultRequest, this.handleModPackGetDefaultRequest.bind(this));
|
|
}
|
|
|
|
async disconnect(code, reason) {
|
|
try {
|
|
await this.connector.disconnect();
|
|
} catch (err) {
|
|
if (!(err instanceof libErrors.SessionLost)) {
|
|
logger.error(`"Unexpected error preparing disconnect:\n${err.stack}`);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* True if the link is connected, not in the dropped state and not in
|
|
* the process of disconnecting.
|
|
* @type {boolean}
|
|
*/
|
|
get connected() {
|
|
return this.connector.connected;
|
|
}
|
|
|
|
async handleModPackGetRequest(request) {
|
|
let { id } = request;
|
|
let modPack = this._controller.modPacks.get(id);
|
|
if (!modPack) {
|
|
throw new libErrors.RequestError(`Mod pack with ID ${id} does not exist`);
|
|
}
|
|
return modPack;
|
|
}
|
|
|
|
async handleModPackGetDefaultRequest() {
|
|
let id = this._controller.config.get("controller.default_mod_pack_id");
|
|
if (id === null) {
|
|
throw new libErrors.RequestError("Default mod pack not set on controller");
|
|
}
|
|
let modPack = this._controller.modPacks.get(id);
|
|
if (!modPack) {
|
|
throw new libErrors.RequestError(`Default mod pack configured (${id}) does not exist`);
|
|
}
|
|
return modPack;
|
|
}
|
|
}
|
|
|
|
module.exports = BaseConnection;
|