org.knowm.xchange.kraken.service.KrakenAccountServiceRaw Maven / Gradle / Ivy

Go to download
package org.knowm.xchange.kraken.service;

import java.io.IOException;
import java.math.BigDecimal;
import java.util.Date;
import java.util.List;
import java.util.Map;
import org.knowm.xchange.Exchange;
import org.knowm.xchange.currency.Currency;
import org.knowm.xchange.currency.CurrencyPair;
import org.knowm.xchange.kraken.KrakenUtils;
import org.knowm.xchange.kraken.dto.account.*;
import org.knowm.xchange.kraken.dto.account.results.*;
import org.knowm.xchange.utils.DateUtils;

/** @author jamespedwards42 */
public class KrakenAccountServiceRaw extends KrakenBaseService {

  /**
   * Constructor
   *
   * @param exchange
   */
  public KrakenAccountServiceRaw(Exchange exchange) {

    super(exchange);
  }

  /**
   * To avoid having to map to Kraken currency codes (e.g., ZUSD) use {@link
   * KrakenAccountService#getAccountInfo} instead.
   *
   * @return Map of Kraken Assets to account balance
   * @throws IOException
   */
  public Map getKrakenBalance() throws IOException {

    KrakenBalanceResult balanceResult =
        kraken.balance(
            exchange.getExchangeSpecification().getApiKey(),
            signatureCreator,
            exchange.getNonceFactory());
    return checkResult(balanceResult);
  }

  public KrakenDepositAddress[] getDepositAddresses(
      String currency, String method, boolean newAddress) throws IOException {
    KrakenDepositAddressResult depositAddressesResult =
        kraken.getDepositAddresses(
            null,
            currency,
            method,
            newAddress ? true : null, // must be sent as NULL and not as false!
            exchange.getExchangeSpecification().getApiKey(),
            signatureCreator,
            exchange.getNonceFactory());
    return checkResult(depositAddressesResult);
  }

  public KrakenDepositMethods[] getDepositMethods(String assetPairs, String assets)
      throws IOException {
    KrakenDepositMethodsResults depositMethods =
        kraken.getDepositMethods(
            assetPairs,
            assets,
            exchange.getExchangeSpecification().getApiKey(),
            signatureCreator,
            exchange.getNonceFactory());
    return checkResult(depositMethods);
  }

  public WithdrawInfo getWithdrawInfo(
      String assetPairs, String assets, String key, BigDecimal amount) throws IOException {
    WithdrawInfoResult withdrawInfoResult =
        kraken.getWithdrawInfo(
            assetPairs,
            assets,
            key,
            amount,
            exchange.getExchangeSpecification().getApiKey(),
            signatureCreator,
            exchange.getNonceFactory());
    return checkResult(withdrawInfoResult);
  }

  public Withdraw withdraw(String assetPairs, String assets, String key, BigDecimal amount)
      throws IOException {
    WithdrawResult withdrawResult =
        kraken.withdraw(
            assetPairs,
            assets,
            key,
            amount,
            exchange.getExchangeSpecification().getApiKey(),
            signatureCreator,
            exchange.getNonceFactory());
    return checkResult(withdrawResult);
  }

  public List getDepositStatus(String assetPairs, String assets, String method)
      throws IOException {
    DepositStatusResult result =
        kraken.getDepositStatus(
            assetPairs,
            assets,
            method,
            exchange.getExchangeSpecification().getApiKey(),
            signatureCreator,
            exchange.getNonceFactory());
    return checkResult(result);
  }

  public List getWithdrawStatus(String assetPairs, String assets, String method)
      throws IOException {
    WithdrawStatusResult result =
        kraken.getWithdrawStatus(
            assetPairs,
            assets,
            method,
            exchange.getExchangeSpecification().getApiKey(),
            signatureCreator,
            exchange.getNonceFactory());
    return checkResult(result);
  }

  /**
   * @param valuationCurrency - Base asset used to determine balance (can be null, defaults to USD).
   *     The asset should be provided in the form of a standard currency code, i.e., EUR. It will be
   *     converted to the appropriate Kraken Asset code.
   * @return KrakenTradeBalanceInfo
   * @throws IOException
   */
  public KrakenTradeBalanceInfo getKrakenTradeBalance(Currency valuationCurrency)
      throws IOException {

    String valuationCurrencyCode = null;

    if (valuationCurrency != null) {
      valuationCurrencyCode = KrakenUtils.getKrakenCurrencyCode(valuationCurrency);
    }

    KrakenTradeBalanceInfoResult balanceResult =
        kraken.tradeBalance(
            null,
            valuationCurrencyCode,
            exchange.getExchangeSpecification().getApiKey(),
            signatureCreator,
            exchange.getNonceFactory());
    return checkResult(balanceResult);
  }

