mockaton/src/server/utils/http-response.js

HTTP Mock Server

import fs, { readFileSync } from 'node:fs'

import { logger } from './logger.js'
import { mimeFor } from './mime.js'
import { HEADER_502 } from '../ApiConstants.js'


export function setHeaders(response, headers) {
	for (let i = 0; i < headers.length; i += 2)
		response.setHeader(headers[i], headers[i + 1])
}

export function sendOK(response) {
	logger.access(response)
	response.end()
}

export function sendHTML(response, html, csp) {
	logger.access(response)
	response.setHeader('Content-Type', mimeFor('html'))
	response.setHeader('Content-Security-Policy', csp)
	response.end(html)
}

export function sendJSON(response, payload) {
	logger.access(response)
	response.setHeader('Content-Type', 'application/json')
	response.end(JSON.stringify(payload))
}

export function sendFile(response, file) {
	logger.access(response)
	response.setHeader('Content-Type', mimeFor(file))
	response.end(readFileSync(file, 'utf8'))
}

export function sendNoContent(response) {
	response.statusCode = 204
	logger.access(response)
	response.end()
}


export function sendBadRequest(response) {
	response.statusCode = 400
	logger.access(response)
	response.end()
}

export function sendNotFound(response) {
	response.statusCode = 404
	logger.access(response)
	response.end()
}

export function sendMockNotFound(response) {
	response.statusCode = 404
	logger.accessMock(response.req.url, '404')
	response.end()
}

export function sendTooLongURI(response) {
	response.statusCode = 414
	logger.access(response)
	response.end()
}

export function sendUnprocessable(response, error) {
	logger.access(response, error)
	response.statusCode = 422
	response.end(error)
}


export function sendInternalServerError(response, error) {
	logger.error(500, response.req.url, error?.message || error, error?.stack || '')
	response.statusCode = 500
	response.end()
}

export function sendBadGateway(response, error) {
	logger.warn('Fallback Proxy Error:', error.cause.message)
	response.statusCode = 502
	response.setHeader(HEADER_502, 1)
	response.end()
}


export async function sendPartialContent(response, range, file) {
	const { size } = await fs.promises.lstat(file)
	let [start, end] = range.replace(/bytes=/, '').split('-').map(n => parseInt(n, 10))
	if (isNaN(end)) end = size - 1
	if (isNaN(start)) start = size - end

	if (start < 0 || start > end || start >= size || end >= size) {
		response.statusCode = 416 // Range Not Satisfiable
		response.setHeader('Content-Range', `bytes */${size}`)
		response.end()
	}
	else {
		response.statusCode = 206 // Partial Content
		response.setHeader('Accept-Ranges', 'bytes')
		response.setHeader('Content-Range', `bytes ${start}-${end}/${size}`)
		response.setHeader('Content-Type', mimeFor(file))
		const reader = fs.createReadStream(file, { start, end })
		reader.on('open', function () {
			this.pipe(response)
		})
		reader.on('error', function (error) {
			sendInternalServerError(response, error)
		})
	}
}