diff --git a/backend/notes.py b/backend/notes.py index 2763cf2..eab55aa 100644 --- a/backend/notes.py +++ b/backend/notes.py @@ -48,6 +48,24 @@ def note_attachments(body: str) -> tuple[int, str | None]: return count, preview +def note_snippet(body: str, limit: int = 240) -> str: + lines = body.splitlines() + start = 0 + for i, line in enumerate(lines): + stripped = line.strip() + if not stripped: + continue + if stripped.startswith("# "): + start = i + 1 + continue + start = i + break + excerpt = "\n".join(lines[start:]).strip() + if len(excerpt) > limit: + excerpt = excerpt[:limit].rstrip() + "…" + return excerpt + + def note_path(username: str, note_id: str) -> str: return f"{username}/{note_id}.md" @@ -293,6 +311,7 @@ async def _attach_meta(notes: list[dict]) -> None: count, preview = 0, None note["attachments"] = count note["image"] = preview + note["snippet"] = note_snippet(body) note.pop("path", None) diff --git a/browser-extension/background.js b/browser-extension/background.js index 7a76082..e5554bd 100644 --- a/browser-extension/background.js +++ b/browser-extension/background.js @@ -24,6 +24,8 @@ browser.runtime.onMessage.addListener((msg, sender) => { return saveNote(msg); case 'upload_attachment': return uploadAttachment(msg); + case 'capture_full_page': + return captureFullPage(msg); default: return Promise.reject(new Error('Unknown message kind')); } @@ -93,6 +95,60 @@ async function uploadAttachment(msg) { return res.json(); } +const MAX_SLICES = 60; + +function sleep(ms) { + return new Promise(function (resolve) { setTimeout(resolve, ms); }); +} + +function loadImage(dataUrl) { + return new Promise(function (resolve, reject) { + const img = new Image(); + img.onload = function () { resolve(img); }; + img.onerror = function () { reject(new Error('Could not decode screenshot slice')); }; + img.src = dataUrl; + }); +} + +async function captureFullPage(msg) { + const tabId = msg.tabId; + const windowId = msg.windowId; + if (tabId == null) throw new Error('No active tab'); + + const dims = await browser.tabs.sendMessage(tabId, { kind: 'get_viewport' }); + const vw = dims.viewportWidth || 1; + const vh = dims.viewportHeight || 1; + const ph = dims.pageHeight || vh; + const dpr = dims.devicePixelRatio || 1; + + const sliceCount = Math.min(MAX_SLICES, Math.ceil(ph / vh)); + if (Math.ceil(ph / vh) > MAX_SLICES) { + throw new Error('Page too tall to capture (' + Math.ceil(ph / vh) + ' viewports)'); + } + + const parts = []; + for (let i = 0; i < sliceCount; i++) { + const y = Math.max(0, Math.min(i * vh, ph - vh)); + await browser.tabs.sendMessage(tabId, { kind: 'scroll_to', y: y }); + await sleep(250); + const dataUrl = await browser.tabs.captureVisibleTab(windowId, { format: 'png' }); + parts.push({ dataUrl: dataUrl, y: y }); + } + + await browser.tabs.sendMessage(tabId, { kind: 'scroll_to', y: dims.scrollY || 0 }).catch(() => {}); + + const canvas = document.createElement('canvas'); + canvas.width = Math.round(vw * dpr); + canvas.height = Math.round(ph * dpr); + const ctx = canvas.getContext('2d'); + for (const p of parts) { + const img = await loadImage(p.dataUrl); + ctx.drawImage(img, 0, Math.round(p.y * dpr)); + } + + return { dataUrl: canvas.toDataURL('image/png') }; +} + function dataUrlToBlob(dataUrl) { const [meta, b64] = dataUrl.split(','); const mime = (meta.match(/data:(.*?);/) || [])[1] || 'image/png'; diff --git a/browser-extension/content.js b/browser-extension/content.js index f4ff8af..4ce8698 100644 --- a/browser-extension/content.js +++ b/browser-extension/content.js @@ -23,7 +23,9 @@ // ── Extension→Page messaging (capture context for the popup) ────────────── browser.runtime.onMessage.addListener(function (msg) { - if (msg && msg.kind === 'get_context') { + if (!msg || typeof msg.kind !== 'string') return Promise.resolve(undefined); + + if (msg.kind === 'get_context') { var selection = ''; try { selection = window.getSelection().toString().trim(); } catch (e) {} return Promise.resolve({ @@ -32,6 +34,26 @@ url: window.location.href || '', }); } + + if (msg.kind === 'get_viewport') { + var ph = Math.max( + document.documentElement ? document.documentElement.scrollHeight : 0, + document.body ? document.body.scrollHeight : 0 + ); + return Promise.resolve({ + viewportWidth: window.innerWidth || document.documentElement.clientWidth || 0, + viewportHeight: window.innerHeight || document.documentElement.clientHeight || 0, + pageHeight: ph, + devicePixelRatio: window.devicePixelRatio || 1, + scrollY: window.scrollY || window.pageYOffset || 0, + }); + } + + if (msg.kind === 'scroll_to') { + window.scrollTo(0, msg.y); + return Promise.resolve({ y: window.scrollY || window.pageYOffset || 0 }); + } + return Promise.resolve(undefined); }); })(); diff --git a/browser-extension/popup.html b/browser-extension/popup.html index 380526e..d0660de 100644 --- a/browser-extension/popup.html +++ b/browser-extension/popup.html @@ -11,7 +11,7 @@