centersl/lobehub
1
0
Fork 0
mirror of https://github.com/lobehub/lobehub.git synced 2026-03-26 13:19:34 +07:00
Code Issues Packages Projects Releases Wiki Activity

🔨 chore: improve gateway (#12643)

Browse Source 
improve gateway
This commit is contained in:
Arvin Xu
2026-03-03 23:26:46 +08:00
committed by GitHub
parent 90714af0dc
commit f223aeb7f4
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
apps/device-gateway/src/index.ts
View File

@@ -18,8 +18,19 @@ export default {
const token = url.searchParams.get('token');
if (!token) return new Response('Missing token', { status: 401 });
let userId: string;
try {
const { userId } = await verifyDesktopToken(env, token);
if (token === env.SERVICE_TOKEN) {
// Service token auth (for CLI debugging)
const uid = url.searchParams.get('userId');
if (!uid) return new Response('Missing userId for service token auth', { status: 400 });
userId = uid;
} else {
// JWT auth (normal desktop flow)
const result = await verifyDesktopToken(env, token);
userId = result.userId;
}
const id = env.DEVICE_GATEWAY.idFromName(`user:${userId}`);
const stub = env.DEVICE_GATEWAY.get(id);
@@ -41,6 +52,7 @@ export default {
}
const body = (await request.clone().json()) as { userId: string };
if (!body.userId) return new Response('Missing userId', { status: 400 });
const id = env.DEVICE_GATEWAY.idFromName(`user:${body.userId}`);
const stub = env.DEVICE_GATEWAY.get(id);
return stub.fetch(request);