NaturalCycles/nodejs-lib

import crypto, { BinaryToTextEncoding } from 'node:crypto'
import type { Base64String, Base64UrlString } from '@naturalcycles/js-lib'

export function md5(s: string | Buffer, outputEncoding: BinaryToTextEncoding = 'hex'): string {
  return hash(s, 'md5', outputEncoding)
}

export function md5AsBuffer(s: string | Buffer): Buffer {
  return hashAsBuffer(s, 'md5')
}

export function sha256(s: string | Buffer, outputEncoding: BinaryToTextEncoding = 'hex'): string {
  return hash(s, 'sha256', outputEncoding)
}

export function sha256AsBuffer(s: string | Buffer): Buffer {
  return hashAsBuffer(s, 'sha256')
}

export function hash(
  s: string | Buffer,
  algorithm: string,
  outputEncoding: BinaryToTextEncoding = 'hex',
): string {
  // todo: cleanup after @types/node is updated
  // https://github.com/nodejs/node/blob/main/doc/changelogs/CHANGELOG_V20.md#crypto-implement-cryptohash
  // Node 20.12+
  if ((crypto as any).hash) {
    return (crypto as any).hash(algorithm, s, outputEncoding)
  }

  return crypto.createHash(algorithm).update(s).digest(outputEncoding)
}

export function hashAsBuffer(s: string | Buffer, algorithm: string): Buffer {
  // todo: cleanup after @types/node is updated
  // Node 20.12+
  if ((crypto as any).hash) {
    return (crypto as any).hash(algorithm, s, 'buffer')
  }

  return crypto.createHash(algorithm).update(s).digest()
}

export function base64(s: string | Buffer): Base64String {
  return (typeof s === 'string' ? Buffer.from(s) : s).toString('base64')
}

export function base64Url(s: string | Buffer): Base64UrlString {
  return (typeof s === 'string' ? Buffer.from(s) : s).toString('base64url')
}

export function base64ToString(strBase64: Base64String): string {
  return Buffer.from(strBase64, 'base64').toString('utf8')
}

export function base64UrlToString(strBase64Url: Base64UrlString): string {
  return Buffer.from(strBase64Url, 'base64url').toString('utf8')
}

export function base64ToBuffer(strBase64: Base64String): Buffer {
  return Buffer.from(strBase64, 'base64')
}

export function base64UrlToBuffer(strBase64Url: Base64UrlString): Buffer {
  return Buffer.from(strBase64Url, 'base64url')
}

export function stringToBase64(s: string): Base64String {
  return Buffer.from(s, 'utf8').toString('base64')
}

export function stringToBase64Url(s: string): Base64UrlString {
  return Buffer.from(s, 'utf8').toString('base64url')
}

export function bufferToBase64(b: Buffer): Base64String {
  return b.toString('base64')
}

export function bufferToBase64Url(b: Buffer): Base64UrlString {
  return b.toString('base64url')
}