@electric-ax/agents-server 0.3.0

package/package.json

@@ -0,0 +1,89 @@
+{
+  "name": "@electric-ax/agents-server",
+  "version": "0.3.0",
+  "description": "Electric Agents entity runtime server",
+  "author": "Durable Stream contributors",
+  "bin": {
+    "electric-agents-server": "./dist/entrypoint.js",
+    "durable-streams-electric-agents-server": "./dist/entrypoint.js"
+  },
+  "license": "Apache-2.0",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/electric-sql/electric.git",
+    "directory": "packages/agents-server"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "type": "module",
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      },
+      "require": {
+        "types": "./dist/index.d.cts",
+        "default": "./dist/index.cjs"
+      }
+    },
+    "./package.json": "./package.json"
+  },
+  "sideEffects": false,
+  "dependencies": {
+    "@anthropic-ai/sdk": "^0.78.0",
+    "@durable-streams/client": "https://pkg.pr.new/durable-streams/durable-streams/@durable-streams/client@350",
+    "@durable-streams/server": "https://pkg.pr.new/durable-streams/durable-streams/@durable-streams/server@350",
+    "@durable-streams/state": "https://pkg.pr.new/durable-streams/durable-streams/@durable-streams/state@350",
+    "@electric-sql/client": "^1.5.17",
+    "@mariozechner/pi-agent-core": "^0.70.2",
+    "@opentelemetry/api": "^1.9.1",
+    "@sinclair/typebox": "^0.34.48",
+    "@whatwg-node/server": "^0.10.18",
+    "ajv": "^8.18.0",
+    "cron-parser": "^5.5.0",
+    "drizzle-orm": "^0.44.0",
+    "fastq": "^1.20.1",
+    "itty-router": "^5.0.23",
+    "lmdb": "^3.5.1",
+    "pino": "^10.3.1",
+    "pino-pretty": "^13.0.0",
+    "postgres": "^3.4.0",
+    "undici": "^7.24.7",
+    "@electric-ax/agents-runtime": "0.1.3"
+  },
+  "devDependencies": {
+    "@types/node": "^22.19.15",
+    "@vitest/coverage-v8": "^4.1.0",
+    "cross-env": "^10.1.0",
+    "drizzle-kit": "^0.31.0",
+    "tsdown": "^0.9.0",
+    "tsx": "^4.19.0",
+    "typescript": "^5.0.0",
+    "vitest": "^4.1.0",
+    "@electric-ax/agents-server-conformance-tests": "0.1.2",
+    "@electric-ax/agents": "0.3.0",
+    "@electric-ax/agents-server-ui": "0.3.0"
+  },
+  "files": [
+    "dist",
+    "drizzle",
+    "src"
+  ],
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "scripts": {
+    "build": "tsdown",
+    "dev": "tsdown --watch",
+    "start": "cross-env-shell \"DATABASE_URL=${DATABASE_URL:-postgresql://electric_agents:electric_agents@localhost:5432/electric_agents} ELECTRIC_URL=${ELECTRIC_URL:-http://localhost:3060} tsx --watch src/entrypoint.ts\"",
+    "test": "vitest run",
+    "coverage": "ELECTRIC_AGENTS_KEEP_BACKEND=1 pnpm exec vitest run --coverage $(find test -name '*.test.ts' ! -name 'conformance.test.ts' | sort) && pnpm exec vitest run test/conformance.test.ts",
+    "typecheck": "tsc --noEmit",
+    "stylecheck": "eslint . --quiet"
+  }
+}

package/src/authenticated-user-format.ts

@@ -0,0 +1,17 @@
+import type { AuthenticatedRequestUser } from './electric-agents-types.js'
+
+function clean(value: string | undefined): string | undefined {
+  const trimmed = value?.trim()
+  return trimmed || undefined
+}
+
+export function formatAuthenticatedUser(
+  user: AuthenticatedRequestUser | null | undefined
+): string | undefined {
+  if (!user) return undefined
+  const email = clean(user.email)
+  const name = clean(user.name)
+  const userId = clean(user.userId)
+  if (name && email) return `${name} <${email}>`
+  return email ?? name ?? userId
+}

package/src/claim-write-token-store.ts

