mirror of
https://github.com/lightningnetwork/lnd.git
synced 2025-02-24 14:50:40 +01:00
ae2e9541a4
We'll want to re-use the PSBT funding functionality from the wallet kit sub server in the main RPC server. To be able to dynamically obtain an instance of the wallet kit server we need to know its name under which it registers itself in the list of sub server. We export the name so we don't have to hard code it in the main server.
83 lines
2.5 KiB
Go
83 lines
2.5 KiB
Go
// +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
|
|
// useable.
|
|
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))
|
|
}
|
|
}
|