import { PublicKey, Connection } from "@uttacoin/web3.js";
import { getHashedName, getNameAccountKey } from "./utils";
import { NameRegistryState, TokenData, Mint } from "./state";

export const TOKEN_TLD = new PublicKey(
  "6NSu2tci4apRKQtt257bAVcvqYjB3zV2H1dWo56vgpa6"
);

export const getTokenInfoFromMint = async (
  connection: Connection,
  mint: PublicKey
) => {
  const nameKey = await getNameAccountKey(
    await getHashedName(mint.toBase58()),
    undefined,
    TOKEN_TLD
  );
  const registry = await NameRegistryState.retrieve(connection, nameKey);
  if (!registry.data) {
    throw new Error("Invalid account data");
  }
  return TokenData.deserialize(registry.data);
};

export const getTokenInfoFromName = async (
  connection: Connection,
  name: string
) => {
  const reverseNameKey = await getNameAccountKey(
    await getHashedName(name),
    undefined,
    TOKEN_TLD
  );
  const reverseRegistry = await NameRegistryState.retrieve(
    connection,
    reverseNameKey
  );
  if (!reverseRegistry.data) {
    throw new Error("Invalid account data");
  }
  const mint = new PublicKey(Mint.deserialize(reverseRegistry.data).mint);
  return await getTokenInfoFromMint(connection, mint);
};