import { randomInt } from 'crypto';
import { isBase58 } from 'class-validator';

const BASE58 = '123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz';
const bs58 = require('base-x')(BASE58);

export function generateBigNumberKey(): string {
  const key = randomInt(0, 140737488355328);
  return encodeBigNumberKey(key);
}

export function decodeBigNumberKey(str: string): number {
  const arr = bs58.decode(str) as Uint8Array;
  const extArr = new Uint8Array([0, 0, ...arr]);

  const view = new DataView(extArr.buffer);
  const res = view.getBigUint64(0);
  return Number(res);
}

export function encodeBigNumberKey(key: number | string): string {
  const buffer = new ArrayBuffer(8);

  const view = new DataView(buffer);
  view.setBigUint64(0, BigInt(key));
  const arr = new Uint8Array(view.buffer);

  return bs58.encode(arr.slice(2));
}

export function isValidBigNumberKey(str: string) {
  return isBase58(str);
}