mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-21 10:27:46 +08:00
100 lines
2.4 KiB
Go
100 lines
2.4 KiB
Go
package cmdenv
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/ipfs/go-ipfs/commands"
|
|
"github.com/ipfs/go-ipfs/core"
|
|
|
|
cmds "github.com/ipfs/go-ipfs-cmds"
|
|
config "github.com/ipfs/go-ipfs-config"
|
|
logging "github.com/ipfs/go-log"
|
|
coreiface "github.com/ipfs/interface-go-ipfs-core"
|
|
options "github.com/ipfs/interface-go-ipfs-core/options"
|
|
)
|
|
|
|
var log = logging.Logger("core/commands/cmdenv")
|
|
|
|
// GetNode extracts the node from the environment.
|
|
func GetNode(env interface{}) (*core.IpfsNode, error) {
|
|
ctx, ok := env.(*commands.Context)
|
|
if !ok {
|
|
return nil, fmt.Errorf("expected env to be of type %T, got %T", ctx, env)
|
|
}
|
|
|
|
return ctx.GetNode()
|
|
}
|
|
|
|
// GetApi extracts CoreAPI instance from the environment.
|
|
func GetApi(env cmds.Environment, req *cmds.Request) (coreiface.CoreAPI, error) {
|
|
ctx, ok := env.(*commands.Context)
|
|
if !ok {
|
|
return nil, fmt.Errorf("expected env to be of type %T, got %T", ctx, env)
|
|
}
|
|
|
|
offline, _ := req.Options["offline"].(bool)
|
|
if !offline {
|
|
offline, _ = req.Options["local"].(bool)
|
|
if offline {
|
|
log.Errorf("Command '%s', --local is deprecated, use --offline instead", strings.Join(req.Path, " "))
|
|
}
|
|
}
|
|
api, err := ctx.GetAPI()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if offline {
|
|
return api.WithOptions(options.Api.Offline(offline))
|
|
}
|
|
|
|
return api, nil
|
|
}
|
|
|
|
// GetConfig extracts the config from the environment.
|
|
func GetConfig(env cmds.Environment) (*config.Config, error) {
|
|
ctx, ok := env.(*commands.Context)
|
|
if !ok {
|
|
return nil, fmt.Errorf("expected env to be of type %T, got %T", ctx, env)
|
|
}
|
|
|
|
return ctx.GetConfig()
|
|
}
|
|
|
|
// GetConfigRoot extracts the config root from the environment
|
|
func GetConfigRoot(env cmds.Environment) (string, error) {
|
|
ctx, ok := env.(*commands.Context)
|
|
if !ok {
|
|
return "", fmt.Errorf("expected env to be of type %T, got %T", ctx, env)
|
|
}
|
|
|
|
return ctx.ConfigRoot, nil
|
|
}
|
|
|
|
// EscNonPrint converts non-printable characters and backslash into Go escape
|
|
// sequences. This is done to display all characters in a string, including
|
|
// those that would otherwise not be displayed or have an undesirable effect on
|
|
// the display.
|
|
func EscNonPrint(s string) string {
|
|
if !needEscape(s) {
|
|
return s
|
|
}
|
|
|
|
esc := strconv.Quote(s)
|
|
// Remove first and last quote, and unescape quotes.
|
|
return strings.ReplaceAll(esc[1:len(esc)-1], `\"`, `"`)
|
|
}
|
|
|
|
func needEscape(s string) bool {
|
|
if strings.ContainsRune(s, '\\') {
|
|
return true
|
|
}
|
|
for _, r := range s {
|
|
if !strconv.IsPrint(r) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|