node.ts

import http from 'http'
import https from 'https'
import FormData from 'form-data'
import { URL } from 'url'
import { GotenbergClientFunction } from '../_types'

/**
 * Little helper to parse url and get needed request method - either HTTP or HTTPS
 * @param url
 */
const parse = (url: string): [URL, typeof http.request | typeof https.request] => {
  const _url = new URL(url)
  const request = _url.protocol === 'http:' ? http.request : https.request
  return [_url, request]
}

/**
 * Perform POST request to Gotenberg API
 */
export function post(
  this: object | null,
  url: string,
  data: FormData,
  headers?: http.OutgoingHttpHeaders
): Promise<NodeJS.ReadableStream> {
  const [_url, request] = parse(url)
  return new Promise((resolve, reject) => {
    const req = request(_url, {
      method: 'POST',
      ...this, // extends with config options
      headers: {
        ...data.getHeaders(),
        ...headers,
        ...(this ? (this as any).headers : null), // extends with config headers
      },
    })

    req.on('error', reject)
    req.on('response', (res) => {
      if (res.statusCode && res.statusCode >= 200 && res.statusCode < 300) {
        resolve(res)
      } else {
        let error = res.statusCode + ' ' + res.statusMessage

        // something is wrong, get error message from Gotenberg
        const chunks: Buffer[] = []
        res.on('data', (chunk: Buffer) => chunks.push(chunk))
        res.on('end', () => {
          try {
            error += ' (' + JSON.parse(Buffer.concat(chunks).toString()).message + ')'
          } catch (err) {
            // ignore
          }
          reject(new Error(error))
        })
      }
    })

    data.pipe(req) // pipe Form data to request
    // pipe should automatically call `req.end()` after stream ends
  })
}

/**
 * Perform POST request to Gotenberg API
 */
export function get(this: object | null, url: string): Promise<NodeJS.ReadableStream> {
  const [_url, request] = parse(url)
  return new Promise((resolve, reject) => {
    const req = request(_url, {
      method: 'GET',
      ...this, // extends with config options
    })

    req.on('error', reject)
    req.on('response', (res) => {
      if (res.statusCode && res.statusCode >= 200 && res.statusCode < 300) {
        resolve(res)
      } else {
        res.resume() // ignore response body
        reject(new Error(res.statusCode + ' ' + res.statusMessage))
      }
    })

    req.end() // send request
  })
}

/**
 * GotenbergClient function implementation.
 *
 * Uses native Nodejs modules `http` and `https`, can accept any options
 * for http -> https://nodejs.org/docs/latest-v10.x/api/http.html#http_http_request_options_callback
 * for https -> https://nodejs.org/docs/latest-v10.x/api/https.html#https_https_request_options_callback
 */
export const client: GotenbergClientFunction = (config?: object) => ({
  post: post.bind(config || null),
  get: get.bind(config || null),
})