feat: 小红书自动评论助手 v1.7.5
- 侧边栏常驻：搜索/自动点击/自动评论全流程 - 4种评论方式：固定池/自定义文字/图片回复/AI按笔记生成 - 按地区搜索、评论记录统计（今日/本月，点击直达笔记） - CardAuth 授权：1天试用倒计时、卡密激活、购买自动发卡 - 公告推送、安装统计、小红书风格图标
This commit is contained in:
88531 committed
commit
8193e6c0c5
17 files changed
+6405
No files matched your search
+659
@@ -0,0 +1,659 @@
|
||||
// 后台脚本
|
||||
// 点击工具栏图标 → 直接在浏览器右侧打开/收起侧边栏（而不是弹窗）
|
||||
if (chrome.sidePanel && chrome.sidePanel.setPanelBehavior) {
|
||||
chrome.sidePanel.setPanelBehavior({ openPanelOnActionClick: true }).catch((error) => {
|
||||
console.log('设置侧边栏行为失败:', error);
|
||||
});
|
||||
}
|
||||
|
||||
// 默认自定义模型配置（OpenAI 兼容接口）
|
||||
const DEFAULT_AI = {
|
||||
apiBaseUrl: 'https://ai.88531.cn/v1',
|
||||
model: 'mimo-v2.6-flash'
|
||||
};
|
||||
|
||||
// ===== CardAuth 授权系统配置 =====
|
||||
const AUTH_BASE = 'https://auth.88531.cn';
|
||||
const AUTH_PROJECT_ID = 3;
|
||||
const AUTH_API_KEY = 'ak_bbf83b76df8219388d7f60b9e67c7db2';
|
||||
const CLIENT_VERSION = '1.7.5';
|
||||
const TRIAL_MS = 24 * 60 * 60 * 1000; // 未授权试用1天
|
||||
|
||||
// 把用户填写的地址规范化为完整的 chat/completions 端点
|
||||
function buildChatCompletionsUrl(baseUrl) {
|
||||
let base = String(baseUrl || '').trim() || DEFAULT_AI.apiBaseUrl;
|
||||
base = base.replace(/\/+$/, '');
|
||||
if (/\/chat\/completions$/i.test(base)) {
|
||||
return base;
|
||||
}
|
||||
if (!/\/v\d+$/i.test(base)) {
|
||||
base += '/v1'; // 未带版本号时自动补 /v1（如 https://api.deepseek.com）
|
||||
}
|
||||
return base + '/chat/completions';
|
||||
}
|
||||
|
||||
// 构造请求体
|
||||
function buildChatBody(model, prompt, maxTokens) {
|
||||
return {
|
||||
model: model,
|
||||
messages: [{ role: 'user', content: prompt }],
|
||||
temperature: 0.7,
|
||||
max_tokens: maxTokens,
|
||||
presence_penalty: 0,
|
||||
frequency_penalty: 0,
|
||||
// 思考型模型（mimo / DeepSeek-R1 等）会先写 reasoning_content，
|
||||
// 不关思考时正文容易被 max_tokens 截断 → 返回空。
|
||||
// 支持的端点直接关掉思考；不支持的端点会报 400，下面自动去掉该参数重试
|
||||
thinking: { type: 'disabled' }
|
||||
};
|
||||
}
|
||||
|
||||
// 从各种 OpenAI 兼容返回结构里取正文（兼容 content 字符串 / 分段数组 / text / output_text 等）
|
||||
function extractChatContent(data) {
|
||||
if (!data || typeof data !== 'object') return '';
|
||||
const pick = (value) => {
|
||||
if (typeof value === 'string') return value.trim();
|
||||
if (Array.isArray(value)) {
|
||||
return value.map(part => {
|
||||
if (typeof part === 'string') return part;
|
||||
if (part && typeof part === 'object') return part.text || part.content || '';
|
||||
return '';
|
||||
}).join('').trim();
|
||||
}
|
||||
return '';
|
||||
};
|
||||
try {
|
||||
const choice = Array.isArray(data.choices) ? data.choices[0] : null;
|
||||
if (choice) {
|
||||
const fromMessage = choice.message ? pick(choice.message.content) : '';
|
||||
if (fromMessage) return fromMessage;
|
||||
const fromText = pick(choice.text);
|
||||
if (fromText) return fromText;
|
||||
}
|
||||
return pick(data.output_text) || pick(data.output);
|
||||
} catch (error) {
|
||||
return '';
|
||||
}
|
||||
}
|
||||
|
||||
chrome.runtime.onInstalled.addListener(() => {
|
||||
console.log('小红书自动评论助手已安装');
|
||||
|
||||
// 初始化存储
|
||||
chrome.storage.local.get(['expandContent', 'commentText', 'apiKey', 'aiPrompt', 'apiBaseUrl', 'modelName'], (result) => {
|
||||
if (result.expandContent === undefined) {
|
||||
chrome.storage.local.set({ expandContent: true });
|
||||
}
|
||||
if (!result.commentText) {
|
||||
chrome.storage.local.set({ commentText: '👍 赞！' });
|
||||
}
|
||||
if (!result.apiKey) {
|
||||
chrome.storage.local.set({ apiKey: '' });
|
||||
}
|
||||
// 自定义模型配置：默认使用 ai.88531.cn/v1 的 mimo-v2.6-flash
|
||||
if (!result.apiBaseUrl) {
|
||||
chrome.storage.local.set({ apiBaseUrl: DEFAULT_AI.apiBaseUrl });
|
||||
}
|
||||
if (!result.modelName) {
|
||||
chrome.storage.local.set({ modelName: DEFAULT_AI.model });
|
||||
}
|
||||
if (!result.aiPrompt) {
|
||||
const defaultPrompt = '根据笔记内容来生成符合笔记的评论。最后只需要输出评论内容，不要输出你的推理，不要输出任何与评论无关的内容，不要输出#相关的内容，不要输出笔记关键词相关的内容。评论不要超过300字。这个是小红书的笔记内容：\n\n{{content}}\n\n';
|
||||
chrome.storage.local.set({ aiPrompt: defaultPrompt });
|
||||
}
|
||||
});
|
||||
|
||||
// 授权系统：登记安装（计入后台安装统计）+ 初始化试用期 + 拉公告
|
||||
authOnInstalled();
|
||||
});
|
||||
|
||||
// 浏览器启动 / 服务工作线程重启时恢复定时任务
|
||||
chrome.runtime.onStartup?.addListener?.(() => {
|
||||
setupAuthAlarms();
|
||||
});
|
||||
|
||||
// ===== CardAuth 授权系统 =====
|
||||
|
||||
// 统一请求封装（CardAuth 统一响应: {code, message, data}）
|
||||
async function authFetch(path, { method = 'GET', body = null, needKey = false } = {}) {
|
||||
const headers = { 'Content-Type': 'application/json' };
|
||||
if (needKey) headers['X-Api-Key'] = AUTH_API_KEY;
|
||||
|
||||
const response = await fetch(AUTH_BASE + path, {
|
||||
method,
|
||||
headers,
|
||||
body: body ? JSON.stringify(body) : undefined
|
||||
});
|
||||
|
||||
let json = {};
|
||||
try {
|
||||
json = await response.json();
|
||||
} catch (error) {
|
||||
json = {};
|
||||
}
|
||||
return {
|
||||
ok: response.ok && json.code === 200,
|
||||
status: response.status,
|
||||
code: json.code,
|
||||
message: json.message || `HTTP ${response.status}`,
|
||||
data: json.data
|
||||
};
|
||||
}
|
||||
|
||||
// 获取/生成设备指纹（授权绑定与安装统计的唯一标识）
|
||||
async function getDeviceId() {
|
||||
const result = await chrome.storage.local.get(['deviceId']);
|
||||
if (result.deviceId) return result.deviceId;
|
||||
const deviceId = 'EXT-' + crypto.randomUUID();
|
||||
await chrome.storage.local.set({ deviceId });
|
||||
console.log('已生成设备指纹:', deviceId);
|
||||
return deviceId;
|
||||
}
|
||||
|
||||
// 安装初始化：试用期起点 + 设备指纹 + 安装上报 + 公告
|
||||
async function authOnInstalled() {
|
||||
try {
|
||||
const result = await chrome.storage.local.get(['trialStartedAt']);
|
||||
if (!result.trialStartedAt) {
|
||||
await chrome.storage.local.set({ trialStartedAt: Date.now() });
|
||||
console.log('试用期开始（1天）');
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('初始化试用期失败:', error);
|
||||
}
|
||||
|
||||
setupAuthAlarms();
|
||||
|
||||
try {
|
||||
await authCheckUpdate(); // 后台计入安装统计 + 检查更新
|
||||
} catch (error) {
|
||||
console.error('安装上报失败:', error);
|
||||
}
|
||||
|
||||
try {
|
||||
await authGetAnnouncements();
|
||||
} catch (error) {
|
||||
console.error('拉取公告失败:', error);
|
||||
}
|
||||
|
||||
// 已有卡密则静默复验
|
||||
try {
|
||||
const { licenseState } = await chrome.storage.local.get(['licenseState']);
|
||||
if (licenseState && licenseState.cardKey && licenseState.valid) {
|
||||
await authVerify(licenseState.cardKey);
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('启动时授权复验失败:', error);
|
||||
}
|
||||
}
|
||||
|
||||
// 注册定时任务：心跳/复验/公告/更新检查
|
||||
function setupAuthAlarms() {
|
||||
if (!chrome.alarms) return;
|
||||
chrome.alarms.create('authHeartbeat', { periodInMinutes: 30 });
|
||||
chrome.alarms.create('authVerify', { periodInMinutes: 10 });
|
||||
chrome.alarms.create('authAnnounce', { periodInMinutes: 60 });
|
||||
chrome.alarms.create('authUpdateCheck', { periodInMinutes: 60 });
|
||||
}
|
||||
|
||||
chrome.alarms?.onAlarm?.addListener(async (alarm) => {
|
||||
try {
|
||||
if (alarm.name === 'authHeartbeat') {
|
||||
await authHeartbeat();
|
||||
} else if (alarm.name === 'authVerify') {
|
||||
const { licenseState } = await chrome.storage.local.get(['licenseState']);
|
||||
if (licenseState && licenseState.cardKey && licenseState.valid) {
|
||||
await authVerify(licenseState.cardKey);
|
||||
}
|
||||
} else if (alarm.name === 'authAnnounce') {
|
||||
await authGetAnnouncements();
|
||||
} else if (alarm.name === 'authUpdateCheck') {
|
||||
await authCheckUpdate();
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('授权定时任务失败:', alarm.name, error);
|
||||
}
|
||||
});
|
||||
|
||||
// 安装上报 / 检查更新（后台「安装统计」自动记录）
|
||||
async function authCheckUpdate() {
|
||||
const machineId = await getDeviceId();
|
||||
const { licenseState } = await chrome.storage.local.get(['licenseState']);
|
||||
const body = {
|
||||
machine_id: machineId,
|
||||
client_version: CLIENT_VERSION,
|
||||
device_info: navigator.userAgent.includes('Edg/') ? 'Edge 插件' : 'Chrome 插件'
|
||||
};
|
||||
if (licenseState && licenseState.cardKey) body.card_key = licenseState.cardKey;
|
||||
|
||||
const res = await authFetch('/api/public/installations/check-update', {
|
||||
method: 'POST',
|
||||
body,
|
||||
needKey: true
|
||||
});
|
||||
|
||||
if (res.ok && res.data) {
|
||||
const updateInfo = {
|
||||
updateAvailable: !!res.data.update_available,
|
||||
updateRequired: !!res.data.update_required,
|
||||
latestVersion: res.data.latest_version || '',
|
||||
downloadUrl: res.data.download_url || '',
|
||||
changelog: res.data.changelog || '',
|
||||
isForce: !!res.data.is_force,
|
||||
checkedAt: Date.now()
|
||||
};
|
||||
await chrome.storage.local.set({ updateInfo });
|
||||
console.log('更新检查:', updateInfo.updateAvailable ? `发现 ${updateInfo.latestVersion}` : '已是最新');
|
||||
}
|
||||
return res;
|
||||
}
|
||||
|
||||
// 心跳：更新「最后上线」时间
|
||||
async function authHeartbeat() {
|
||||
const machineId = await getDeviceId();
|
||||
const { licenseState } = await chrome.storage.local.get(['licenseState']);
|
||||
const body = {
|
||||
machine_id: machineId,
|
||||
client_version: CLIENT_VERSION
|
||||
};
|
||||
if (licenseState && licenseState.cardKey) body.card_key = licenseState.cardKey;
|
||||
|
||||
return authFetch('/api/public/installations/heartbeat', {
|
||||
method: 'POST',
|
||||
body,
|
||||
needKey: true
|
||||
});
|
||||
}
|
||||
|
||||
// 授权验证：cardKey 存在则校验并更新缓存；首次调用即绑定设备
|
||||
async function authVerify(cardKey) {
|
||||
const key = String(cardKey || '').trim();
|
||||
if (!key) return { ok: false, message: '卡密不能为空' };
|
||||
|
||||
const machineId = await getDeviceId();
|
||||
const res = await authFetch('/api/public/verify', {
|
||||
method: 'POST',
|
||||
body: {
|
||||
card_key: key,
|
||||
machine_id: machineId,
|
||||
client_version: CLIENT_VERSION,
|
||||
device_info: '小红书评论插件 ' + CLIENT_VERSION
|
||||
},
|
||||
needKey: true
|
||||
});
|
||||
|
||||
if (!res.ok) {
|
||||
// 网络/服务异常时不清除现有授权，仅报错
|
||||
return { ok: false, message: res.message || '验证失败' };
|
||||
}
|
||||
|
||||
const data = res.data || {};
|
||||
const licenseState = {
|
||||
valid: !!data.valid,
|
||||
cardKey: key,
|
||||
message: data.message || '',
|
||||
expireTime: data.expire_time || '',
|
||||
isPermanent: !!data.is_permanent,
|
||||
remainingDays: data.remaining_days ?? null,
|
||||
boundAt: data.bound_at || '',
|
||||
verifiedAt: Date.now()
|
||||
};
|
||||
await chrome.storage.local.set({ licenseState });
|
||||
console.log('授权验证:', licenseState.valid ? '有效' : '无效', licenseState.message);
|
||||
return { ok: true, licenseState };
|
||||
}
|
||||
|
||||
// 拉取公告并缓存
|
||||
async function authGetAnnouncements() {
|
||||
const res = await authFetch('/api/public/announcements');
|
||||
if (res.ok && res.data) {
|
||||
const announcements = {
|
||||
list: res.data.list || [],
|
||||
popup: res.data.popup || [],
|
||||
fetchedAt: Date.now()
|
||||
};
|
||||
await chrome.storage.local.set({ announcements });
|
||||
return { ok: true, announcements };
|
||||
}
|
||||
return { ok: false, message: res.message || '拉取公告失败' };
|
||||
}
|
||||
|
||||
// 读取授权状态（valid授权 > 1天试用 > 锁定）
|
||||
async function authGetStatus() {
|
||||
const { licenseState, trialStartedAt } = await chrome.storage.local.get(['licenseState', 'trialStartedAt']);
|
||||
|
||||
if (licenseState && licenseState.valid) {
|
||||
return {
|
||||
state: 'authorized',
|
||||
license: {
|
||||
expireTime: licenseState.expireTime,
|
||||
isPermanent: licenseState.isPermanent,
|
||||
remainingDays: licenseState.remainingDays,
|
||||
cardKey: licenseState.cardKey
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
const start = trialStartedAt || Date.now();
|
||||
if (!trialStartedAt) {
|
||||
await chrome.storage.local.set({ trialStartedAt: start });
|
||||
}
|
||||
const remainingMs = start + TRIAL_MS - Date.now();
|
||||
if (remainingMs > 0) {
|
||||
return {
|
||||
state: 'trial',
|
||||
remainingMs,
|
||||
remainingHours: Math.max(1, Math.ceil(remainingMs / 3600000)),
|
||||
trialEndAt: start + TRIAL_MS
|
||||
};
|
||||
}
|
||||
return { state: 'locked' };
|
||||
}
|
||||
|
||||
// 创建购买订单（contact_info 带机器码 → 支付成功后系统自动发卡绑定）
|
||||
async function authCreateOrder(payType) {
|
||||
const pay = payType === 'wxpay' ? 'wxpay' : 'alipay';
|
||||
|
||||
// 取项目第一个套餐
|
||||
const typesRes = await authFetch(`/api/public/projects/${AUTH_PROJECT_ID}/card-types`);
|
||||
if (!typesRes.ok || !Array.isArray(typesRes.data) || typesRes.data.length === 0) {
|
||||
return { ok: false, message: '暂无在售套餐，请稍后再试' };
|
||||
}
|
||||
const cardType = typesRes.data[0];
|
||||
|
||||
const machineId = await getDeviceId();
|
||||
const res = await authFetch('/api/public/orders', {
|
||||
method: 'POST',
|
||||
body: {
|
||||
project_id: AUTH_PROJECT_ID,
|
||||
card_type_id: cardType.id,
|
||||
amount: Number(cardType.price),
|
||||
pay_type: pay,
|
||||
contact_info: `MACHINE:${machineId}`
|
||||
}
|
||||
});
|
||||
|
||||
if (!res.ok || !res.data) {
|
||||
return { ok: false, message: res.message || '创建订单失败' };
|
||||
}
|
||||
|
||||
// 记录进行中的订单，便于侧边栏恢复轮询
|
||||
const pendingOrder = {
|
||||
orderNo: res.data.order_no,
|
||||
payType: pay,
|
||||
amount: Number(cardType.price),
|
||||
packageName: cardType.name || '',
|
||||
createdAt: Date.now()
|
||||
};
|
||||
await chrome.storage.local.set({ pendingOrder });
|
||||
|
||||
return { ok: true, order: pendingOrder, payUrl: res.data.pay_url };
|
||||
}
|
||||
|
||||
// 查询订单（支付+自动发卡后返回 card_key；拿到卡密自动激活）
|
||||
async function authQueryOrder(orderNo) {
|
||||
const res = await authFetch(`/api/public/orders/query?order_no=${encodeURIComponent(orderNo)}`);
|
||||
if (!res.ok || !res.data) {
|
||||
return { ok: false, message: res.message || '订单查询失败' };
|
||||
}
|
||||
|
||||
const data = res.data;
|
||||
const result = {
|
||||
ok: true,
|
||||
paid: data.status === 'paid' || !!data.card_key,
|
||||
status: data.status,
|
||||
cardKey: data.card_key || '',
|
||||
expireTime: data.expire_time || ''
|
||||
};
|
||||
|
||||
// 自动发卡成功 → 自动激活绑定本机
|
||||
if (result.cardKey) {
|
||||
const verifyRes = await authVerify(result.cardKey);
|
||||
result.activated = !!(verifyRes.ok && verifyRes.licenseState && verifyRes.licenseState.valid);
|
||||
result.licenseState = verifyRes.licenseState;
|
||||
if (result.activated) {
|
||||
await chrome.storage.local.remove(['pendingOrder']);
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
// 监听来自内容脚本的消息
|
||||
chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
|
||||
if (request.action === 'getCurrentNote') {
|
||||
handleGetCurrentNote(request, sendResponse);
|
||||
return true; // 保持消息通道开放
|
||||
}
|
||||
|
||||
if (request.action === 'saveNote') {
|
||||
handleSaveNote(request.noteData, sendResponse);
|
||||
return true; // 保持消息通道开放
|
||||
}
|
||||
|
||||
if (request.action === 'callAiApi') {
|
||||
handleCallAiApi(request, sendResponse);
|
||||
return true; // 保持消息通道开放
|
||||
}
|
||||
|
||||
if (request.action === 'getNotes') {
|
||||
handleGetNotes(sendResponse);
|
||||
return true;
|
||||
}
|
||||
|
||||
if (request.action === 'deleteNote') {
|
||||
handleDeleteNote(request.noteId, sendResponse);
|
||||
return true;
|
||||
}
|
||||
|
||||
// ===== 授权系统消息 =====
|
||||
if (request.action === 'authStatus') {
|
||||
authGetStatus().then((status) => sendResponse({ ok: true, status }));
|
||||
return true;
|
||||
}
|
||||
|
||||
if (request.action === 'authActivate') {
|
||||
authVerify(request.cardKey).then((res) => sendResponse(res));
|
||||
return true;
|
||||
}
|
||||
|
||||
if (request.action === 'authCreateOrder') {
|
||||
authCreateOrder(request.payType).then((res) => sendResponse(res));
|
||||
return true;
|
||||
}
|
||||
|
||||
if (request.action === 'authQueryOrder') {
|
||||
authQueryOrder(request.orderNo).then((res) => sendResponse(res));
|
||||
return true;
|
||||
}
|
||||
|
||||
if (request.action === 'authGetAnnouncements') {
|
||||
authGetAnnouncements().then((res) => sendResponse(res)).catch((error) => {
|
||||
sendResponse({ ok: false, message: error.message });
|
||||
});
|
||||
return true;
|
||||
}
|
||||
|
||||
if (request.action === 'authCheckUpdate') {
|
||||
authCheckUpdate()
|
||||
.then(async () => {
|
||||
const { updateInfo } = await chrome.storage.local.get(['updateInfo']);
|
||||
sendResponse({ ok: true, updateInfo: updateInfo || null });
|
||||
})
|
||||
.catch((error) => sendResponse({ ok: false, message: error.message }));
|
||||
return true;
|
||||
}
|
||||
});
|
||||
|
||||
// 调用自定义模型（OpenAI 兼容接口），由后台执行以规避页面 CORS / 混合内容限制
|
||||
async function handleCallAiApi(request, sendResponse) {
|
||||
const apiKey = String(request.apiKey || '').trim();
|
||||
if (!apiKey) {
|
||||
sendResponse({ ok: false, error: '未配置API密钥' });
|
||||
return;
|
||||
}
|
||||
|
||||
const url = buildChatCompletionsUrl(request.baseUrl);
|
||||
const model = String(request.model || '').trim() || DEFAULT_AI.model;
|
||||
const prompt = String(request.prompt || '');
|
||||
const maxTokens = Number(request.maxTokens) > 0 ? Number(request.maxTokens) : 300;
|
||||
const timeoutMs = Number(request.timeoutMs) > 0 ? Number(request.timeoutMs) : 90000;
|
||||
|
||||
const controller = new AbortController();
|
||||
const timer = setTimeout(() => controller.abort(), timeoutMs);
|
||||
|
||||
try {
|
||||
const body = buildChatBody(model, prompt, maxTokens);
|
||||
|
||||
const sendOnce = async (payload) => {
|
||||
const response = await fetch(url, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'Authorization': `Bearer ${apiKey}`
|
||||
},
|
||||
body: JSON.stringify(payload),
|
||||
signal: controller.signal
|
||||
});
|
||||
let data = {};
|
||||
try {
|
||||
data = await response.json();
|
||||
} catch (parseError) {
|
||||
data = {};
|
||||
}
|
||||
return { response: response, data: data };
|
||||
};
|
||||
|
||||
let result = await sendOnce(body);
|
||||
|
||||
// 400/422：端点不认某些参数，按报错修正后重试一次
|
||||
if (!result.response.ok && (result.response.status === 400 || result.response.status === 422)) {
|
||||
const message = (result.data && result.data.error && result.data.error.message) || '';
|
||||
let repaired = false;
|
||||
|
||||
if (body.thinking && /thinking|unknown|unrecognized|extra|invalid|not allowed|parameter/i.test(message)) {
|
||||
delete body.thinking;
|
||||
repaired = true;
|
||||
}
|
||||
if (body.max_tokens && /max_tokens/i.test(message) && /max_completion_tokens|unknown|unrecognized|invalid|not allowed/i.test(message)) {
|
||||
body.max_completion_tokens = body.max_tokens;
|
||||
delete body.max_tokens;
|
||||
repaired = true;
|
||||
}
|
||||
|
||||
if (repaired) {
|
||||
console.log('参数不被端点接受，修正后重试:', message);
|
||||
result = await sendOnce(body);
|
||||
}
|
||||
}
|
||||
|
||||
if (!result.response.ok) {
|
||||
sendResponse({
|
||||
ok: false,
|
||||
status: result.response.status,
|
||||
error: (result.data && result.data.error && result.data.error.message) || `HTTP ${result.response.status}`,
|
||||
url: url,
|
||||
model: model
|
||||
});
|
||||
return;
|
||||
}
|
||||
|
||||
let content = extractChatContent(result.data);
|
||||
|
||||
// 正文为空：基本是 max_tokens 被思考过程吃掉 → 提高上限重试一次
|
||||
if (!content) {
|
||||
const choice = result.data && Array.isArray(result.data.choices) ? result.data.choices[0] : null;
|
||||
const truncated = !!(choice && choice.finish_reason === 'length');
|
||||
const hasReasoning = !!(choice && choice.message && typeof choice.message.reasoning_content === 'string' && choice.message.reasoning_content.length > 0);
|
||||
|
||||
if (truncated || hasReasoning) {
|
||||
body.max_tokens = Math.max(4096, maxTokens * 4);
|
||||
console.log('正文为空，提高 max_tokens 到', body.max_tokens, '重试');
|
||||
result = await sendOnce(body);
|
||||
if (result.response.ok) {
|
||||
content = extractChatContent(result.data);
|
||||
}
|
||||
}
|
||||
|
||||
if (!content) {
|
||||
sendResponse({
|
||||
ok: false,
|
||||
error: hasReasoning
|
||||
? '模型只返回了思考过程，没有返回正文（可提高 max_tokens 或换非思考型模型）'
|
||||
: 'AI返回的内容为空',
|
||||
url: url,
|
||||
model: model
|
||||
});
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
sendResponse({ ok: true, content: content, url: url, model: model });
|
||||
} catch (error) {
|
||||
sendResponse({
|
||||
ok: false,
|
||||
error: error.name === 'AbortError' ? `请求超时（${Math.round(timeoutMs / 1000)}秒）` : error.message,
|
||||
url: url,
|
||||
model: model
|
||||
});
|
||||
} finally {
|
||||
clearTimeout(timer);
|
||||
}
|
||||
}
|
||||
|
||||
async function handleGetCurrentNote(request, sendResponse) {
|
||||
try {
|
||||
sendResponse({
|
||||
success: true,
|
||||
message: '获取当前笔记成功'
|
||||
});
|
||||
} catch (error) {
|
||||
console.error('获取当前笔记失败:', error);
|
||||
sendResponse({ success: false, error: error.message });
|
||||
}
|
||||
}
|
||||
|
||||
async function handleSaveNote(request, sendResponse) {
|
||||
try {
|
||||
sendResponse({ success: true, message: '保存笔记成功' });
|
||||
} catch (error) {
|
||||
console.error('保存笔记失败:', error);
|
||||
sendResponse({ success: false, error: error.message });
|
||||
}
|
||||
}
|
||||
|
||||
async function handleGetNotes(sendResponse) {
|
||||
try {
|
||||
sendResponse({
|
||||
success: true,
|
||||
notes: [],
|
||||
tags: []
|
||||
});
|
||||
} catch (error) {
|
||||
console.error('获取笔记失败:', error);
|
||||
sendResponse({ success: false, error: error.message });
|
||||
}
|
||||
}
|
||||
|
||||
async function handleDeleteNote(request, sendResponse) {
|
||||
try {
|
||||
sendResponse({ success: true, message: '删除笔记成功' });
|
||||
} catch (error) {
|
||||
console.error('删除笔记失败:', error);
|
||||
sendResponse({ success: false, error: error.message });
|
||||
}
|
||||
}
|
||||
|
||||
// 存储变化监听
|
||||
chrome.storage.onChanged.addListener((changes, namespace) => {
|
||||
if (namespace === 'local') {
|
||||
console.log('存储数据发生变化:', changes);
|
||||
}
|
||||
});
|
||||
|
||||
// 标签页更新监听
|
||||
chrome.tabs.onUpdated.addListener((tabId, changeInfo, tab) => {
|
||||
if (changeInfo.status === 'complete' && tab.url && tab.url.includes('xiaohongshu.com')) {
|
||||
// 当小红书页面加载完成时，可以在这里执行一些操作
|
||||
console.log('小红书页面加载完成');
|
||||
}
|
||||
});
|
||||
Reference in new issue
Block a user