fimfic2epub/src/fetch.js

import isNode from 'detect-node'

function fetchNode (url, responseType) {
  const request = require('request')
  if (url.indexOf('/') === 0) {
    url = 'http://www.fimfiction.net' + url
  }
  return new Promise((resolve, reject) => {
    request({
      url: url,
      encoding: responseType ? null : 'utf8',
      headers: {
        referer: 'http://www.fimfiction.net/',
        cookie: 'view_mature=true'
      }
    }, (error, response, body) => {
      if (error) {
        reject(error)
        return
      }
      // let type = response.headers['content-type']
      resolve(body)
    })
  })
}

export default function fetch (url, responseType) {
  if (url.indexOf('//') === 0) {
    url = 'http:' + url
  }
  if (url.indexOf('/') === 0) {
    url = 'https://fimfiction.net' + url
  }

  if (isNode) {
    return fetchNode(url, responseType)
  }
  return new Promise((resolve, reject) => {
    if (typeof window.fetch === 'function') {
      const headers = new Headers()
      headers.append('cookie', 'view_mature=true')
      window.fetch(url, {
        method: 'GET',
        mode: 'cors',
        credentials: 'include',
        cache: 'default',
        headers: headers,
        redirect: 'follow'
      }).then((response) => {
        if (responseType === 'blob') {
          response.blob().then(resolve, reject)
        } else if (responseType === 'arraybuffer') {
          response.arrayBuffer().then(resolve, reject)
        } else {
          response.text().then(resolve, reject)
        }
      }).catch((err) => {
        reject(new Error('Error fetching ' + url + ' (' + err + ')'))
      })
    } else {
      let x = new XMLHttpRequest()
      x.open('get', url, true)
      if (responseType) {
        x.responseType = responseType
      }
      x.onload = function () {
        resolve(x.response)
      }
      x.onerror = function () {
        reject(new Error('Error fetching ' + url))
      }
      x.send()
    }
  })
}