import type { Token } from "@saberhq/token-utils";
import { TokenAmount } from "@saberhq/token-utils";
import { useMemo } from "react";

/**
 * Parses a {@link TokenAmount}.
 * @param token The token.
 * @param valueStr The string representation of the amount.
 * @returns
 */
export const useTokenAmount = (
  token: Token | null | undefined,
  valueStr: string | null | undefined
): TokenAmount | null | undefined => {
  return useMemo(() => {
    if (!token) {
      return token;
    }
    if (valueStr === null || valueStr === undefined) {
      return valueStr;
    }
    if (!valueStr) {
      return null;
    }
    try {
      return TokenAmount.parse(token, valueStr);
    } catch (e) {
      return null;
    }
  }, [token, valueStr]);
};