| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609 |
- import { chatTranscriptRouter } from './chatTranscriptApi.js'
- import {DesktopEnvironments} from './desktopEnvironments.js'
- import {DesktopHostHub} from './desktopHosts.js'
- import type {GrokAuth} from './grokAuth.js'
- import {GrokCloud} from './grokCloud.js'
- import {WorkspaceInspector} from './workspaceInspector.js'
- import {WorkspaceRoutines} from './workspaceRoutines.js'
- import { WorkspacePlugins } from './botPlugins/workspacePlugins.js'
- import { nativeMessageFileText } from '../shared/messageFiles.js'
- import { hermesBotRelay } from './hermesBotRelay.js'
- import { remoteAgentExports } from './workspaceRemoteAgents.js'
- import { randomBytes, createHash } from 'node:crypto'
- import { mkdirSync, readFileSync, writeFileSync } from 'node:fs'
- import { createServer as createHttpServer, type Server as HttpServer } from 'node:http'
- import { createServer as createHttpsServer } from 'node:https'
- import { join } from 'node:path'
- import { bodyParser } from '@koa/bodyparser'
- import Koa from 'koa'
- import { RealtimeAPI } from './realtimeApi.js'
- import type { ServerConfig } from './config.js'
- import { loadServerConfig } from './config.js'
- import { errorMessage, HttpError } from './errors.js'
- import { NodePairingStore } from './pairing.js'
- import { LocalAuthStore, UpstreamServiceSession } from './localAuth.js'
- import { AccountLoginPairingStore } from './accountPairing.js'
- import { UpstreamProfileIdentityService } from './profileIdentities.js'
- import { createApiRouter } from './routes.js'
- import {
- applySecurityHeaders,
- CsrfProtection,
- isAllowedHostHeader,
- isExactOrigin,
- } from './security.js'
- import { UpstreamHttpError, UpstreamClient } from './upstream.js'
- import { UploadStore } from './uploads.js'
- import { WorkspaceStore } from './workspaceStore.js'
- import { WorkspaceNodes } from './workspaceGateway.js'
- import { WorkspaceRuntime } from './workspaceRuntime.js'
- import { WorkspaceMemorySynthesis } from './workspaceMemorySynthesis.js'
- import { WorkspaceAssets } from './workspaceAssets.js'
- import { workspaceRouter } from './workspaceRoutes.js'
- import { SystemUpdateManager } from './updateManager.js'
- import { PushCoordinator } from './pushCoordinator.js'
- import {
- APNsConfigurationManager,
- type APNsConfigurationSnapshot,
- } from './apnsConfiguration.js'
- import {
- FCMConfigurationManager,
- type FCMConfigurationSnapshot,
- } from './fcmConfiguration.js'
- import { OpenVikingConfigurationManager } from './openVikingConfiguration.js'
- import { OpenVikingService } from './openVikingService.js'
- import { PushCoordinatorEventAdapter } from './pushEventAdapter.js'
- import {
- AllowedHostsConfigurationManager,
- type AllowedHostsConfigurationSnapshot,
- } from './allowedHostsConfiguration.js'
- import {
- ChatPushJobManager,
- HermesChatNotificationResolver,
- } from './pushEvents.js'
- import { ChatCacheCoordinator, ChatCacheStore } from './chatCache.js'
- import {LocalVmService} from './localVm.js'
- import {SharedComputers} from './sharedComputers.js'
- import {ComputerControlService} from './computerControls.js'
- import { RunnerHub } from './runnerHub.js'
- import { HermesBridgeManager } from './hermesBridge.js'
- export interface ApplicationOptions {
- hermesBridge?: HermesBridgeManager
- config?: ServerConfig
- fetchImpl?: typeof fetch
- grokFetch?:typeof fetch
- csrfSecret?: Buffer
- pairings?: NodePairingStore
- uploads?: UploadStore
- updates?: SystemUpdateManager
- auth?: LocalAuthStore
- pluginFetch?: typeof fetch
- upstreamSession?: UpstreamServiceSession
- accountPairings?: AccountLoginPairingStore
- profileIdentities?: UpstreamProfileIdentityService
- push?: PushCoordinator
- apnsConfiguration?: APNsConfigurationManager
- fcmConfiguration?: FCMConfigurationManager
- openVikingConfiguration?: OpenVikingConfigurationManager
- allowedHostsConfiguration?: AllowedHostsConfigurationManager
- chatCache?: ChatCacheCoordinator
- }
- export interface ApplicationRuntime {
- hermesBridge: HermesBridgeManager
- runners: RunnerHub
- workspace: WorkspaceStore
- workspaceRuntime: WorkspaceRuntime
- realtime: RealtimeAPI
- app: Koa
- config: ServerConfig
- csrf: CsrfProtection
- pairings: NodePairingStore
- upstream: UpstreamClient
- uploads: UploadStore
- updates: SystemUpdateManager
- auth: LocalAuthStore
- upstreamSession: UpstreamServiceSession
- accountPairings: AccountLoginPairingStore
- profileIdentities: UpstreamProfileIdentityService
- push: PushCoordinator
- apnsConfiguration: APNsConfigurationManager
- fcmConfiguration: FCMConfigurationManager
- openVikingConfiguration: OpenVikingConfigurationManager
- openVikingService: OpenVikingService
- allowedHostsConfiguration: AllowedHostsConfigurationManager
- pushEventCoordinator: PushCoordinatorEventAdapter
- chatPushJobs: ChatPushJobManager
- chatCache?: ChatCacheCoordinator
- desktopEnvironments:DesktopEnvironments
- desktopHosts:DesktopHostHub
- grokAuth:GrokAuth
- workspaceRoutines:WorkspaceRoutines
- workspacePlugins:WorkspacePlugins
- localVm: LocalVmService
- close(): void
- }
- function isMutation(method: string): boolean {
- return !['GET', 'HEAD', 'OPTIONS'].includes(method)
- }
- function persistentCsrfSecret(home: string, override?: Buffer): Buffer {
- if (override) return override
- const path = join(home, 'csrf-secret.bin')
- try {
- const secret = readFileSync(path)
- if (secret.byteLength === 32) return secret
- throw new Error('CSRF secret has an invalid length')
- } catch (error) {
- if ((error as NodeJS.ErrnoException).code !== 'ENOENT') throw error
- }
- mkdirSync(home, { recursive: true, mode: 0o700 })
- const generated = randomBytes(32)
- try {
- writeFileSync(path, generated, { mode: 0o600, flag: 'wx' })
- return generated
- } catch (error) {
- // Another 15300 process can finish first during launchd hand-off. Never
- // rotate its secret; use the one it safely wrote instead.
- if ((error as NodeJS.ErrnoException).code === 'EEXIST') return readFileSync(path)
- throw error
- }
- }
- export function createApplication(options: ApplicationOptions = {}): ApplicationRuntime {
- const config = options.config ?? loadServerConfig()
- const runtimeAllowedHosts = new Set(config.allowedHosts)
- config.allowedHosts = runtimeAllowedHosts
- const app = new Koa()
- app.proxy = false
- const csrf = new CsrfProtection(persistentCsrfSecret(config.home, options.csrfSecret), Boolean(config.tlsCert))
- const pairings = options.pairings ?? new NodePairingStore(config.home)
- const upstream = new UpstreamClient(config.upstream, options.fetchImpl, Boolean(config.tlsCert))
- const uploads = options.uploads ?? new UploadStore(config.home)
- const updates = options.updates ?? new SystemUpdateManager(config)
- const auth = options.auth ?? new LocalAuthStore(config.home, Boolean(config.tlsCert))
- const configuredUpstreamCredentials = config.upstreamUsername && config.upstreamPassword
- ? { username: config.upstreamUsername, password: config.upstreamPassword }
- : undefined
- const upstreamSession = options.upstreamSession ?? new UpstreamServiceSession(
- upstream,
- () => auth.upstreamCredentials(configuredUpstreamCredentials),
- )
- let chatCache = options.chatCache
- if (!chatCache) {
- chatCache = new ChatCacheCoordinator(
- new ChatCacheStore(
- config.home,
- auth.soleUserIDForLegacyMigration(),
- ),
- upstreamSession,
- config.chatCacheMode ?? 'prefer-local',
- )
- }
- const accountPairings = options.accountPairings ?? new AccountLoginPairingStore()
- const profileIdentities = options.profileIdentities
- ?? new UpstreamProfileIdentityService(config, upstreamSession)
- const initialAPNsSettings: APNsConfigurationSnapshot = config.apnsSettings ?? {
- source: config.apns || config.apnsConfigurationError ? 'environment' : 'none',
- editable: !(config.apns || config.apnsConfigurationError),
- ...(config.apns ? {
- input: {
- ...config.apns,
- environments: [...(config.apns.environments ?? ['development', 'production'])],
- },
- config: config.apns,
- } : {}),
- warnings: [],
- ...(config.apnsConfigurationError ? { configurationError: config.apnsConfigurationError } : {}),
- }
- const apnsConfiguration = options.apnsConfiguration
- ?? new APNsConfigurationManager(config.home, initialAPNsSettings)
- const initialFCMSettings: FCMConfigurationSnapshot = config.fcmSettings ?? {
- source: config.fcm || config.fcmConfigurationError ? 'environment' : 'none',
- editable: !(config.fcm || config.fcmConfigurationError),
- ...(config.fcm ? { input: { ...config.fcm }, config: config.fcm } : {}),
- warnings: [],
- ...(config.fcmConfigurationError ? { configurationError: config.fcmConfigurationError } : {}),
- }
- const fcmConfiguration = options.fcmConfiguration
- ?? new FCMConfigurationManager(config.home, initialFCMSettings)
- const openVikingConfiguration = options.openVikingConfiguration
- ?? new OpenVikingConfigurationManager(config.home)
- const initialAllowedHostsSettings: AllowedHostsConfigurationSnapshot = config.allowedHostsSettings ?? {
- source: runtimeAllowedHosts.size ? 'environment' : 'none',
- hosts: [...runtimeAllowedHosts],
- editableHosts: [],
- environmentHosts: [...runtimeAllowedHosts],
- }
- const allowedHostsConfiguration = options.allowedHostsConfiguration
- ?? new AllowedHostsConfigurationManager(config.home, initialAllowedHostsSettings, runtimeAllowedHosts)
- const push = options.push ?? new PushCoordinator({
- home: config.home,
- apns: config.apns,
- apnsConfigurationError: config.apnsConfigurationError,
- fcm: config.fcm,
- fcmConfigurationError: config.fcmConfigurationError,
- isUserActive: userID => auth.isUserActive(userID),
- userAuthorizationVersion: userID => auth.pushAuthorizationVersion(userID),
- })
- const pushEventCoordinator = new PushCoordinatorEventAdapter(push)
- const chatPushJobs = new ChatPushJobManager(config, upstreamSession, pushEventCoordinator, {
- resolver: new HermesChatNotificationResolver(
- upstreamSession,
- (localUserID, prompt) => push.promptDigest(localUserID, prompt),
- ),
- })
- // The application owns group events. No Dashboard plugin is queried.
- const workspace = new WorkspaceStore(config.home)
- const openVikingService = new OpenVikingService(workspace, openVikingConfiguration)
- const workspaceNodes = new WorkspaceNodes(workspace, config, { url: config.upstream, client: upstream, session: upstreamSession }, pairings.nodeID)
- workspaceNodes.sourceAllowed = (owner, nodeId, profile) => auth.canUseSource(owner, nodeId, profile)
- const runners=new RunnerHub(workspace,auth,workspaceNodes.local)
- runners.composeDesktops.desktops.push(...(config.composeDesktops??[]))
- const sharedComputers=new SharedComputers(workspace,auth,workspaceNodes,runners)
- const computerControls=new ComputerControlService(workspace,auth,workspaceNodes,runners)
- const localVm=new LocalVmService(workspace,auth,workspaceNodes,runners,sharedComputers,config)
- runners.localVmAllowed=(id,runnerId)=>localVm.allowed(id,runnerId)
- runners.controlAllowed=(id,runnerId)=>computerControls.allowed(id,runnerId)
- workspaceNodes.runnerTarget=(owner,nodeId,computer)=>runners.target(owner,nodeId,computer)
- const workspaceRuntime = new WorkspaceRuntime(workspace, workspaceNodes, uploads, owner => auth.isUserActive(owner), owner => auth.pushAuthorizationVersion(owner) ?? 0, openVikingService)
- const hermesBridge=options.hermesBridge??new HermesBridgeManager(config,upstreamSession,{isIdle:()=>workspaceRuntime.idleForUpdate&&runners.idleForUpdate&&realtime.broker.idleForUpdate})
- const workspaceMemory = new WorkspaceMemorySynthesis(workspaceRuntime)
- workspaceMemory.start()
- const grokCloud=new GrokCloud(workspace,auth,workspaceNodes,sharedComputers,options.grokFetch)
- const desktopEnvironments=new DesktopEnvironments(workspace,auth,workspaceNodes)
- const desktopHosts=new DesktopHostHub(workspace,auth)
- workspaceRuntime.desktopEnvironments=desktopEnvironments
- grokCloud.preferDesktop=(owner,agent)=>!!desktopEnvironments.selected(owner,agent)
- grokCloud.beforeSelection=(owner,id)=>desktopEnvironments.assertIdle(owner,id)
- workspaceRuntime.cloud=grokCloud
- const workspaceInspector=new WorkspaceInspector(workspace,auth)
- workspaceRuntime.inspector=workspaceInspector
- const workspaceRoutines=new WorkspaceRoutines(workspace,auth,workspaceNodes,workspaceRuntime)
- const workspacePlugins=new WorkspacePlugins(workspace,workspaceNodes,auth,workspaceRuntime,config.home,options.pluginFetch)
- workspaceRuntime.plugins=workspacePlugins
- workspaceRuntime.retireHelper=(owner,helper)=>runners.retireHelper(owner,helper)
- workspaceRuntime.onTeamCreated = (owner, team) => {
- try { push.setGroupSubscription(owner, team.id, true, team.lastSeq) } catch { /* Optional notifications do not undo a team. */ }
- }
- const workspaceAssets = new WorkspaceAssets(workspace, workspaceNodes, config.home)
- workspaceRuntime.publishArtifact=(owner,message,agent,name,bytes)=>workspaceAssets.publish(owner,message,agent,name,bytes)
- workspaceRuntime.onMessage = (owner, message) => workspaceAssets.archive(owner, message)
- workspaceRuntime.onNotify = (owner, c, run, message, interaction) => {
- if (c.kind === 'group' && !push.isGroupSubscribed(owner,c.id)) return
- push.enqueueNotification({ kind: interaction ? interaction.kind === 'approval' ? 'chat.approval.requested' : 'chat.clarification.requested' : run.status === 'failed' ? 'chat.failed' : 'chat.completed', eventID: `workspace:${interaction?.id ?? run.id}`, localUserID: owner, title: c.name, body: (interaction?.message || message?.content || run.error || '回复完成').slice(0,180), collapseID: `workspace:${c.id}:${run.conversationTaskId ?? 'direct'}`, data: { conversationId:c.id, ...(run.conversationTaskId ? { conversationTaskId: run.conversationTaskId } : {}), workspace:'1' }, ...(c.kind === 'group' ? {roomID:c.id} : {}) })
- }
- const realtime = new RealtimeAPI(config, auth, csrf, pairings, upstream, upstreamSession, {
- coordinator: pushEventCoordinator,
- resolver: new HermesChatNotificationResolver(upstreamSession, (user, prompt) => push.promptDigest(user, prompt)),
- }, (owner, profile, sessionID) =>
- chatCache?.store.ownsSession(owner, profile, sessionID) ?? false)
- realtime.broker.protectedSession = id => workspace.ownsUpstream(id)
- realtime.transcriptsAvailable = chatCache.store.transcripts.enabled
- realtime.broker.onNativeEvent = (owner, profile, storedId, frame) => {
- chatCache?.observe(owner, profile, storedId, frame)
- if (owner.startsWith('device:')) return
- if (!['message.complete', 'attachment.staged'].includes(frame.type)) return
- const data = frame.payload ?? {}, text = frame.type === 'attachment.staged' ? JSON.stringify(data) : nativeMessageFileText(data)
- const messageId = String(chatCache.store.projectedMessageID(owner,profile,storedId) ?? data.row_id ?? data.message_id ?? data.id ?? createHash('sha256').update(text).digest('hex'))
- void workspaceAssets.archiveText(owner, text, 'local', profile, storedId, messageId, frame.type === 'attachment.staged' ? 'user' : 'agent').catch(() => {})
- }
- workspace.nativeMessageForFile = (owner, file) => {
- const message = chatCache?.store.fileSourceMessage(owner, file.profile ?? 'default', file.conversationId ?? '', file.messageId ?? '')
- if (!message) return undefined
- return {
- id: file.messageId!, conversationId: file.conversationId!, seq: 0,
- role: message.role === 'assistant' ? 'assistant' : message.role === 'user' ? 'user' : 'system',
- content: nativeMessageFileText(message), reasoning: '', status: 'complete',
- attachments: [], tools: [], createdAt: file.createdAt,
- }
- }
- realtime.broker.onNativeGlobalEvent = (owner, type, upstreamInvalidation) => chatCache?.observeGlobal(owner, type, upstreamInvalidation)
- chatCache.onSynchronized = (owner, profile, sessionID) => {
- realtime.broker.publishOwnerSessionsChanged(owner, 'cache.synced', profile, sessionID)
- }
- realtime.broker.onNativeCommand = (owner, profile, storedId, method, params) => {
- chatCache?.command(owner, profile, storedId, method, params)
- }
- realtime.broker.nativeSourceCursor = (owner,profile,id,runtime,epoch)=>chatCache.store.transcripts.sourceCursor(owner,profile,id,runtime,epoch)
- realtime.broker.onNativeRoute = (owner, profile, storedId, runtimeId, running, snapshot) => {
- chatCache?.route(owner, profile, storedId, runtimeId, snapshot?undefined:running)
- if(snapshot)chatCache.store.recordEvent(owner,profile,storedId,{type:'route.resumed',payload:snapshot})
- }
- chatPushJobs.setTransportFactory(realtime.recoveryTransport, job => realtime.ownsPushJob(job.id))
- try {
- uploads.cleanupUncommitted()
- } catch {
- // A locked cleanup should not prevent startup; upload operations will
- // still surface the underlying storage error when used.
- }
- app.use(async (ctx, next) => {
- try {
- await next()
- } catch (error) {
- if (error instanceof UpstreamHttpError) {
- ctx.status = error.response.status
- ctx.type = 'application/json; charset=utf-8'
- ctx.body = { error: `Hermes returned HTTP ${error.response.status}`, code: error.code }
- return
- }
- const status = error instanceof HttpError
- ? error.status
- : typeof (error as { status?: unknown })?.status === 'number'
- ? Number((error as { status: number }).status)
- : 500
- const expose = status < 500 || error instanceof HttpError
- ctx.status = status
- ctx.type = 'application/json; charset=utf-8'
- ctx.body = {
- error: expose ? errorMessage(error) : 'Internal server error',
- code: error instanceof HttpError ? error.code : 'internal_error',
- }
- if (status >= 500 && !config.production) ctx.app.emit('error', error, ctx)
- }
- })
- app.use(async (ctx, next) => {
- applySecurityHeaders(ctx, Boolean(config.tlsCert), config.allowedHosts)
- if (!isAllowedHostHeader(ctx.get('host'), config)) {
- throw new HttpError(421, 'Request Host is not allowed', 'invalid_host')
- }
- await next()
- })
- app.use(async (ctx, next) => {
- if (ctx.path.startsWith('/api/app/') && isMutation(ctx.method)) {
- const secure = ctx.secure || Boolean(config.tlsCert)
- if (!isExactOrigin(ctx.get('origin'), ctx.get('host'), secure, config.allowedHosts)) {
- throw new HttpError(403, 'Request Origin is not allowed', 'invalid_origin')
- }
- if (!csrf.verify(ctx.get('cookie'), ctx.get('x-csrf-token'))) {
- throw new HttpError(403, 'CSRF token is invalid or missing', 'invalid_csrf')
- }
- }
- await next()
- })
- app.use(runners.middleware())
- app.use(desktopHosts.middleware())
- app.use(async (ctx, next) => {
- const user = auth.current(ctx)
- if (user && user.role !== 'admin' && (ctx.path.startsWith('/api/') || ctx.path.startsWith('/node/') || ctx.path.startsWith('/ws/') || ctx.path.startsWith('/Users/') || ctx.path.startsWith('/attachments/'))) {
- const path = ctx.path
- const publicPath = ['/api/status', '/api/auth/providers', '/api/auth/me', '/api/profiles', '/api/app/bootstrap', '/api/app/login', '/api/app/logout'].includes(path)
- const accountPath = ['/api/account/credentials', '/api/app/account/credentials', '/api/app/account/avatar'].includes(path)
- const workspacePath = /^\/api\/app\/(?:capabilities|agents|conversations|runs|interactions|events|uploads|files|message-files|computers|bot-tools)(?:\/|$)/.test(path)
- && path !== '/api/app/agents/remote'
- const pushPath = /^\/api\/(?:app\/)?push\/v1\//.test(path)
- const identityRead = path === '/api/app/server-identity' && ['GET', 'HEAD'].includes(ctx.method)
- if (!publicPath && !accountPath && !workspacePath && !pushPath && !identityRead)
- throw new HttpError(403, '子账号只能使用 Bot 模式', 'bot_mode_required')
- }
- await next()
- })
- app.use(hermesBotRelay(workspaceNodes, auth, csrf))
- app.use(remoteAgentExports(workspace,workspaceNodes,pairings,auth))
- app.use(realtime.middleware())
- const parseBody = (limit: string) => bodyParser({
- encoding: 'utf-8',
- enableTypes: ['json'],
- parsedMethods: ['POST', 'PUT', 'PATCH', 'DELETE'],
- jsonLimit: limit,
- onError: (error, ctx) => {
- throw new HttpError(ctx.status === 413 || (error as { status?: number }).status === 413 ? 413 : 400, 'Invalid JSON request body', 'invalid_json_body')
- },
- })
- const ordinaryBody = parseBody('2mb'), workspaceBody = parseBody('4mb')
- app.use((ctx,next) => (/^\/api\/app\/(agents|conversations)(\/|$)/.test(ctx.path) || ctx.path === '/api/app/account/avatar' ? workspaceBody : ordinaryBody)(ctx,next))
- app.use(async (ctx, next) => {
- ctx.state.localAuth = auth
- ctx.state.upstreamSession = upstreamSession
- const anonymous = ctx.path === '/api/app/bootstrap'
- || ctx.path === '/api/app/setup'
- || ctx.path === '/api/app/login'
- || ctx.path === '/api/status'
- || ctx.path === '/api/auth/providers'
- || ctx.path === '/auth/password-login'
- || ctx.path.startsWith('/api/pair/v1/')
- || ctx.path.startsWith('/api/account-pair/v1/')
- || ctx.path === '/healthz'
- || ctx.path === '/readyz'
- if (ctx.path.startsWith('/api/app/') && !anonymous) {
- const allowPasswordChange = ctx.path === '/api/app/account/credentials'
- || ctx.path === '/api/app/logout'
- ctx.state.localUser = auth.require(ctx, allowPasswordChange)
- const adminOnly = ctx.path.startsWith('/api/app/admin/')
- || ctx.path.startsWith('/api/app/system/')
- || ctx.path.startsWith('/api/app/plugins/')
- || ctx.path.startsWith('/api/app/pairings')
- || ctx.path.startsWith('/api/app/paired-devices')
- || ctx.path.startsWith('/api/app/groups/nodes')
- if (adminOnly) ctx.state.localUser = auth.requireAdmin(ctx)
- }
- const gatewayCompatibility = (ctx.path.startsWith('/api/') && !ctx.path.startsWith('/api/app/')
- && !ctx.path.startsWith('/api/pair/'))
- || ctx.path === '/auth/logout'
- if (gatewayCompatibility && !anonymous) {
- const allowPasswordChange = ctx.path === '/api/auth/me' || ctx.path === '/api/profiles'
- || ctx.path === '/api/account/credentials'
- ctx.state.localUser = auth.require(ctx, allowPasswordChange)
- if (isMutation(ctx.method)) {
- const origin = ctx.get('origin')
- if (origin && !isExactOrigin(origin, ctx.get('host'), ctx.secure || Boolean(config.tlsCert), config.allowedHosts)) {
- throw new HttpError(403, 'Request Origin is not allowed', 'invalid_origin')
- }
- }
- }
- const user = ctx.state.localUser as { id: string } | undefined
- if (user) {
- await upstream.withReadScope(`user:${user.id}:${auth.pushAuthorizationVersion(user.id)}`, ctx.get('x-yaoyao-cache') === 'bypass', next)
- } else await next()
- })
- for(const router of [desktopEnvironments.router(),grokCloud.authorization.router(),grokCloud.router(),workspaceInspector.router(),workspaceRoutines.router(),workspacePlugins.router()]){app.use(router.routes());app.use(router.allowedMethods())}
- const sharedComputerRouter=sharedComputers.router();app.use(sharedComputerRouter.routes());app.use(sharedComputerRouter.allowedMethods())
- const localVmRouter=localVm.router();app.use(localVmRouter.routes());app.use(localVmRouter.allowedMethods())
- const computerRouter=computerControls.router();app.use(computerRouter.routes());app.use(computerRouter.allowedMethods())
- const applicationRouter = workspaceRouter(workspace, workspaceRuntime, workspaceNodes, workspaceAssets, uploads, auth, push, csrf)
- app.use(runners.adminRouter().routes())
- app.use(desktopHosts.adminRouter().routes())
- app.use(applicationRouter.routes())
- app.use(async (ctx, next) => {
- if (ctx.path.includes('/plugins/yaoyao') || ctx.path.startsWith('/api/app/groups')) throw new HttpError(410, '请使用新版聊天接口', 'retired_plugin_api')
- const id = /\/sessions\/([^/]+)/.exec(ctx.path)?.[1]
- if (id && workspace.ownsUpstream(decodeURIComponent(id))) throw new HttpError(404, '会话不存在', 'session_not_found')
- await next()
- // Native history is a separate surface, including search and profile projections.
- if (ctx.path.includes('/sessions') || ctx.path.endsWith('/profiles')) {
- let value: any = ctx.body
- const buffered = Buffer.isBuffer(value)
- try {
- if (buffered) value = JSON.parse(value.toString())
- const filter = (v: any): any => {
- if (Array.isArray(v)) return v.filter(x => !x || typeof x !== 'object' || (x.source !== 'yaoyao_workspace' && !workspace.ownsUpstream(String(x.id ?? x.session_id ?? x.sessionId ?? '')))).map(filter)
- if (v && typeof v === 'object') return Object.fromEntries(Object.entries(v).map(([k,x]) => [k, filter(x)]))
- return v
- }
- ctx.body = buffered ? Buffer.from(JSON.stringify(filter(value))) : filter(value)
- } catch { /* Preserve non-JSON error responses. */ }
- }
- })
- app.use(chatTranscriptRouter(chatCache, auth).routes())
- const router = createApiRouter({
- onChatListChanged: (owner,profile,id) => realtime.broker.publishOwnerSessionsChanged(owner,'local-state',profile,id),
- onServerIdentityChanged: identity => realtime.broker.publishServerIdentity(identity, config.upstream.href),
- onUserAccessChanged: async owner => {
- for (const c of workspace.list<{ id: string }>(owner, 'conversation')) await workspaceRuntime.stopConversation(owner, c.id)
- },
- workspace,
- config,
- csrf,
- upstream,
- pairings,
- uploads,
- updates,
- auth,
- upstreamSession,
- accountPairings,
- profileIdentities,
- push,
- apnsConfiguration,
- fcmConfiguration,
- openVikingConfiguration,
- openVikingService,
- allowedHostsConfiguration,
- hermesBridge,
- chatCache,
- })
- app.use(router.routes())
- app.use(router.allowedMethods({ throw: false }))
- app.use(async (ctx, next) => {
- if (ctx.path === '/api' || ctx.path.startsWith('/api/') || ctx.path.startsWith('/ws/')) {
- ctx.status = 404
- ctx.type = 'application/json; charset=utf-8'
- ctx.body = { error: 'Route not found', code: 'not_found' }
- return
- }
- await next()
- })
- return {
- runners,
- hermesBridge,
- localVm,
- workspace,
- workspaceRuntime,
- workspaceRoutines,
- workspacePlugins,
- desktopEnvironments,
- desktopHosts,
- grokAuth:grokCloud.authorization,
- realtime,
- app,
- config,
- csrf,
- pairings,
- upstream,
- uploads,
- updates,
- auth,
- upstreamSession,
- accountPairings,
- profileIdentities,
- push,
- apnsConfiguration,
- fcmConfiguration,
- openVikingConfiguration,
- openVikingService,
- allowedHostsConfiguration,
- pushEventCoordinator,
- chatPushJobs,
- chatCache,
- close: () => {
- workspaceMemory.close()
- runners.close()
- workspaceAssets.close()
- desktopEnvironments.close()
- desktopHosts.close()
- grokCloud.authorization.close()
- workspaceRoutines.close()
- workspacePlugins.close()
- workspaceRuntime.close()
- workspaceNodes.close()
- upstream.close()
- realtime.close()
- chatPushJobs.stop()
- push.close()
- uploads.close()
- workspace.close()
- chatCache?.close()
- chatCache?.store.close()
- },
- }
- }
- export interface NodeServerRuntime {
- server: HttpServer
- close(closeConnections?: boolean): Promise<void>
- }
- export function createNodeServer(runtime: ApplicationRuntime): NodeServerRuntime {
- const { config } = runtime
- const server: HttpServer = config.tlsCert && config.tlsKey
- ? createHttpsServer({
- cert: readFileSync(config.tlsCert),
- key: readFileSync(config.tlsKey),
- }, runtime.app.callback())
- : createHttpServer(runtime.app.callback())
- const synchronizePushObservers = (enabled = runtime.push.isAnyProviderEnabled()) => {
- if (enabled) {
- runtime.chatPushJobs.start()
- } else {
- runtime.chatPushJobs.stop()
- }
- }
- server.once('listening',()=>{const address=server.address();if(address&&typeof address!=='string')void runtime.localVm.start(`${config.tlsCert?'https':'http'}://127.0.0.1:${address.port}`).catch(error=>console.error('本地虚拟机服务启动失败',error.message))})
- runtime.workspaceRuntime.start()
- runtime.workspaceRoutines.start()
- runtime.grokAuth.start()
- synchronizePushObservers()
- const removePushConfigurationListener = runtime.push.onEnabledChange(synchronizePushObservers)
- const removeWebSockets = runtime.realtime.rejectLegacyUpgrades(server)
- return {
- server,
- close: async (closeConnections = false) => {
- await runtime.localVm.close()
- removePushConfigurationListener()
- removeWebSockets()
- runtime.realtime.close()
- if (closeConnections) server.closeAllConnections()
- await new Promise<void>((resolve, reject) => {
- if (!server.listening) {
- resolve()
- return
- }
- server.close((error) => error ? reject(error) : resolve())
- })
- runtime.close()
- },
- }
- }
|