initial commit
This commit is contained in:
77
fluxer_api/src/gateway/GatewayController.ts
Normal file
77
fluxer_api/src/gateway/GatewayController.ts
Normal file
@@ -0,0 +1,77 @@
|
||||
/*
|
||||
* Copyright (C) 2026 Fluxer Contributors
|
||||
*
|
||||
* This file is part of Fluxer.
|
||||
*
|
||||
* Fluxer is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* Fluxer is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with Fluxer. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
import type {HonoApp} from '~/App';
|
||||
import {Config} from '~/Config';
|
||||
import {FluxerAPIError} from '~/Errors';
|
||||
import {RateLimitMiddleware} from '~/middleware/RateLimitMiddleware';
|
||||
import {RateLimitConfigs} from '~/RateLimitConfig';
|
||||
|
||||
function parseTokenType(raw: string): 'user' | 'bot' | 'unknown' {
|
||||
if (raw.startsWith('flx_')) return 'user';
|
||||
const dotIndex = raw.indexOf('.');
|
||||
if (dotIndex > 0 && dotIndex < raw.length - 1) {
|
||||
const beforeDot = raw.slice(0, dotIndex);
|
||||
if (/^\d+$/.test(beforeDot)) return 'bot';
|
||||
}
|
||||
return 'unknown';
|
||||
}
|
||||
|
||||
function extractToken(authHeader: string | null): string {
|
||||
if (!authHeader) return '';
|
||||
const lower = authHeader.toLowerCase();
|
||||
if (lower.startsWith('bot ')) return authHeader.slice(4).trim();
|
||||
if (lower.startsWith('bearer ')) return authHeader.slice(7).trim();
|
||||
return authHeader.trim();
|
||||
}
|
||||
|
||||
export const GatewayController = (app: HonoApp) => {
|
||||
app.get('/gateway/bot', RateLimitMiddleware(RateLimitConfigs.GATEWAY_BOT_INFO), async (ctx) => {
|
||||
const token = extractToken(ctx.req.header('Authorization') || null);
|
||||
|
||||
if (!token) {
|
||||
throw new FluxerAPIError({
|
||||
code: 'MISSING_AUTHORIZATION',
|
||||
message: 'Bot token required',
|
||||
status: 401,
|
||||
});
|
||||
}
|
||||
|
||||
if (parseTokenType(token) !== 'bot') {
|
||||
throw new FluxerAPIError({
|
||||
code: 'INVALID_AUTH_TOKEN',
|
||||
message: 'Gateway bot endpoint only accepts bot tokens',
|
||||
status: 401,
|
||||
});
|
||||
}
|
||||
|
||||
await ctx.get('botAuthService').validateBotToken(token);
|
||||
|
||||
return ctx.json({
|
||||
url: Config.endpoints.gateway,
|
||||
shards: 1,
|
||||
session_start_limit: {
|
||||
total: 1000,
|
||||
remaining: 999,
|
||||
reset_after: 14400000,
|
||||
max_concurrency: 1,
|
||||
},
|
||||
});
|
||||
});
|
||||
};
|
||||
Reference in New Issue
Block a user