Tron Blockchain: Javascript (NodeJs) Function to Get the Frozen Balance (Staked TRX) for Energy and Bandwidth

function getFrozonBalance(network, address) {
  network = network.toLowerCase();
  let url = null;
  if (network === "mainnet") {
    url = "https://api.trongrid.io";
  } else if (network === "shasta") {
    url = "https://api.shasta.trongrid.io";
  } else if (network === "nile") {
    url = "https://nile.trongrid.io";
  } 
  const tronWeb = new TronWeb({
    fullHost: url
  });
  return new Promise((resolve, reject) => {
    tronWeb.trx.getAccount(address).then(
      result => {
        let data = {
          "bandwidth": 0,
          "energy": 0
        }
        if (!result || typeof result === "undefined") {
          resolve(data);
        }
        if (result.frozen != null && typeof result.frozen !== "undefined") {
          for (var i = 0; i < result.frozen.length; ++ i) {
            data["bandwidth"] += parseFloat(tronWeb.fromSun(result.frozen[i].frozen_balance));
          }
        }
        if (result.account_resource != null && typeof result.account_resource !== "undefined") {
          if (result.account_resource.frozen_balance_for_energy != null && typeof result.account_resource.frozen_balance_for_energy !== "undefined") {
            data["energy"] += parseFloat(tronWeb.fromSun(result.account_resource.frozen_balance_for_energy.frozen_balance));
          }
        }
        resolve(data);
      }
    );
  });
}