info.bitrich.xchangestream.bitget.BitgetStreamingAuthHelper Maven / Gradle / Ivy

Go to download
package info.bitrich.xchangestream.bitget;

import java.nio.charset.StandardCharsets;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.time.Instant;
import java.util.Base64;
import javax.crypto.Mac;
import javax.crypto.SecretKey;
import javax.crypto.spec.SecretKeySpec;
import lombok.experimental.UtilityClass;
import org.knowm.xchange.service.BaseParamsDigest;

@UtilityClass
public class BitgetStreamingAuthHelper {

  /** Generates signature based on timestamp */
  public String sign(Instant timestamp, String secretString) {
    final SecretKey secretKey =
        new SecretKeySpec(
            secretString.getBytes(StandardCharsets.UTF_8), BaseParamsDigest.HMAC_SHA_256);
    Mac mac = createMac(secretKey, secretKey.getAlgorithm());

    String payloadToSign = String.format("%sGET/user/verify", timestamp.getEpochSecond());
    mac.update(payloadToSign.getBytes(StandardCharsets.UTF_8));

    return Base64.getEncoder().encodeToString(mac.doFinal());
  }

  private Mac createMac(SecretKey secretKey, String hmacString) {
    try {
      Mac mac = Mac.getInstance(hmacString);
      mac.init(secretKey);
      return mac;
    } catch (InvalidKeyException e) {
      throw new IllegalArgumentException("Invalid key for hmac initialization.", e);
    } catch (NoSuchAlgorithmException e) {
      throw new RuntimeException(
          "Illegal algorithm for post body digest. Check the implementation.");
    }
  }
}