atracker-mcp-server 1.0.0 → 1.1.0

package/README.md

@@ -105,6 +105,14 @@ npm install -g atracker-mcp-server
 - `get_settings` — Get tracker settings
 - `update_settings` — Update tracker settings
 
+
+### Geo2IP Bundles
+
+- `geo_status` — Current geo bundle versions, sizes, and last download status
+- `geo_check_update` — Check the platform for newer bundle versions (returns delta/full info)
+- `geo_update` — Download and apply latest bundles (`categories?`, `force?`); polls until done
+- `geo_lookup` — Test IP lookup with resolved geo/asn/threat data and per-field sources
+
 ## Environment Variables
 
 | Variable | Required | Description |

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "atracker-mcp-server",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "description": "MCP Server for ATracker self-hosted ad tracker",
   "type": "module",
   "bin": {

package/src/index.js

@@ -11,6 +11,7 @@ import { registerDomainTools } from './tools/domains.js';
 import { registerReportTools } from './tools/reports.js';
 import { registerStatusTools } from './tools/status.js';
 import { registerSettingsTools } from './tools/settings.js';
+import { registerGeoTools } from './tools/geo.js';
 
 const url = process.env.ATRACKER_URL;
 const token = process.env.ATRACKER_TOKEN;
@@ -26,7 +27,7 @@ const client = createClient(url, token);
 
 const server = new McpServer({
   name: 'atracker',
-  version: '1.0.0',
+  version: '1.1.0',
 });
 
 registerCampaignTools(server, client);
@@ -38,6 +39,7 @@ registerDomainTools(server, client);
 registerReportTools(server, client);
 registerStatusTools(server, client);
 registerSettingsTools(server, client);
+registerGeoTools(server, client);
 
 const transport = new StdioServerTransport();
 await server.connect(transport);

package/src/tools/geo.js

@@ -0,0 +1,78 @@
+const CATEGORIES = ['geo', 'asn', 'threat'];
+const POLL_INTERVAL_MS = 1000;
+const POLL_TIMEOUT_MS = 30 * 60 * 1000;
+
+async function waitForDownload(client, categories) {
+  const targets = categories && categories.length ? categories : CATEGORIES;
+  const startedAt = Date.now();
+  // eslint-disable-next-line no-constant-condition
+  while (true) {
+    if (Date.now() - startedAt > POLL_TIMEOUT_MS) {
+      throw new Error('Timed out waiting for geo bundle download');
+    }
+    const state = await client.get('/geo2ip/download-status');
+    const summaries = targets.map((c) => ({ category: c, ...(state[c] || {}) }));
+    const errored = summaries.find((s) => s.status === 'error');
+    if (errored) throw new Error(`Download failed for ${errored.category}: ${errored.error || 'unknown'}`);
+    const active = summaries.find((s) => !['done', 'idle'].includes(s.status));
+    if (!active) return summaries;
+    await new Promise((r) => setTimeout(r, POLL_INTERVAL_MS));
+  }
+}
+
+export function registerGeoTools(server, client) {
+  server.tool(
+    'geo_status',
+    'Get current geo2ip bundle versions, sizes, and last download status for all three categories (geo, asn, threat).',
+    {},
+    async () => {
+      const data = await client.get('/geo2ip/bundle-status');
+      return { content: [{ type: 'text', text: JSON.stringify(data, null, 2) }] };
+    }
+  );
+
+  server.tool(
+    'geo_check_update',
+    'Check the ATracker platform for newer geo bundle versions. Returns current versions, available manifests (including delta info), and the list of categories that have updates.',
+    {},
+    async () => {
+      const data = await client.post('/geo2ip/check-updates', {});
+      return { content: [{ type: 'text', text: JSON.stringify(data, null, 2) }] };
+    }
+  );
+
+  server.tool(
+    'geo_update',
+    'Download and apply the latest geo bundles from the platform. Optionally limit to specific categories (geo|asn|threat). Polls until completion and returns the final bundle status.',
+    {
+      categories: { type: 'array', items: { type: 'string', enum: CATEGORIES }, description: 'Optional subset of categories to update' },
+      force: { type: 'boolean', description: 'Re-download even if bundles are up to date' },
+    },
+    async (args = {}) => {
+      const body = {};
+      if (Array.isArray(args.categories) && args.categories.length) body.categories = args.categories;
+      if (args.force) body.force = true;
+      await client.post('/geo2ip/download-bundle', body);
+      const summaries = await waitForDownload(client, body.categories);
+      const finalStatus = await client.get('/geo2ip/bundle-status');
+      return {
+        content: [{
+          type: 'text',
+          text: JSON.stringify({ summaries, status: finalStatus }, null, 2),
+        }],
+      };
+    }
+  );
+
+  server.tool(
+    'geo_lookup',
+    'Look up geo, ASN, and threat info for an IPv4/IPv6 address using the locally loaded bundles. Returns resolved data with per-field source attribution.',
+    {
+      ip: { type: 'string', description: 'IPv4 or IPv6 address to look up' },
+    },
+    async ({ ip }) => {
+      const data = await client.post('/geo2ip/test-lookup', { ip });
+      return { content: [{ type: 'text', text: JSON.stringify(data, null, 2) }] };
+    }
+  );
+}