@moneypot/hub 1.0.0 → 1.1.0

package/README.md

@@ -1,8 +1,10 @@
 # @moneypot/hub
 
-This library implements a controller server that manages users and balances across any number of Moneypot casinos.
+@moneypot/hub is our official game server that integrates with any number of Moneypot casinos.
 
-You can use it to quickly create your own controller.
+- Extend it with custom tables and game logic.
+- Give it an api key for each controller you've registered on each casino.
+- It will automatically sync users, their balances, deposits, and withdrawals.
 
 Example implementations:
 
@@ -57,3 +59,13 @@ startAndListen(options)
   })
   .catch(console.error);
 ```
+
+## Dashboard
+
+When the server is running, visit its admin dashboard at the `/dashboard` route.
+
+You'll need an api key from your hub database:
+
+```sql
+insert into hub.api_key default values returning key;
+```

package/dist/src/index.js

@@ -1,7 +1,7 @@
 import PgUpgradeSchema, { DatabaseAheadError, } from "@moneypot/pg-upgrade-schema";
 import * as db from "./db/index.js";
 import config from "./config.js";
-import * as server from "./server/index.js";
+import { createHubServer } from "./server/index.js";
 import { initializeTransferProcessors } from "./process-transfers.js";
 import { join } from "path";
 import { logger, setLogger } from "./logger.js";
@@ -10,6 +10,10 @@ async function initialize(options) {
     if (options.logger) {
         setLogger(options.logger);
     }
+    if (options.signal.aborted) {
+        logger.info("Initialization aborted by graceful shutdown");
+        return;
+    }
     const pgClient = db.getPgClient(config.SUPERUSER_DATABASE_URL);
     await pgClient.connect();
     try {
@@ -28,6 +32,10 @@ async function initialize(options) {
             process.exit(1);
         }
     }
+    if (options.signal.aborted) {
+        logger.info("Initialization aborted by graceful shutdown");
+        return;
+    }
     if (options.userDatabaseMigrationsPath) {
         await PgUpgradeSchema.default({
             pgClient,
@@ -36,7 +44,13 @@ async function initialize(options) {
         });
     }
     await pgClient.end();
-    initializeTransferProcessors();
+    if (options.signal.aborted) {
+        logger.info("Initialization aborted by graceful shutdown");
+        return;
+    }
+    initializeTransferProcessors({
+        signal: options.signal,
+    });
 }
 export async function startAndListen(options) {
     if (options.userDatabaseMigrationsPath &&
@@ -46,18 +60,46 @@ export async function startAndListen(options) {
     if (!options.exportSchemaSDLPath.startsWith("/")) {
         throw new Error(`exportSchemaSDLPath must be an absolute path, got ${options.exportSchemaSDLPath}`);
     }
-    await initialize({
-        userDatabaseMigrationsPath: options.userDatabaseMigrationsPath,
-        logger: options.logger,
-    });
-    return server
-        .listen({
+    const abortController = new AbortController();
+    let isShuttingDown = false;
+    const hubServer = createHubServer({
         configureApp: options.configureApp,
         plugins: options.plugins,
         exportSchemaSDLPath: options.exportSchemaSDLPath,
         extraPgSchemas: options.extraPgSchemas,
-    })
-        .then(() => {
+        abortSignal: abortController.signal,
+    });
+    const gracefulShutdown = async () => {
+        if (isShuttingDown) {
+            console.warn("Already shutting down.");
+            return;
+        }
+        isShuttingDown = true;
+        console.log("Shutting down gracefully...");
+        abortController.abort();
+        const cleanupTasks = [
+            db.postgraphilePool.end(),
+            db.superuserPool.end(),
+            db.notifier.close(),
+            hubServer.shutdown(),
+        ];
+        try {
+            await Promise.all(cleanupTasks);
+            console.log("All cleanup tasks completed. Shutdown successful.");
+        }
+        catch (err) {
+            console.error("Error during cleanup:", err);
+            process.exit(1);
+        }
+    };
+    process.on("SIGINT", gracefulShutdown);
+    process.on("SIGTERM", gracefulShutdown);
+    await initialize({
+        userDatabaseMigrationsPath: options.userDatabaseMigrationsPath,
+        logger: options.logger,
+        signal: abortController.signal,
+    });
+    return hubServer.listen().then(() => {
         return {
             port: config.PORT,
         };

package/dist/src/plugins/hub-add-casino.js

@@ -47,7 +47,8 @@ export const HubAddCasinoPlugin = makeExtendSchemaPlugin((build) => {
             Mutation: {
                 hubAddCasino(_, { $input }) {
                     const $identity = context().get("identity");
-                    const $casinoId = sideEffect([$input, $identity], ([input, identity]) => {
+                    const $abortSignal = context().get("abortSignal");
+                    const $casinoId = sideEffect([$input, $identity, $abortSignal], ([input, identity, abortSignal]) => {
                         return withPgPoolTransaction(superuserPool, async (pgClient) => {
                             if (identity?.kind !== "operator") {
                                 throw new GraphQLError("Unauthorized");
@@ -140,7 +141,10 @@ export const HubAddCasinoPlugin = makeExtendSchemaPlugin((build) => {
                                 graphqlClient,
                                 casinoId: casino.id,
                             });
-                            startTransferProcessor({ casinoId: casino.id });
+                            startTransferProcessor({
+                                casinoId: casino.id,
+                                signal: abortSignal,
+                            });
                             return casino.id;
                         });
                     });

package/dist/src/process-transfers.d.ts

@@ -1,7 +1,10 @@
 import * as db from "./db/index.js";
 export declare function casinoIdsInProcess(): string[];
-export declare function startTransferProcessor({ casinoId, }: {
+export declare function startTransferProcessor({ casinoId, signal, }: {
     casinoId: db.DbCasino["id"];
+    signal: AbortSignal;
 }): void;
 export declare function stopTransferProcessor(casinoId: string): void;
-export declare function initializeTransferProcessors(): void;
+export declare function initializeTransferProcessors({ signal, }: {
+    signal: AbortSignal;
+}): void;

package/dist/src/process-transfers.js

@@ -63,7 +63,7 @@ const MP_CLAIM_TRANSFER = gql(`
 const casinoMap = new Map();
 const MIN_BACKOFF_TIME = 5000;
 const MAX_BACKOFF_TIME = 30 * 1000;
