MyCrypto/common/libs/wallet/privkey.js

70 lines
1.7 KiB
JavaScript

// @flow
import BaseWallet from './base';
import {
privateToPublic,
publicToAddress,
toChecksumAddress
} from 'ethereumjs-util';
import { randomBytes } from 'crypto';
import { pkeyToKeystore } from 'libs/keystore';
import { signRawTxWithPrivKey, signMessageWithPrivKey } from 'libs/signing';
import type { RawTx } from 'libs/validators';
export default class PrivKeyWallet extends BaseWallet {
privKey: Buffer;
pubKey: Buffer;
address: Buffer;
constructor(privkey: Buffer) {
super();
this.privKey = privkey;
this.pubKey = privateToPublic(this.privKey);
this.address = publicToAddress(this.pubKey);
}
getAddress(): Promise<any> {
return Promise.resolve(
toChecksumAddress(`0x${this.address.toString('hex')}`)
);
}
getPrivateKey() {
return this.privKey.toString('hex');
}
static generate() {
return new PrivKeyWallet(randomBytes(32));
}
toKeystore(password: string): Promise<any> {
return new Promise(resolve => {
this.getNakedAddress().then(address => {
resolve(pkeyToKeystore(this.privKey, address, password));
});
});
}
unlock(): Promise<any> {
return Promise.resolve();
}
signRawTransaction(rawTx: RawTx): Promise<any> {
return new Promise((resolve, reject) => {
try {
resolve(signRawTxWithPrivKey(this.privKey, rawTx));
} catch (err) {
reject(err);
}
});
}
signMessage(msg: string, address: string, date: string): Promise<any> {
return new Promise((resolve, reject) => {
try {
resolve(signMessageWithPrivKey(this.privKey, msg, address, date));
} catch (err) {
reject(err);
}
});
}
}