/** * voko-server 的 composition root:读取 env、创建连接、装配服务,并统一管理资源生命周期。 */ import { fileURLToPath } from 'node:url' import { config as loadDotenv } from 'dotenv' import { createAuth, createUserRepo } from '@drama/auth' import { createBillingRepo, createLedger } from '@drama/billing' import { createDb } from '@drama/db' import { createLLM, createStaticKeyProvider } from '@drama/llm' import { createLogger } from '@drama/observability' import { createAccessRequests, createAgentAccess, createOwnerSettings, createWebhooks, nextAllowedSendTime, nextLocalHour } from '@voko/access' import { createMemoryStore, createRateLimiter } from '@drama/ratelimit' import { createEscrow, createTips, verifyConservation } from '@voko/clearing' import { createAgents, createEvents, createGame, createGuilds, createHelp, createJobs, createJudge, createTasks } from '@voko/exchange' import { readEnv } from '../env.js' import { createEmailPorts } from './email.js' import { createGoogleOauthPort } from './google-oauth.js' import { loadSkillFiles } from './skill-files.js' import { createDevOauthPort, createHmacTokenSigner, createScryptHasher } from './security.js' /** * @param {{ service: string }} opts */ export async function createRuntime({ service }) { // 1. 加载 .env 并解析配置。 loadDotenv({ path: fileURLToPath(new URL('../../.env', import.meta.url)) }) const env = readEnv(process.env) const logger = createLogger({ service, level: env.logLevel, format: env.nodeEnv === 'production' ? 'json' : 'pretty', }) // 2. 连接与基础端口。 const db = createDb({ url: env.databaseUrl }) const { authEmail, accessNotifier, ownerNotifier } = createEmailPorts({ env, db, logger }) const tokenSigner = createHmacTokenSigner({ secret: env.authTokenSecret }) const hasher = createScryptHasher() const oauth = env.oauth.googleClientId === null ? createDevOauthPort() : createGoogleOauthPort({ clientId: env.oauth.googleClientId, clientSecret: env.oauth.googleClientSecret }) // 3. 领域服务。 const userRepo = createUserRepo(db) const auth = createAuth({ userRepo, email: authEmail, oauth, tokenSigner, hasher }) const access = createAccessRequests({ db, notifier: accessNotifier }) const agents = createAgents() const guilds = createGuilds({ db }) const escrow = createEscrow({ db }) // 大模型最小装配:一个静态 key 直连供应商,不接成本、审计与扣费端口(Alpha 只有平台裁判两处调用)。 const llm = createLLM({ provider: env.llm.provider, model: env.llm.model, keyProvider: createStaticKeyProvider({ api_key: env.llm.apiKey, base_url: env.llm.baseUrl }), }) const judge = createJudge({ chat: (request) => llm.chat(request) }) const tasks = createTasks({ db, agents, escrow, judge }) const events = createEvents({ db }) const ownerSettings = createOwnerSettings({ db }) const help = createHelp({ db, mailer: ownerNotifier, // 求助邮件:主人关了就不发;在安静时段内推迟到结束时刻。 async sendAfter(ctx, { user_id, now }) { const settings = await ownerSettings.get(ctx, { user_id }) return settings.email_help ? nextAllowedSendTime(now, settings) : null }, }) const tips = createTips({ db }) const webhooks = createWebhooks({ db, secret_key: env.webhookSecretKey }) const game = createGame({ db }) /** * 给主人发日报:过去 24 小时没有任何动静(无结算、无在手任务、无待回复求助)就不打扰。 * @param {object} ctx * @param {{ user_id: string }} input */ async function sendDigest(ctx, { user_id }) { const digest = await game.digestFor(ctx, { owner_user_id: user_id, since: new Date(Date.now() - 86400_000) }) const quiet = digest.open_help === 0 && digest.agents.every((agent) => agent.settled === 0 && agent.active_tasks === 0) if (!quiet) await ownerNotifier.sendDigest(ctx, { user_id, digest }) } const jobs = createJobs({ db, tasks, help, sendDigest, runDailyGame: (ctx) => game.runDaily(ctx), conservationProblems: () => verifyConservation(db), logger }) /** * 给每个有 Agent 且开着日报的主人排下一次日报(当地 09:00)。重复调用无害。 * @param {object} ctx */ async function scheduleDigests(ctx) { const now = new Date() const owners = await db('vk_agents').whereNotNull('owner_user_id').distinct('owner_user_id') for (const { owner_user_id } of owners) { const settings = await ownerSettings.get(ctx, { user_id: owner_user_id }) if (settings.email_digest) await jobs.enqueue(ctx, { kind: 'daily_digest', ref_id: owner_user_id, run_at: nextLocalHour(now, settings.timezone, 9) }) } } /** * 运营给公会预算调账:记到公司 Agent 的钱包,reason=admin_adjustment,note 必填。 * @param {{ wallet_user_id: string, amount: number, note: string, idempotency_key: string }} input */ async function grantBudget({ wallet_user_id, amount, note, idempotency_key }) { return createLedger(createBillingRepo(db)).grant({ user_id: wallet_user_id, amount, source: 'admin', reason: 'admin_adjustment', note, idempotency_key: `vk:budget:${idempotency_key}` }) } /** * 给已结算任务的赢家打赏;收款事件与动账同事务。 * @param {object} ctx * @param {{ agent: import('@voko/exchange').Agent, task_id: string, amount: number, client_request_key: string }} input */ async function tipTask(ctx, { agent, task_id, amount, client_request_key }) { const seen = await tasks.get(ctx, { agent, task_id }) if (!seen.ok) return seen return tips.send(ctx, { task_id, from_agent_id: agent.id, from_wallet: agent.wallet_user_id, amount, client_request_key, async onSent(trx, { to_wallet }) { const recipient = await agents.getByWallet(ctx, trx, { wallet_user_id: to_wallet }) await events.emit(ctx, trx, { guild_id: seen.task.guild_id, task_id, type: 'tip.received', target_agent_id: recipient.id }) }, }) } // 公开 API 按 Agent 限流(接入说明:读 120/分、写 40/分、发任务 30/小时)。单进程部署用内存 store,多实例时换 MySQL 或 Redis store。 const rateLimiter = createRateLimiter({ store: createMemoryStore(), rules: [ { endpoint: 'read', by: 'anon', capacity: 120, refillPerSec: 120 / 60 }, { endpoint: 'write', by: 'anon', capacity: 40, refillPerSec: 40 / 60 }, { endpoint: 'publish', by: 'anon', capacity: 30, refillPerSec: 30 / 3600 }, ], }) const agentAccess = createAgentAccess({ db, agents, public_url: env.publicUrl, notifier: ownerNotifier, // 启动积分 100:在认领事务里记账,幂等键由 agent_id 唯一决定。 async grantStarter(_ctx, trx, { agent_id, wallet_user_id }) { await createLedger(createBillingRepo(db, trx)).grant({ user_id: wallet_user_id, amount: 100, source: 'voko', credit_kind: 'admin', reason: 'vk_starter', ref_type: 'vk_agent', ref_id: agent_id, idempotency_key: `vk:starter:${agent_id}`, }) }, }) /** * 注册:验证码校验与建用户在同一事务里完成。 * @param {{ email: string, password: string, verification_code: string, locale: string }} input */ async function register(input) { return db.transaction(async (trx) => { const trxAuth = createAuth({ userRepo: createUserRepo(trx), email: authEmail, oauth, tokenSigner, hasher }) await trxAuth.verifyCode({ email: input.email, code: input.verification_code, purpose: 'register' }) return trxAuth.register({ email: input.email, password: input.password, locale: input.locale, email_verified_at: new Date() }) }) } return { env, logger, services: { db, auth, userRepo, access, agents, guilds, agentAccess, escrow, tasks, events, jobs, help, ownerSettings, webhooks, game, skill: loadSkillFiles(), sendDigest, scheduleDigests, grantBudget, tipTask, rateLimiter, register, oauthEnabled: env.oauth.googleClientId !== null }, async close() { await db.destroy() }, } }