mirror of
https://github.com/jeffusion/gitea-ai-assistant.git
synced 2026-03-27 10:05:50 +00:00
291 lines
9.9 KiB
TypeScript
291 lines
9.9 KiB
TypeScript
import { Hono } from 'hono';
|
||
import { configManager } from '../config/config-manager';
|
||
import { CONFIG_FIELDS, CONFIG_GROUPS, type ConfigFieldMeta } from '../config/config-schema';
|
||
import { getNotificationManager } from '../services/notification-manager';
|
||
import type { NotificationProvider } from '../services/notification/types';
|
||
import { logger } from '../utils/logger';
|
||
|
||
// ── Constants ────────────────────────────────────────────────────────────────
|
||
|
||
const MASKED_VALUE = '••••••••';
|
||
|
||
/** Number fields that must be integers (decimal not allowed). */
|
||
const INTEGER_FIELDS = new Set([
|
||
'REVIEW_MAX_PARALLEL_RUNS',
|
||
'REVIEW_MAX_FILES_PER_RUN',
|
||
'REVIEW_MAX_FILE_CONTENT_CHARS',
|
||
'REVIEW_COMMAND_TIMEOUT_MS',
|
||
'FEW_SHOT_EXAMPLES_COUNT',
|
||
'MAX_REFLECTION_ROUNDS',
|
||
]);
|
||
|
||
/** Fast lookup from envKey → field metadata. */
|
||
const FIELDS_MAP = new Map<string, ConfigFieldMeta>(CONFIG_FIELDS.map((f) => [f.envKey, f]));
|
||
const TESTABLE_PROVIDERS = new Set<NotificationProvider>(['feishu', 'wecom']);
|
||
const NOTIFICATION_TEST_HISTORY_LIMIT = 30;
|
||
|
||
type NotificationTestRecord = {
|
||
id: string;
|
||
provider: string;
|
||
status: 'success' | 'error';
|
||
message: string;
|
||
timestamp: string;
|
||
};
|
||
|
||
const notificationTestHistory: NotificationTestRecord[] = [];
|
||
|
||
function appendNotificationTestRecord(
|
||
provider: string,
|
||
status: 'success' | 'error',
|
||
message: string
|
||
): void {
|
||
notificationTestHistory.unshift({
|
||
id: `${Date.now()}-${Math.random().toString(36).slice(2, 8)}`,
|
||
provider,
|
||
status,
|
||
message,
|
||
timestamp: new Date().toISOString(),
|
||
});
|
||
|
||
if (notificationTestHistory.length > NOTIFICATION_TEST_HISTORY_LIMIT) {
|
||
notificationTestHistory.splice(NOTIFICATION_TEST_HISTORY_LIMIT);
|
||
}
|
||
}
|
||
|
||
// ── Helpers ──────────────────────────────────────────────────────────────────
|
||
|
||
/**
|
||
* Validate a single field value against its metadata.
|
||
* Returns an error message string, or `null` if valid.
|
||
*/
|
||
function validateField(field: ConfigFieldMeta, key: string, value: string): string | null {
|
||
switch (field.type) {
|
||
case 'url': {
|
||
try {
|
||
new URL(value);
|
||
} catch {
|
||
return `${field.label}（${key}）必须是有效的 URL`;
|
||
}
|
||
return null;
|
||
}
|
||
case 'enum': {
|
||
if (field.enumValues && !field.enumValues.includes(value)) {
|
||
return `${field.label}（${key}）必须是以下值之一: ${field.enumValues.join(', ')}`;
|
||
}
|
||
return null;
|
||
}
|
||
case 'boolean': {
|
||
if (value !== 'true' && value !== 'false') {
|
||
return `${field.label}（${key}）必须是布尔值`;
|
||
}
|
||
return null;
|
||
}
|
||
case 'number': {
|
||
const num = Number(value);
|
||
if (Number.isNaN(num)) {
|
||
return `${field.label}（${key}）必须是有效的数字`;
|
||
}
|
||
if (INTEGER_FIELDS.has(key) && !Number.isInteger(num)) {
|
||
return `${field.label}（${key}）必须是整数`;
|
||
}
|
||
if (field.min !== undefined && num < field.min) {
|
||
return `${field.label}（${key}）不能小于 ${field.min}`;
|
||
}
|
||
if (field.max !== undefined && num > field.max) {
|
||
return `${field.label}（${key}）不能大于 ${field.max}`;
|
||
}
|
||
return null;
|
||
}
|
||
default:
|
||
// string, text — no special validation
|
||
return null;
|
||
}
|
||
}
|
||
|
||
// ── Router ───────────────────────────────────────────────────────────────────
|
||
|
||
export const configRouter = new Hono();
|
||
|
||
/**
|
||
* GET / — Return all configuration groups, fields with metadata,
|
||
* effective values, and source. Sensitive fields are masked.
|
||
*/
|
||
configRouter.get('/', (c) => {
|
||
const rawValues = configManager.getAllRawValues();
|
||
|
||
const groups = CONFIG_GROUPS.map((group) => {
|
||
const groupFields = CONFIG_FIELDS.filter((f) => f.group === group.key);
|
||
|
||
const fields = groupFields.map((field) => {
|
||
const rawValue = rawValues[field.envKey];
|
||
const hasValue = rawValue !== undefined && rawValue !== '';
|
||
const source = configManager.getSource(field.envKey);
|
||
const value = field.sensitive && hasValue ? MASKED_VALUE : rawValue;
|
||
|
||
return {
|
||
envKey: field.envKey,
|
||
label: field.label,
|
||
description: field.description,
|
||
type: field.type,
|
||
sensitive: field.sensitive,
|
||
...(field.enumValues !== undefined && { enumValues: field.enumValues }),
|
||
...(field.min !== undefined && { min: field.min }),
|
||
...(field.max !== undefined && { max: field.max }),
|
||
...(field.defaultValue !== undefined && { defaultValue: field.defaultValue }),
|
||
value,
|
||
hasValue,
|
||
source,
|
||
};
|
||
});
|
||
|
||
return {
|
||
key: group.key,
|
||
label: group.label,
|
||
description: group.description,
|
||
icon: group.icon,
|
||
fields,
|
||
};
|
||
});
|
||
|
||
return c.json({ groups });
|
||
});
|
||
|
||
/**
|
||
* PUT / — Validate and persist configuration updates.
|
||
* Masked sentinel ('••••••••') for sensitive fields is silently skipped.
|
||
* Empty string '' causes the key to be reset (deleted from DB).
|
||
*/
|
||
configRouter.put('/', async (c) => {
|
||
try {
|
||
const body = await c.req.json<Record<string, unknown>>();
|
||
|
||
if (typeof body !== 'object' || body === null || Array.isArray(body)) {
|
||
return c.json({ message: '保存配置失败', error: '请求体必须是 JSON 对象' }, 400);
|
||
}
|
||
|
||
const updates: Record<string, string> = {};
|
||
const errors: string[] = [];
|
||
|
||
for (const [key, rawValue] of Object.entries(body)) {
|
||
const field = FIELDS_MAP.get(key);
|
||
|
||
if (!field) {
|
||
errors.push(`未知配置项: ${key}`);
|
||
continue;
|
||
}
|
||
const value = String(rawValue ?? '');
|
||
|
||
// Skip masked sentinel for sensitive fields — do not overwrite with mask
|
||
if (field.sensitive && value === MASKED_VALUE) {
|
||
continue;
|
||
}
|
||
|
||
// Empty string → reset (ConfigManager deletes the key)
|
||
if (value === '') {
|
||
updates[key] = '';
|
||
continue;
|
||
}
|
||
|
||
const fieldError = validateField(field, key, value);
|
||
if (fieldError) {
|
||
errors.push(fieldError);
|
||
continue;
|
||
}
|
||
|
||
updates[key] = value;
|
||
}
|
||
|
||
if (errors.length > 0) {
|
||
return c.json({ message: '保存配置失败', error: errors.join('; ') }, 400);
|
||
}
|
||
|
||
await configManager.setOverrides(updates);
|
||
return c.json({ success: true, message: '配置已保存' });
|
||
} catch (error: unknown) {
|
||
const errMsg = error instanceof Error ? error.message : String(error);
|
||
logger.error('保存配置失败:', error);
|
||
return c.json({ message: '保存配置失败', error: errMsg }, 500);
|
||
}
|
||
});
|
||
|
||
/**
|
||
* POST /reset — Remove specified keys from DB (revert to default).
|
||
*/
|
||
configRouter.post('/reset', async (c) => {
|
||
try {
|
||
const { keys } = await c.req.json<{ keys: unknown }>();
|
||
|
||
if (!Array.isArray(keys) || !keys.every((k): k is string => typeof k === 'string')) {
|
||
return c.json({ message: '保存配置失败', error: 'keys 必须是字符串数组' }, 400);
|
||
}
|
||
|
||
const unknownKeys = keys.filter((k) => !FIELDS_MAP.has(k));
|
||
if (unknownKeys.length > 0) {
|
||
return c.json(
|
||
{ message: '保存配置失败', error: `未知配置项: ${unknownKeys.join(', ')}` },
|
||
400
|
||
);
|
||
}
|
||
|
||
await configManager.resetKeys(keys);
|
||
return c.json({ success: true, message: '配置已重置' });
|
||
} catch (error: unknown) {
|
||
const errMsg = error instanceof Error ? error.message : String(error);
|
||
logger.error('重置配置失败:', error);
|
||
return c.json({ message: '保存配置失败', error: errMsg }, 500);
|
||
}
|
||
});
|
||
|
||
configRouter.post('/notification/test', async (c) => {
|
||
try {
|
||
let body: Record<string, unknown>;
|
||
try {
|
||
const parsed = await c.req.json();
|
||
if (typeof parsed !== 'object' || parsed === null || Array.isArray(parsed)) {
|
||
appendNotificationTestRecord('unknown', 'error', '请求体必须是 JSON 对象');
|
||
return c.json({ message: '发送测试通知失败', error: '请求体必须是 JSON 对象' }, 400);
|
||
}
|
||
body = parsed;
|
||
} catch {
|
||
appendNotificationTestRecord('unknown', 'error', '请求体必须是 JSON 对象');
|
||
return c.json({ message: '发送测试通知失败', error: '请求体必须是 JSON 对象' }, 400);
|
||
}
|
||
|
||
const provider = typeof body.provider === 'string' ? body.provider : '';
|
||
|
||
if (!TESTABLE_PROVIDERS.has(provider as NotificationProvider)) {
|
||
appendNotificationTestRecord(
|
||
provider || 'unknown',
|
||
'error',
|
||
'provider 必须是 feishu 或 wecom'
|
||
);
|
||
return c.json({ message: '发送测试通知失败', error: 'provider 必须是 feishu 或 wecom' }, 400);
|
||
}
|
||
|
||
const notificationManager = getNotificationManager();
|
||
const providerName = provider as NotificationProvider;
|
||
|
||
if (!notificationManager.hasService(providerName)) {
|
||
appendNotificationTestRecord(providerName, 'error', `${providerName} 未启用或未配置`);
|
||
return c.json({ message: '发送测试通知失败', error: `${providerName} 未启用或未配置` }, 400);
|
||
}
|
||
|
||
await notificationManager.sendTestMessage(providerName);
|
||
appendNotificationTestRecord(providerName, 'success', `${providerName} 测试通知已发送`);
|
||
|
||
return c.json({
|
||
success: true,
|
||
message: `${providerName} 测试通知已发送`,
|
||
});
|
||
} catch (error: unknown) {
|
||
const errMsg = error instanceof Error ? error.message : String(error);
|
||
appendNotificationTestRecord('unknown', 'error', errMsg);
|
||
logger.error('发送测试通知失败:', error);
|
||
return c.json({ message: '发送测试通知失败', error: errMsg }, 500);
|
||
}
|
||
});
|
||
|
||
configRouter.get('/notification/test/history', (c) => {
|
||
return c.json({ data: notificationTestHistory });
|
||
});
|