fix(open-meteo-mcp-server): simplify billing to avoid the errors (#67)

Co-authored-by: Jiří Spilka <[email protected]>

Author: ivanvia7

open-meteo-mcp-server/src/billing.ts

@@ -1,28 +1,81 @@
-
 import { Actor, log } from 'apify';
-import { readFile } from 'node:fs/promises';
-import path from 'node:path';
 
-// Load billable events from .actor/pay_per_event.json (sync at startup)
-let BILLABLE_EVENTS: Record<string, any> = {};
-const payPerEventPath = path.resolve(__dirname, '../.actor/pay_per_event.json');
-async function loadBillableEvents() {
-    try {
-        const data = await readFile(payPerEventPath, 'utf-8');
-        BILLABLE_EVENTS = JSON.parse(data);
-        log.info('Loaded billable events from pay_per_event.json');
-    } catch (err) {
-        log.error('Failed to load billable events:', { error: err });
+const TOOL_PRICING: Record<string, { eventPriceUsd: number; eventTitle: string }> = {
+    "actor-start": {
+        eventPriceUsd: 0.1,
+        eventTitle: "Price for Actor start",
+    },
+    "weather_forecast": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "Weather forecast tool",
+    },
+    "weather_archive": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "Weather archive tool",
+    },
+    "air_quality": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "Air quality tool",
+    },
+    "marine_weather": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "Marine weather tool",
+    },
+    "elevation": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "Elevation tool",
+    },
+    "geocoding": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "Geocoding tool",
+    },
+    "dwd_icon_forecast": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "DWD ICON forecast tool",
+    },
+    "gfs_forecast": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "NOAA GFS forecast tool",
+    },
+    "meteofrance_forecast": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "Météo-France forecast tool",
+    },
+    "ecmwf_forecast": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "ECMWF forecast tool",
+    },
+    "jma_forecast": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "JMA forecast tool",
+    },
+    "metno_forecast": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "MET Norway forecast tool",
+    },
+    "gem_forecast": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "Environment Canada GEM forecast tool",
+    },
+    "flood_forecast": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "Flood forecast tool",
+    },
+    "seasonal_forecast": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "Seasonal forecast tool",
+    },
+    "climate_projection": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "Climate projection tool",
+    },
+    "ensemble_forecast": {
+        eventPriceUsd: 0.0001,
+        eventTitle: "Ensemble forecast tool",
     }
-}
-// Immediately load on module import
-await loadBillableEvents();
+};
 /**
- * This module handles billing for different types of protocol requests in the MCP server.
- * It defines a function to charge users based on the type of protocol method invoked.
- */
 
-/**
  * Charges the user for a message request based on the method type.
  * Supported method types are mapped to specific billing events.
  *
@@ -34,17 +87,17 @@ export async function chargeMessageRequest(request: { method: string; params?: {
 
     if (method === 'tools/call' && request?.params?.tool) {
         const toolName = request.params.tool;
-        if (BILLABLE_EVENTS[toolName]) {
+        if (TOOL_PRICING[toolName]) {
             await Actor.charge({ eventName: toolName });
-            const event = BILLABLE_EVENTS[toolName];
+            const event = TOOL_PRICING[toolName];
             log.info(`Charged for tool call: ${toolName} ($${event.eventPriceUsd}) - ${event.eventTitle}`);
         } else {
             await Actor.charge({ eventName: 'tool-call' });
             log.info(`Charged for tool call (unknown tool): ${toolName}`);
         }
-    } else if (BILLABLE_EVENTS[method]) {
+    } else if (TOOL_PRICING[method]) {
         await Actor.charge({ eventName: method });
-        const event = BILLABLE_EVENTS[method];
+        const event = TOOL_PRICING[method];
         log.info(`Charged for event: ${method} ($${event.eventPriceUsd}) - ${event.eventTitle}`);
     } else {
         log.info(`Not charging for method: ${method}`);

open-meteo-mcp-server/src/server.ts

@@ -95,7 +95,9 @@ async function mcpPostHandler(req: Request, res: Response) {
             await server.connect(transport);
 
             // Charge for the request
-            await chargeMessageRequest(req.body);
+            const { method, params } = req.body;
+            const chargeParams = params && typeof params.tool === 'string' ? { tool: params.tool } : undefined;
+            await chargeMessageRequest({ method, params: chargeParams });
 
             await transport.handleRequest(req, res, req.body);
             return; // Already handled
@@ -113,7 +115,9 @@ async function mcpPostHandler(req: Request, res: Response) {
         }
 
         // Charge for the request
-        await chargeMessageRequest(req.body);
+        const { method, params } = req.body;
+        const chargeParams = params && typeof params.tool === 'string' ? { tool: params.tool } : undefined;
+        await chargeMessageRequest({ method, params: chargeParams });
         // Handle the request with existing transport - no need to reconnect
         // The existing transport is already connected to the server
         await transport.handleRequest(req, res, req.body);