markdown-it/benchmark/benchmark.mjs

#!/usr/bin/env node
/* eslint no-console:0 */

import { createRequire } from 'node:module'
const require = createRequire(import.meta.url)

const fs        = require('fs')
const util      = require('util')
const Benchmark = require('benchmark')
const ansi      = require('ansi')
const cursor    = ansi(process.stdout)

const IMPLS = []

for (const name of fs.readdirSync(new URL('./implementations', import.meta.url)).sort()) {
  const filepath = new URL(`./implementations/${name}/index.mjs`, import.meta.url)
  const code = (await import(filepath))

  IMPLS.push({
    name: name,
    code: code
  })
}

const SAMPLES = []

fs.readdirSync(new URL('./samples', import.meta.url)).sort().forEach(sample => {
  const filepath = new URL(`./samples/${sample}`, import.meta.url)

  const content = {}

  content.string = fs.readFileSync(filepath, 'utf8')

  const title = `(${content.string.length} bytes)`

  function onComplete () { cursor.write('\n') }

  const suite = new Benchmark.Suite(
    title,
    {
      onStart: () => { console.log('\nSample: %s %s', sample, title) },
      onComplete: onComplete
    }
  )

  IMPLS.forEach(function (impl) {
    suite.add(
      impl.name,
      {
        onCycle: function onCycle (event) {
          cursor.horizontalAbsolute()
          cursor.eraseLine()
          cursor.write(' > ' + event.target)
        },
        onComplete: onComplete,
        fn: function () { impl.code.run(content.string) }
      }
    )
  })

  SAMPLES.push({
    name: sample.split('.')[0],
    title: title,
    content: content,
    suite: suite
  })
})


function select (patterns) {
  const result = []

  if (!(patterns instanceof Array)) {
    patterns = [patterns]
  }

  function checkName (name) {
    return patterns.length === 0 || patterns.some(function (regexp) {
      return regexp.test(name)
    })
  }

  SAMPLES.forEach(function (sample) {
    if (checkName(sample.name)) {
      result.push(sample)
    }
  })

  return result
}


function run (files) {
  const selected = select(files)

  if (selected.length > 0) {
    console.log('Selected samples: (%d of %d)', selected.length, SAMPLES.length)
    selected.forEach(function (sample) {
      console.log(' > %s', sample.name)
    })
  } else {
    console.log('There isn\'t any sample matches any of these patterns: %s', util.inspect(files))
  }

  selected.forEach(function (sample) {
    sample.suite.run()
  })
}

run(process.argv.slice(2).map(source => new RegExp(source, 'i')))