⚡ ”new” scripts
This commit is contained in:
144
bin/§pdftk-data.mjs
Executable file
144
bin/§pdftk-data.mjs
Executable file
@@ -0,0 +1,144 @@
|
||||
#!/usr/bin/env nodejsscript
|
||||
/* jshint esversion: 11,-W097, -W040, module: true, node: true, expr: true, undef: true *//* global echo, $, pipe, s, fetch, cyclicLoop */
|
||||
const tmp= "tmp-"; //pdftk has issue reading from /tmp, so need to put it somewhere to the same folder :-(
|
||||
if(!s.which("pdftk")) $.error("pdftk not found");
|
||||
|
||||
$.api()
|
||||
.version("2025-08-05")
|
||||
.describe([
|
||||
"A small wrapper around 'pdftk' to extract data from pdf into the JSON.",
|
||||
"Or to update the PDF with data from JSON.",
|
||||
"Bookmars are converted to JSON object with key=PageNumber, value= Title (no. of spaces= level-1).",
|
||||
])
|
||||
.option("--debug", "Debug mode")
|
||||
.command("extract <file_pdf> [file_info]", "Extract data from PDF.")
|
||||
.action(function extract(file_pdf, file_info, { debug }){
|
||||
if(!s.test("-f", file_pdf)) $.error("PDF File not found");
|
||||
if(!file_info) file_info= filename(file_pdf) + ".json";
|
||||
|
||||
const temp= `${tmp}${tmpname(file_pdf)}.info` ;
|
||||
s.run`pdftk ${file_pdf} dump_data_utf8 output ${temp}`;
|
||||
const info= infoToJSON(temp);
|
||||
if(!debug) s.rm(temp);
|
||||
s.echo(info).to(file_info);
|
||||
$.exit(0);
|
||||
})
|
||||
.command("update <file_pdf> [file_info]", "Update PDF with data from JSON.")
|
||||
.action(function update(file_pdf, file_info, { debug }){
|
||||
if(!s.test("-f", file_pdf)) $.error("PDF File not found");
|
||||
if(!file_info) file_info= filename(file_pdf) + ".json";
|
||||
if(!s.test("-f", file_info)) $.error("Info File not found");
|
||||
|
||||
const info= infoFromJSON(file_info);
|
||||
const temp= `${tmp}${tmpname(file_pdf)}.info`;
|
||||
s.echo(info).to(temp);
|
||||
const tmp_pdf= `${tmp}${tmpname(file_pdf)}.pdf`;
|
||||
s.cp(file_pdf, tmp_pdf);
|
||||
s.run`pdftk ${tmp_pdf} update_info_utf8 ${temp} output ${file_pdf}`;
|
||||
if(!debug){
|
||||
s.rm(tmp_pdf);
|
||||
s.rm(temp);
|
||||
}
|
||||
$.exit(0);
|
||||
})
|
||||
.command("convert <file_info>", "Converts between JSON and raw text.")
|
||||
.action(function convert(file_info){
|
||||
if(!s.test("-f", file_info)) $.error("Info File not found");
|
||||
const ext= file_info.slice(file_info.lastIndexOf("."));
|
||||
const info= ext===".json" ? infoFromJSON(file_info) : infoToJSON(file_info);
|
||||
echo(info);
|
||||
$.exit(0);
|
||||
})
|
||||
.parse();
|
||||
|
||||
function filename(path){ return path.slice(path.lastIndexOf("/")+1, path.lastIndexOf(".")); }
|
||||
function tmpname(path){ return filename(path) + "-" + Date.now(); }
|
||||
function infoFromJSON(file_info){
|
||||
const info= s.cat(file_info).xargs(JSON.parse);
|
||||
const output= [];
|
||||
info.Bookmark= Object.entries(info.Bookmark)
|
||||
.map(/** @param {[string, string]} _ */([PageNumber, Title])=> {
|
||||
const level= Title.search(/[^ ]/);
|
||||
return {
|
||||
PageNumber,
|
||||
Title: Title.slice(level),
|
||||
Level: level+1,
|
||||
};
|
||||
});
|
||||
for(const [key, value] of Object.entries(info)){
|
||||
if(Array.isArray(value)){
|
||||
const records= value.flatMap(pipe(
|
||||
Object.entries,
|
||||
entries=> [key+"Begin"].concat(entries.map(([subkey, value])=> `${key}${subkey}: ${value}`)),
|
||||
));
|
||||
output.push(...records);
|
||||
continue;
|
||||
}
|
||||
if(typeof value==="object"){
|
||||
const records= Object.entries(value).flatMap(([subkey, value])=> [
|
||||
key+"Begin",
|
||||
`${key}Key: ${subkey}`,
|
||||
`${key}Value: ${value}`,
|
||||
]);
|
||||
output.push(...records);
|
||||
continue;
|
||||
}
|
||||
output.push(`${key}: ${value}`);
|
||||
}
|
||||
return output.join("\n");
|
||||
}
|
||||
/** @returns {Record<string, unknown>} */
|
||||
function infoToJSON(file_info){
|
||||
const output= new Map();
|
||||
const data= s.cat(file_info).split("\n");
|
||||
let line= 0;
|
||||
const isEnd= line=> line>=data.length;
|
||||
for(; line<data.length; line++){
|
||||
const content= data[line].trim();
|
||||
if(!content) continue;
|
||||
|
||||
if("InfoBegin"===content){
|
||||
const info= output.has("Info") ? output.get("Info") : {};
|
||||
const curr= [];
|
||||
while(++line){
|
||||
if(isEnd(line)) break;
|
||||
const content= data[line].trim();
|
||||
if(!content){ line++; continue; }
|
||||
const [key, value]= content.split(": ");
|
||||
const index= ["InfoKey", "InfoValue"].indexOf(key);
|
||||
if(index===-1){ line--; break; }
|
||||
curr[index]= value;
|
||||
}
|
||||
const [key, value]= curr;
|
||||
info[key]= value;
|
||||
output.set("Info", info);
|
||||
continue;
|
||||
}
|
||||
if(content.endsWith("Begin")){
|
||||
const name= content.replace("Begin", "");
|
||||
const output_curr= output.has(name) ? output.get(name) : [];
|
||||
const curr= {};
|
||||
while(++line){
|
||||
if(isEnd(line)) break;
|
||||
const content= data[line].trim();
|
||||
if(!content){ line++; continue; }
|
||||
const [key, value]= content.split(": ");
|
||||
if(!key.startsWith(name) || key === (name+"Begin")){ line--; break; }
|
||||
curr[key.replace(name, "")]= value;
|
||||
}
|
||||
output_curr.push(curr);
|
||||
output.set(name, output_curr);
|
||||
continue;
|
||||
}
|
||||
const [key, value]= content.split(": ");
|
||||
output.set(key, value);
|
||||
}
|
||||
output.set("Bookmark", pipe(
|
||||
items=> items.map(({ PageNumber, Title, Level })=> ([PageNumber, " ".repeat(Number(Level)-1) + Title])),
|
||||
Object.fromEntries,
|
||||
)(output.get("Bookmark") || []));
|
||||
return pipe(
|
||||
Object.fromEntries,
|
||||
o=> JSON.stringify(o, null, "\t"),
|
||||
)(output.entries());
|
||||
}
|
||||
Reference in New Issue
Block a user