badge-size/lib/fetch.js

Displays the size of a given file in your repository

const got = require('got')

/**
 * Fetch file to stat from Github.
 *
 * @param {object} baton
 */
async function fetch(baton) {
  try {
    const res = await got[baton.compression ? 'get' : 'head'](baton.url, {
      headers: {
        'accept-encoding': 'identity'
      }
    })

    baton.originalSize = Number(res.headers['content-length'])
    baton.size = baton.originalSize
    baton.data = res.body
  }
  catch (err) {
    baton.err = new Error('Unknown path')
    throw baton
  }
}

module.exports = fetch