@@ -0,0 +1,74 @@
+import { randomUUID } from 'node:crypto'
+
+interface ActiveClaimWriteToken {
+  token: string
+  consumerId: string
+}
+
+export class ClaimWriteTokenStore {
+  private readonly claimsByStream = new Map<string, ActiveClaimWriteToken>()
+  private readonly streamByConsumer = new Map<string, string>()
+
+  mint(service: string, streamPath: string, consumerId: string): string {
+    const streamKey = this.streamKey(service, streamPath)
+    const consumerKey = this.consumerKey(service, consumerId)
+    const previousClaimForStream = this.claimsByStream.get(streamKey)
+    if (previousClaimForStream) {
+      this.streamByConsumer.delete(
+        this.consumerKey(service, previousClaimForStream.consumerId)
+      )
+    }
+
+    const previousStreamForConsumer = this.streamByConsumer.get(consumerKey)
+    if (previousStreamForConsumer) {
+      this.claimsByStream.delete(previousStreamForConsumer)
+    }
+
+    const token = randomUUID()
+    this.claimsByStream.set(streamKey, { token, consumerId })
+    this.streamByConsumer.set(consumerKey, streamKey)
+    return token
+  }
+
+  isValid(service: string, streamPath: string, token: string): boolean {
+    return (
+      this.claimsByStream.get(this.streamKey(service, streamPath))?.token ===
+      token
+    )
+  }
+
+  owns(service: string, streamPath: string, consumerId: string): boolean {
+    return (
+      this.claimsByStream.get(this.streamKey(service, streamPath))
+        ?.consumerId === consumerId
+    )
+  }
+
+  clearStream(service: string, streamPath: string): void {
+    const streamKey = this.streamKey(service, streamPath)
+    const activeClaim = this.claimsByStream.get(streamKey)
+    if (!activeClaim) return
+
+    this.claimsByStream.delete(streamKey)
+    this.streamByConsumer.delete(
+      this.consumerKey(service, activeClaim.consumerId)
+    )
+  }
+
+  clearConsumer(service: string, consumerId: string): void {
+    const consumerKey = this.consumerKey(service, consumerId)
+    const streamKey = this.streamByConsumer.get(consumerKey)
+    if (!streamKey) return
+
+    this.streamByConsumer.delete(consumerKey)
+    this.claimsByStream.delete(streamKey)
+  }
+
+  private streamKey(service: string, streamPath: string): string {
+    return `${service}\0${streamPath}`
+  }
+
+  private consumerKey(service: string, consumerId: string): string {
+    return `${service}\0${consumerId}`
+  }
+}

package/src/db/index.ts

@@ -0,0 +1,53 @@
+import { existsSync } from 'node:fs'
+import { dirname, resolve } from 'node:path'
+import { fileURLToPath } from 'node:url'
+import { drizzle } from 'drizzle-orm/postgres-js'
+import { migrate } from 'drizzle-orm/postgres-js/migrator'
+import postgres from 'postgres'
+import * as schema from './schema.js'
+
+export type DrizzleDB = ReturnType<typeof drizzle<typeof schema>>
+export type PgClient = ReturnType<typeof postgres>
+
+export function createDb(postgresUrl: string): {
+  db: DrizzleDB
+  client: PgClient
+} {
+  const poolMax = Number(process.env.ELECTRIC_AGENTS_PG_POOL_MAX ?? `100`)
+  const client = postgres(postgresUrl, {
+    max: poolMax,
+    fetch_types: false,
+  })
+  const db = drizzle(client, { schema })
+  return { db, client }
+}
+
+export function resolveMigrationsFolder(fromUrl = import.meta.url): string {
+  const here = dirname(fileURLToPath(fromUrl))
+  const candidates = [
+    resolve(here, `../../drizzle`),
+    resolve(here, `../drizzle`),
+    resolve(process.cwd(), `packages/agents-server/drizzle`),
+  ]
+
+  const folder = candidates.find((candidate) => existsSync(candidate))
+  if (!folder) {
+    throw new Error(
+      `Could not locate agent-server migrations directory from ${fromUrl}`
+    )
+  }
+
+  return folder
+}
+
+export async function runMigrations(postgresUrl: string): Promise<void> {
+  const migrationClient = postgres(postgresUrl, {
+    max: 1,
+    onnotice: () => {},
+  })
+  const db = drizzle(migrationClient)
+  await migrate(db, {
+    migrationsFolder: resolveMigrationsFolder(),
+  })
+  await migrationClient.end()
+}