import type { ErrorType } from '../errors/utils.js'
import type { Bytes, Hex } from '../types/data.js'

export type TrimOptions = {
  dir?: 'left' | 'right' | undefined
}

export type TrimReturnType<value extends Bytes | Hex> = value extends Hex
  ? Hex
  : Bytes

export type TrimErrorType = ErrorType

export function trim<value extends Bytes | Hex>(
  hexOrBytes: value,
  options: TrimOptions = {},
): TrimReturnType<value> {
  const { dir = 'left' } = options

  let data: any =
    typeof hexOrBytes === 'string' ? hexOrBytes.replace('0x', '') : hexOrBytes

  let sliceLength = 0
  for (let i = 0; i < data.length - 1; i++) {
    if (data[dir === 'left' ? i : data.length - i - 1].toString() === '0')
      sliceLength++
    else break
  }
  data =
    dir === 'left'
      ? data.slice(sliceLength)
      : data.slice(0, data.length - sliceLength)

  if (typeof hexOrBytes === 'string') {
    if (data.length === 1 && dir === 'right') data = `${data}0`
    return `0x${
      data.length % 2 === 1 ? `0${data}` : data
    }` as TrimReturnType<value>
  }
  return data as TrimReturnType<value>
}