[Share] Export pdf text of items to notes 输出附件PDF全文到笔记

[Stream-L]

Description

I felt it was possible to write a script based on this one, combined with BN, but failed.

感觉可以可以基于这个脚本写结合BN的脚本，但是我试了一下失败了

If you need to adjust the location where a script adds notes, you can refer to the two methods in this script.

如果你需要调整脚本添加笔记的位置可以参考这个脚本设置的两种方式

Event

None

Operation

Script

Data

/**
 * 脚本功能：将选中条目的PDF附件全文提取并复制到独立笔记中
 * 
 * 使用说明：
 * 1. 选中一个或多个包含PDF附件的条目
 * 2. 运行此脚本
 * 3. 脚本会提取所有PDF的全文，并创建一个独立笔记（或为每个条目创建子笔记）
 * 4. 笔记将保存在当前分类中
 */

if (item) return; // 仅对所有选中项执行一次

const Zotero = require("Zotero");
const window = require("window");
const console = require("console");

// 使用传入的 items 或 item 变量
let selectedItems = [];
if (typeof items !== "undefined" && items.length > 0) {
  selectedItems = items;
} else if (typeof item !== "undefined") {
  selectedItems = [item];
} else {
  window.alert("请先选择要处理的文献");
  return;
}

// 询问用户创建方式
let createMode = window.prompt(
  "请选择笔记创建方式：\n\n1. 合并到一个独立笔记（所有PDF合并）\n2. 为每个条目创建子笔记\n\n请输入 1 或 2："
);

if (!createMode || (createMode !== '1' && createMode !== '2')) {
  window.alert("未选择有效选项，操作终止");
  return;
}

// 创建进度窗口
let progressWindow = new Zotero.ProgressWindow({ closeOnClick: false });
progressWindow.changeHeadline("提取PDF全文到笔记");
progressWindow.show();

let progressIcon = 'chrome://zotero/skin/spinner-16px.png';
let itemProgress = new progressWindow.ItemProgress(progressIcon, "PDF全文提取");
itemProgress.setProgress(0);
itemProgress.setText("⏳ 正在准备提取PDF全文...");
progressWindow.show();

/**
 * 提取单个PDF附件的全文
 */
async function extractPDFText(attachment) {
  try {
    // 使用Zotero的PDFWorker提取全文
    let result = await Zotero.PDFWorker.getFullText(attachment.id, null, true);
    let text = result && result.text ? result.text : '';
    return text.trim();
  } catch (error) {
    console.error(`提取PDF全文失败: ${attachment.id}`, error);
    return null;
  }
}

/**
 * 模式1: 合并所有PDF全文到一个独立笔记
 */
async function createMergedNote(items) {
  itemProgress.setProgress(10);
  itemProgress.setText("⏳ 正在提取PDF全文...");
  
  let parts = [];
  let processedCount = 0;
  let totalPDFs = 0;
  
  // 先统计总共有多少个PDF
  for (let item of items) {
    if (item.isAttachment()) continue;
    let attachments = Zotero.Items.get(item.getAttachments()).filter(a => a.isPDFAttachment());
    totalPDFs += attachments.length;
  }
  
  if (totalPDFs === 0) {
    itemProgress.setError();
    itemProgress.setText("❌ 未找到任何PDF附件");
    progressWindow.startCloseTimer(3000);
    return;
  }
  
  itemProgress.setText(`⏳ 找到 ${totalPDFs} 个PDF，开始提取...`);
  
  for (let item of items) {
    // 跳过附件类型的项目
    if (item.isAttachment()) {
      console.log(`跳过附件项目: ${item.id}`);
      continue;
    }
    
    let itemTitle = item.getField("title") || "无标题";
    let attachments = Zotero.Items.get(item.getAttachments()).filter(a => a.isPDFAttachment());
    
    for (let att of attachments) {
      processedCount++;
      let progress = 10 + Math.floor((processedCount / totalPDFs) * 70);
      itemProgress.setProgress(progress);
      
      let fileName = att.attachmentFilename || att.getField('title') || `PDF-${att.id}`;
      itemProgress.setText(`⏳ 提取中 (${processedCount}/${totalPDFs}): ${fileName.substring(0, 30)}...`);
      
      let text = await extractPDFText(att);
      
      if (text) {
        // 添加条目标题作为分隔
        parts.push(`<h2>${Zotero.Utilities.text2html(itemTitle)}</h2>`);
        parts.push(`<h3>📄 ${Zotero.Utilities.text2html(fileName)}</h3>`);
        parts.push(`<pre style="white-space: pre-wrap; font-family: monospace; font-size: 12px;">${Zotero.Utilities.text2html(text)}</pre>`);
        parts.push(`<hr/>`);
      } else {
        parts.push(`<p><em>⚠️ 无法提取: ${Zotero.Utilities.text2html(fileName)}</em></p>`);
      }
    }
  }
  
  if (parts.length === 0) {
    itemProgress.setError();
    itemProgress.setText("❌ 未能提取任何PDF内容");
    progressWindow.startCloseTimer(3000);
    return;
  }
  
  itemProgress.setProgress(85);
  itemProgress.setText("⏳ 正在创建笔记...");
  
  // 创建独立笔记
  let libraryID = items[0].libraryID;
  let library = Zotero.Libraries.get(libraryID);
  
  if (!library.editable) {
    itemProgress.setError();
    itemProgress.setText("❌ 该库为只读状态，无法创建笔记");
    progressWindow.startCloseTimer(3000);
    return;
  }
  
  // 获取当前分类
  let collectionID = null;
  try {
    let zoteroPane = Zotero.getActiveZoteroPane();
    if (zoteroPane) {
      let selectedCollection = zoteroPane.getSelectedCollection(true);
      if (selectedCollection) {
        collectionID = selectedCollection;
      }
    }
  } catch (e) {
    console.log("无法获取当前分类:", e);
  }
  
  let combinedHTML = `<h1>📚 PDF全文合集</h1>\n<p><em>提取时间: ${new Date().toLocaleString('zh-CN')}</em></p>\n<hr/>\n` + parts.join('\n');
  
  let note = new Zotero.Item('note');
  note.libraryID = libraryID;
  note.parentID = null; // 独立笔记
  if (collectionID) {
    note.addToCollection(collectionID);
  }
  
  note.setNote(combinedHTML);
  await note.saveTx();
  
  itemProgress.setProgress(100);
  itemProgress.setText(`✅ 成功创建笔记，已提取 ${processedCount} 个PDF`);
  progressWindow.startCloseTimer(3000);
}

