mirror of
https://github.com/lightningnetwork/lnd.git
synced 2024-11-19 09:53:54 +01:00
85 lines
2.6 KiB
Go
85 lines
2.6 KiB
Go
//go:build walletrpc
|
|
// +build walletrpc
|
|
|
|
package walletrpc
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/lightningnetwork/lnd/lnrpc"
|
|
)
|
|
|
|
// createNewSubServer is a helper method that will create the new WalletKit RPC
|
|
// sub server given the main config dispatcher method. If we're unable to find
|
|
// the config that is meant for us in the config dispatcher, then we'll exit
|
|
// with an error.
|
|
func createNewSubServer(configRegistry lnrpc.SubServerConfigDispatcher) (
|
|
*WalletKit, lnrpc.MacaroonPerms, error) {
|
|
|
|
// We'll attempt to look up the config that we expect, according to our
|
|
// SubServerName name. If we can't find this, then we'll exit with an
|
|
// error, as we're unable to properly initialize ourselves without this
|
|
// config.
|
|
walletKitServerConf, ok := configRegistry.FetchConfig(SubServerName)
|
|
if !ok {
|
|
return nil, nil, fmt.Errorf("unable to find config for "+
|
|
"subserver type %s", SubServerName)
|
|
}
|
|
|
|
// Now that we've found an object mapping to our service name, we'll
|
|
// ensure that it's the type we need.
|
|
config, ok := walletKitServerConf.(*Config)
|
|
if !ok {
|
|
return nil, nil, fmt.Errorf("wrong type of config for "+
|
|
"subserver %s, expected %T got %T", SubServerName,
|
|
&Config{}, walletKitServerConf)
|
|
}
|
|
|
|
// Before we try to make the new WalletKit service instance, we'll
|
|
// perform some sanity checks on the arguments to ensure that they're
|
|
// usable.
|
|
switch {
|
|
case config.MacService != nil && config.NetworkDir == "":
|
|
return nil, nil, fmt.Errorf("NetworkDir must be set to " +
|
|
"create WalletKit RPC server")
|
|
|
|
case config.FeeEstimator == nil:
|
|
return nil, nil, fmt.Errorf("FeeEstimator must be set to " +
|
|
"create WalletKit RPC server")
|
|
|
|
case config.Wallet == nil:
|
|
return nil, nil, fmt.Errorf("Wallet must be set to create " +
|
|
"WalletKit RPC server")
|
|
|
|
case config.KeyRing == nil:
|
|
return nil, nil, fmt.Errorf("KeyRing must be set to create " +
|
|
"WalletKit RPC server")
|
|
|
|
case config.Sweeper == nil:
|
|
return nil, nil, fmt.Errorf("Sweeper must be set to create " +
|
|
"WalletKit RPC server")
|
|
|
|
case config.Chain == nil:
|
|
return nil, nil, fmt.Errorf("Chain must be set to create " +
|
|
"WalletKit RPC server")
|
|
}
|
|
|
|
return New(config)
|
|
}
|
|
|
|
func init() {
|
|
subServer := &lnrpc.SubServerDriver{
|
|
SubServerName: SubServerName,
|
|
NewGrpcHandler: func() lnrpc.GrpcHandler {
|
|
return &ServerShell{}
|
|
},
|
|
}
|
|
|
|
// If the build tag is active, then we'll register ourselves as a
|
|
// sub-RPC server within the global lnrpc package namespace.
|
|
if err := lnrpc.RegisterSubServer(subServer); err != nil {
|
|
panic(fmt.Sprintf("failed to register sub server driver '%s': %v",
|
|
SubServerName, err))
|
|
}
|
|
}
|