@jeffusion/bungee-core 6.8.1 → 6.9.0

package/dist/config-storage-worker.js

@@ -6874,6 +6874,10 @@ function validateGlobalPolicies(object, context) {
   if (object.body_parser_limit !== undefined && (typeof object.body_parser_limit !== "string" || !isBodyParserLimit(object.body_parser_limit))) {
     context.add(typeof object.body_parser_limit === "string" ? "invalid_value" : "invalid_type", "body_parser_limit", "Invalid body parser limit");
   }
+  const maxConnections = object.websocket_max_connections;
+  if (maxConnections !== undefined && maxConnections !== null && (!Number.isSafeInteger(maxConnections) || maxConnections < 1 || maxConnections > 2147483647)) {
+    context.add("invalid_value", "websocket_max_connections", "Expected null or an integer between 1 and 2147483647");
+  }
   validateAuth(object.auth, "auth", context);
   if (object.publication !== undefined) {
     for (const error of validatePublicationPolicy(object.publication)) {
@@ -7243,7 +7247,7 @@ function validatePluginOptions(name, options, path, catalog, context) {
 }
 
 // src/config-storage/compiler.ts
-var GLOBAL_KEYS = ["log_level", "body_parser_limit", "auth", "logging", "publication"];
+var GLOBAL_KEYS = ["log_level", "body_parser_limit", "auth", "logging", "publication", "websocket_max_connections"];
 var SERVICE_KEYS = ["health_check", "failover", "load_balancing"];
 var ROUTE_KEYS = [
   "request",
@@ -8776,13 +8780,15 @@ function replaceActiveMaterialization(db, aggregate) {
   const settingsColumns = new Set(db.query("PRAGMA table_info(settings)").all().map(({ name }) => name));
   const publicationColumn = settingsColumns.has("publication_json") ? ",publication_json" : "";
   const publicationValue = settingsColumns.has("publication_json") ? ",?" : "";
+  const hasWebSocketLimit = settingsColumns.has("websocket_max_connections_json");
   db.run(`INSERT INTO settings
-    (id,log_level,body_parser_limit,auth_json,logging_json${publicationColumn}) VALUES (1,?,?,?,?${publicationValue})`, [
+    (id,log_level,body_parser_limit,auth_json,logging_json${publicationColumn}${hasWebSocketLimit ? ",websocket_max_connections_json" : ""}) VALUES (1,?,?,?,?${publicationValue}${hasWebSocketLimit ? ",?" : ""})`, [
     logical.log_level ?? null,
     logical.body_parser_limit ?? null,
     logical.auth === undefined ? null : canonicalJson(logical.auth),
     logical.logging === undefined ? null : canonicalJson(logical.logging),
-    ...settingsColumns.has("publication_json") ? [logical.publication === undefined ? null : canonicalJson(logical.publication)] : []
+    ...settingsColumns.has("publication_json") ? [logical.publication === undefined ? null : canonicalJson(logical.publication)] : [],
+    ...hasWebSocketLimit ? [logical.websocket_max_connections === undefined ? null : canonicalJson(logical.websocket_max_connections)] : []
   ]);
   writeServices(db, logical.services);
   writeRoutes(db, logical.routes);
@@ -9437,11 +9443,23 @@ var SERVICE_RESERVED = new Set(["id", "position", "name", "endpoints", "service_
 var ROUTE_RESERVED = new Set(["id", "position", "path", "service_id", "service", "service_references", "endpoints", "plugins"]);
 function readSettings(db) {
   const hasPublication = sqliteAll(db, "PRAGMA table_info(settings)").some(({ name }) => name === "publication_json");
-  const rows = sqliteAll(db, `SELECT log_level,body_parser_limit,auth_json,logging_json${hasPublication ? ",publication_json" : ",NULL AS publication_json"} FROM settings`);
+  const rows = sqliteAll(db, `SELECT log_level,body_parser_limit,auth_json,logging_json${hasPublication ? ",publication_json" : ",NULL AS publication_json"},websocket_max_connections_json FROM settings`);
   const row = rows[0];
   if (rows.length !== 1 || row === undefined)
     throw new ConfigRepositoryError("schema_corrupt", "settings singleton is broken");
+  let websocketLimit;
+  if (row.websocket_max_connections_json !== null) {
+    try {
+      websocketLimit = JSON.parse(row.websocket_max_connections_json);
+    } catch {
+      throw new ConfigRepositoryError("schema_corrupt", "WebSocket connection limit is invalid");
+    }
+    if (JSON.stringify(websocketLimit) !== row.websocket_max_connections_json) {
+      throw new ConfigRepositoryError("schema_corrupt", "WebSocket connection limit is not canonical");
+    }
+  }
   return {
+    ...row.websocket_max_connections_json === null ? {} : { websocket_max_connections: websocketLimit },
     ...row.log_level === null ? {} : { log_level: requireOptionalGlobalScalar(row.log_level, "log_level") },
     ...row.body_parser_limit === null ? {} : {
       body_parser_limit: requireOptionalGlobalScalar(row.body_parser_limit, "body_parser_limit")
@@ -9844,6 +9862,20 @@ function databaseSchemaDescriptor(db) {
   });
 }
 
+// src/config-storage/migrations/websocket-connections.ts
+var WEBSOCKET_CONNECTIONS_MIGRATION = {
+  version: 6,
+  name: "websocket_connections",
+  up(db) {
+    db.run(`ALTER TABLE settings ADD COLUMN websocket_max_connections_json TEXT
+      CHECK(websocket_max_connections_json IS NULL OR websocket_max_connections_json='null'
+        OR (json_valid(websocket_max_connections_json)
+          AND json_type(websocket_max_connections_json)='integer'
+          AND json_extract(websocket_max_connections_json,'$') BETWEEN 1 AND 2147483647
+          AND websocket_max_connections_json=CAST(json_extract(websocket_max_connections_json,'$') AS TEXT)))`);
+  }
+};
+
 // src/config-storage/migrations/service-references.ts
 var SERVICE_REFERENCES_MIGRATION = {
   version: 5,
@@ -10265,7 +10297,7 @@ var PUBLICATION_RETIREMENT_MIGRATION = {
 };
 
 // src/config-storage/migrations/plan.ts
-var CONFIG_MIGRATIONS = [CONFIG_BASELINE, PUBLICATION_RETIREMENT_MIGRATION, UI_MENUS_MIGRATION, MANAGEMENT_PROTECTION_MIGRATION, SERVICE_REFERENCES_MIGRATION];
+var CONFIG_MIGRATIONS = [CONFIG_BASELINE, PUBLICATION_RETIREMENT_MIGRATION, UI_MENUS_MIGRATION, MANAGEMENT_PROTECTION_MIGRATION, SERVICE_REFERENCES_MIGRATION, WEBSOCKET_CONNECTIONS_MIGRATION];
 var CONFIG_SCHEMA_VERSION = CONFIG_MIGRATIONS.at(-1).version;
 
 // src/config-storage/schema-fingerprint.ts