@bsv/sdk/src/storage/StorageUtils.ts

BSV Blockchain Software Development Kit

import { sha256 } from '../primitives/Hash.js'
import { toHex, fromBase58Check, toBase58Check, toArray } from '../primitives/utils.js'

/**
 * Takes a UHRP URL and removes any prefixes.
 * @param {string} URL - The UHRP URL.
 * @returns {string} - Normalized URL.
 */
export const normalizeURL = (URL: string): string => {
  if (URL.toLowerCase().startsWith('uhrp:')) URL = URL.slice(5)
  if (URL.startsWith('//')) URL = URL.slice(2)
  return URL
}

/**
 * Generates a UHRP URL from a given SHA-256 hash.
 * @param {number[]} hash - 32-byte SHA-256 hash.
 * @returns {string} - Base58Check encoded URL.
 */
export const getURLForHash = (hash: number[]): string => {
  if (hash.length !== 32) {
    throw new Error('Hash length must be 32 bytes (sha256)')
  }
  return toBase58Check(hash, toArray('ce00', 'hex'))
}

/**
 * Generates a UHRP URL for a given file.
 * @param {number[] | string} file - File content as number array or string.
 * @returns {string} - Base58Check encoded URL.
 */
export const getURLForFile = (file: number[]): string => {
  const hash = sha256(file)
  return getURLForHash(hash)
}

/**
 * Extracts the hash from a UHRP URL.
 * @param {string} URL - UHRP URL.
 * @returns {number[]} - Extracted SHA-256 hash.
 */
export const getHashFromURL = (URL: string): number[] => {
  URL = normalizeURL(URL)
  const { data, prefix } = fromBase58Check(URL, undefined, 2)
  if (data.length !== 32) {
    throw new Error('Invalid length!')
  }
  if (toHex(prefix as number[]) !== 'ce00') {
    throw new Error('Bad prefix')
  }
  return data as number[]
}

/**
 * Checks if a URL is a valid UHRP URL.
 * @param {string} URL - The URL to validate.
 * @returns {boolean} - True if valid, false otherwise.
 */
export const isValidURL = (URL: string): boolean => {
  try {
    getHashFromURL(URL)
    return true
  } catch (e) {
    return false
  }
}