Skip to content

Commit df5dc37

Browse files
hacdiashacdias
committed
wip: client
1 parent ec97069 commit df5dc37

File tree

5 files changed

+196
-51
lines changed

5 files changed

+196
-51
lines changed

‎client.go‎

Lines changed: 108 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -5,23 +5,43 @@ import (
55
"encoding/json"
66
"errors"
77
"fmt"
8+
"io"
89
"os"
910
"time"
1011

1112
"github.com/ipfs/boxo/ipns"
1213
"github.com/ipfs/boxo/routing/http/client"
1314
"github.com/ipfs/boxo/routing/http/types"
15+
"github.com/ipfs/boxo/routing/http/types/iter"
1416
"github.com/ipfs/go-cid"
1517
"github.com/libp2p/go-libp2p/core/peer"
1618
)
1719

18-
func findProviders(ctx context.Context, key cid.Cid, endpoint string, prettyOutput bool) error {
20+
type askClient struct {
21+
drc *client.Client
22+
out io.Writer
23+
pretty bool
24+
}
25+
26+
func newAskClient(endpoint string, prettyOutput bool, out io.Writer) (*askClient, error) {
1927
drc, err := client.New(endpoint)
2028
if err != nil {
21-
return err
29+
return nil, err
2230
}
2331

24-
recordsIter, err := drc.FindProviders(ctx, key)
32+
if out == nil {
33+
out = os.Stdout
34+
}
35+
36+
return &askClient{
37+
drc: drc,
38+
pretty: prettyOutput,
39+
out: out,
40+
}, nil
41+
}
42+
43+
func (a *askClient) findProviders(ctx context.Context, key cid.Cid) error {
44+
recordsIter, err := a.drc.FindProviders(ctx, key)
2545
if err != nil {
2646
return err
2747
}
@@ -40,24 +60,24 @@ func findProviders(ctx context.Context, key cid.Cid, endpoint string, prettyOutp
4060
return nil
4161
}
4262

43-
if prettyOutput {
63+
if a.pretty {
4464
switch res.Val.GetSchema() {
4565
case types.SchemaPeer:
4666
record := res.Val.(*types.PeerRecord)
47-
fmt.Fprintln(os.Stdout, record.ID)
48-
fmt.Fprintln(os.Stdout, "\tProtocols:", record.Protocols)
49-
fmt.Fprintln(os.Stdout, "\tAddresses:", record.Addrs)
67+
fmt.Fprintln(a.out, record.ID)
68+
fmt.Fprintln(a.out, "\tProtocols:", record.Protocols)
69+
fmt.Fprintln(a.out, "\tAddresses:", record.Addrs)
5070
default:
5171
// This is an unknown schema. Let's just print it raw.
52-
err := json.NewEncoder(os.Stdout).Encode(res.Val)
72+
err := json.NewEncoder(a.out).Encode(res.Val)
5373
if err != nil {
5474
return err
5575
}
5676
}
5777

58-
fmt.Fprintln(os.Stdout)
78+
fmt.Fprintln(a.out)
5979
} else {
60-
err := json.NewEncoder(os.Stdout).Encode(res.Val)
80+
err := json.NewEncoder(a.out).Encode(res.Val)
6181
if err != nil {
6282
return err
6383
}
@@ -67,13 +87,56 @@ func findProviders(ctx context.Context, key cid.Cid, endpoint string, prettyOutp
6787
return nil
6888
}
6989

70-
func findPeers(ctx context.Context, pid peer.ID, endpoint string, prettyOutput bool) error {
71-
drc, err := client.New(endpoint)
90+
func (a *askClient) provide(ctx context.Context, records ...*types.AnnouncementRecord) error {
91+
for _, rec := range records {
92+
err := rec.Verify()
93+
if err != nil {
94+
return err
95+
}
96+
}
97+
98+
recordsIter, err := a.drc.ProvideRecords(ctx, records...)
7299
if err != nil {
73100
return err
74101
}
102+
defer recordsIter.Close()
103+
return a.printProvideResult(recordsIter)
104+
}
105+
106+
func (a *askClient) printProvideResult(recordsIter iter.ResultIter[*types.AnnouncementResponseRecord]) error {
107+
for recordsIter.Next() {
108+
res := recordsIter.Val()
75109

76-
recordsIter, err := drc.FindPeers(ctx, pid)
110+
// Check for error, but do not complain if we exceeded the timeout. We are
111+
// expecting that to happen: we explicitly defined a timeout.
112+
if res.Err != nil {
113+
if !errors.Is(res.Err, context.DeadlineExceeded) {
114+
return res.Err
115+
}
116+
117+
return nil
118+
}
119+
120+
if a.pretty {
121+
if res.Val.Error != "" {
122+
fmt.Fprintf(a.out, "Error: %s", res.Val.Error)
123+
} else {
124+
fmt.Fprintf(a.out, "TTL: %s", res.Val.TTL)
125+
}
126+
fmt.Fprintln(a.out)
127+
} else {
128+
err := json.NewEncoder(a.out).Encode(res.Val)
129+
if err != nil {
130+
return err
131+
}
132+
}
133+
}
134+
135+
return nil
136+
}
137+
138+
func (a *askClient) findPeers(ctx context.Context, pid peer.ID) error {
139+
recordsIter, err := a.drc.FindPeers(ctx, pid)
77140
if err != nil {
78141
return err
79142
}
@@ -92,13 +155,13 @@ func findPeers(ctx context.Context, pid peer.ID, endpoint string, prettyOutput b
92155
return nil
93156
}
94157

95-
if prettyOutput {
96-
fmt.Fprintln(os.Stdout, res.Val.ID)
97-
fmt.Fprintln(os.Stdout, "\tProtocols:", res.Val.Protocols)
98-
fmt.Fprintln(os.Stdout, "\tAddresses:", res.Val.Addrs)
99-
fmt.Fprintln(os.Stdout)
158+
if a.pretty {
159+
fmt.Fprintln(a.out, res.Val.ID)
160+
fmt.Fprintln(a.out, "\tProtocols:", res.Val.Protocols)
161+
fmt.Fprintln(a.out, "\tAddresses:", res.Val.Addrs)
162+
fmt.Fprintln(a.out)
100163
} else {
101-
err := json.NewEncoder(os.Stdout).Encode(res.Val)
164+
err := json.NewEncoder(a.out).Encode(res.Val)
102165
if err != nil {
103166
return err
104167
}
@@ -108,18 +171,30 @@ func findPeers(ctx context.Context, pid peer.ID, endpoint string, prettyOutput b
108171
return nil
109172
}
110173

111-
func getIPNS(ctx context.Context, name ipns.Name, endpoint string, prettyOutput bool) error {
112-
drc, err := client.New(endpoint)
174+
func (a *askClient) providePeer(ctx context.Context, records ...*types.AnnouncementRecord) error {
175+
for _, rec := range records {
176+
err := rec.Verify()
177+
if err != nil {
178+
return err
179+
}
180+
}
181+
182+
recordsIter, err := a.drc.ProvidePeerRecords(ctx, records...)
113183
if err != nil {
114184
return err
115185
}
186+
defer recordsIter.Close()
116187

117-
rec, err := drc.GetIPNS(ctx, name)
188+
return a.printProvideResult(recordsIter)
189+
}
190+
191+
func (a *askClient) getIPNS(ctx context.Context, name ipns.Name) error {
192+
rec, err := a.drc.GetIPNS(ctx, name)
118193
if err != nil {
119194
return err
120195
}
121196

122-
if prettyOutput {
197+
if a.pretty {
123198
v, err := rec.Value()
124199
if err != nil {
125200
return err
@@ -135,19 +210,19 @@ func getIPNS(ctx context.Context, name ipns.Name, endpoint string, prettyOutput
135210
return err
136211
}
137212

138-
fmt.Printf("/ipns/%s\n", name)
213+
fmt.Fprintf(a.out, "/ipns/%s\n", name)
139214

140215
// Since [client.Client.GetIPNS] verifies if the retrieved record is valid, we
141216
// do not need to verify it again. However, if you were not using this specific
142217
// client, but using some other tool, you should always validate the IPNS Record
143218
// using the [ipns.Validate] or [ipns.ValidateWithName] functions.
144-
fmt.Println("\tSignature Validated")
145-
fmt.Println("\tValue:", v.String())
146-
fmt.Println("\tSequence:", seq)
147-
fmt.Println("\tValidityType : EOL/End-of-Life")
148-
fmt.Println("\tValidity:", eol.Format(time.RFC3339))
219+
fmt.Fprintln(a.out, "\tSignature Validated")
220+
fmt.Fprintln(a.out, "\tValue:", v.String())
221+
fmt.Fprintln(a.out, "\tSequence:", seq)
222+
fmt.Fprintln(a.out, "\tValidityType : EOL/End-of-Life")
223+
fmt.Fprintln(a.out, "\tValidity:", eol.Format(time.RFC3339))
149224
if ttl, err := rec.TTL(); err == nil {
150-
fmt.Println("\tTTL:", ttl.String())
225+
fmt.Fprintln(a.out, "\tTTL:", ttl.String())
151226
}
152227

153228
return nil
@@ -158,20 +233,15 @@ func getIPNS(ctx context.Context, name ipns.Name, endpoint string, prettyOutput
158233
return err
159234
}
160235

161-
_, err = os.Stdout.Write(raw)
236+
_, err = a.out.Write(raw)
162237
return err
163238
}
164239

165-
func putIPNS(ctx context.Context, name ipns.Name, record []byte, endpoint string) error {
166-
drc, err := client.New(endpoint)
167-
if err != nil {
168-
return err
169-
}
170-
240+
func (a *askClient) putIPNS(ctx context.Context, name ipns.Name, record []byte) error {
171241
rec, err := ipns.UnmarshalRecord(record)
172242
if err != nil {
173243
return err
174244
}
175245

176-
return drc.PutIPNS(ctx, name, rec)
246+
return a.drc.PutIPNS(ctx, name, rec)
177247
}

‎go.mod‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ go 1.21
55
require (
66
github.com/CAFxX/httpcompression v0.0.9
77
github.com/felixge/httpsnoop v1.0.4
8-
github.com/ipfs/boxo v0.18.1-0.20240220135834-45d50f539dc4
8+
github.com/ipfs/boxo v0.18.1-0.20240306111355-f261eac52f4c
99
github.com/ipfs/go-cid v0.4.1
1010
github.com/ipfs/go-log/v2 v2.5.1
1111
github.com/libp2p/go-libp2p v0.33.0

‎go.sum‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -186,8 +186,8 @@ github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpO
186186
github.com/huin/goupnp v1.3.0 h1:UvLUlWDNpoUdYzb2TCn+MuTWtcjXKSza2n6CBdQ0xXc=
187187
github.com/huin/goupnp v1.3.0/go.mod h1:gnGPsThkYa7bFi/KWmEysQRf48l2dvR5bxr2OFckNX8=
188188
github.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8=
189-
github.com/ipfs/boxo v0.18.1-0.20240220135834-45d50f539dc4 h1:2okDVJqJ8Zqnud+JX4rItjtV/7mHzGXJ27BAvI2GwW0=
190-
github.com/ipfs/boxo v0.18.1-0.20240220135834-45d50f539dc4/go.mod h1:pIZgTWdm3k3pLF9Uq6MB8JEcW07UDwNJjlXW1HELW80=
189+
github.com/ipfs/boxo v0.18.1-0.20240306111355-f261eac52f4c h1:e0PwRwv2XN6K1PBFxaUJfvCTEtXXW4A7d3d/bcoxwRk=
190+
github.com/ipfs/boxo v0.18.1-0.20240306111355-f261eac52f4c/go.mod h1:pIZgTWdm3k3pLF9Uq6MB8JEcW07UDwNJjlXW1HELW80=
191191
github.com/ipfs/go-cid v0.0.3/go.mod h1:GHWU/WuQdMPmIosc4Yn1bcCT7dSeX4lBafM7iqUPQvM=
192192
github.com/ipfs/go-cid v0.4.1 h1:A/T3qGvxi4kpKWWcPC/PgbvDA2bjVLO7n4UeVwnbs/s=
193193
github.com/ipfs/go-cid v0.4.1/go.mod h1:uQHwDeX4c6CtyrFwdqyhpNcxVewur1M7l7fNU7LKwZk=

‎main.go‎

Lines changed: 78 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,13 @@
11
package main
22

33
import (
4+
"encoding/json"
45
"errors"
56
"log"
67
"os"
78

89
"github.com/ipfs/boxo/ipns"
10+
"github.com/ipfs/boxo/routing/http/types"
911
"github.com/ipfs/go-cid"
1012
"github.com/libp2p/go-libp2p/core/peer"
1113
"github.com/multiformats/go-multibase"
@@ -85,7 +87,39 @@ func main() {
8587
if err != nil {
8688
return err
8789
}
88-
return findProviders(ctx.Context, c, ctx.String("endpoint"), ctx.Bool("pretty"))
90+
cl, err := newAskClient(ctx.String("endpoint"), ctx.Bool("pretty"), os.Stdout)
91+
if err != nil {
92+
return err
93+
}
94+
return cl.findProviders(ctx.Context, c)
95+
},
96+
},
97+
{
98+
Name: "provide",
99+
Usage: "provide <record-path...>",
100+
UsageText: "Provide a one or multiple announcement records to the network",
101+
Action: func(ctx *cli.Context) error {
102+
if ctx.NArg() < 1 {
103+
return errors.New("invalid command, see help")
104+
}
105+
records := []*types.AnnouncementRecord{}
106+
for _, recordPath := range ctx.Args().Slice() {
107+
recordData, err := os.ReadFile(recordPath)
108+
if err != nil {
109+
return err
110+
}
111+
var record *types.AnnouncementRecord
112+
err = json.Unmarshal(recordData, &record)
113+
if err != nil {
114+
return err
115+
}
116+
records = append(records, record)
117+
}
118+
cl, err := newAskClient(ctx.String("endpoint"), ctx.Bool("pretty"), os.Stdout)
119+
if err != nil {
120+
return err
121+
}
122+
return cl.provide(ctx.Context, records...)
89123
},
90124
},
91125
{
@@ -101,7 +135,39 @@ func main() {
101135
if err != nil {
102136
return err
103137
}
104-
return findPeers(ctx.Context, pid, ctx.String("endpoint"), ctx.Bool("pretty"))
138+
cl, err := newAskClient(ctx.String("endpoint"), ctx.Bool("pretty"), os.Stdout)
139+
if err != nil {
140+
return err
141+
}
142+
return cl.findPeers(ctx.Context, pid)
143+
},
144+
},
145+
{
146+
Name: "providepeers",
147+
Usage: "providepeers <record-path...>",
148+
UsageText: "Provide a one or multiple peer announcement records to the network",
149+
Action: func(ctx *cli.Context) error {
150+
if ctx.NArg() < 1 {
151+
return errors.New("invalid command, see help")
152+
}
153+
records := []*types.AnnouncementRecord{}
154+
for _, recordPath := range ctx.Args().Slice() {
155+
recordData, err := os.ReadFile(recordPath)
156+
if err != nil {
157+
return err
158+
}
159+
var record *types.AnnouncementRecord
160+
err = json.Unmarshal(recordData, &record)
161+
if err != nil {
162+
return err
163+
}
164+
records = append(records, record)
165+
}
166+
cl, err := newAskClient(ctx.String("endpoint"), ctx.Bool("pretty"), os.Stdout)
167+
if err != nil {
168+
return err
169+
}
170+
return cl.providePeer(ctx.Context, records...)
105171
},
106172
},
107173
{
@@ -117,7 +183,11 @@ func main() {
117183
if err != nil {
118184
return err
119185
}
120-
return getIPNS(ctx.Context, name, ctx.String("endpoint"), ctx.Bool("pretty"))
186+
cl, err := newAskClient(ctx.String("endpoint"), ctx.Bool("pretty"), os.Stdout)
187+
if err != nil {
188+
return err
189+
}
190+
return cl.getIPNS(ctx.Context, name)
121191
},
122192
},
123193
{
@@ -139,7 +209,11 @@ func main() {
139209
if err != nil {
140210
return err
141211
}
142-
return putIPNS(ctx.Context, name, recBytes, ctx.String("endpoint"))
212+
cl, err := newAskClient(ctx.String("endpoint"), ctx.Bool("pretty"), os.Stdout)
213+
if err != nil {
214+
return err
215+
}
216+
return cl.putIPNS(ctx.Context, name, recBytes)
143217
},
144218
},
145219
},

0 commit comments

Comments
 (0)