openshard-registry 1.0.1

package/README.md

@@ -0,0 +1,18 @@
+# @openshard/registry
+
+> Peer discovery and liveness monitoring service for the OpenShard network.
+
+The **OpenShard Registry** is a lightweight, high-performance Fastify server that tracks active seller nodes, advertises AI offerings, and prunes offline peers automatically.
+
+## API Endpoints
+
+- `POST /peers/register`: Registers a new seller node with its public endpoint, peer ID, merchant address, and available AI offerings.
+- `POST /peers/heartbeat`: Periodic liveness ping sent by active sellers (pruned after 90 seconds of inactivity).
+- `GET /peers`: Lists all live, healthy peers sorted by recency and filtered by supported models.
+- `GET /metrics`: Observability endpoint exporting standard Prometheus metrics.
+
+## Running
+
+```bash
+npm run dev --workspace=packages/registry
+```

package/package.json

@@ -0,0 +1,18 @@
+{
+  "name": "openshard-registry",
+  "version": "1.0.1",
+  "description": "Discovery & registry service — lets buyers find sellers",
+  "main": "src/index.js",
+  "type": "module",
+  "scripts": {
+    "dev": "node --watch src/index.js",
+    "start": "node src/index.js"
+  },
+  "dependencies": {
+    "openshard-shared": "*",
+    "@fastify/cors": "^10.0.1",
+    "fastify": "^5.2.1"
+  },
+  "keywords": [],
+  "license": "ISC"
+}

package/src/index.js

