feat: AI-powered email verification code recognition

mail-worker/src/email/email.js

@@ -10,6 +10,7 @@ import emailUtils from '../utils/email-utils';
 import roleService from '../service/role-service';
 import userService from '../service/user-service';
 import telegramService from '../service/telegram-service';
+import aiService from '../service/ai-service';
 
 export async function email(message, env, ctx) {
 
@@ -89,6 +90,7 @@ export async function email(message, env, ctx) {
 		}
 
 		const toName = email.to.find(item => item.address === message.to)?.name || '';
+		const code = await aiService.extractCode({ env }, email);
 
 		const params = {
 			toEmail: message.to,
@@ -96,6 +98,7 @@ export async function email(message, env, ctx) {
 			sendEmail: email.from.address,
 			name: email.from.name || emailUtils.getName(email.from.address),
 			subject: email.subject,
+			code,
 			content: email.html,
 			text: email.text,
 			cc: email.cc ? JSON.stringify(email.cc) : '[]',

mail-worker/src/entity/email.js

@@ -7,6 +7,7 @@ export const email = sqliteTable('email', {
 	accountId: integer('account_id').notNull(),
 	userId: integer('user_id').notNull(),
 	subject: text('subject'),
+	code: text('code').default('').notNull(),
 	text: text('text'),
 	content: text('content'),
 	cc: text('cc').default('[]'),

mail-worker/src/init/init.js

@@ -34,6 +34,12 @@ const dbInit = {
 	},
 
 	async v3_0DB(c) {
+		try {
+			await c.env.db.prepare(`ALTER TABLE email ADD COLUMN code TEXT NOT NULL DEFAULT '';`).run();
+		} catch (e) {
+			console.warn(`跳过字段：${e.message}`);
+		}
+
 		try {
 			await c.env.db.batch([
 				c.env.db.prepare(`ALTER TABLE setting ADD COLUMN black_subject TEXT NOT NULL DEFAULT '';`),

mail-worker/src/service/ai-service.js

@@ -0,0 +1,46 @@
+import emailUtils from '../utils/email-utils';
+
+const aiService = {
+	async extractCode(c, email) {
+		const ai = c.env.AI || c.env.ai;
+
+		if (!ai) {
+			return '';
+		}
+
+		try {
+			const subject = email.subject || '';
+			const text = emailUtils.formatText(email.text || '');
+			const htmlText = emailUtils.htmlToText(email.html || '');
+			const body = (htmlText || text).slice(0, 6000);
+
+			if (!subject && !body) {
+				return '';
+			}
+
+			const result = await ai.run(c.env.ai_model || '@cf/meta/llama-3.1-8b-instruct', {
+				messages: [
+					{
+						role: 'system',
+						content: 'You extract verification codes from emails. Return only JSON like {"code":"123456"} or {"code":""}. Do not explain.'
+					},
+					{
+						role: 'user',
+						content: `Subject: ${subject}\n\n${body}`
+					}
+				],
+				temperature: 0,
+				max_tokens: 32
+			});
+
+			const content = typeof result === 'string' ? result : result?.response || '';
+			const json = JSON.parse(content);
+			return typeof json.code === 'string' ? json.code.trim().slice(0, 64) : '';
+		} catch (e) {
+			console.error('验证码提取失败: ', e);
+			return '';
+		}
+	}
+};
+
+export default aiService;