@axiom-lattice/cli-a2a 0.1.1

package/src/server/index.ts

@@ -0,0 +1,124 @@
+/**
+ * A2A HTTP Server
+ *
+ * Creates an Express server with standard A2A endpoints:
+ *  - GET /.well-known/agent-card.json  → Agent Card
+ *  - POST /a2a/jsonrpc                 → JSON-RPC transport
+ *  - POST /a2a/rest                    → REST transport
+ *  - GET /health                       → Health check
+ */
+
+import express, { type RequestHandler } from 'express';
+import { AGENT_CARD_PATH } from '@a2a-js/sdk';
+import { DefaultRequestHandler, InMemoryTaskStore } from '@a2a-js/sdk/server';
+import {
+  jsonRpcHandler,
+  restHandler,
+  UserBuilder,
+} from '@a2a-js/sdk/server/express';
+
+import type { AgentConfig } from '../config/types.js';
+import { buildAgentCard } from './agent-card.js';
+import { createExecutor } from '../executors/index.js';
+import { logger } from '../logger.js';
+
+const log = logger.child('server');
+
+// ─── Types ──────────────────────────────────────────────────────────────────
+
+export interface ServerHandle {
+  app: ReturnType<typeof express>;
+  server: ReturnType<ReturnType<typeof express>['listen']>;
+  shutdown(): Promise<void>;
+}
+
+// ─── Server Factory ─────────────────────────────────────────────────────────
+
+export async function createA2AServer(config: Required<AgentConfig>): Promise<ServerHandle> {
+  const srv = config.server;
+  const port = srv.port ?? 3000;
+  const hostname = srv.hostname ?? '0.0.0.0';
+  const advertiseHost = srv.advertiseHost ?? 'localhost';
+  const advertiseProto = srv.advertiseProtocol ?? 'http';
+
+  // 1. Create executor
+  const executor = createExecutor(config);
+  await executor.initialize();
+
+  // 2. Build agent card
+  const agentCard = buildAgentCard(config);
+
+  // 3. A2A request handler
+  const taskStore = new InMemoryTaskStore();
+  const requestHandler = new DefaultRequestHandler(agentCard, taskStore, executor);
+
+  // 4. Express app
+  const app = express();
+
+  // A2A-Version header
+  app.use((_req, res, next) => {
+    res.setHeader('A2A-Version', '0.3');
+    next();
+  });
+
+  // Health check
+  app.get('/health', (_req, res) => {
+    res.json({ status: 'healthy', agent: agentCard.name, provider: config.provider });
+  });
+
+  // Agent card — dynamic URL rewriting for reverse proxy support
+  const serveAgentCard: RequestHandler = (req, res) => {
+    const host = req.headers.host || `${advertiseHost}:${port}`;
+    const proto = (req.headers['x-forwarded-proto'] as string) || advertiseProto;
+    const dynamicBase = `${proto}://${host}`;
+    res.json({
+      ...agentCard,
+      url: `${dynamicBase}/a2a/jsonrpc`,
+      additionalInterfaces: [
+        { transport: 'JSONRPC', url: `${dynamicBase}/a2a/jsonrpc` },
+        { transport: 'REST', url: `${dynamicBase}/a2a/rest` },
+      ],
+    });
+  };
+
+  app.get(`/${AGENT_CARD_PATH}`, serveAgentCard);
+  app.get('/.well-known/agent.json', serveAgentCard);
+  app.get('/.well-known/agent-json', serveAgentCard);
+
+  // A2A transport handlers
+  app.use('/a2a/jsonrpc', jsonRpcHandler({ requestHandler, userBuilder: UserBuilder.noAuthentication }));
+  app.use('/a2a/rest', restHandler({ requestHandler, userBuilder: UserBuilder.noAuthentication }));
+
+  // 5. Start listening
+  const httpServer = app.listen(port, hostname, () => {
+    log.info('A2A server started', { bind: hostname, port, proto: advertiseProto });
+
+    const banner = [
+      '',
+      '╔══════════════════════════════════════════════════════════════╗',
+      `║       CLI A2A Gateway — ${config.provider.padEnd(28)}║`,
+      '╠══════════════════════════════════════════════════════════════╣',
+      `║  Agent:         ${agentCard.name}`,
+      `║  Bind:          ${hostname}:${port}`,
+      `║  Agent Card:    ${advertiseProto}://${advertiseHost}:${port}/${AGENT_CARD_PATH}`,
+      `║  JSON-RPC:      ${advertiseProto}://${advertiseHost}:${port}/a2a/jsonrpc`,
+      `║  REST:          ${advertiseProto}://${advertiseHost}:${port}/a2a/rest`,
+      `║  Health:        ${advertiseProto}://${advertiseHost}:${port}/health`,
+      '╠══════════════════════════════════════════════════════════════╣',
+      '║  Ready. Press Ctrl+C to stop.                               ║',
+      '╚══════════════════════════════════════════════════════════════╝',
+    ].join('\n');
+    console.log(banner);
+  });
+
+  // 6. Return handle
+  return {
+    app,
+    server: httpServer,
+    async shutdown() {
+      httpServer.close();
+      await executor.shutdown();
+      log.info('Server shut down');
+    },
+  };
+}

package/tsconfig.json

@@ -0,0 +1,21 @@
+{
+  "compilerOptions": {
+    "target": "ES2020",
+    "module": "preserve",
+    "lib": ["ES2020"],
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "strict": true,
+    "moduleResolution": "Bundler",
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "resolveJsonModule": true,
+    "declaration": true,
+    "declarationMap": true,
+    "types": ["node", "jest"],
+    "sourceMap": true
+  },
+  "include": ["src/**/*.ts"],
+  "exclude": ["node_modules", "dist", "__tests__"]
+}