@@ -0,0 +1,172 @@
+import Fastify from 'fastify';
+import cors from '@fastify/cors';
+import { REGISTRY_PORT, isAlive, errorBody, PrometheusRegistry } from 'openshard-shared';
+
+// ─── In-memory peer store ─────────────────────────────────────────────────────
+const peers = new Map();
+const _startTime = Date.now();
+
+// ─── Prometheus metrics ───────────────────────────────────────────────────────
+const reg = new PrometheusRegistry();
+const m = {
+  peersTotal: reg.gauge('openshard_registry_peers_total', 'Total registered peers'),
+  peersAlive: reg.gauge('openshard_registry_peers_alive', 'Live peers (seen within 60s)'),
+  registrations: reg.counter('openshard_registry_registrations_total', 'Total peer registration calls'),
+  heartbeats: reg.counter('openshard_registry_heartbeats_total', 'Total heartbeat calls'),
+  uptimeSeconds: reg.gauge('openshard_registry_uptime_seconds', 'Registry uptime in seconds'),
+  pruned: reg.counter('openshard_registry_pruned_total', 'Total stale peers pruned')
+};
+
+// ─── Prune dead peers every 90 seconds ───────────────────────────────────────
+setInterval(() => {
+  let pruned = 0;
+  for (const [id, peer] of peers) {
+    if (!isAlive(peer)) {
+      console.log(`[registry] pruning stale peer: ${id.slice(0, 12)}…`);
+      peers.delete(id);
+      pruned++;
+    }
+  }
+  if (pruned > 0) m.pruned.inc({}, pruned);
+}, 90_000);
+
+// ─── Fastify instance ─────────────────────────────────────────────────────────
+const app = Fastify({ logger: true });
+app.addHook('onRequest', (req, reply, done) => {
+  reply.header('Access-Control-Allow-Private-Network', 'true');
+  reply.header('Access-Control-Allow-Origin', req.headers.origin || '*');
+  reply.header('Access-Control-Allow-Credentials', 'true');
+  reply.header('Access-Control-Allow-Methods', 'GET,PUT,POST,DELETE,OPTIONS');
+  reply.header('Access-Control-Allow-Headers', 'Content-Type, Authorization, x-anteseed-pin-peer, anthropic-version, x-402-payment-auth, x-402-channel-id, x-openshard-buyer-address');
+  if (req.method === 'OPTIONS') {
+    return reply.status(200).send();
+  }
+  done();
+});
+await app.register(cors, { origin: true, credentials: true });
+
+// ── GET /health ───────────────────────────────────────────────────────────────
+app.get('/health', async () => ({
+  status: 'ok',
+  peers: peers.size,
+  alive: [...peers.values()].filter(isAlive).length
+}));
+
+// ── GET /metrics  (Prometheus) ────────────────────────────────────────────────
+app.get('/metrics', async (req, reply) => {
+  const alive = [...peers.values()].filter(isAlive).length;
+  m.peersTotal.set({}, peers.size);
+  m.peersAlive.set({}, alive);
+  m.uptimeSeconds.set({}, Math.floor((Date.now() - _startTime) / 1000));
+  reply.header('content-type', 'text/plain; version=0.0.4; charset=utf-8');
+  return reply.send(reg.format());
+});
+
+// ── POST /peers/register ──────────────────────────────────────────────────────
+app.post('/peers/register', {
+  schema: {
+    body: {
+      type: 'object',
+      required: ['peerId', 'endpoint', 'offerings'],
+      properties: {
+        peerId: { type: 'string' },
+        endpoint: { type: 'string' },
+        offerings: { type: 'array' },
+        merchantAddress: { type: 'string' }
+      }
+    }
+  }
+}, async (req, reply) => {
+  const { peerId, endpoint, offerings, merchantAddress } = req.body;
+
+  if (!peerId || !endpoint || !Array.isArray(offerings)) {
+    return reply.status(400).send(errorBody('bad_request', 'peerId, endpoint, and offerings are required'));
+  }
+
+  const now = Date.now();
+  const existing = peers.get(peerId);
+
+  peers.set(peerId, {
+    peerId,
+    endpoint,
+    offerings,
+    merchantAddress: merchantAddress ?? '0xYOUR_MERCHANT_ADDRESS_HERE',
+    registeredAt: existing?.registeredAt ?? now,
+    lastSeen: now
+  });
+
+  m.registrations.inc();
+  app.log.info(`[registry] registered: ${peerId.slice(0, 12)}… (${offerings.length} offerings)`);
+  return { ok: true, peerId };
+});
+
+// ── POST /peers/heartbeat ─────────────────────────────────────────────────────
+app.post('/peers/heartbeat', {
+  schema: {
+    body: {
+      type: 'object',
+      required: ['peerId'],
+      properties: { peerId: { type: 'string' } }
+    }
+  }
+}, async (req, reply) => {
+  const { peerId } = req.body;
+  const peer = peers.get(peerId);
+
+  if (!peer) {
+    return reply.status(404).send(errorBody('not_found', 'Peer not registered. Call /peers/register first.'));
+  }
+
+  peer.lastSeen = Date.now();
+  m.heartbeats.inc();
+  return { ok: true };
+});
+
+// ── GET /peers ────────────────────────────────────────────────────────────────
+app.get('/peers', async (req) => {
+  const { model } = req.query;
+  let results = [...peers.values()].filter(isAlive);
+
+  if (model) {
+    results = results.filter(p =>
+      p.offerings?.some(o => o.services?.includes(model))
+    );
+  }
+
+  // Sort by most recently seen
+  results.sort((a, b) => b.lastSeen - a.lastSeen);
+  return { peers: results };
+});
+
+// ── GET /peers/:peerId ────────────────────────────────────────────────────────
+app.get('/peers/:peerId', async (req, reply) => {
+  const peer = peers.get(req.params.peerId);
+  if (!peer || !isAlive(peer)) {
+    return reply.status(404).send(errorBody('not_found', 'Peer not found or offline'));
+  }
+  return peer;
+});
+
+// ── DELETE /peers/:peerId ─────────────────────────────────────────────────────
+app.delete('/peers/:peerId', async (req, reply) => {
+  if (!peers.has(req.params.peerId)) {
+    return reply.status(404).send(errorBody('not_found', 'Peer not found'));
+  }
+  peers.delete(req.params.peerId);
+  return { ok: true };
+});
+
+// ─── Start ────────────────────────────────────────────────────────────────────
+try {
+  await app.listen({ port: REGISTRY_PORT, host: '0.0.0.0' });
+  console.log(`
+🐜 OpenShard Registry  http://0.0.0.0:${REGISTRY_PORT}
+
+  GET  /peers           list live peers
+  GET  /metrics         Prometheus metrics
+  GET  /health          liveness
+`);
+} catch (err) {
+  app.log.error(err);
+  process.exit(1);
+}