* Copyright 2019 ConsenSys AG.
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 * http://www.apache.org/licenses/LICENSE-2.0
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
package tech.pegasys.ethsigner.core.requesthandler.sendtransaction.transaction;

import tech.pegasys.ethsigner.core.jsonrpc.JsonRpcRequest;
import tech.pegasys.ethsigner.core.requesthandler.sendtransaction.NonceProvider;

import java.math.BigInteger;

import org.web3j.utils.Base64String;

public class BesuPrivateNonceProvider implements NonceProvider {

  private final VertxNonceRequestTransmitter vertxNonceRequestTransmitter;
  private final String accountAddress;
  private final Base64String privacyGroupId;

  public BesuPrivateNonceProvider(
      final String accountAddress,
      final Base64String privacyGroupId,
      final VertxNonceRequestTransmitter vertxNonceRequestTransmitter) {
    this.vertxNonceRequestTransmitter = vertxNonceRequestTransmitter;
    this.accountAddress = accountAddress;
    this.privacyGroupId = privacyGroupId;

  public BigInteger getNonce() {
    final JsonRpcRequest request = generateRequest();
    return vertxNonceRequestTransmitter.requestNonce(request);

  protected JsonRpcRequest generateRequest() {
    final JsonRpcRequest request = new JsonRpcRequest("2.0", "priv_getTransactionCount");
    request.setParams(new Object[] {accountAddress, privacyGroupId});
    return request;