forked from ipfs/kubo
/
daemon.go
102 lines (87 loc) · 2.11 KB
/
daemon.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package daemon
import (
"encoding/json"
"fmt"
"net"
core "github.com/jbenet/go-ipfs/core"
"github.com/jbenet/go-ipfs/core/commands"
u "github.com/jbenet/go-ipfs/util"
ma "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-multiaddr"
)
// DaemonListener listens to an initialized IPFS node and can send it commands instead of
// starting up a new set of connections
type DaemonListener struct {
node *core.IpfsNode
list net.Listener
closed bool
}
//Command accepts user input and can be sent to the running IPFS node
type Command struct {
Command string
Args []string
Opts map[string]interface{}
}
func NewDaemonListener(ipfsnode *core.IpfsNode, addr *ma.Multiaddr) (*DaemonListener, error) {
network, host, err := addr.DialArgs()
if err != nil {
return nil, err
}
list, err := net.Listen(network, host)
if err != nil {
return nil, err
}
fmt.Println("New daemon listener initialized.")
return &DaemonListener{
node: ipfsnode,
list: list,
}, nil
}
func NewCommand() *Command {
return &Command{
Opts: make(map[string]interface{}),
}
}
func (dl *DaemonListener) Listen() {
fmt.Println("listen.")
for {
conn, err := dl.list.Accept()
fmt.Println("Loop!")
if err != nil {
if !dl.closed {
u.PErr("DaemonListener Accept: %v\n", err)
}
return
}
go dl.handleConnection(conn)
}
}
func (dl *DaemonListener) handleConnection(conn net.Conn) {
defer conn.Close()
dec := json.NewDecoder(conn)
var command Command
err := dec.Decode(&command)
if err != nil {
fmt.Fprintln(conn, err)
return
}
u.DOut("Got command: %v\n", command)
switch command.Command {
case "add":
err = commands.Add(dl.node, command.Args, command.Opts, conn)
case "cat":
err = commands.Cat(dl.node, command.Args, command.Opts, conn)
case "ls":
err = commands.Ls(dl.node, command.Args, command.Opts, conn)
case "pin":
err = commands.Pin(dl.node, command.Args, command.Opts, conn)
default:
err = fmt.Errorf("Invalid Command: '%s'", command.Command)
}
if err != nil {
fmt.Fprintln(conn, err)
}
}
func (dl *DaemonListener) Close() error {
dl.closed = true
return dl.list.Close()
}