Skip to content

Commit cb0394f

Browse files
authored
allow goat firehose to accept source url as positional arg (#980)
2 parents aa8c2ee + d7166a4 commit cb0394f

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

cmd/goat/firehose.go

+16-1
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ import (
44
"bytes"
55
"context"
66
"encoding/json"
7+
"errors"
78
"fmt"
89
"log/slog"
910
"net/http"
@@ -78,7 +79,21 @@ func runFirehose(cctx *cli.Context) error {
7879
CollectionFilter: cctx.StringSlice("collection"),
7980
}
8081

81-
relayHost := cctx.String("relay-host")
82+
var relayHost string
83+
if cctx.IsSet("relay-host") {
84+
if cctx.Args().Len() != 0 {
85+
return errors.New("error: unused positional args")
86+
}
87+
relayHost = cctx.String("relay-host")
88+
} else {
89+
if cctx.Args().Len() == 1 {
90+
relayHost = cctx.Args().First()
91+
} else if cctx.Args().Len() > 1 {
92+
return errors.New("can only have at most one relay-host")
93+
} else {
94+
relayHost = cctx.String("relay-host")
95+
}
96+
}
8297

8398
dialer := websocket.DefaultDialer
8499
u, err := url.Parse(relayHost)

0 commit comments

Comments
 (0)