/**
 * 模式2: 为每个条目创建子笔记
 */
async function createChildNotes(items) {
  itemProgress.setProgress(10);
  itemProgress.setText("⏳ 正在为每个条目创建子笔记...");
  
  let processedCount = 0;
  let successCount = 0;
  let totalItems = items.filter(item => !item.isAttachment()).length;
  
  if (totalItems === 0) {
    itemProgress.setError();
    itemProgress.setText("❌ 未找到有效的条目");
    progressWindow.startCloseTimer(3000);
    return;
  }
  
  for (let item of items) {
    // 跳过附件类型的项目
    if (item.isAttachment()) {
      console.log(`跳过附件项目: ${item.id}`);
      continue;
    }
    
    processedCount++;
    let progress = 10 + Math.floor((processedCount / totalItems) * 80);
    itemProgress.setProgress(progress);
    
    let itemTitle = item.getField("title") || "无标题";
    itemProgress.setText(`⏳ 处理中 (${processedCount}/${totalItems}): ${itemTitle.substring(0, 30)}...`);
    
    // 获取PDF附件
    let attachments = Zotero.Items.get(item.getAttachments()).filter(a => a.isPDFAttachment());
    
    if (attachments.length === 0) {
      console.log(`条目无PDF附件: ${itemTitle}`);
      continue;
    }
    
    // 合并该条目所有PDF的全文
    let parts = [];
    for (let att of attachments) {
      let fileName = att.attachmentFilename || att.getField('title') || `PDF-${att.id}`;
      let text = await extractPDFText(att);
      
      if (text) {
        parts.push(`<h3>📄 ${Zotero.Utilities.text2html(fileName)}</h3>`);
        parts.push(`<pre style="white-space: pre-wrap; font-family: monospace; font-size: 12px;">${Zotero.Utilities.text2html(text)}</pre>`);
        if (attachments.length > 1) parts.push(`<hr/>`);
      }
    }
    
    if (parts.length === 0) {
      console.log(`条目PDF提取失败: ${itemTitle}`);
      continue;
    }
    
    // 创建子笔记
    try {
      let noteHTML = `<h2>${Zotero.Utilities.text2html(itemTitle)}</h2>\n<p><em>提取时间: ${new Date().toLocaleString('zh-CN')}</em></p>\n<hr/>\n` + parts.join('\n');
      
      let note = new Zotero.Item('note');
      note.libraryID = item.libraryID;
      note.parentID = item.id; // 设置为子笔记
      note.setNote(noteHTML);
      await note.saveTx();
      
      successCount++;
    } catch (error) {
      console.error(`创建子笔记失败: ${itemTitle}`, error);
    }
  }
  
  itemProgress.setProgress(100);
  itemProgress.setText(`✅ 成功为 ${successCount} 个条目创建子笔记`);
  progressWindow.startCloseTimer(3000);
}

// 根据用户选择执行相应模式
if (createMode === '1') {
  createMergedNote(selectedItems).catch((error) => {
    console.error("创建合并笔记时出错:", error);
    itemProgress.setError();
    itemProgress.setText(`❌ 错误: ${error.message}`);
    progressWindow.startCloseTimer(5000);
  });
} else if (createMode === '2') {
  createChildNotes(selectedItems).catch((error) => {
    console.error("创建子笔记时出错:", error);
    itemProgress.setError();
    itemProgress.setText(`❌ 错误: ${error.message}`);
    progressWindow.startCloseTimer(5000);
  });
}

Anything else

No response