wagesj45/Sortana
1
0
Fork 0
Code Issues Pull requests Projects Releases 5 Packages Wiki Activity Actions

Add Markdown conversion option

Browse source
This commit is contained in:
Jordan Wages 2025-07-05 02:26:58 -05:00
commit 200c03c875
6 changed files with 37 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

23
background.js
View file

@ -22,6 +22,8 @@ let iconTimer = null;
let timingStats = { count: 0, mean: 0, m2: 0, total: 0, last: -1 };
let currentStart = 0;
let logGetTiming = true;
let htmlToMarkdown = false;
let TurndownService = null;
function setIcon(path) {
if (browser.browserAction) {
@ -70,7 +72,17 @@ function collectText(part, bodyParts, attachments) {
attachments.push(`${name} (${ct}, ${part.size || byteSize(body)} bytes)`);
} else if (ct.startsWith("text/html")) {
const doc = new DOMParser().parseFromString(body, 'text/html');
bodyParts.push(replaceInlineBase64(doc.body.textContent || ""));
if (htmlToMarkdown && TurndownService) {
try {
const td = new TurndownService();
const md = td.turndown(doc.body.innerHTML || body);
bodyParts.push(replaceInlineBase64(`[HTML Body converted to Markdown]\n${md}`));
} catch (e) {
bodyParts.push(replaceInlineBase64(doc.body.textContent || ""));
}
} else {
bodyParts.push(replaceInlineBase64(doc.body.textContent || ""));
}
} else {
bodyParts.push(replaceInlineBase64(body));
}
@ -213,16 +225,19 @@ async function clearCacheForMessages(idsInput) {
try {
AiClassifier = await import(browser.runtime.getURL("modules/AiClassifier.js"));
logger.aiLog("AiClassifier imported", {debug: true});
const td = await import(browser.runtime.getURL("resources/js/turndown.js"));
TurndownService = td.default || td.TurndownService;
} catch (e) {
console.error("failed to import AiClassifier", e);
return;
}
try {
const store = await storage.local.get(["endpoint", "templateName", "customTemplate", "customSystemPrompt", "aiParams", "debugLogging", "aiRules"]);
const store = await storage.local.get(["endpoint", "templateName", "customTemplate", "customSystemPrompt", "aiParams", "debugLogging", "htmlToMarkdown", "aiRules"]);
logger.setDebug(store.debugLogging);
await AiClassifier.setConfig(store);
await AiClassifier.init();
htmlToMarkdown = store.htmlToMarkdown === true;
const savedStats = await storage.local.get('classifyStats');
if (savedStats.classifyStats && typeof savedStats.classifyStats === 'object') {
Object.assign(timingStats, savedStats.classifyStats);
@ -254,6 +269,10 @@ async function clearCacheForMessages(idsInput) {
});
logger.aiLog("aiRules updated from storage change", {debug: true}, aiRules);
}
if (changes.htmlToMarkdown) {
htmlToMarkdown = changes.htmlToMarkdown.newValue === true;
logger.aiLog("htmlToMarkdown updated from storage change", {debug: true}, htmlToMarkdown);
}
});
} catch (err) {
logger.aiLog("failed to load config", {level: 'error'}, err);