pub fun main(): UFix64 {
  return getBalance(0x01)
}
pub fun getBalance(_ address: Address): UFix64 {
  let account = getAccount(address)
  return account.balance
}

// Snippet from FlowServiceAccount that is called when calling `.balance` through a direct contract function invoke

pub fun defaultTokenBalance(_ acct: PublicAccount): UFix64 {
    var balance = 0.0
    if let balanceRef = acct
        .getCapability(/public/flowTokenBalance)
        .borrow<&FlowToken.Vault{FungibleToken.Balance}>(){               // <---- lets say you are here
            balance = balanceRef.balance
        }

    return balance
}