'use strict';

const fs = require('fs');
const format = require('human-format');
const promisifyEvent = require('promisify-event');
const memwatch = require('node-memwatch');
const SAXParser = require('../../packages/parse5-sax-parser/lib');

main();

async function main() {
    let parsedDataSize = 0;
    let maxMemUsage = 0;
    let startDate = null;
    let endDate = null;
    const heapDiffMeasurement = new memwatch.HeapDiff();
    let heapDiff = null;

    memwatch.on('stats', stats => {
        maxMemUsage = Math.max(maxMemUsage, stats['current_base']);
    });

    startDate = new Date();

    const parserPromise = parse().then(dataSize => {
        parsedDataSize = dataSize;
        endDate = new Date();
        heapDiff = heapDiffMeasurement.end();
    });

    await Promise.all([
        parserPromise,
        promisifyEvent(memwatch, 'stats') // NOTE: we need at least one `stats` result
    ]);

    printResults(parsedDataSize, startDate, endDate, heapDiff, maxMemUsage);
}

async function parse() {
    const data = fs.readFileSync('../test/data/huge-page/huge-page.html', 'utf8');
    let parsedDataSize = 0;
    const stream = new SAXParser();

    for (let i = 0; i < 200; i++) {
        parsedDataSize += data.length;
        stream.write(data);
    }

    stream.end();

    await promisifyEvent(stream, 'finish');

    return parsedDataSize;
}

function getDuration(startDate, endDate) {
    const scale = new format.Scale({
        seconds: 1,
        minutes: 60,
        hours: 3600
    });

    return format((endDate - startDate) / 1000, { scale: scale });
}

function printResults(parsedDataSize, startDate, endDate, heapDiff, maxMemUsage) {
    console.log('Input data size:', format(parsedDataSize, { unit: 'B' }));
    console.log('Duration: ', getDuration(startDate, endDate));
    console.log('Memory before: ', heapDiff.before.size);
    console.log('Memory after: ', heapDiff.after.size);
    console.log('Memory max: ', format(maxMemUsage, { unit: 'B' }));
}