routes.js

const Router = require('express').Router;
const router = new Router();
const Parser = require('@postlight/parser');
const ParserCustomizer = require('./customizer');

ParserCustomizer.customize(Parser);

router.route('/').get((req, res) => {
    res.json({
        message: 'Welcome to 🚀mercury-parser-api API! Endpoint: /parser',
    });
});

router.route('/parser').get(async (req, res) => {
    let result = { message: 'No URL was provided' };
    if (req.query.url) {
        try {
            const contentType = req.query.contentType || 'html';
            let headers = new Object();
            if (typeof req.query.headers !== 'undefined') {
                headers = JSON.parse(req.query.headers);
            }
            result = await Parser.parse(req.query.url, {
                contentType,
                headers,
            });
        } catch (error) {
            result = { error: true, messages: error.message };
        }
    }
    return res.json(result);
});

router.route('/parse-html').post(async (req, res) => {
    let result = { message: 'Invalid request body, both url and html are required' };
    if (req.body.url && req.body.html) {
        try {
            result = await Parser.parse(req.body.url, {
                html: req.body.html
            });
        } catch (error) {
            result = { error: true, messages: error.message };
        }
    }
    return res.json(result);
});

module.exports = router;