org.fisco.bcos.web3j.console.WalletManager Maven / Gradle / Ivy

Go to download
package org.fisco.bcos.web3j.console;

import static org.fisco.bcos.web3j.utils.Console.exitError;

import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import org.fisco.bcos.web3j.crypto.CipherException;
import org.fisco.bcos.web3j.crypto.Credentials;
import org.fisco.bcos.web3j.crypto.WalletUtils;
import org.fisco.bcos.web3j.utils.Console;

/** Common functions used by the wallet console tools. */
abstract class WalletManager {

  final IODevice console;

  WalletManager() {
    console = new ConsoleDevice();
  }

  WalletManager(IODevice console) {
    this.console = console;
  }

  String getPassword(String initialPrompt) {
    while (true) {
      char[] input1 = console.readPassword(initialPrompt);
      char[] input2 = console.readPassword("Please re-enter the password: ");

      if (Arrays.equals(input1, input2)) {
        String s = new String(input1);
        Arrays.fill(input1, ' ');
        Arrays.fill(input2, ' ');
        return s;
      } else {
        console.printf("Sorry, passwords did not match\n");
      }
    }
  }

  String getDestinationDir() {
    String defaultDir = WalletUtils.getTestnetKeyDirectory();
    String destinationDir =
        console.readLine("Please enter a destination directory location [" + defaultDir + "]: ");
    if (destinationDir.equals("")) {
      return defaultDir;
    } else if (destinationDir.startsWith("~")) {
      return System.getProperty("user.home") + destinationDir.substring(1);
    } else {
      return destinationDir;
    }
  }

  File createDir(String destinationDir) {
    File destination = new File(destinationDir);

    if (!destination.exists()) {
      console.printf("Creating directory: " + destinationDir + " ...");
      if (!destination.mkdirs()) {
        Console.exitError(
            "Unable to create destination directory [" + destinationDir + "], exiting...");
      } else {
        console.printf("complete\n");
      }
    }

    return destination;
  }

  Credentials getCredentials(File walletFile) {
    if (!walletFile.exists() || !walletFile.isFile()) {
      exitError("Unable to read wallet file: " + walletFile);
    }
    return loadWalletFile(walletFile);
  }

  private Credentials loadWalletFile(File walletFile) {
    while (true) {
      char[] password = console.readPassword("Please enter your existing wallet file password: ");
      String currentPassword = new String(password);
      Arrays.fill(password, ' ');
      try {
        return WalletUtils.loadCredentials(currentPassword, walletFile);
      } catch (CipherException e) {
        console.printf("Invalid password specified\n");
      } catch (IOException e) {
        exitError("Unable to load wallet file: " + walletFile + "\n" + e.getMessage());
      }
    }
  }
}