450 lines
12 KiB
Go
450 lines
12 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"path"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/btcsuite/btcutil"
|
||
|
|
"github.com/lightningnetwork/lnd/lncfg"
|
||
|
|
"github.com/urfave/cli"
|
||
|
|
"gopkg.in/macaroon.v2"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
// defaultLncliDir is the default directory to store the profile file
|
||
|
|
// in. This defaults to:
|
||
|
|
// C:\Users\<username>\AppData\Local\Lncli\ on Windows
|
||
|
|
// ~/.lncli/ on Linux
|
||
|
|
// ~/Library/Application Support/Lncli/ on MacOS
|
||
|
|
defaultLncliDir = btcutil.AppDataDir("lncli", false)
|
||
|
|
|
||
|
|
// defaultProfileFile is the full, absolute path of the profile file.
|
||
|
|
defaultProfileFile = path.Join(defaultLncliDir, "profiles.json")
|
||
|
|
)
|
||
|
|
|
||
|
|
var profileSubCommand = cli.Command{
|
||
|
|
Name: "profile",
|
||
|
|
Category: "Profiles",
|
||
|
|
Usage: "Create and manage lncli profiles",
|
||
|
|
Description: `
|
||
|
|
Profiles for lncli are an easy and comfortable way to manage multiple
|
||
|
|
nodes from the command line by storing node specific parameters like RPC
|
||
|
|
host, network, TLS certificate path or macaroons in a named profile.
|
||
|
|
|
||
|
|
To use a predefined profile, just use the '--profile=myprofile' (or
|
||
|
|
short version '-p=myprofile') with any lncli command.
|
||
|
|
|
||
|
|
A default profile can also be defined, lncli will then always use the
|
||
|
|
connection/node parameters from that profile instead of the default
|
||
|
|
values.
|
||
|
|
|
||
|
|
WARNING: Setting a default profile changes the default behavior of
|
||
|
|
lncli! To disable the use of the default profile for a single command,
|
||
|
|
set '--profile= '.
|
||
|
|
|
||
|
|
The profiles are stored in a file called profiles.json in the user's
|
||
|
|
home directory, for example:
|
||
|
|
C:\Users\<username>\AppData\Local\Lncli\profiles.json on Windows
|
||
|
|
~/.lncli/profiles.json on Linux
|
||
|
|
~/Library/Application Support/Lncli/profiles.json on MacOS
|
||
|
|
`,
|
||
|
|
Subcommands: []cli.Command{
|
||
|
|
profileListCommand,
|
||
|
|
profileAddCommand,
|
||
|
|
profileRemoveCommand,
|
||
|
|
profileSetDefaultCommand,
|
||
|
|
profileUnsetDefaultCommand,
|
||
|
|
profileAddMacaroonCommand,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
var profileListCommand = cli.Command{
|
||
|
|
Name: "list",
|
||
|
|
Usage: "Lists all lncli profiles",
|
||
|
|
Action: profileList,
|
||
|
|
}
|
||
|
|
|
||
|
|
func profileList(_ *cli.Context) error {
|
||
|
|
f, err := loadProfileFile(defaultProfileFile)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
printJSON(f)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
var profileAddCommand = cli.Command{
|
||
|
|
Name: "add",
|
||
|
|
Usage: "Add a new profile",
|
||
|
|
ArgsUsage: "name",
|
||
|
|
Description: `
|
||
|
|
Add a new named profile to the main profiles.json. All global options
|
||
|
|
(see 'lncli --help') passed into this command are stored in that named
|
||
|
|
profile.
|
||
|
|
`,
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
cli.StringFlag{
|
||
|
|
Name: "name",
|
||
|
|
Usage: "the name of the new profile",
|
||
|
|
},
|
||
|
|
cli.BoolFlag{
|
||
|
|
Name: "default",
|
||
|
|
Usage: "set the new profile to be the default profile",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Action: profileAdd,
|
||
|
|
}
|
||
|
|
|
||
|
|
func profileAdd(ctx *cli.Context) error {
|
||
|
|
if ctx.NArg() == 0 && ctx.NumFlags() == 0 {
|
||
|
|
return cli.ShowCommandHelp(ctx, "add")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Load the default profile file or create a new one if it doesn't exist
|
||
|
|
// yet.
|
||
|
|
f, err := loadProfileFile(defaultProfileFile)
|
||
|
|
switch {
|
||
|
|
case err == errNoProfileFile:
|
||
|
|
f = &profileFile{}
|
||
|
|
_ = os.MkdirAll(path.Dir(defaultProfileFile), 0700)
|
||
|
|
|
||
|
|
case err != nil:
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Create a profile struct from all the global options.
|
||
|
|
profile, err := profileFromContext(ctx, true)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("could not load global options: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Finally, all that's left is to get the profile name from either
|
||
|
|
// positional argument or flag.
|
||
|
|
args := ctx.Args()
|
||
|
|
switch {
|
||
|
|
case ctx.IsSet("name"):
|
||
|
|
profile.Name = ctx.String("name")
|
||
|
|
case args.Present():
|
||
|
|
profile.Name = args.First()
|
||
|
|
default:
|
||
|
|
return fmt.Errorf("name argument missing")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Is there already a profile with that name?
|
||
|
|
for _, p := range f.Profiles {
|
||
|
|
if p.Name == profile.Name {
|
||
|
|
return fmt.Errorf("a profile with the name %s already "+
|
||
|
|
"exists", profile.Name)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Do we need to update the default entry to be this one?
|
||
|
|
if ctx.Bool("default") {
|
||
|
|
f.Default = profile.Name
|
||
|
|
}
|
||
|
|
|
||
|
|
// All done, store the updated profile file.
|
||
|
|
f.Profiles = append(f.Profiles, profile)
|
||
|
|
if err = saveProfileFile(defaultProfileFile, f); err != nil {
|
||
|
|
return fmt.Errorf("error writing profile file %s: %v",
|
||
|
|
defaultProfileFile, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Printf("Profile %s added to file %s.\n", profile.Name,
|
||
|
|
defaultProfileFile)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
var profileRemoveCommand = cli.Command{
|
||
|
|
Name: "remove",
|
||
|
|
Usage: "Remove a profile",
|
||
|
|
ArgsUsage: "name",
|
||
|
|
Description: `Remove the specified profile from the profile file.`,
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
cli.StringFlag{
|
||
|
|
Name: "name",
|
||
|
|
Usage: "the name of the profile to delete",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Action: profileRemove,
|
||
|
|
}
|
||
|
|
|
||
|
|
func profileRemove(ctx *cli.Context) error {
|
||
|
|
if ctx.NArg() == 0 && ctx.NumFlags() == 0 {
|
||
|
|
return cli.ShowCommandHelp(ctx, "remove")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Load the default profile file.
|
||
|
|
f, err := loadProfileFile(defaultProfileFile)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("could not load profile file: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get the profile name from either positional argument or flag.
|
||
|
|
var (
|
||
|
|
args = ctx.Args()
|
||
|
|
name string
|
||
|
|
found = false
|
||
|
|
)
|
||
|
|
switch {
|
||
|
|
case ctx.IsSet("name"):
|
||
|
|
name = ctx.String("name")
|
||
|
|
case args.Present():
|
||
|
|
name = args.First()
|
||
|
|
default:
|
||
|
|
return fmt.Errorf("name argument missing")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Create a copy of all profiles but don't include the one to delete.
|
||
|
|
newProfiles := make([]*profileEntry, 0, len(f.Profiles)-1)
|
||
|
|
for _, p := range f.Profiles {
|
||
|
|
// Skip the one we want to delete.
|
||
|
|
if p.Name == name {
|
||
|
|
found = true
|
||
|
|
|
||
|
|
if p.Name == f.Default {
|
||
|
|
fmt.Println("Warning: removing default profile.")
|
||
|
|
}
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
// Keep all others.
|
||
|
|
newProfiles = append(newProfiles, p)
|
||
|
|
}
|
||
|
|
|
||
|
|
// If what we were looking for didn't exist in the first place, there's
|
||
|
|
// no need for updating the file.
|
||
|
|
if !found {
|
||
|
|
return fmt.Errorf("profile with name %s not found in file",
|
||
|
|
name)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Great, everything updated, now let's save the file.
|
||
|
|
f.Profiles = newProfiles
|
||
|
|
return saveProfileFile(defaultProfileFile, f)
|
||
|
|
}
|
||
|
|
|
||
|
|
var profileSetDefaultCommand = cli.Command{
|
||
|
|
Name: "setdefault",
|
||
|
|
Usage: "Set the default profile",
|
||
|
|
ArgsUsage: "name",
|
||
|
|
Description: `
|
||
|
|
Set a specified profile to be used as the default profile.
|
||
|
|
|
||
|
|
WARNING: Setting a default profile changes the default behavior of
|
||
|
|
lncli! To disable the use of the default profile for a single command,
|
||
|
|
set '--profile= '.
|
||
|
|
`,
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
cli.StringFlag{
|
||
|
|
Name: "name",
|
||
|
|
Usage: "the name of the profile to set as default",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Action: profileSetDefault,
|
||
|
|
}
|
||
|
|
|
||
|
|
func profileSetDefault(ctx *cli.Context) error {
|
||
|
|
if ctx.NArg() == 0 && ctx.NumFlags() == 0 {
|
||
|
|
return cli.ShowCommandHelp(ctx, "setdefault")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Load the default profile file.
|
||
|
|
f, err := loadProfileFile(defaultProfileFile)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("could not load profile file: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get the profile name from either positional argument or flag.
|
||
|
|
var (
|
||
|
|
args = ctx.Args()
|
||
|
|
name string
|
||
|
|
found = false
|
||
|
|
)
|
||
|
|
switch {
|
||
|
|
case ctx.IsSet("name"):
|
||
|
|
name = ctx.String("name")
|
||
|
|
case args.Present():
|
||
|
|
name = args.First()
|
||
|
|
default:
|
||
|
|
return fmt.Errorf("name argument missing")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Make sure the new default profile actually exists.
|
||
|
|
for _, p := range f.Profiles {
|
||
|
|
if p.Name == name {
|
||
|
|
found = true
|
||
|
|
f.Default = p.Name
|
||
|
|
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// If the default profile doesn't exist, there's no need for updating
|
||
|
|
// the file.
|
||
|
|
if !found {
|
||
|
|
return fmt.Errorf("profile with name %s not found in file",
|
||
|
|
name)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Great, everything updated, now let's save the file.
|
||
|
|
return saveProfileFile(defaultProfileFile, f)
|
||
|
|
}
|
||
|
|
|
||
|
|
var profileUnsetDefaultCommand = cli.Command{
|
||
|
|
Name: "unsetdefault",
|
||
|
|
Usage: "Unsets the default profile",
|
||
|
|
Description: `
|
||
|
|
Disables the use of a default profile and restores lncli to its original
|
||
|
|
behavior.
|
||
|
|
`,
|
||
|
|
Action: profileUnsetDefault,
|
||
|
|
}
|
||
|
|
|
||
|
|
func profileUnsetDefault(_ *cli.Context) error {
|
||
|
|
// Load the default profile file.
|
||
|
|
f, err := loadProfileFile(defaultProfileFile)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("could not load profile file: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Save the file with the flag disabled.
|
||
|
|
f.Default = ""
|
||
|
|
return saveProfileFile(defaultProfileFile, f)
|
||
|
|
}
|
||
|
|
|
||
|
|
var profileAddMacaroonCommand = cli.Command{
|
||
|
|
Name: "addmacaroon",
|
||
|
|
Usage: "Add a macaroon to a profile's macaroon jar",
|
||
|
|
ArgsUsage: "macaroon-name",
|
||
|
|
Description: `
|
||
|
|
Add an additional macaroon specified by the global option --macaroonpath
|
||
|
|
to an existing profile's macaroon jar.
|
||
|
|
|
||
|
|
If no profile is selected, the macaroon is added to the default profile
|
||
|
|
(if one exists). To add a macaroon to a specific profile, use the global
|
||
|
|
--profile=myprofile option.
|
||
|
|
|
||
|
|
If multiple macaroons exist in a profile's macaroon jar, the one to use
|
||
|
|
can be specified with the global option --macfromjar=xyz.
|
||
|
|
`,
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
cli.StringFlag{
|
||
|
|
Name: "name",
|
||
|
|
Usage: "the name of the macaroon",
|
||
|
|
},
|
||
|
|
cli.BoolFlag{
|
||
|
|
Name: "default",
|
||
|
|
Usage: "set the new macaroon to be the default " +
|
||
|
|
"macaroon in the jar",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Action: profileAddMacaroon,
|
||
|
|
}
|
||
|
|
|
||
|
|
func profileAddMacaroon(ctx *cli.Context) error {
|
||
|
|
if ctx.NArg() == 0 && ctx.NumFlags() == 0 {
|
||
|
|
return cli.ShowCommandHelp(ctx, "addmacaroon")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Load the default profile file or create a new one if it doesn't exist
|
||
|
|
// yet.
|
||
|
|
f, err := loadProfileFile(defaultProfileFile)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("could not load profile file: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Finally, all that's left is to get the profile name from either
|
||
|
|
// positional argument or flag.
|
||
|
|
var (
|
||
|
|
args = ctx.Args()
|
||
|
|
profileName string
|
||
|
|
macName string
|
||
|
|
)
|
||
|
|
switch {
|
||
|
|
case ctx.IsSet("name"):
|
||
|
|
macName = ctx.String("name")
|
||
|
|
case args.Present():
|
||
|
|
macName = args.First()
|
||
|
|
default:
|
||
|
|
return fmt.Errorf("name argument missing")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Make sure the user actually set a macaroon path to use.
|
||
|
|
if !ctx.GlobalIsSet("macaroonpath") {
|
||
|
|
return fmt.Errorf("macaroonpath global option missing")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Find out which profile we should add the macaroon. The global flag
|
||
|
|
// takes precedence over the default profile.
|
||
|
|
if f.Default != "" {
|
||
|
|
profileName = f.Default
|
||
|
|
}
|
||
|
|
if ctx.GlobalIsSet("profile") {
|
||
|
|
profileName = ctx.GlobalString("profile")
|
||
|
|
}
|
||
|
|
if len(strings.TrimSpace(profileName)) == 0 {
|
||
|
|
return fmt.Errorf("no profile specified and no default " +
|
||
|
|
"profile exists")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Is there a profile with that name?
|
||
|
|
var selectedProfile *profileEntry
|
||
|
|
for _, p := range f.Profiles {
|
||
|
|
if p.Name == profileName {
|
||
|
|
selectedProfile = p
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if selectedProfile == nil {
|
||
|
|
return fmt.Errorf("profile with name %s not found", profileName)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Does a macaroon with that name already exist?
|
||
|
|
for _, m := range selectedProfile.Macaroons.Jar {
|
||
|
|
if m.Name == macName {
|
||
|
|
return fmt.Errorf("a macaroon with the name %s "+
|
||
|
|
"already exists", macName)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Do we need to update the default entry to be this one?
|
||
|
|
if ctx.Bool("default") {
|
||
|
|
selectedProfile.Macaroons.Default = macName
|
||
|
|
}
|
||
|
|
|
||
|
|
// Now load and possibly encrypt the macaroon file.
|
||
|
|
macPath := lncfg.CleanAndExpandPath(ctx.GlobalString("macaroonpath"))
|
||
|
|
macBytes, err := ioutil.ReadFile(macPath)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("unable to read macaroon path: %v", err)
|
||
|
|
}
|
||
|
|
mac := &macaroon.Macaroon{}
|
||
|
|
if err = mac.UnmarshalBinary(macBytes); err != nil {
|
||
|
|
return fmt.Errorf("unable to decode macaroon: %v", err)
|
||
|
|
}
|
||
|
|
macEntry := &macaroonEntry{
|
||
|
|
Name: macName,
|
||
|
|
}
|
||
|
|
if err = macEntry.storeMacaroon(mac, nil); err != nil {
|
||
|
|
return fmt.Errorf("unable to store macaroon: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// All done, store the updated profile file.
|
||
|
|
selectedProfile.Macaroons.Jar = append(
|
||
|
|
selectedProfile.Macaroons.Jar, macEntry,
|
||
|
|
)
|
||
|
|
if err = saveProfileFile(defaultProfileFile, f); err != nil {
|
||
|
|
return fmt.Errorf("error writing profile file %s: %v",
|
||
|
|
defaultProfileFile, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Printf("Macaroon %s added to profile %s in file %s.\n", macName,
|
||
|
|
selectedProfile.Name, defaultProfileFile)
|
||
|
|
return nil
|
||
|
|
}
|