urltomarkdown/index.js

108 lines
2.9 KiB
JavaScript
Raw Normal View History

2022-01-08 17:05:05 +00:00
const https = require('https');
const turndown = require('turndown');
const { Readability } = require('@mozilla/readability');
const JSDOM = require('jsdom').JSDOM;
2022-01-13 14:40:21 +00:00
const common_filters = require('./url_to_markdown_common_filters');
2022-01-30 20:14:46 +00:00
const table_to_markdown = require('./html_table_to_markdown.js');
const validURL = require('@7c/validurl');
const express = require('express');
2022-01-09 15:20:59 +00:00
const rateLimit = require('express-rate-limit');
const port = process.env.PORT;
2022-01-09 15:20:59 +00:00
2022-01-30 11:50:17 +00:00
const app = express();
const service = new turndown();
2022-01-09 15:20:59 +00:00
const rateLimiter = rateLimit({
windowMs: 30 * 1000,
max: 5,
message: 'Rate limit exceeded',
headers: true
});
app.use(rateLimiter);
2022-01-08 17:05:05 +00:00
app.use(express.urlencoded({
2022-01-30 11:50:17 +00:00
extended: true,
limit: '10mb'
}));
2022-01-09 15:20:59 +00:00
2022-01-08 17:05:05 +00:00
app.get('/', (req, res) => {
url = req.query.url;
if (url && validURL(url)) {
send_headers(res);
2022-01-29 19:52:39 +00:00
read_url(url, res);
2022-01-08 17:35:43 +00:00
} else {
res.status(400).send("Please specify a valid url query parameter");
2022-01-08 17:35:43 +00:00
}
2022-01-08 17:05:05 +00:00
});
app.post('/', function(req, res) {
2022-01-30 11:50:17 +00:00
let html = req.body.html;
let url = req.body.url;
if (!html) {
res.status(400).send("Please provide a POST parameter called html");
} else {
2022-01-30 21:20:23 +00:00
try {
2022-01-30 11:50:17 +00:00
let document = new JSDOM(html);
let markdown = process_dom(url, document, res);
send_headers(res);
res.send(markdown);
2022-01-30 21:20:23 +00:00
} catch (error) {
2022-01-30 15:04:54 +00:00
res.status(400).send("Could not parse that document");
2022-01-30 21:20:23 +00:00
}
}
});
2022-01-08 17:05:05 +00:00
app.listen(port, () => {
})
function send_headers(res) {
res.header("Access-Control-Allow-Origin", '*');
res.header("Access-Control-Expose-Headers", 'X-Title');
res.header("Content-Type", 'text/markdown');
}
2022-01-30 11:50:17 +00:00
function process_dom(url, document, res) {
let title = document.window.document.querySelector('title');
if (title)
res.header("X-Title", encodeURIComponent(title.textContent));
let reader = new Readability(document.window.document);
2022-01-30 15:04:54 +00:00
let readable = reader.parse().content;
let replacement = {placeholders:[], tables:[]}
readable = format_tables(readable, replacement);
let markdown = service.turndown(readable);
for (let i=0;i<replacement.placeholders.length;i++) {
markdown = markdown.replace(replacement.placeholders[i], replacement.tables[i]);
}
2022-01-30 11:50:17 +00:00
let result = (url) ? common_filters.filter(url, markdown) : markdown;
return result;
}
2022-01-08 17:05:05 +00:00
function read_url(url, res) {
JSDOM.fromURL(url).then((document)=>{
2022-01-30 11:50:17 +00:00
let markdown = process_dom(url, document, res);
res.send(markdown);
}).catch((error)=> {
res.status(400).send("Sorry, could not fetch and convert that URL");
2022-01-08 17:05:05 +00:00
});
}
2022-01-30 15:04:54 +00:00
function format_tables(html, replacements) {
const tables = html.match(/(<table[^>]*>(?:.|\n)*?<\/table>)/gi);
2022-01-30 21:05:14 +00:00
if (tables) {
for (let t=0;t<tables.length;t++) {
let table = tables[t];
let markdown = table_to_markdown.convert(table);
let placeholder = "urltomarkdowntableplaceholder"+t+Math.random();
replacements.placeholders[t] = placeholder;
replacements.tables[t] = markdown;
html = html.replace(table, "<p>"+placeholder+"</p>");
}
2022-01-30 15:04:54 +00:00
}
return html;
}