You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
2 years ago
|
package version
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"regexp"
|
||
|
|
"strings"
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
reRelease *regexp.Regexp
|
||
|
|
reDev *regexp.Regexp
|
||
|
|
reOnce sync.Once
|
||
|
|
uapCbs map[string]func() string
|
||
|
|
)
|
||
|
|
|
||
|
|
func UserAgent() string {
|
||
|
|
uaVersion := defaultVersion
|
||
|
|
|
||
|
|
reOnce.Do(func() {
|
||
|
|
reRelease = regexp.MustCompile(`^(v[0-9]+\.[0-9]+)\.[0-9]+$`)
|
||
|
|
reDev = regexp.MustCompile(`^(v[0-9]+\.[0-9]+)\.[0-9]+`)
|
||
|
|
})
|
||
|
|
|
||
|
|
if matches := reRelease.FindAllStringSubmatch(Version, 1); len(matches) > 0 {
|
||
|
|
uaVersion = matches[0][1]
|
||
|
|
} else if matches := reDev.FindAllStringSubmatch(Version, 1); len(matches) > 0 {
|
||
|
|
uaVersion = matches[0][1] + "-dev"
|
||
|
|
}
|
||
|
|
|
||
|
|
res := &strings.Builder{}
|
||
|
|
fmt.Fprintf(res, "buildkit/%s", uaVersion)
|
||
|
|
for pname, pver := range uapCbs {
|
||
|
|
fmt.Fprintf(res, " %s/%s", pname, pver())
|
||
|
|
}
|
||
|
|
|
||
|
|
return res.String()
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetUserAgentProduct sets a callback to get the version of a product to be
|
||
|
|
// included in the User-Agent header. The callback is called every time the
|
||
|
|
// User-Agent header is generated. Caller must ensure that the callback is
|
||
|
|
// cached if it is expensive to compute.
|
||
|
|
func SetUserAgentProduct(name string, cb func() (version string)) {
|
||
|
|
if uapCbs == nil {
|
||
|
|
uapCbs = make(map[string]func() string)
|
||
|
|
}
|
||
|
|
uapCbs[name] = cb
|
||
|
|
}
|