ipfs.go 4.56 KB
Newer Older
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
1 2 3
package main

import (
4
	"errors"
Juan Batiz-Benet's avatar
refmt  
Juan Batiz-Benet committed
5
	"fmt"
6
	"os"
7
	"runtime/pprof"
8

Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
9 10
	flag "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/gonuts/flag"
	commander "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/commander"
11
	ma "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-multiaddr"
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
12 13

	config "github.com/jbenet/go-ipfs/config"
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
14
	core "github.com/jbenet/go-ipfs/core"
15
	daemon "github.com/jbenet/go-ipfs/daemon"
Juan Batiz-Benet's avatar
refmt  
Juan Batiz-Benet committed
16
	u "github.com/jbenet/go-ipfs/util"
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
17 18
)

Juan Batiz-Benet's avatar
go lint  
Juan Batiz-Benet committed
19
// The IPFS command tree. It is an instance of `commander.Command`.
Juan Batiz-Benet's avatar
refmt  
Juan Batiz-Benet committed
20 21 22 23 24 25 26 27 28 29 30 31 32 33
var CmdIpfs = &commander.Command{
	UsageLine: "ipfs [<flags>] <command> [<args>]",
	Short:     "global versioned p2p merkledag file system",
	Long: `ipfs - global versioned p2p merkledag file system

Basic commands:

    add <path>    Add an object to ipfs.
    cat <ref>     Show ipfs object data.
    ls <ref>      List links from an object.
    refs <ref>    List link hashes from an object.

Tool commands:

Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
34 35 36
    config        Manage configuration.
    version       Show ipfs version information.
    commands      List all available commands.
Juan Batiz-Benet's avatar
refmt  
Juan Batiz-Benet committed
37 38 39

Advanced Commands:

Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
40
    mount         Mount an ipfs read-only mountpoint.
41
    serve         Serve an interface to ipfs.
Juan Batiz-Benet's avatar
refmt  
Juan Batiz-Benet committed
42 43 44 45 46

Use "ipfs help <command>" for more information about a command.
`,
	Run: ipfsCmd,
	Subcommands: []*commander.Command{
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
47
		cmdIpfsAdd,
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
48
		cmdIpfsCat,
49
		cmdIpfsLs,
Juan Batiz-Benet's avatar
refs  
Juan Batiz-Benet committed
50
		cmdIpfsRefs,
51
		cmdIpfsConfig,
Juan Batiz-Benet's avatar
refmt  
Juan Batiz-Benet committed
52 53
		cmdIpfsVersion,
		cmdIpfsCommands,
54
		cmdIpfsMount,
55
		cmdIpfsInit,
verokarhu's avatar
verokarhu committed
56
		cmdIpfsServe,
Jeromy's avatar
Jeromy committed
57
		cmdIpfsRun,
58
		cmdIpfsName,
59
		cmdIpfsBootstrap,
Juan Batiz-Benet's avatar
refmt  
Juan Batiz-Benet committed
60 61 62 63
	},
	Flag: *flag.NewFlagSet("ipfs", flag.ExitOnError),
}

Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
64
// log is the command logger
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
65
var log = u.Logger("cmd/ipfs")
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
66

Jeromy's avatar
Jeromy committed
67
func init() {
Shanti Bouchez-Mongardé's avatar
Shanti Bouchez-Mongardé committed
68
	config, err := config.PathRoot()
69
	if err != nil {
70 71
		u.POut("Failure initializing the default Config Directory: ", err)
		os.Exit(1)
72 73
	}
	CmdIpfs.Flag.String("c", config, "specify config directory")
Jeromy's avatar
Jeromy committed
74 75
}

Juan Batiz-Benet's avatar
refmt  
Juan Batiz-Benet committed
76 77 78 79 80
func ipfsCmd(c *commander.Command, args []string) error {
	u.POut(c.Long)
	return nil
}

Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
81
func main() {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
82 83 84
	u.Debug = false

	// setup logging
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
85
	// u.SetupLogging() done in an init() block now.
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
86 87 88 89 90 91 92 93 94 95 96

	// 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()
97
	}
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
98 99

	err := CmdIpfs.Dispatch(os.Args[1:])
Juan Batiz-Benet's avatar
refmt  
Juan Batiz-Benet committed
100 101
	if err != nil {
		if len(err.Error()) > 0 {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
102
			fmt.Fprintf(os.Stderr, "ipfs %s: %v\n", os.Args[1], err)
Juan Batiz-Benet's avatar
refmt  
Juan Batiz-Benet committed
103 104 105 106
		}
		os.Exit(1)
	}
	return
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
107
}
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
108

109
// localNode constructs a node
110
func localNode(confdir string, online bool) (*core.IpfsNode, error) {
Shanti Bouchez-Mongardé's avatar
Shanti Bouchez-Mongardé committed
111
	filename, err := config.Filename(confdir)
112 113 114 115 116
	if err != nil {
		return nil, err
	}

	cfg, err := config.Load(filename)
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
117 118 119 120
	if err != nil {
		return nil, err
	}

121
	return core.NewIpfsNode(cfg, online)
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
122
}
Jeromy's avatar
Jeromy committed
123 124

// Gets the config "-c" flag from the command, or returns
125
// the default configuration root directory
126
func getConfigDir(c *commander.Command) (string, error) {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
127 128 129

	// use the root cmd (that's where config is specified)
	for ; c.Parent != nil; c = c.Parent {
130
	}
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
131 132 133 134 135

	// flag should be defined on root.
	param := c.Flag.Lookup("c").Value.Get().(string)
	if param != "" {
		return u.TildeExpansion(param)
136
	}
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
137 138 139 140 141 142 143 144

	return config.PathRoot()
}

func getConfig(c *commander.Command) (*config.Config, error) {
	confdir, err := getConfigDir(c)
	if err != nil {
		return nil, err
Jeromy's avatar
Jeromy committed
145
	}
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
146 147 148 149

	filename, err := config.Filename(confdir)
	if err != nil {
		return nil, err
150
	}
151

Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
152
	return config.Load(filename)
Jeromy's avatar
Jeromy committed
153
}
154 155 156 157 158 159 160 161 162 163 164

// cmdContext is a wrapper structure that keeps a node, a daemonlistener, and
// a config directory together. These three are needed for most commands.
type cmdContext struct {
	node      *core.IpfsNode
	daemon    *daemon.DaemonListener
	configDir string
}

// setupCmdContext initializes a cmdContext structure from a given command.
func setupCmdContext(c *commander.Command, online bool) (cc cmdContext, err error) {
165 166 167 168 169
	rootCmd := c
	for ; rootCmd.Parent != nil; rootCmd = c.Parent {
	}

	cc.configDir, err = getConfigDir(rootCmd)
170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204
	if err != nil {
		return
	}

	cc.node, err = localNode(cc.configDir, online)
	if err != nil {
		return
	}

	cc.daemon, err = setupDaemon(cc.configDir, cc.node)
	if err != nil {
		return
	}

	return
}

// setupDaemon sets up the daemon corresponding to given node.
func setupDaemon(confdir string, node *core.IpfsNode) (*daemon.DaemonListener, error) {
	if node.Config.Addresses.API == "" {
		return nil, errors.New("no config.Addresses.API endpoint supplied")
	}

	maddr, err := ma.NewMultiaddr(node.Config.Addresses.API)
	if err != nil {
		return nil, err
	}

	dl, err := daemon.NewDaemonListener(node, maddr, confdir)
	if err != nil {
		return nil, err
	}
	go dl.Listen()
	return dl, nil
}