@0xopenseeddev/registry 1.0.0

package/package.json

@@ -0,0 +1,18 @@
+{
+  "name": "@0xopenseeddev/registry",
+  "version": "1.0.0",
+  "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": {
+    "@0xopenseeddev/shared": "*",
+    "@fastify/cors": "^10.0.1",
+    "fastify": "^5.2.1"
+  },
+  "keywords": [],
+  "license": "ISC"
+}

package/src/index.js

@@ -0,0 +1,161 @@
+import Fastify from 'fastify';
+import cors from '@fastify/cors';
+import { REGISTRY_PORT, isAlive, errorBody, PrometheusRegistry } from '@0xopenseeddev/shared';
+
+// ─── In-memory peer store ─────────────────────────────────────────────────────
+const peers = new Map();
+const _startTime = Date.now();
+
+// ─── Prometheus metrics ───────────────────────────────────────────────────────
+const reg = new PrometheusRegistry();
+const m = {
+  peersTotal: reg.gauge('openseed_registry_peers_total', 'Total registered peers'),
+  peersAlive: reg.gauge('openseed_registry_peers_alive', 'Live peers (seen within 60s)'),
+  registrations: reg.counter('openseed_registry_registrations_total', 'Total peer registration calls'),
+  heartbeats: reg.counter('openseed_registry_heartbeats_total', 'Total heartbeat calls'),
+  uptimeSeconds: reg.gauge('openseed_registry_uptime_seconds', 'Registry uptime in seconds'),
+  pruned: reg.counter('openseed_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 });
+await app.register(cors, { origin: '*' });
+
+// ── 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 ?? '0x742d35Cc6634C0532925a3b844Bc454e4438f44e',
+    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(`
+🐜 OpenSeed 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);
+}