hal2dy / html-scan

const fs = require('fs')

const { join } = require('path')

const { Readable } = require('stream')

module.exports = (input, callback) => {

  if (typeof input === 'string') {

    // Input as file

    const path = join(__dirname, '../', input)

    if (!fs.existsSync(path)) {

      console.error(`Cannot read file ${path}`)

      return

    }

    const html = fs.readFileSync(path, 'utf8')

    callback(html)

  } else if (input instanceof Readable) {

    // Input as stream

    input.on('data', chunk => {

      callback(chunk.toString())

    })

    input.on('error', err => {

      console.error(err)

    })

  } else {

    // Unknown input

    console.error('Invalid Input')

  }

}