ciat-777
487bd23a02
- Implemented `CliProxyApiAdapter` with detection logic for CLIProxyAPI. - Added tests for `CliProxyApiAdapter` to verify detection and response handling. - Created `OneHubAdapter` with methods for fetching models and user groups. - Added tests for `OneHubAdapter` to ensure correct fallback and response parsing. - Developed `Sub2ApiAdapter` with detection and user info fetching capabilities. - Added comprehensive tests for `Sub2ApiAdapter` covering various scenarios. - Introduced `TitleHintPlatform` type and detection logic based on HTML title. - Implemented tests for title hint detection to identify platforms from URLs.
55 lines
2.0 KiB
TypeScript
55 lines
2.0 KiB
TypeScript
import { OneApiAdapter } from './oneApi.js';
|
|
|
|
export class OneHubAdapter extends OneApiAdapter {
|
|
readonly platformName: string = 'one-hub';
|
|
|
|
async detect(url: string): Promise<boolean> {
|
|
const normalized = url.toLowerCase();
|
|
return normalized.includes('onehub') || normalized.includes('one-hub');
|
|
}
|
|
|
|
/**
|
|
* OneHub model discovery: try /v1/models first, fall back to /api/available_model.
|
|
* The /api/available_model endpoint returns { data: { model_name: { price: ... }, ... } }
|
|
* where the keys are model names.
|
|
*/
|
|
override async getModels(baseUrl: string, apiToken: string, platformUserId?: number): Promise<string[]> {
|
|
let openAiModels: string[] = [];
|
|
try {
|
|
openAiModels = await super.getModels(baseUrl, apiToken, platformUserId);
|
|
} catch {}
|
|
if (openAiModels.length > 0) return openAiModels;
|
|
|
|
try {
|
|
const res = await this.fetchJson<any>(`${baseUrl}/api/available_model`, {
|
|
headers: { Authorization: `Bearer ${apiToken}` },
|
|
});
|
|
const payload = res?.data && typeof res.data === 'object' ? res.data : res;
|
|
if (payload && typeof payload === 'object' && !Array.isArray(payload)) {
|
|
const models = Object.keys(payload).filter(Boolean);
|
|
if (models.length > 0) return models;
|
|
}
|
|
} catch {}
|
|
|
|
return [];
|
|
}
|
|
|
|
/**
|
|
* OneHub user groups: /api/user_group_map returns { data: { group_name: ratio, ... } }
|
|
*/
|
|
override async getUserGroups(baseUrl: string, accessToken: string): Promise<string[]> {
|
|
try {
|
|
const res = await this.fetchJson<any>(`${baseUrl}/api/user_group_map`, {
|
|
headers: { Authorization: `Bearer ${accessToken}` },
|
|
});
|
|
const source = res?.data || res;
|
|
if (source && typeof source === 'object' && !Array.isArray(source)) {
|
|
const groups = Object.keys(source).map((k) => k.trim()).filter(Boolean);
|
|
if (groups.length > 0) return Array.from(new Set(groups));
|
|
}
|
|
} catch {}
|
|
|
|
return super.getUserGroups(baseUrl, accessToken);
|
|
}
|
|
}
|