导出 poe.com 聊天到 Notion，支持图片上传(PicList)+隐私开关+单条导出
// ==UserScript==
// @name Poe to Notion Exporter (with PicList)
// @namespace https://github.com/wyih/poe-to-notion
// @version 0.1
// @description 导出 poe.com 聊天到 Notion，支持图片上传(PicList)+隐私开关+单条导出
// @author Wyih
// @match https://poe.com/*
// @connect api.notion.com
// @connect 127.0.0.1
// @connect *
// @grant GM_xmlhttpRequest
// @grant GM_setValue
// @grant GM_getValue
// @grant GM_registerMenuCommand
// @grant GM_addStyle
// @license MIT
// ==/UserScript==
(function () {
'use strict';
// ============ 基本配置 ============
const PICLIST_URL = "http://127.0.0.1:36677/upload";
const ASSET_PLACEHOLDER_PREFIX = "PICLIST_WAITING::";
const MAX_TEXT_LENGTH = 2000;
// 简单语言判断
const isZH = (navigator.language || navigator.userLanguage || '').startsWith('zh');
const LABEL = isZH ? {
saveAll: '📥 保存到 Notion',
processing: '🕵️ 处理中...',
saving: '💾 保存中...',
done: '✅ 已保存',
error: '❌ 出错',
user: 'User',
bot: 'Assistant',
privacyOn: '👁️',
privacyOff: '🚫',
singleExportTitle: '仅导出此条对话（含紧随其后的回复）',
configMenu: '⚙️ 设置 Notion Token/DB',
privacyHint: '点击切换：是否导出此条内容',
} : {
saveAll: '📥 Save to Notion',
processing: '🕵️ Processing...',
saving: '💾 Saving...',
done: '✅ Saved',
error: '❌ Error',
user: 'User',
bot: 'Assistant',
privacyOn: '👁️',
privacyOff: '🚫',
singleExportTitle: 'Export only this message (and following reply)',
configMenu: '⚙️ Config Notion Token/DB',
privacyHint: 'Toggle: export or skip this message',
};
// ============ 0. PicList 心跳检测 ============
function checkPicListConnection() {
GM_xmlhttpRequest({
method: "GET",
url: "http://127.0.0.1:36677/heartbeat",
timeout: 2000,
onload: (res) => {
if (res.status === 200) console.log("✅ PicList 连接正常");
},
onerror: () => console.warn("❌ PicList 未连接（可忽略，仅影响图片上传）")
});
}
setTimeout(checkPicListConnection, 3000);
// ============ 1. Notion 配置 ============
function getConfig() {
return {
token: GM_getValue('poe_notion_token', ''),
dbId: GM_getValue('poe_notion_db_id', '')
};
}
function promptConfig() {
const token = prompt('请输入 Notion Integration Secret:', GM_getValue('poe_notion_token', ''));
if (token) {
const dbId = prompt('请输入 Notion Database ID:', GM_getValue('poe_notion_db_id', ''));
if (dbId) {
GM_setValue('poe_notion_token', token);
GM_setValue('poe_notion_db_id', dbId);
alert('配置已保存 ✅');
}
}
}
GM_registerMenuCommand(LABEL.configMenu, promptConfig);
// ============ 2. 样式 ============
GM_addStyle(`
#poe-notion-saver-btn {
position: fixed;
bottom: 20px;
right: 20px;
z-index: 9999;
background-color: #0066CC;
color: white;
border: none;
border-radius: 6px;
padding: 10px 16px;
cursor: pointer;
box-shadow: 0 4px 12px rgba(0,0,0,0.15);
font-family: sans-serif;
font-weight: 600;
font-size: 14px;
transition: all 0.2s;
}
#poe-notion-saver-btn:hover { background-color: #0052a3; transform: translateY(-2px); }
#poe-notion-saver-btn.loading { background-color: #666; cursor: wait; }
.poe-message-bubble {
position: relative; /* 确保绝对定位的工具条以气泡为参照 */
}
.poe-tool-group {
z-index: 9500;
display: flex;
gap: 6px;
opacity: 0;
transition: opacity 0.15s ease-in-out;
background: #fff;
padding: 4px 6px;
border-radius: 999px;
box-shadow: 0 2px 5px rgba(0,0,0,0.18);
border: 1px solid rgba(0,0,0,0.06);
position: absolute;
top: -10px; /* 稍微顶出气泡一点 */
right: 8px; /* 一律贴右上角 */
}
.poe-message-bubble:hover .poe-tool-group {
opacity: 1;
}
.poe-tool-group .poe-icon-btn {
cursor: pointer;
font-size: 16px;
line-height: 24px;
user-select: none;
width: 26px;
height: 26px;
text-align: center;
border-radius: 50%;
transition: background 0.15s, color 0.15s, transform 0.1s;
display: flex;
align-items: center;
justify-content: center;
color: #555;
}
.poe-tool-group .poe-icon-btn:hover {
background: rgba(0,0,0,0.06);
color: #000;
transform: translateY(-1px);
}
.poe-tool-group .poe-privacy-toggle[data-skip="true"] {
color: #d93025;
background: #fce8e6;
}
.poe-icon-btn.processing { cursor: wait; color: #1a73e8; background: #e8f0fe; }
.poe-icon-btn.success { color: #188038 !important; background: #e6f4ea; }
.poe-icon-btn.error { color: #d93025 !important; background: #fce8e6; }
.poe-tool-group .poe-icon-btn {
cursor: pointer;
font-size: 16px;
line-height: 24px;
user-select: none;
width: 26px;
height: 26px;
text-align: center;
border-radius: 50%;
transition: background 0.2s;
display: flex;
align-items: center;
justify-content: center;
color: #555;
}
.poe-tool-group .poe-icon-btn:hover {
background: rgba(0,0,0,0.08);
color: #000;
}
.poe-tool-group .poe-privacy-toggle[data-skip="true"] {
color: #d93025;
background: #fce8e6;
}
.poe-icon-btn.processing { cursor: wait; color: #1a73e8; background: #e8f0fe; }
.poe-icon-btn.processing span { display: block; animation: spin 1s linear infinite; }
.poe-icon-btn.success { color: #188038 !important; background: #e6f4ea; }
.poe-icon-btn.error { color: #d93025 !important; background: #fce8e6; }
@keyframes spin { 0% { transform: rotate(0deg); } 100% { transform: rotate(360deg); } }
`);
// ============ 3. DOM 工具 & UI 注入 ============
function injectMessageTools() {
// 找到所有 Markdown 容器，再往上找“气泡”
const markdownContainers = document.querySelectorAll('[class^="Markdown_markdownContainer"]');
markdownContainers.forEach(container => {
let bubble = container;
while (bubble && !bubble.className.includes('MessageBubble')) {
bubble = bubble.parentElement;
}
if (!bubble) return;
// 标记方便样式控制
if (!bubble.classList.contains('poe-message-bubble')) {
bubble.classList.add('poe-message-bubble');
}
// 已经有工具栏就跳过
if (bubble.querySelector('.poe-tool-group')) return;
const group = document.createElement('div');
group.className = 'poe-tool-group';
// 隐私按钮
const privacyBtn = document.createElement('div');
privacyBtn.className = 'poe-icon-btn poe-privacy-toggle';
privacyBtn.title = LABEL.privacyHint;
privacyBtn.setAttribute('data-skip', 'false');
const privacyIcon = document.createElement('span');
privacyIcon.textContent = LABEL.privacyOn;
privacyBtn.appendChild(privacyIcon);
privacyBtn.onclick = (e) => {
e.stopPropagation();
const isSkipping = privacyBtn.getAttribute('data-skip') === 'true';
if (isSkipping) {
privacyBtn.setAttribute('data-skip', 'false');
privacyIcon.textContent = LABEL.privacyOn;
bubble.setAttribute('data-privacy-skip', 'false');
} else {
privacyBtn.setAttribute('data-skip', 'true');
privacyIcon.textContent = LABEL.privacyOff;
bubble.setAttribute('data-privacy-skip', 'true');
}
};
// 单条导出按钮
const singleBtn = document.createElement('div');
singleBtn.className = 'poe-icon-btn';
singleBtn.title = LABEL.singleExportTitle;
const exportIcon = document.createElement('span');
exportIcon.textContent = '📤';
singleBtn.appendChild(exportIcon);
singleBtn.onclick = (e) => {
e.stopPropagation();
handleSingleExport(bubble, singleBtn, exportIcon);
};
group.appendChild(privacyBtn);
group.appendChild(singleBtn);
// 插在 bubble 顶部
bubble.insertBefore(group, bubble.firstChild);
});
}
// ============ 4. 资源处理（PicList 上传） ============
function convertBlobImageToBuffer(blobUrl) {
return new Promise((resolve, reject) => {
const img = document.querySelector(`img[src="${blobUrl}"]`);
if (!img || !img.complete || img.naturalWidth === 0) return reject("图片加载失败");
try {
const canvas = document.createElement('canvas');
canvas.width = img.naturalWidth;
canvas.height = img.naturalHeight;
canvas.getContext('2d').drawImage(img, 0, 0);
canvas.toBlob(b => {
if (!b) return reject("Canvas 失败");
b.arrayBuffer().then(buf => resolve({ buffer: buf, type: b.type }));
}, 'image/png');
} catch (e) { reject(e.message); }
});
}
function fetchAssetAsArrayBuffer(url) {
return new Promise((resolve, reject) => {
if (url.startsWith('blob:')) {
convertBlobImageToBuffer(url)
.then(resolve)
.catch(() => {
GM_xmlhttpRequest({
method: "GET",
url,
responseType: 'arraybuffer',
onload: r => {
if (r.status === 200) {
resolve({
buffer: r.response,
type: 'application/octet-stream'
});
} else reject("blob fetch fail");
}
});
});
} else {
GM_xmlhttpRequest({
method: "GET",
url,
responseType: 'arraybuffer',
onload: r => {
if (r.status === 200) {
const m = r.responseHeaders.match(/content-type:\s*(.*)/i);
resolve({ buffer: r.response, type: m ? m[1] : undefined });
} else reject("http fetch fail");
}
});
}
});
}
function uploadToPicList(obj, filename) {
return new Promise((resolve, reject) => {
if (!obj.buffer) return reject("空文件");
let finalFilename = filename.split('?')[0];
const mime = (obj.type || '').split(';')[0].trim().toLowerCase();
if (!finalFilename.includes('.') || finalFilename.length - finalFilename.lastIndexOf('.') > 6) {
const mimeMap = {
'application/pdf': '.pdf',
'application/msword': '.doc',
'application/vnd.openxmlformats-officedocument.wordprocessingml.document': '.docx',
'image/png': '.png',
'image/jpeg': '.jpg',
'image/webp': '.webp'
};
if (mimeMap[mime]) finalFilename += mimeMap[mime];
}
const boundary = "----PoeNotionBoundary" + Math.random().toString(36).substring(2);
const preData =
`--${boundary}\r\n` +
`Content-Disposition: form-data; name="file"; filename="${finalFilename.replace(/"/g, '')}"\r\n` +
`Content-Type: ${mime || 'application/octet-stream'}\r\n\r\n`;
const blob = new Blob([preData, obj.buffer, `\r\n--${boundary}--\r\n`]);
GM_xmlhttpRequest({
method: "POST",
url: PICLIST_URL,
headers: { "Content-Type": `multipart/form-data; boundary=${boundary}` },
data: blob,
onload: (res) => {
try {
const r = JSON.parse(res.responseText);
if (r.success && r.result && r.result[0]) resolve(r.result[0]);
else reject(r.message || 'PicList error');
} catch (e) { reject(e.message); }
},
onerror: () => reject("PicList 网络错误")
});
});
}
async function processAssets(blocks, statusCallback) {
const tasks = [];
const map = new Map();
blocks.forEach((b, i) => {
let urlObj = null;
if (b.type === 'image' && b.image?.external?.url?.startsWith(ASSET_PLACEHOLDER_PREFIX)) {
urlObj = b.image.external;
} else if (b.type === 'file' && b.file?.external?.url?.startsWith(ASSET_PLACEHOLDER_PREFIX)) {
urlObj = b.file.external;
}
if (!urlObj) return;
const [_, name, realUrl] = urlObj.url.split('::');
if (realUrl.startsWith('blob:') && b.type === 'file') {
// 文件 + blob：放弃上传，改为文本提示
b.type = "paragraph";
b.paragraph = {
rich_text: [{
type: "text",
text: { content: `📄 [本地文件未上传] ${name}` },
annotations: { color: "gray", italic: true }
}]
};
delete b.file;
return;
}
const task = fetchAssetAsArrayBuffer(realUrl)
.then(buf => uploadToPicList(buf, name))
.then(u => ({ i, url: u, name, ok: true }))
.catch(e => ({ i, err: e, name, ok: false }));
tasks.push(task);
map.set(i, b);
});
if (tasks.length) {
statusCallback(`⏳ Uploading ${tasks.length}...`);
const res = await Promise.all(tasks);
res.forEach(r => {
const blk = map.get(r.i);
if (!blk) return;
if (r.ok) {
if (blk.type === 'image') {
blk.image.external.url = r.url;
} else if (blk.type === 'file') {
blk.file.external.url = r.url;
blk.file.name = r.name || "File";
}
} else {
console.error('Upload fail', r.name, r.err);
blk.type = "paragraph";
blk.paragraph = {
rich_text: [{
type: "text",
text: { content: `⚠️ Upload Failed: ${r.name}` },
annotations: { color: "red" }
}]
};
delete blk.image;
delete blk.file;
}
});
}
return blocks;
}
// ============ 5. DOM→Notion 解析 ============
const NOTION_LANGUAGES = new Set([
"bash", "c", "c++", "css", "go", "html", "java", "javascript",
"json", "kotlin", "markdown", "php", "python", "ruby", "rust",
"shell", "sql", "swift", "typescript", "yaml", "r", "plain text"
]);
function mapLanguageToNotion(lang) {
if (!lang) return "plain text";
lang = lang.toLowerCase().trim();
if (lang === "js") return "javascript";
if (lang === "py") return "python";
if (NOTION_LANGUAGES.has(lang)) return lang;
return "plain text";
}
function detectLanguageFromPre(preNode) {
const code = preNode.querySelector('code');
if (code && code.className) {
const m = code.className.match(/language-([\w-]+)/);
if (m) return mapLanguageToNotion(m[1]);
}
return "plain text";
}
function splitTextSafe(text) {
const chunks = [];
let remaining = text;
while (remaining.length > 0) {
if (remaining.length <= MAX_TEXT_LENGTH) {
chunks.push(remaining);
break;
}
let idx = remaining.lastIndexOf('\n', MAX_TEXT_LENGTH - 1);
if (idx === -1) idx = MAX_TEXT_LENGTH;
else idx += 1;
chunks.push(remaining.slice(0, idx));
remaining = remaining.slice(idx);
}
return chunks;
}
function parseInlineNodes(nodes) {
const rt = [];
function tr(n, s = {}) {
if (n.nodeType === 3) {
const full = n.textContent;
if (!full) return;
for (let i = 0; i < full.length; i += MAX_TEXT_LENGTH) {
rt.push({
type: "text",
text: { content: full.slice(i, i + MAX_TEXT_LENGTH), link: s.link },
annotations: {
bold: !!s.bold,
italic: !!s.italic,
code: !!s.code,
color: "default"
}
});
}
} else if (n.nodeType === 1) {
const latex = n.getAttribute('data-latex-source') || n.getAttribute('data-math');
if (latex) {
rt.push({
type: "equation",
equation: { expression: latex.trim() }
});
return;
}
const ns = { ...s };
if (['B', 'STRONG'].includes(n.tagName)) ns.bold = true;
if (['I', 'EM'].includes(n.tagName)) ns.italic = true;
if (n.tagName === 'CODE') ns.code = true;
if (n.tagName === 'A') ns.link = { url: n.href };
n.childNodes.forEach(c => tr(c, ns));
}
}
nodes.forEach(n => tr(n));
return rt;
}
function processNodesToBlocks(nodes) {
const blocks = [];
const buf = [];
const flush = () => {
if (!buf.length) return;
const rt = parseInlineNodes(buf);
if (rt.length) {
blocks.push({
object: "block",
type: "paragraph",
paragraph: { rich_text: rt }
});
}
buf.length = 0;
};
const fileExtRegex = /\.(pdf|zip|docx?|xlsx?|pptx?|csv|txt|md|html?|rar|7z|tar|gz|iso|exe|apk|dmg|json|xml|epub|R|Rmd|qmd)(\?|$)/i;
Array.from(nodes).forEach(n => {
if (['SCRIPT', 'STYLE', 'SVG'].includes(n.nodeName)) return;
const isElement = n.nodeType === 1;
// 块级公式
if (isElement) {
const isMathTag = n.hasAttribute('data-math') || n.hasAttribute('data-latex-source');
const isBlockLayout =
n.tagName === 'DIV' ||
n.classList.contains('math-block') ||
n.classList.contains('katex-display');
if (isMathTag && isBlockLayout) {
const latex = n.getAttribute('data-latex-source') || n.getAttribute('data-math');
if (latex) {
flush();
blocks.push({
object: "block",
type: "equation",
equation: { expression: latex.trim() }
});
return;
}
}
}
// 行内缓冲
if (
n.nodeType === 3 ||
['B', 'I', 'CODE', 'SPAN', 'A', 'STRONG', 'EM'].includes(n.nodeName)
) {
if (
isElement &&
n.tagName === 'A' &&
(n.hasAttribute('download') ||
(n.href && (n.href.includes('blob:') || fileExtRegex.test(n.href))))
) {
flush();
const fn = (n.innerText || 'file').trim();
blocks.push({
object: "block",
type: "file",
file: {
type: "external",
name: fn.slice(0, 60),
external: { url: `${ASSET_PLACEHOLDER_PREFIX}${fn}::${n.href}` }
}
});
return;
}
buf.push(n);
return;
}
if (isElement) {
flush();
const t = n.tagName;
if (t === 'P') {
blocks.push(...processNodesToBlocks(n.childNodes));
} else if (t === 'IMG') {
if (n.src) {
blocks.push({
object: "block",
type: "image",
image: {
type: "external",
external: {
url: `${ASSET_PLACEHOLDER_PREFIX}image.png::${n.src}`
}
}
});
}
} else if (t === 'PRE') {
const codeText = n.textContent || '';
const lang = detectLanguageFromPre(n);
const chunks = splitTextSafe(codeText);
const rich = chunks.map(c => ({
type: "text",
text: { content: c }
}));
blocks.push({
object: "block",
type: "code",
code: { rich_text: rich, language: lang }
});
} else if (/^H[1-6]$/.test(t)) {
const level = t[1] < 4 ? t[1] : 3;
blocks.push({
object: "block",
type: `heading_${level}`,
[`heading_${level}`]: { rich_text: parseInlineNodes(n.childNodes) }
});
} else if (t === 'BLOCKQUOTE') {
blocks.push({
object: "block",
type: "quote",
quote: { rich_text: parseInlineNodes(n.childNodes) }
});
} else if (t === 'UL' || t === 'OL') {
const tp = t === 'UL' ? 'bulleted_list_item' : 'numbered_list_item';
Array.from(n.children).forEach(li => {
if (li.tagName !== 'LI') return;
const liBlocks = processNodesToBlocks(li.childNodes);
if (!liBlocks.length) return;
let richText;
let children = [];
const first = liBlocks[0];
if (first.type === 'paragraph' && first.paragraph?.rich_text?.length) {
richText = first.paragraph.rich_text;
children = liBlocks.slice(1);
} else {
richText = parseInlineNodes(li.childNodes);
children = liBlocks;
}
const listBlock = {
object: "block",
type: tp,
[tp]: { rich_text: richText }
};
if (children.length) {
listBlock[tp].children = children;
}
blocks.push(listBlock);
});
} else if (t === 'TABLE') {
const rows = Array.from(n.querySelectorAll('tr'));
if (rows.length) {
const tb = {
object: "block",
type: "table",
table: { table_width: 1, children: [] }
};
let max = 0;
rows.forEach(r => {
const cs = Array.from(r.querySelectorAll('td,th'));
max = Math.max(max, cs.length);
tb.table.children.push({
object: "block",
type: "table_row",
table_row: {
cells: cs.map(c => parseInlineNodes(c.childNodes))
}
});
});
tb.table.table_width = max;
blocks.push(tb);
}
} else {
blocks.push(...processNodesToBlocks(n.childNodes));
}
}
});
flush();
return blocks;
}
// ============ 6. 从 Poe 抓取消息 → Notion blocks ============
function getRoleFromBubble(bubble) {
// 沿用你原 exporter 的逻辑：看 leftSide/rightSide
let p = bubble;
while (p && p !== document.body) {
if (p.className && p.className.includes('leftSide')) return LABEL.bot;
if (p.className && p.className.includes('rightSide')) return LABEL.user;
p = p.parentElement;
}
// 兜底：如果包含 right/left 文本
const cls = bubble.className || '';
if (cls.includes('right')) return LABEL.user;
return LABEL.bot;
}
function getAllMessageBubbles() {
const list = [];
const markdownContainers = document.querySelectorAll('[class^="Markdown_markdownContainer"]');
markdownContainers.forEach(container => {
let bubble = container;
while (bubble && !bubble.className.includes('MessageBubble')) {
bubble = bubble.parentElement;
}
if (!bubble) return;
if (!list.includes(bubble)) list.push(bubble);
});
return list;
}
function getChatBlocksFromBubbles(targetBubbles = null) {
const bubbles = targetBubbles || getAllMessageBubbles();
const blocks = [];
bubbles.forEach(bubble => {
const skip = bubble.getAttribute('data-privacy-skip') === 'true';
const role = getRoleFromBubble(bubble);
// 隐私：直接放一个 callout + divider
if (skip) {
blocks.push({
object: "block",
type: "callout",
callout: {
rich_text: [{
type: "text",
text: {
content: (isZH
? `🚫 此 ${role} 内容已标记为隐私，未导出。`
: `🚫 This ${role} message is marked as private and not exported.`)
},
annotations: { color: "gray", italic: true }
}],
icon: { emoji: "🔒" },
color: "gray_background"
}
});
blocks.push({ object: "block", type: "divider", divider: {} });
return;
}
// 1) 角色标题（User / Assistant）
blocks.push({
object: "block",
type: "heading_3",
heading_3: {
rich_text: [{ type: "text", text: { content: role } }]
}
});
// 2) 文本部分（markdown）
const container = bubble.querySelector('[class^="Markdown_markdownContainer"]');
if (container) {
const clone = container.cloneNode(true);
// 防守：清理我们自己的工具条（虽然一般不在这里）
clone.querySelectorAll('.poe-tool-group').forEach(e => e.remove());
blocks.push(...processNodesToBlocks(clone.childNodes));
} else {
const text = bubble.innerText || '';
if (text.trim()) {
blocks.push({
object: "block",
type: "paragraph",
paragraph: {
rich_text: [{
type: "text",
text: { content: text.slice(0, MAX_TEXT_LENGTH) }
}]
}
});
}
}
// 3) markdown 内嵌图片（包括 GPT-Image / Seedream 生成的图）
// 说明：因为我们在 processNodesToBlocks 里把 SPAN 当作“行内节点”，
// 不会往下递归到 <img>，所以这里额外扫一次 markdown 里的 img。
const markdownImgs = bubble.querySelectorAll(
'.Markdown_markdownContainer__Tz3HQ img'
);
markdownImgs.forEach(img => {
const url = img.src;
if (!url) return;
let name = 'image.png';
try {
const u = new URL(url);
const pathname = u.pathname || '';
const base = pathname.split('/').pop() || '';
if (base) {
const qIdx = base.indexOf('.');
name = qIdx > -1 ? base.slice(0, qIdx) + base.slice(qIdx) : base;
}
} catch (_) {}
blocks.push({
object: "block",
type: "image",
image: {
type: "external",
external: {
url: `${ASSET_PLACEHOLDER_PREFIX}${name || 'image.png'}::${url}`
}
}
});
});
// 4) 附件图片（用户上传的 Attachments_attachments__x_H2Q）
const attachmentImgs = bubble.querySelectorAll('.Attachments_attachments__x_H2Q img');
attachmentImgs.forEach(img => {
const url = img.src;
if (!url) return;
let name = 'image.png';
try {
const u = new URL(url);
const pathname = u.pathname || '';
const base = pathname.split('/').pop() || '';
if (base) {
const qIdx = base.indexOf('.');
name = qIdx > -1 ? base.slice(0, qIdx) + base.slice(qIdx) : base;
}
} catch (_) {}
blocks.push({
object: "block",
type: "image",
image: {
type: "external",
external: {
url: `${ASSET_PLACEHOLDER_PREFIX}${name || 'image.png'}::${url}`
}
}
});
});
// 5) 每条气泡之后加 divider
blocks.push({ object: "block", type: "divider", divider: {} });
});
return blocks;
}
function getChatTitleFromFirstBubble() {
const bubbles = getAllMessageBubbles();
if (!bubbles.length) return 'Poe Chat';
const first = bubbles[0];
const text = (first.innerText || '').replace(/\s+/g, ' ').trim();
return text ? text.slice(0, 60) : 'Poe Chat';
}
// ============ 7. Notion 上传 ============
function appendBlocksBatch(pageId, blocks, token, statusCallback) {
if (!blocks.length) {
statusCallback(LABEL.done);
setTimeout(() => statusCallback(null), 2500);
return;
}
GM_xmlhttpRequest({
method: "PATCH",
url: `https://api.notion.com/v1/blocks/${pageId}/children`,
headers: {
"Authorization": `Bearer ${token}`,
"Content-Type": "application/json",
"Notion-Version": "2022-06-28"
},
data: JSON.stringify({
children: blocks.slice(0, 90)
}),
onload: (res) => {
if (res.status === 200) {
appendBlocksBatch(pageId, blocks.slice(90), token, statusCallback);
} else {
console.error(res.responseText);
statusCallback(LABEL.error);
}
},
onerror: () => statusCallback(LABEL.error)
});
}
function createPageAndUpload(title, blocks, token, dbId, statusCallback) {
GM_xmlhttpRequest({
method: "POST",
url: "https://api.notion.com/v1/pages",
headers: {
"Authorization": `Bearer ${token}`,
"Content-Type": "application/json",
"Notion-Version": "2022-06-28"
},
data: JSON.stringify({
parent: { database_id: dbId },
properties: {
"Name": { title: [{ text: { content: title } }] },
"Date": { date: { start: new Date().toISOString() } },
"URL": { url: location.href }
},
children: blocks.slice(0, 90)
}),
onload: (res) => {
if (res.status === 200) {
const pageId = JSON.parse(res.responseText).id;
appendBlocksBatch(pageId, blocks.slice(90), token, statusCallback);
} else {
console.error(res.responseText);
statusCallback(LABEL.error);
alert(res.responseText);
}
},
onerror: () => statusCallback(LABEL.error)
});
}
// ============ 8. 导出主逻辑 ============
async function executeExport(blocks, title, btnOrLabelUpdater, iconElem) {
const { token, dbId } = getConfig();
if (!token || !dbId) {
promptConfig();
return;
}
const isGlobalBtn = btnOrLabelUpdater && btnOrLabelUpdater.id === 'poe-notion-saver-btn';
const updateStatus = (msg) => {
if (!btnOrLabelUpdater) return;
if (btnOrLabelUpdater.classList && btnOrLabelUpdater.classList.contains('poe-icon-btn') && iconElem) {
if (msg && msg.includes('Saved')) {
btnOrLabelUpdater.classList.remove('processing');
btnOrLabelUpdater.classList.add('success');
iconElem.textContent = '✅';
setTimeout(() => {
btnOrLabelUpdater.classList.remove('success');
iconElem.textContent = '📤';
}, 2000);
} else if (msg && (msg.includes('Fail') || msg.includes('Error') || msg.includes('错'))) {
btnOrLabelUpdater.classList.remove('processing');
btnOrLabelUpdater.classList.add('error');
iconElem.textContent = '❌';
} else if (msg) {
btnOrLabelUpdater.classList.add('processing');
btnOrLabelUpdater.classList.remove('success', 'error');
iconElem.textContent = '⏳';
}
} else if (isGlobalBtn) {
if (msg === null) btnOrLabelUpdater.textContent = LABEL.saveAll;
else btnOrLabelUpdater.textContent = msg;
}
};
if (isGlobalBtn) {
btnOrLabelUpdater.classList.add('loading');
btnOrLabelUpdater.textContent = LABEL.processing;
} else if (btnOrLabelUpdater && iconElem) {
updateStatus('Processing...');
}
try {
blocks = await processAssets(blocks, updateStatus);
if (isGlobalBtn) btnOrLabelUpdater.textContent = LABEL.saving;
createPageAndUpload(title, blocks, token, dbId, updateStatus);
} catch (e) {
console.error(e);
if (isGlobalBtn) btnOrLabelUpdater.textContent = LABEL.error;
if (btnOrLabelUpdater && iconElem) updateStatus(LABEL.error);
alert(e.message || e);
} finally {
if (isGlobalBtn) btnOrLabelUpdater.classList.remove('loading');
}
}
function handleFullExport() {
const btn = document.getElementById('poe-notion-saver-btn');
const blocks = getChatBlocksFromBubbles(null);
executeExport(blocks, getChatTitleFromFirstBubble(), btn);
}
function handleSingleExport(bubble, iconBtn, iconElem) {
const bubbles = getAllMessageBubbles();
const idx = bubbles.indexOf(bubble);
const targets = [];
if (idx >= 0) {
targets.push(bubble);
// 如果下一条是对方的回复，也一起导出（类似 Gemini 逻辑）
if (idx + 1 < bubbles.length) {
const next = bubbles[idx + 1];
if (next.getAttribute('data-privacy-skip') !== 'true') {
targets.push(next);
}
}
} else {
targets.push(bubble);
}
const blocks = getChatBlocksFromBubbles(targets);
const title = (bubble.innerText || '').replace(/\s+/g, ' ').slice(0, 60) || getChatTitleFromFirstBubble();
executeExport(blocks, title, iconBtn, iconElem);
}
function tryInit() {
if (!document.getElementById('poe-notion-saver-btn')) {
const btn = document.createElement('button');
btn.id = 'poe-notion-saver-btn';
btn.textContent = LABEL.saveAll;
btn.onclick = handleFullExport;
document.body.appendChild(btn);
}
injectMessageTools();
}
setInterval(tryInit, 1500);
})();