-async function listenForNewCasinos() {
+async function listenForNewCasinos({ signal }) {
     const pgClient = new pg.Client(config.SUPERUSER_DATABASE_URL);
     await pgClient.connect();
     const NewCasinoPayload = z.object({
@@ -89,11 +89,15 @@ async function listenForNewCasinos() {
                     logger.error("Error parsing new casino notification:", result.error);
                     return;
                 }
-                startTransferProcessor({ casinoId: result.data.id });
+                startTransferProcessor({ casinoId: result.data.id, signal });
                 break;
             }
         }
     });
+    signal.addEventListener("abort", () => {
+        pgClient.removeAllListeners("notification");
+        pgClient.end();
+    });
     await pgClient.query("LISTEN new_casino");
 }
 async function processTransfer({ casinoId, controllerId, transfer, graphqlClient, }) {
@@ -191,14 +195,18 @@ async function processTransfer({ casinoId, controllerId, transfer, graphqlClient
         }
     }
 }
-async function processTransfersUntilEmpty({ afterCursor, graphqlClient, casinoInfo, }) {
+async function processTransfersUntilEmpty({ afterCursor, graphqlClient, casinoInfo, signal, }) {
     let hasNextPage = true;
     const timeout = (ms) => new Promise((res) => setTimeout(res, ms));
-    while (hasNextPage) {
+    while (hasNextPage && !signal.aborted) {
         await processWithdrawalRequests({
             casinoId: casinoInfo.id,
             graphqlClient,
         });
+        if (signal.aborted) {
+            logger.info(`[processTransfersUntilEmpty] Aborted by graceful shutdown.`);
+            break;
+        }
         const data = await graphqlClient.request(PAGINATE_TRANSFERS, {
             controllerId: casinoInfo.controller_id,
             limit: 10,
@@ -219,6 +227,10 @@ async function processTransfersUntilEmpty({ afterCursor, graphqlClient, casinoIn
                 logger.error(`transfer ${transfer.id} is not an ExperienceTransfer, skipping`);
                 continue;
             }
+            if (signal.aborted) {
+                logger.info(`[processTransfersUntilEmpty] Aborted by graceful shutdown.`);
+                break;
+            }
             await processTransfer({
                 casinoId: casinoInfo.id,
                 controllerId: casinoInfo.controller_id,
@@ -232,6 +244,10 @@ async function processTransfersUntilEmpty({ afterCursor, graphqlClient, casinoIn
         }
         hasNextPage = data.transfersByHolder.pageInfo.hasNextPage;
         afterCursor = data.transfersByHolder.pageInfo.endCursor || undefined;
+        if (signal.aborted) {
+            logger.info(`[processTransfersUntilEmpty] Aborted by graceful shutdown.`);
+            break;
+        }
         await timeout(1000);
     }
     return afterCursor;
@@ -292,7 +308,11 @@ async function processWithdrawals({ casinoId, graphqlClient, }) {
 export function casinoIdsInProcess() {
     return Array.from(casinoMap.keys());
 }
-export function startTransferProcessor({ casinoId, }) {
+export function startTransferProcessor({ casinoId, signal, }) {
+    if (signal.aborted) {
+        logger.info(`[startTransferProcessor] AbortSignal aborted. Not starting processor for casino ${casinoId}`);
+        return;
+    }
     logger.info(`starting processor for casino ${casinoId}`);
     if (casinoMap.has(casinoId)) {
         throw new Error(`processor already running for casino ${casinoId}`);
@@ -321,7 +341,7 @@ export function startTransferProcessor({ casinoId, }) {
         processorState.emitter.on("once", () => {
             shouldStop = true;
         });
-        while (!shouldStop) {
+        while (!shouldStop && !signal.aborted) {
             try {
                 const now = Date.now();
                 const timeToWait = Math.max(0, processorState.backoffTime - (now - processorState.lastAttempt));
@@ -344,11 +364,19 @@ export function startTransferProcessor({ casinoId, }) {
                     graphqlUrl: casino.graphql_url,
                     apiKey: casinoSecret.api_key,
                 });
+                if (signal.aborted) {
+                    logger.info(`[startTransferProcessor] Aborted by graceful shutdown.`);
+                    break;
+                }
                 await jwtService.refreshCasinoJwksTask(superuserPool, {
                     casinoId: casino.id,
                     graphqlClient,
                 });
                 if (!upsertedCurrencies) {
+                    if (signal.aborted) {
+                        logger.info(`[startTransferProcessor] Aborted by graceful shutdown.`);
+                        break;
+                    }
                     const currencies = await graphqlClient
                         .request(GET_CURRENCIES)
                         .then((res) => {
@@ -362,11 +390,20 @@ export function startTransferProcessor({ casinoId, }) {
                     }
                     upsertedCurrencies = true;
                 }
+                if (signal.aborted) {
+                    logger.info(`[startTransferProcessor] Aborted by graceful shutdown.`);
+                    break;
+                }
                 cursor = await processTransfersUntilEmpty({
                     afterCursor: cursor,
                     graphqlClient,
                     casinoInfo: { ...casino, ...casinoSecret },
+                    signal,
                 });
+                if (signal.aborted) {
+                    logger.info(`[startTransferProcessor] Aborted by graceful shutdown.`);
+                    break;
+                }
                 await processWithdrawals({ casinoId: casino.id, graphqlClient });
                 processorState.backoffTime = MIN_BACKOFF_TIME;
             }
@@ -376,7 +413,7 @@ export function startTransferProcessor({ casinoId, }) {
                 logger.debug(`Next retry for casino ${casinoId} in ${processorState.backoffTime}ms`);
             }
         }
-        logger.debug(`processor stopped for casino ${casinoId}`);
+        logger.info(`processor stopped for casino ${casinoId}`);
         casinoMap.delete(casinoId);
     })();
 }
@@ -388,7 +425,7 @@ export function stopTransferProcessor(casinoId) {
     }
     processorState.emitter.emit("stop");
 }
-export function initializeTransferProcessors() {
+export function initializeTransferProcessors({ signal, }) {
     (async () => {
         try {
             const casinos = await db.listCasinos(superuserPool);
@@ -402,9 +439,9 @@ export function initializeTransferProcessors() {
                     logger.warn(`${casino.id} has localhost endpoint "${casino.graphql_url}" while NODE_ENV=production.`);
                 }
                 logger.info(`Starting casino processor for "${casino.name}" at "${casino.graphql_url}"`);
-                startTransferProcessor({ casinoId: casino.id });
+                startTransferProcessor({ casinoId: casino.id, signal });
             }
-            await listenForNewCasinos();
+            await listenForNewCasinos({ signal });
         }
         catch (e) {
             logger.error(`Error initializing transfer processors:`, e);

package/dist/src/server/graphile.config.d.ts

@@ -15,6 +15,7 @@ type PluginIdentity = {
 };
 export type PluginContext = Grafast.Context & {
     identity?: PluginIdentity;
+    abortSignal: AbortSignal;
 };
 export declare const requiredPlugins: readonly GraphileConfig.Plugin[];
 declare global {
@@ -25,9 +26,10 @@ declare global {
     }
 }
 export declare const defaultPlugins: readonly GraphileConfig.Plugin[];
-export declare function createPreset({ plugins, exportSchemaSDLPath, extraPgSchemas, }: {
+export declare function createPreset({ plugins, exportSchemaSDLPath, extraPgSchemas, abortSignal, }: {
     plugins: readonly GraphileConfig.Plugin[];
     exportSchemaSDLPath: string;
     extraPgSchemas: string[];
+    abortSignal: AbortSignal;
 }): GraphileConfig.Preset;
 export {};

package/dist/src/server/graphile.config.js

@@ -38,7 +38,7 @@ export const defaultPlugins = [
     HubClaimFaucetPlugin,
     customPgOmitArchivedPlugin("deleted"),
 ];
-export function createPreset({ plugins, exportSchemaSDLPath, extraPgSchemas, }) {
+export function createPreset({ plugins, exportSchemaSDLPath, extraPgSchemas, abortSignal, }) {
     if (!exportSchemaSDLPath.startsWith("/")) {
         throw new Error("exportSchemaSDLPath must be an absolute path");
     }
@@ -120,6 +120,7 @@ export function createPreset({ plugins, exportSchemaSDLPath, extraPgSchemas, })
                 return {
                     pgSettings,
                     identity: pluginIdentity,
+                    abortSignal,
                 };
             },
         },

package/dist/src/server/index.d.ts

@@ -1,2 +1,8 @@
 import { ServerOptions } from "../index.js";
-export declare function listen({ configureApp, plugins, exportSchemaSDLPath, extraPgSchemas, }: Pick<ServerOptions, "plugins" | "exportSchemaSDLPath" | "extraPgSchemas" | "configureApp">): Promise<void>;
+export type HubServer = {
+    listen: () => Promise<void>;
+    shutdown: () => Promise<void>;
+};
+export declare function createHubServer({ configureApp, plugins, exportSchemaSDLPath, extraPgSchemas, abortSignal, }: Pick<ServerOptions, "plugins" | "exportSchemaSDLPath" | "extraPgSchemas" | "configureApp"> & {
+    abortSignal: AbortSignal;
+}): HubServer;

package/dist/src/server/index.js

@@ -1,4 +1,4 @@
-import { createServer } from "node:http";
+import { createServer as createNodeServer } from "node:http";
 import { grafserv } from "grafserv/express/v4";
 import postgraphile from "postgraphile";
 import { createPreset, defaultPlugins } from "./graphile.config.js";
@@ -43,27 +43,37 @@ function createExpressServer() {
     });
     return app;
 }
-export async function listen({ configureApp, plugins, exportSchemaSDLPath, extraPgSchemas, }) {
+export function createHubServer({ configureApp, plugins, exportSchemaSDLPath, extraPgSchemas, abortSignal, }) {
     const expressServer = createExpressServer();
     const preset = createPreset({
         plugins: plugins ?? defaultPlugins,
         exportSchemaSDLPath,
         extraPgSchemas: extraPgSchemas ?? [],
+        abortSignal,
     });
     const pgl = postgraphile.default(preset);
     const serv = pgl.createServ(grafserv);
     if (configureApp) {
         configureApp(expressServer);
     }
-    const server = createServer(expressServer);
-    server.on("error", (e) => {
+    const nodeServer = createNodeServer(expressServer);
+    nodeServer.on("error", (e) => {
         logger.error(e);
     });
-    serv.addTo(expressServer, server).catch((e) => {
+    serv.addTo(expressServer, nodeServer).catch((e) => {
         logger.error(e);
         process.exit(1);
     });
-    return new Promise((resolve) => {
-        server.listen(config.PORT, resolve);
-    });
+    return {
+        listen: () => {
+            return new Promise((resolve) => {
+                nodeServer.listen(config.PORT, resolve);
+            });
+        },
+        shutdown: () => {
+            return new Promise((resolve) => {
+                nodeServer.close(() => resolve());
+            });
+        },
+    };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@moneypot/hub",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "author": "moneypot.com",
   "homepage": "https://moneypot.com/hub",
   "keywords": [