forked from TrueCloudLab/restic
72 lines
1.6 KiB
Go
72 lines
1.6 KiB
Go
package rest
|
|
|
|
import (
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/restic/restic/internal/errors"
|
|
"github.com/restic/restic/internal/options"
|
|
)
|
|
|
|
// Config contains all configuration necessary to connect to a REST server.
|
|
type Config struct {
|
|
URL *url.URL
|
|
Connections uint `option:"connections" help:"set a limit for the number of concurrent connections (default: 5)"`
|
|
}
|
|
|
|
func init() {
|
|
options.Register("rest", Config{})
|
|
}
|
|
|
|
// NewConfig returns a new Config with the default values filled in.
|
|
func NewConfig() Config {
|
|
return Config{
|
|
Connections: 5,
|
|
}
|
|
}
|
|
|
|
// ParseConfig parses the string s and extracts the REST server URL.
|
|
func ParseConfig(s string) (interface{}, error) {
|
|
if !strings.HasPrefix(s, "rest:") {
|
|
return nil, errors.New("invalid REST backend specification")
|
|
}
|
|
|
|
s = prepareURL(s)
|
|
|
|
u, err := url.Parse(s)
|
|
if err != nil {
|
|
return nil, errors.WithStack(err)
|
|
}
|
|
|
|
cfg := NewConfig()
|
|
cfg.URL = u
|
|
return cfg, nil
|
|
}
|
|
|
|
// StripPassword removes the password from the URL
|
|
// If the repository location cannot be parsed as a valid URL, it will be returned as is
|
|
// (it's because this function is used for logging errors)
|
|
func StripPassword(s string) string {
|
|
scheme := s[:5]
|
|
s = prepareURL(s)
|
|
|
|
u, err := url.Parse(s)
|
|
if err != nil {
|
|
return scheme + s
|
|
}
|
|
|
|
if _, set := u.User.Password(); !set {
|
|
return scheme + s
|
|
}
|
|
|
|
// a password was set: we replace it with ***
|
|
return scheme + strings.Replace(u.String(), u.User.String()+"@", u.User.Username()+":***@", 1)
|
|
}
|
|
|
|
func prepareURL(s string) string {
|
|
s = s[5:]
|
|
if !strings.HasSuffix(s, "/") {
|
|
s += "/"
|
|
}
|
|
return s
|
|
}
|