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;

mail-worker/wrangler-action.toml

@@ -22,6 +22,9 @@ id = "${KV_NAMESPACE_ID}" # 使用占位符引用环境变量
 binding = "r2"
 bucket_name = "${R2_BUCKET_NAME}" # 使用占位符引用环境变量
 
+[ai]
+binding = "AI"
+
 [assets]
 binding = "assets"		#静态资源绑定名默认不可修改
 directory = "./dist"	#前端vue项目打包的静态资源存放位置,默认dist
@@ -33,6 +36,7 @@ crons = ["0 16 * * *"]	#定时任务每天晚上12点执行
 
 
 [vars]
+ai_model = "@cf/meta/llama-3.1-8b-instruct"
 #orm_log = false
 domain = "${DOMAIN}"				#邮件域名可可配置多个 示例: ["example1.com","example2.com"]
 admin = "${ADMIN}"				#管理员的邮箱	示例: admin@example.com

mail-worker/wrangler-dev.toml

@@ -22,6 +22,9 @@ id = "2io01d4b299e481b9de060ece9e7785c"
 #binding = "r2"
 #bucket_name = "email"
 
+[ai]
+binding = "AI"
+
 [assets]
 binding = "assets"
 directory = "./dist"
@@ -30,6 +33,7 @@ run_worker_first = true
 
 
 [vars]
+ai_model = "@cf/meta/llama-3.1-8b-instruct"
 orm_log = false
 domain = ["example.com", "example2.com", "example3.com", "example4.com"]
 admin = "admin@example.com"

mail-worker/wrangler-test.toml

@@ -20,6 +20,9 @@ id = "9be10cd058e04c55b526f8c0c116f50c"					#kv数据库id
 #binding = "r2"			#r2对象存储绑定名默认不可修改
 #bucket_name = "test-email"		#r2对象存储桶的名字
 
+[ai]
+binding = "AI"
+
 [assets]
 binding = "assets"		#静态资源绑定名默认不可修改
 directory = "./dist"	#前端vue项目打包的静态资源存放位置,默认dist
@@ -31,6 +34,7 @@ crons = ["0 16 * * *"]	#定时任务每天晚上12点执行
 
 
 [vars]
+ai_model = "@cf/meta/llama-3.1-8b-instruct"
 orm_log = false
 domain = ["example.com"]				#邮件域名可可配置多个 示例: ["example1.com","example2.com"]
 admin = "admin@example.com"				#管理员的邮箱	示例: admin@example.com

mail-worker/wrangler.toml

@@ -19,6 +19,9 @@ enabled = true
 #binding = "r2"			#r2对象存储绑定名默认不可修改
 #bucket_name = ""		#r2对象存储桶的名字
 
+[ai]
+binding = "AI"
+
 [assets]
 binding = "assets"		#静态资源绑定名默认不可修改
 directory = "./dist"	#前端vue项目打包的静态资源存放位置,默认dist
@@ -29,7 +32,8 @@ run_worker_first = true
 crons = ["0 16 * * *"]	#定时任务每天晚上12点执行
 
 
-#[vars]
+[vars]
+ai_model = "@cf/meta/llama-3.1-8b-instruct"
 #orm_log = false
 #domain = []				#邮件域名可可配置多个 示例: ["example1.com","example2.com"]
 #admin = ""				#管理员的邮箱	示例: admin@example.com