Ben Woosley
7 years ago
committed by
GitHub
GitHub
6 changed files with 414 additions and 361 deletions
@ -0,0 +1,85 @@ |
|||||
|
import split2 from 'split2' |
||||
|
import { spawn } from 'child_process' |
||||
|
import EventEmitter from 'events' |
||||
|
import config from '../config' |
||||
|
import { mainLog, lndLog, lndLogGetLevel } from '../../utils/log' |
||||
|
|
||||
|
class Neutrino extends EventEmitter { |
||||
|
constructor(alias, autopilot) { |
||||
|
super() |
||||
|
this.alias = alias |
||||
|
this.autopilot = autopilot |
||||
|
this.process = null |
||||
|
} |
||||
|
|
||||
|
start() { |
||||
|
if (this.process) { |
||||
|
throw new Error('Neutrino process with PID ${this.process.pid} already exists.') |
||||
|
} |
||||
|
|
||||
|
const lndConfig = config.lnd() |
||||
|
mainLog.info('Starting lnd in neutrino mode') |
||||
|
mainLog.debug(' > lndPath', lndConfig.lndPath) |
||||
|
mainLog.debug(' > lightningRpc:', lndConfig.lightningRpc) |
||||
|
mainLog.debug(' > lightningHost:', lndConfig.lightningHost) |
||||
|
mainLog.debug(' > cert:', lndConfig.cert) |
||||
|
mainLog.debug(' > macaroon:', lndConfig.macaroon) |
||||
|
|
||||
|
const neutrinoArgs = [ |
||||
|
`--configfile=${lndConfig.configPath}`, |
||||
|
`${this.autopilot ? '--autopilot.active' : ''}`, |
||||
|
`${this.alias ? `--alias=${this.alias}` : ''}` |
||||
|
] |
||||
|
|
||||
|
this.process = spawn(lndConfig.lndPath, neutrinoArgs) |
||||
|
.on('error', error => this.emit('error', error)) |
||||
|
.on('close', code => { |
||||
|
this.emit('close', code) |
||||
|
this.process = null |
||||
|
}) |
||||
|
|
||||
|
// Listen for when neutrino prints odata to stderr.
|
||||
|
this.process.stderr.pipe(split2()).on('data', line => { |
||||
|
if (process.env.NODE_ENV === 'development') { |
||||
|
lndLog[lndLogGetLevel(line)](line) |
||||
|
} |
||||
|
}) |
||||
|
|
||||
|
// Listen for when neutrino prints data to stdout.
|
||||
|
this.process.stdout.pipe(split2()).on('data', line => { |
||||
|
if (process.env.NODE_ENV === 'development') { |
||||
|
lndLog[lndLogGetLevel(line)](line) |
||||
|
} |
||||
|
|
||||
|
// gRPC started.
|
||||
|
if (line.includes('gRPC proxy started') && line.includes('password')) { |
||||
|
this.emit('grpc-proxy-started') |
||||
|
} |
||||
|
|
||||
|
// Wallet opened.
|
||||
|
if (line.includes('gRPC proxy started') && !line.includes('password')) { |
||||
|
this.emit('wallet-opened') |
||||
|
} |
||||
|
|
||||
|
// LND is all caught up to the blockchain.
|
||||
|
if (line.includes('Chain backend is fully synced')) { |
||||
|
this.emit('fully-synced') |
||||
|
} |
||||
|
|
||||
|
// Pass current block height progress to front end for loading state UX
|
||||
|
if (line.includes('Caught up to height') || line.includes('Catching up block hashes')) { |
||||
|
this.emit('got-block-height', line) |
||||
|
} |
||||
|
}) |
||||
|
return this.process |
||||
|
} |
||||
|
|
||||
|
stop() { |
||||
|
if (this.process) { |
||||
|
this.process.kill() |
||||
|
this.process = null |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
export default Neutrino |
||||
@ -0,0 +1,28 @@ |
|||||
|
import { lookup } from 'ps-node' |
||||
|
import { mainLog } from '../../utils/log' |
||||
|
|
||||
|
/** |
||||
|
* Check to see if an LND process is running. |
||||
|
* @return {Promise} Boolean indicating wether an existing lnd process was found on the host machine. |
||||
|
*/ |
||||
|
const isLndRunning = () => { |
||||
|
return new Promise((resolve, reject) => { |
||||
|
mainLog.info('Looking for existing lnd process') |
||||
|
lookup({ command: 'lnd' }, (err, results) => { |
||||
|
// There was an error checking for the LND process.
|
||||
|
if (err) { |
||||
|
return reject(err) |
||||
|
} |
||||
|
|
||||
|
if (!results.length) { |
||||
|
// An LND process was found, no need to start our own.
|
||||
|
mainLog.info('Existing lnd process not found') |
||||
|
return resolve(false) |
||||
|
} |
||||
|
mainLog.info('Found existing lnd process') |
||||
|
return resolve(true) |
||||
|
}) |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
export default isLndRunning |
||||
@ -0,0 +1,217 @@ |
|||||
|
import { app, ipcMain, dialog } from 'electron' |
||||
|
import Store from 'electron-store' |
||||
|
import lnd from './lnd' |
||||
|
import Neutrino from './lnd/lib/neutrino' |
||||
|
import { mainLog } from './utils/log' |
||||
|
|
||||
|
/** |
||||
|
* @class ZapController |
||||
|
* |
||||
|
* The ZapController class coordinates actions between the the main nand renderer processes. |
||||
|
*/ |
||||
|
class ZapController { |
||||
|
/** |
||||
|
* Create a new ZapController instance. |
||||
|
* @param {BrowserWindow} mainWindow BrowserWindow instance to interact with |
||||
|
* @param {String|Promise} mode String or Promise that resolves to the desired run mode. Valid options are: |
||||
|
* - 'internal': start a new lnd process. |
||||
|
* - 'external': connect to an existing lnd process. |
||||
|
*/ |
||||
|
constructor(mainWindow, mode) { |
||||
|
this.mode = mode |
||||
|
|
||||
|
// Variable to hold the main window instance.
|
||||
|
this.mainWindow = mainWindow |
||||
|
|
||||
|
// Keep a reference any neutrino process started by us.
|
||||
|
this.neutrino = null |
||||
|
|
||||
|
// Time for the splash screen to remain visible.
|
||||
|
this.splashScreenTime = 500 |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* Initialize the application. |
||||
|
*/ |
||||
|
init() { |
||||
|
if (process.env.HOT) { |
||||
|
const port = process.env.PORT || 1212 |
||||
|
this.mainWindow.loadURL(`http://localhost:${port}/dist/index.html`) |
||||
|
} else { |
||||
|
this.mainWindow.loadURL(`file://${__dirname}/dist/index.html`) |
||||
|
} |
||||
|
|
||||
|
// Register IPC listeners so that we can react to instructions coming from the app.
|
||||
|
this._registerIpcListeners() |
||||
|
|
||||
|
// Show the window as soon as the application has finished loading.
|
||||
|
this.mainWindow.webContents.on('did-finish-load', async () => { |
||||
|
this.mainWindow.show() |
||||
|
this.mainWindow.focus() |
||||
|
mainLog.timeEnd('Time until app is visible') |
||||
|
mainLog.time('Time until we know the run mode') |
||||
|
|
||||
|
Promise.resolve(this.mode) |
||||
|
.then(mode => { |
||||
|
const timeUntilWeKnowTheRunMode = mainLog.timeEnd('Time until we know the run mode') |
||||
|
return setTimeout(() => { |
||||
|
if (mode === 'external') { |
||||
|
// If lnd is already running, create and subscribe to the Lightning grpc object.
|
||||
|
this.startGrpc() |
||||
|
this.sendMessage('successfullyCreatedWallet') |
||||
|
} else { |
||||
|
// Otherwise, start the onboarding process.
|
||||
|
this.sendMessage('startOnboarding') |
||||
|
mainLog.timeEnd('Time until onboarding has started') |
||||
|
} |
||||
|
}, timeUntilWeKnowTheRunMode < this.splashScreenTime ? this.splashScreenTime : 0) |
||||
|
}) |
||||
|
.catch(mainLog.error) |
||||
|
}) |
||||
|
|
||||
|
this.mainWindow.on('closed', () => { |
||||
|
this.mainWindow = null |
||||
|
|
||||
|
// shut down zap when a user closes the window
|
||||
|
app.quit() |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* Send a message to the main window. |
||||
|
* @param {string} msg message to send. |
||||
|
* @param {[type]} data additional data to acompany the message. |
||||
|
*/ |
||||
|
sendMessage(msg, data) { |
||||
|
mainLog.info('Sending message to renderer process: %o', { msg, data }) |
||||
|
this.mainWindow.webContents.send(msg, data) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* Create and subscribe to the Lightning grpc object. |
||||
|
*/ |
||||
|
startGrpc() { |
||||
|
mainLog.info('Starting gRPC...') |
||||
|
try { |
||||
|
const { lndSubscribe, lndMethods } = lnd.initLnd() |
||||
|
|
||||
|
// Subscribe to bi-directional streams
|
||||
|
lndSubscribe(this.mainWindow) |
||||
|
|
||||
|
// Listen for all gRPC restful methods
|
||||
|
ipcMain.on('lnd', (event, { msg, data }) => { |
||||
|
lndMethods(event, msg, data) |
||||
|
}) |
||||
|
|
||||
|
this.sendMessage('grpcConnected') |
||||
|
} catch (error) { |
||||
|
dialog.showMessageBox({ |
||||
|
type: 'error', |
||||
|
message: `Unable to connect to lnd. Please check your lnd node and try again: ${error}` |
||||
|
}) |
||||
|
app.quit() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* Create and subscribe to the WalletUnlocker grpc object. |
||||
|
*/ |
||||
|
startWalletUnlocker() { |
||||
|
mainLog.info('Starting wallet unlocker...') |
||||
|
try { |
||||
|
const walletUnlockerMethods = lnd.initWalletUnlocker() |
||||
|
|
||||
|
// Listen for all gRPC restful methods
|
||||
|
ipcMain.on('walletUnlocker', (event, { msg, data }) => { |
||||
|
walletUnlockerMethods(event, msg, data) |
||||
|
}) |
||||
|
|
||||
|
this.sendMessage('walletUnlockerStarted') |
||||
|
} catch (error) { |
||||
|
dialog.showMessageBox({ |
||||
|
type: 'error', |
||||
|
message: `Unable to start lnd wallet unlocker. Please check your lnd node and try again: ${error}` |
||||
|
}) |
||||
|
app.quit() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* Starts the LND node and attach event listeners. |
||||
|
* @param {string} alias Alias to assign to the lnd node. |
||||
|
* @param {boolean} autopilot True if autopilot should be enabled. |
||||
|
* @return {Neutrino} Neutrino instance. |
||||
|
*/ |
||||
|
startLnd(alias, autopilot) { |
||||
|
this.neutrino = new Neutrino(alias, autopilot) |
||||
|
|
||||
|
this.neutrino.on('error', error => { |
||||
|
mainLog.error(`Got error from lnd process: ${error})`) |
||||
|
dialog.showMessageBox({ |
||||
|
type: 'error', |
||||
|
message: `lnd error: ${error}` |
||||
|
}) |
||||
|
}) |
||||
|
|
||||
|
this.neutrino.on('close', code => { |
||||
|
mainLog.info(`Lnd process has shut down (code ${code})`) |
||||
|
app.quit() |
||||
|
}) |
||||
|
|
||||
|
this.neutrino.on('grpc-proxy-started', () => { |
||||
|
mainLog.info('gRPC proxy started') |
||||
|
this.startWalletUnlocker() |
||||
|
}) |
||||
|
|
||||
|
this.neutrino.on('wallet-opened', () => { |
||||
|
mainLog.info('Wallet opened') |
||||
|
this.startGrpc() |
||||
|
this.sendMessage('lndSyncing') |
||||
|
}) |
||||
|
|
||||
|
this.neutrino.on('fully-synced', () => { |
||||
|
mainLog.info('Neutrino fully synced') |
||||
|
this.sendMessage('lndSynced') |
||||
|
}) |
||||
|
|
||||
|
this.neutrino.on('got-block-height', line => { |
||||
|
this.sendMessage('lndStdout', line) |
||||
|
}) |
||||
|
|
||||
|
this.neutrino.start() |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* Add IPC event listeners... |
||||
|
*/ |
||||
|
_registerIpcListeners() { |
||||
|
ipcMain.on('startLnd', (event, options = {}) => { |
||||
|
const store = new Store({ name: 'connection' }) |
||||
|
store.store = { |
||||
|
type: options.connectionType, |
||||
|
host: options.connectionHost, |
||||
|
cert: options.connectionCert, |
||||
|
macaroon: options.connectionMacaroon, |
||||
|
alias: options.alias, |
||||
|
autopilot: options.autopilot |
||||
|
} |
||||
|
mainLog.info('Saved lnd config to:', store.path) |
||||
|
|
||||
|
if (options.connectionType === 'local') { |
||||
|
mainLog.info('Starting new lnd instance') |
||||
|
mainLog.debug(' > alias:', options.alias) |
||||
|
mainLog.debug(' > autopilot:', options.autopilot) |
||||
|
this.startLnd(options.alias, options.autopilot) |
||||
|
} else { |
||||
|
mainLog.info('Connecting to custom lnd instance') |
||||
|
mainLog.debug(' > connectionHost:', options.connectionHost) |
||||
|
mainLog.debug(' > connectionCert:', options.connectionCert) |
||||
|
mainLog.debug(' > connectionMacaroon:', options.connectionMacaroon) |
||||
|
this.startGrpc() |
||||
|
this.sendMessage('successfullyCreatedWallet') |
||||
|
} |
||||
|
}) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
export default ZapController |
||||
Loading…
Reference in new issue