mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-24 20:07:45 +08:00
refactor(cmd/ipfs2) rename to match cmd/ipfs1
This commit is contained in:
Brian Tiger Chow
Juan Batiz-Benet
parent
b3e7cbb9a5
commit
db7a42e73c
@ -1,177 +1,14 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"runtime/pprof"
|
||||
|
||||
cmds "github.com/jbenet/go-ipfs/commands"
|
||||
cmdsCli "github.com/jbenet/go-ipfs/commands/cli"
|
||||
cmdsHttp "github.com/jbenet/go-ipfs/commands/http"
|
||||
"github.com/jbenet/go-ipfs/config"
|
||||
"github.com/jbenet/go-ipfs/core"
|
||||
commands "github.com/jbenet/go-ipfs/core/commands2"
|
||||
daemon "github.com/jbenet/go-ipfs/daemon2"
|
||||
u "github.com/jbenet/go-ipfs/util"
|
||||
)
|
||||
|
||||
// log is the command logger
|
||||
var log = u.Logger("cmd/ipfs")
|
||||
|
||||
func main() {
|
||||
args := os.Args[1:]
|
||||
root := Root
|
||||
|
||||
req, err := cmdsCli.Parse(args, root)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// if the CLI-specific root doesn't contain the command, use the general root
|
||||
if len(req.Path()) == 0 {
|
||||
root = commands.Root
|
||||
req, err = cmdsCli.Parse(args, root)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
cmd, err := root.Get(req.Path())
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
options, err := getOptions(req, root)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
if help, found := options.Option("help"); found && help.(bool) {
|
||||
fmt.Println(cmd.Help)
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
if debug, found := options.Option("debug"); found && debug.(bool) {
|
||||
u.Debug = true
|
||||
|
||||
// if debugging, setup profiling.
|
||||
if u.Debug {
|
||||
ofi, err := os.Create("cpu.prof")
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return
|
||||
}
|
||||
pprof.StartCPUProfile(ofi)
|
||||
defer ofi.Close()
|
||||
defer pprof.StopCPUProfile()
|
||||
}
|
||||
}
|
||||
|
||||
configPath, err := getConfigRoot(options)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
conf, err := getConfig(configPath)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
ctx := req.Context()
|
||||
ctx.ConfigRoot = configPath
|
||||
ctx.Config = conf
|
||||
|
||||
if _, found := options.Option("encoding"); !found {
|
||||
if req.Command().Format != nil {
|
||||
req.SetOption("encoding", cmds.Text)
|
||||
} else {
|
||||
req.SetOption("encoding", cmds.JSON)
|
||||
}
|
||||
}
|
||||
|
||||
var res cmds.Response
|
||||
if root == Root {
|
||||
res = root.Call(req)
|
||||
|
||||
} else {
|
||||
local, found := options.Option("local")
|
||||
|
||||
if (!found || !local.(bool)) && daemon.Locked(configPath) {
|
||||
res, err = cmdsHttp.Send(req)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
} else {
|
||||
node, err := core.NewIpfsNode(conf, false)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
ctx.Node = node
|
||||
|
||||
res = root.Call(req)
|
||||
}
|
||||
}
|
||||
|
||||
if res.Error() != nil {
|
||||
fmt.Println(res.Error().Error())
|
||||
|
||||
if cmd.Help != "" && res.Error().Code == cmds.ErrClient {
|
||||
// TODO: convert from markdown to ANSI terminal format?
|
||||
fmt.Println(cmd.Help)
|
||||
}
|
||||
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
_, err = io.Copy(os.Stdout, res)
|
||||
if err != nil {
|
||||
fmt.Println(err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
func getOptions(req cmds.Request, root *cmds.Command) (cmds.Request, error) {
|
||||
tempReq := cmds.NewRequest(req.Path(), req.Options(), nil, nil, nil)
|
||||
|
||||
options, err := root.GetOptions(tempReq.Path())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = tempReq.ConvertOptions(options)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return tempReq, nil
|
||||
}
|
||||
|
||||
func getConfigRoot(req cmds.Request) (string, error) {
|
||||
if opt, found := req.Option("config"); found {
|
||||
return opt.(string), nil
|
||||
}
|
||||
|
||||
configPath, err := config.PathRoot()
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
return configPath, nil
|
||||
}
|
||||
|
||||
func getConfig(path string) (*config.Config, error) {
|
||||
configFile, err := config.Filename(path)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return config.Load(configFile)
|
||||
var Root = &cmds.Command{
|
||||
Options: commands.Root.Options,
|
||||
Help: commands.Root.Help,
|
||||
Subcommands: map[string]*cmds.Command{
|
||||
"daemon": Daemon,
|
||||
},
|
||||
}
|
||||
|
||||
177
cmd/ipfs2/main.go
Normal file
177
cmd/ipfs2/main.go
Normal file
@ -0,0 +1,177 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"runtime/pprof"
|
||||
|
||||
cmds "github.com/jbenet/go-ipfs/commands"
|
||||
cmdsCli "github.com/jbenet/go-ipfs/commands/cli"
|
||||
cmdsHttp "github.com/jbenet/go-ipfs/commands/http"
|
||||
"github.com/jbenet/go-ipfs/config"
|
||||
"github.com/jbenet/go-ipfs/core"
|
||||
commands "github.com/jbenet/go-ipfs/core/commands2"
|
||||
daemon "github.com/jbenet/go-ipfs/daemon2"
|
||||
u "github.com/jbenet/go-ipfs/util"
|
||||
)
|
||||
|
||||
// log is the command logger
|
||||
var log = u.Logger("cmd/ipfs")
|
||||
|
||||
func main() {
|
||||
args := os.Args[1:]
|
||||
root := Root
|
||||
|
||||
req, err := cmdsCli.Parse(args, root)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// if the CLI-specific root doesn't contain the command, use the general root
|
||||
if len(req.Path()) == 0 {
|
||||
root = commands.Root
|
||||
req, err = cmdsCli.Parse(args, root)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
cmd, err := root.Get(req.Path())
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
options, err := getOptions(req, root)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
if help, found := options.Option("help"); found && help.(bool) {
|
||||
fmt.Println(cmd.Help)
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
if debug, found := options.Option("debug"); found && debug.(bool) {
|
||||
u.Debug = true
|
||||
|
||||
// if debugging, setup profiling.
|
||||
if u.Debug {
|
||||
ofi, err := os.Create("cpu.prof")
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return
|
||||
}
|
||||
pprof.StartCPUProfile(ofi)
|
||||
defer ofi.Close()
|
||||
defer pprof.StopCPUProfile()
|
||||
}
|
||||
}
|
||||
|
||||
configPath, err := getConfigRoot(options)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
conf, err := getConfig(configPath)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
ctx := req.Context()
|
||||
ctx.ConfigRoot = configPath
|
||||
ctx.Config = conf
|
||||
|
||||
if _, found := options.Option("encoding"); !found {
|
||||
if req.Command().Format != nil {
|
||||
req.SetOption("encoding", cmds.Text)
|
||||
} else {
|
||||
req.SetOption("encoding", cmds.JSON)
|
||||
}
|
||||
}
|
||||
|
||||
var res cmds.Response
|
||||
if root == Root {
|
||||
res = root.Call(req)
|
||||
|
||||
} else {
|
||||
local, found := options.Option("local")
|
||||
|
||||
if (!found || !local.(bool)) && daemon.Locked(configPath) {
|
||||
res, err = cmdsHttp.Send(req)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
} else {
|
||||
node, err := core.NewIpfsNode(conf, false)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
ctx.Node = node
|
||||
|
||||
res = root.Call(req)
|
||||
}
|
||||
}
|
||||
|
||||
if res.Error() != nil {
|
||||
fmt.Println(res.Error().Error())
|
||||
|
||||
if cmd.Help != "" && res.Error().Code == cmds.ErrClient {
|
||||
// TODO: convert from markdown to ANSI terminal format?
|
||||
fmt.Println(cmd.Help)
|
||||
}
|
||||
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
_, err = io.Copy(os.Stdout, res)
|
||||
if err != nil {
|
||||
fmt.Println(err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
func getOptions(req cmds.Request, root *cmds.Command) (cmds.Request, error) {
|
||||
tempReq := cmds.NewRequest(req.Path(), req.Options(), nil, nil, nil)
|
||||
|
||||
options, err := root.GetOptions(tempReq.Path())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = tempReq.ConvertOptions(options)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return tempReq, nil
|
||||
}
|
||||
|
||||
func getConfigRoot(req cmds.Request) (string, error) {
|
||||
if opt, found := req.Option("config"); found {
|
||||
return opt.(string), nil
|
||||
}
|
||||
|
||||
configPath, err := config.PathRoot()
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
return configPath, nil
|
||||
}
|
||||
|
||||
func getConfig(path string) (*config.Config, error) {
|
||||
configFile, err := config.Filename(path)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return config.Load(configFile)
|
||||
}
|
||||
@ -1,14 +0,0 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
cmds "github.com/jbenet/go-ipfs/commands"
|
||||
commands "github.com/jbenet/go-ipfs/core/commands2"
|
||||
)
|
||||
|
||||
var Root = &cmds.Command{
|
||||
Options: commands.Root.Options,
|
||||
Help: commands.Root.Help,
|
||||
Subcommands: map[string]*cmds.Command{
|
||||
"daemon": Daemon,
|
||||
},
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user