pdf-tools/tools/fieldRename.ts

import { type PDFAcroField, type PDFField, PDFName, PDFString } from "pdf-lib";
import { loadPdf, loadPdfForm, savePdf } from "util/saveLoadPdf.ts";
import { callWithArgPrompt } from "util/call.ts";
import { TerminalBlock } from "../cli/TerminalLayout.ts";
import { forceArgs } from "../cli/forceArgs.ts";
import { colorize } from "../cli/style.ts";
import { cliLog, cliPrompt } from "../cli/prompts.ts";
import { multiSelectMenuInteractive } from "../cli/selectMenu.ts";

// const thing = PDFAcroField.prototype.getFullyQualifiedName;
// PDFAcroField.prototype.getFullyQualifiedName = function () {
//     const name = thing.call(this)
//     // if (name?.includes('langauge'))
//     console.log(name)
//     return name;
// }

// const thing = PDFHexString.prototype.copyBytesInto
// PDFHexString.prototype.copyBytesInto = function (buffer: Uint8Array, offset: number) {
//     console.log((this as any).value)

//     const result = thing.call(this, buffer, offset)
//     return result;
// }

async function renameFields(
  path: string,
  pattern: string | RegExp,
  change: string,
) {
  if (typeof pattern === "string") pattern = new RegExp(pattern);
  const form = await loadPdfForm(path);
  const fields = form.getFields();
  let changesMade = false;
  for (const field of fields) {
    const name = field.getName();
    if (pattern.test(name)) {
      console.log(name + " %cfound", "color: red");
      const segments = name.split(".");
      const matchingSegments = segments.filter((s) => pattern.test(s));
      let cField: PDFAcroField | undefined = field.acroField;
      while (cField) {
        if (
          cField.getPartialName() &&
          matchingSegments.includes(cField.getPartialName()!)
        ) {
          const mName = cField.getPartialName()?.replace(pattern, change);
          if (mName) {
            changesMade = true;
            cField.dict.set(PDFName.of("T"), PDFString.of(mName));
            // console.log(cField.getPartialName())
          }
        }
        cField = cField.getParent();
        // console.log(cField?.getPartialName())
      }
      console.log(field.getName());
      // const newName = name.replace(pattern, change);
      // console.log("Change to: %c" + newName, "color: yellow");
      // if (confirm('Ok?')) {
      //     let parent = field.acroField.getParent();
      //     field.acroField.setPartialName(segments.pop())
      //     while (parent && segments.length) {
      //         console.log(parent.getPartialName())
      //         parent.setPartialName(segments.pop())
      //         parent = parent.getParent();
      //     }
      //     changesMade = true;
      //     console.log(field.getName())
      //     // dict.set(PDFName.of("T"), PDFHexString.fromText(newName))
      //     console.log("%cDone!", "color: lime")
      // }
      // break;
    }
  }
  if (changesMade) {
    savePdf(form.doc, path);
  }
}

function applyRename(
  field: PDFField,
  name: string,
  pattern: RegExp,
  change: string,
) {
  const segments = name.split(".");
  const matchingSegments = segments.filter((s) => pattern.test(s));
  let cField: PDFAcroField | undefined = field.acroField;
  while (cField) {
    if (
      cField.getPartialName() &&
      matchingSegments.includes(cField.getPartialName()!)
    ) {
      const mName = cField.getPartialName()?.replace(pattern, change);
      if (mName) {
        cField.dict.set(PDFName.of("T"), PDFString.of(mName));
        // console.log(cField.getPartialName())
      }
    }
    cField = cField.getParent();
    // console.log(cField?.getPartialName())
  }
}

function evaluateChange(change: string, match: RegExpExecArray) {
  return change.replace(
    /\$(\d+)(i?)/g,
    (_, i, indexed) =>
      indexed
        ? (parseInt(match[i]) ? (parseInt(match[i]) - 1).toString() : match[i])
        : match[i],
  );
}

class RenameFields implements ITool {
  name = "renamefields";
  description = "Renames fields in a PDF form";
  block: TerminalBlock | undefined;

  setBlock(block: TerminalBlock) {
    this.block = block;
  }

  help(standalone = false) {
    cliLog(
      "Usage: renamefields <pdfPath> <pattern> <change>",
      standalone ? undefined : this.block,
    );
  }
  async run(pdfPath: string = "", pattern: string = "", change: string = "") {
    if (!this.block) {
      this.block = new TerminalBlock();
    }
    this.block.setPreserveHistory(true);

    [pdfPath, pattern, change] = await forceArgs(
      [pdfPath, pattern, change],
      [
        ["Please provide path to PDF:", (p) => !!p && p.endsWith(".pdf")],
        "Please provide search string:",
        "Please provide requested change:",
      ],
      this.block,
    );

    const patternRegex = new RegExp(pattern);

    const pdf = await loadPdf(pdfPath);
    const form = pdf.getForm();
    const fields = form.getFields();

    const foundUpdates: [string, callback][] = [];

    for (const field of fields) {
      const name = field.getName();
      const match = patternRegex.exec(name);
      if (match) {
        const toChange = evaluateChange(change, match);
        foundUpdates.push([
          `${colorize(name, "red")} -> ${colorize(toChange, "green")}`,
          () => {
            applyRename(field, name, patternRegex, toChange);
          },
        ]);
      }
    }

    if (foundUpdates.length) {
      cliLog("Found updates:", this.block);
      await multiSelectMenuInteractive(
        "Please select an option to apply",
        foundUpdates,
        { terminalBlock: this.block },
      );
    }

    const path = await cliPrompt(
      "Save to path (or hit enter to keep current):",
      this.block,
    );
    await savePdf(pdf, path || pdfPath);
  }
}
export default new RenameFields();

if (import.meta.main) {
  // await call(renameFields)
  // while (!path || !path.endsWith('.pdf')) path = prompt("Please provide path to PDF:") || '';
  // while (!pattern) pattern = prompt("Please provide search string:") || '';
  // while (!change) change = prompt("Please provide requested change:") || '';
  await callWithArgPrompt(renameFields, [
    ["Please provide path to PDF:", (p) => !!p && p.endsWith(".pdf")],
    "Please provide search string:",
    "Please provide requested change:",
  ]);
}