97 lines
1.7 KiB
Go
97 lines
1.7 KiB
Go
package cliutil
|
|
|
|
import (
|
|
"net/http"
|
|
"net/url"
|
|
"regexp"
|
|
"strings"
|
|
|
|
logging "github.com/ipfs/go-log/v2"
|
|
"github.com/multiformats/go-multiaddr"
|
|
manet "github.com/multiformats/go-multiaddr/net"
|
|
)
|
|
|
|
var log = logging.Logger("cliutil")
|
|
|
|
var (
|
|
infoWithToken = regexp.MustCompile("^[a-zA-Z0-9\\-_]+?\\.[a-zA-Z0-9\\-_]+?\\.([a-zA-Z0-9\\-_]+)?:.+$")
|
|
)
|
|
|
|
type APIInfo struct {
|
|
Addr string
|
|
Token []byte
|
|
}
|
|
|
|
func ParseApiInfo(s string) APIInfo {
|
|
var tok []byte
|
|
|
|
if infoWithToken.Match([]byte(s)) {
|
|
sp := strings.SplitN(s, ":", 2)
|
|
tok = []byte(sp[0])
|
|
s = sp[1]
|
|
}
|
|
|
|
return APIInfo{
|
|
Addr: s,
|
|
Token: tok,
|
|
}
|
|
}
|
|
|
|
func ParseApiInfoMulti(s string) []APIInfo {
|
|
var apiInfos []APIInfo
|
|
|
|
allAddrs := strings.SplitN(s, ",", -1)
|
|
|
|
for _, addr := range allAddrs {
|
|
apiInfos = append(apiInfos, ParseApiInfo(addr))
|
|
}
|
|
|
|
return apiInfos
|
|
}
|
|
|
|
func (a APIInfo) DialArgs(version string) (string, error) {
|
|
ma, err := multiaddr.NewMultiaddr(a.Addr)
|
|
if err == nil {
|
|
_, addr, err := manet.DialArgs(ma)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return "ws://" + addr + "/rpc/" + version, nil
|
|
}
|
|
|
|
_, err = url.Parse(a.Addr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return a.Addr + "/rpc/" + version, nil
|
|
}
|
|
|
|
func (a APIInfo) Host() (string, error) {
|
|
ma, err := multiaddr.NewMultiaddr(a.Addr)
|
|
if err == nil {
|
|
_, addr, err := manet.DialArgs(ma)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return addr, nil
|
|
}
|
|
|
|
spec, err := url.Parse(a.Addr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return spec.Host, nil
|
|
}
|
|
|
|
func (a APIInfo) AuthHeader() http.Header {
|
|
if len(a.Token) != 0 {
|
|
headers := http.Header{}
|
|
headers.Add("Authorization", "Bearer "+string(a.Token))
|
|
return headers
|
|
}
|
|
log.Warn("API Token not set and requested, capabilities might be limited.")
|
|
return nil
|
|
}
|