  /**
   * Retrieves the user's trade balance using the default currency ZUSD to determine the balance.
   *
   * @return KrakenTradeBalanceInfo
   * @throws IOException
   */
  public KrakenTradeBalanceInfo getKrakenTradeBalance() throws IOException {

    return getKrakenTradeBalance(null);
  }

  /**
   * Retrieves the full account Ledger which represents all account asset activity.
   *
   * @return
   * @throws IOException
   */
  public Map getKrakenLedgerInfo() throws IOException {

    return getKrakenLedgerInfo(null, null, null, null);
  }

  /**
   * Retrieves a fraction of the ledger entries (usually a "page" of 50 entries) between the start
   * date and the end date. The ledger records the activity (trades, deposit, withdrawals) of the
   * account for all assets.
   *
   * @param assets Set of assets to restrict output to (can be null, defaults to all)
   * @param ledgerType {@link LedgerType} to retrieve (can be null, defaults to all types)
   * @param startTime Start Unix timestamp or ledger id of results (can be null)
   * @param endTime End Unix timestamp or ledger id of results (can be null)
   * @param offset Result offset (can be null)
   * @return
   * @throws IOException
   */
  public Map getKrakenPartialLedgerInfo(
      LedgerType ledgerType, String startTime, String endTime, Long offset, Currency... assets)
      throws IOException {
    String ledgerTypeString = (ledgerType == null) ? "all" : ledgerType.toString().toLowerCase();
    KrakenLedgerResult ledgerResult =
        kraken.ledgers(
            null,
            delimitAssets(assets),
            ledgerTypeString,
            startTime,
            endTime,
            offset,
            exchange.getExchangeSpecification().getApiKey(),
            signatureCreator,
            exchange.getNonceFactory());
    return checkResult(ledgerResult).getLedgerMap();
  }

  /**
   * Retrieves all ledger entries between the start date and the end date. This method iterates over
   * ledger pages until it has retrieved all entries between the start date and the end date. The
   * ledger records the activity (trades, deposit, withdrawals) of the account for all assets.
   *
   * @param assets Set of assets to restrict output to (can be null, defaults to all)
   * @param ledgerType {@link LedgerType} to retrieve (can be null, defaults to all types)
   * @param start Start unix timestamp or ledger id of results (can be null)
   * @param end End unix timestamp or ledger id of results (can be null)
   * @param offset Result offset (can be null)
   * @return
   * @throws IOException
   */
  public Map getKrakenLedgerInfo(
      LedgerType ledgerType, Date start, Date end, Long offset, Currency... assets)
      throws IOException {

    String startTime = null;
    String endTime = null;
    long longOffset = 0;

    if (start != null) {
      startTime = String.valueOf(DateUtils.toUnixTime(start));
    }
    if (end != null) {
      endTime = String.valueOf(DateUtils.toUnixTime(end));
    }
    if (offset != null) {
      longOffset = offset;
    }

    Map fullLedgerMap =
        getKrakenPartialLedgerInfo(ledgerType, startTime, endTime, offset, assets);
    Map lastLedgerMap = fullLedgerMap;

    while (!lastLedgerMap.isEmpty()) {
      longOffset += lastLedgerMap.size();
      lastLedgerMap =
          getKrakenPartialLedgerInfo(ledgerType, startTime, endTime, longOffset, assets);
      if (lastLedgerMap.size() == 1 && fullLedgerMap.keySet().containsAll(lastLedgerMap.keySet())) {
        break;
      }
      fullLedgerMap.putAll(lastLedgerMap);
    }
    return fullLedgerMap;
  }

  public KrakenWebsocketToken getKrakenWebsocketToken() throws IOException {
    KrakenWebsocketTokenResult tokenResult =
        kraken.getWebsocketToken(
            null,
            null,
            exchange.getExchangeSpecification().getApiKey(),
            signatureCreator,
            exchange.getNonceFactory());

    return checkResult(tokenResult);
  }

  public Map queryKrakenLedger(String... ledgerIds) throws IOException {

    KrakenQueryLedgerResult ledgerResult =
        kraken.queryLedgers(
            createDelimitedString(ledgerIds),
            exchange.getExchangeSpecification().getApiKey(),
            signatureCreator,
            exchange.getNonceFactory());

    return checkResult(ledgerResult);
  }

  public KrakenTradeVolume getTradeVolume(CurrencyPair... currencyPairs) throws IOException {
    KrakenTradeVolumeResult result =
        kraken.tradeVolume(
            delimitAssetPairs(currencyPairs),
            exchange.getExchangeSpecification().getApiKey(),
            signatureCreator,
            exchange.getNonceFactory());
    return checkResult(result);
  }
}