JimmyMow
7 years ago
committed by
GitHub
GitHub
10 changed files with 190 additions and 91 deletions
@ -1,36 +1,118 @@ |
|||||
import fs from 'fs' |
// @flow
|
||||
|
|
||||
import grpc from 'grpc' |
import grpc from 'grpc' |
||||
import { loadSync } from '@grpc/proto-loader' |
import { loadSync } from '@grpc/proto-loader' |
||||
import walletUnlockerMethods from './walletUnlockerMethods' |
import StateMachine from 'javascript-state-machine' |
||||
|
import LndConfig from './config' |
||||
|
import { getDeadline, validateHost, createSslCreds, createMacaroonCreds } from './util' |
||||
|
import methods from './walletUnlockerMethods' |
||||
import { mainLog } from '../utils/log' |
import { mainLog } from '../utils/log' |
||||
|
|
||||
export const initWalletUnlocker = lndConfig => { |
/** |
||||
const walletUnlockerObj = walletUnlocker(lndConfig) |
* Creates an LND grpc client lightning service. |
||||
const walletUnlockerMethodsCallback = (event, msg, data) => |
* @returns {WalletUnlocker} |
||||
walletUnlockerMethods(lndConfig, walletUnlockerObj, mainLog, event, msg, data) |
*/ |
||||
|
class WalletUnlocker { |
||||
|
service: any |
||||
|
lndConfig: LndConfig |
||||
|
_fsm: StateMachine |
||||
|
|
||||
return walletUnlockerMethodsCallback |
// Transitions provided by the state machine.
|
||||
} |
connect: any |
||||
|
disconnect: any |
||||
|
terminate: any |
||||
|
is: any |
||||
|
can: any |
||||
|
state: string |
||||
|
|
||||
|
constructor(lndConfig: LndConfig) { |
||||
|
this.service = null |
||||
|
this.lndConfig = lndConfig |
||||
|
|
||||
|
// Initialize the state machine.
|
||||
|
this._fsm() |
||||
|
} |
||||
|
|
||||
|
// ------------------------------------
|
||||
|
// FSM Callbacks
|
||||
|
// ------------------------------------
|
||||
|
|
||||
|
/** |
||||
|
* Connect to the gRPC interface and verify it is functional. |
||||
|
* @return {Promise<rpc.lnrpc.WalletUnlocker>} |
||||
|
*/ |
||||
|
async onBeforeConnect() { |
||||
|
mainLog.info('Connecting to WalletUnlocker gRPC service') |
||||
|
const { rpcProtoPath, host, cert, macaroon } = this.lndConfig |
||||
|
|
||||
|
// Verify that the host is valid before creating a gRPC client that is connected to it.
|
||||
|
return await validateHost(host).then(async () => { |
||||
|
// Load the gRPC proto file.
|
||||
|
// The following options object closely approximates the existing behavior of grpc.load.
|
||||
|
// See https://github.com/grpc/grpc-node/blob/master/packages/grpc-protobufjs/README.md
|
||||
|
const options = { |
||||
|
keepCase: true, |
||||
|
longs: Number, |
||||
|
enums: String, |
||||
|
defaults: true, |
||||
|
oneofs: true |
||||
|
} |
||||
|
const packageDefinition = loadSync(rpcProtoPath, options) |
||||
|
|
||||
export const walletUnlocker = lndConfig => { |
// Load gRPC package definition as a gRPC object hierarchy.
|
||||
const lndCert = fs.readFileSync(lndConfig.cert) |
const rpc = grpc.loadPackageDefinition(packageDefinition) |
||||
const credentials = grpc.credentials.createSsl(lndCert) |
|
||||
|
// Create ssl and macaroon credentials to use with the gRPC client.
|
||||
// Load the gRPC proto file.
|
const [sslCreds, macaroonCreds] = await Promise.all([ |
||||
// The following options object closely approximates the existing behavior of grpc.load
|
createSslCreds(cert), |
||||
// See https://github.com/grpc/grpc-node/blob/master/packages/grpc-protobufjs/README.md
|
createMacaroonCreds(macaroon) |
||||
const options = { |
]) |
||||
keepCase: true, |
const credentials = grpc.credentials.combineChannelCredentials(sslCreds, macaroonCreds) |
||||
longs: Number, |
|
||||
enums: String, |
// Create a new gRPC client instance.
|
||||
defaults: true, |
this.service = new rpc.lnrpc.WalletUnlocker(host, credentials) |
||||
oneofs: true |
|
||||
|
// Wait for the gRPC connection to be established.
|
||||
|
return new Promise((resolve, reject) => { |
||||
|
this.service.waitForReady(getDeadline(5), err => { |
||||
|
if (err) { |
||||
|
this.service.close() |
||||
|
return reject(err) |
||||
|
} |
||||
|
return resolve() |
||||
|
}) |
||||
|
}) |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* Discomnnect the gRPC service. |
||||
|
*/ |
||||
|
onBeforeDisconnect() { |
||||
|
mainLog.info('Disconnecting from WalletUnlocker gRPC service') |
||||
|
if (this.service) { |
||||
|
this.service.close() |
||||
|
} |
||||
} |
} |
||||
const packageDefinition = loadSync(lndConfig.rpcProtoPath, options) |
|
||||
|
|
||||
// Load gRPC package definition as a gRPC object hierarchy.
|
// ------------------------------------
|
||||
const rpc = grpc.loadPackageDefinition(packageDefinition) |
// Helpers
|
||||
|
// ------------------------------------
|
||||
|
|
||||
// Instantiate a new connection to the WalletUnlocker interface.
|
/** |
||||
return new rpc.lnrpc.WalletUnlocker(lndConfig.host, credentials) |
* Hook up lnd restful methods. |
||||
|
*/ |
||||
|
registerMethods(event: Event, msg: string, data: any) { |
||||
|
return methods(this.service, mainLog, event, msg, data, this.lndConfig) |
||||
|
} |
||||
} |
} |
||||
|
|
||||
|
StateMachine.factory(WalletUnlocker, { |
||||
|
init: 'ready', |
||||
|
transitions: [ |
||||
|
{ name: 'connect', from: 'ready', to: 'connected' }, |
||||
|
{ name: 'disconnect', from: 'connected', to: 'ready' } |
||||
|
] |
||||
|
}) |
||||
|
|
||||
|
export default WalletUnlocker |
||||
|
|||||
Loading…
Reference in new issue