93 lines
1.6 KiB
Go
93 lines
1.6 KiB
Go
package overrides
|
|
|
|
import (
|
|
"runtime"
|
|
"strings"
|
|
|
|
"go.arsenm.dev/lure/distro"
|
|
"go.arsenm.dev/lure/internal/cpu"
|
|
)
|
|
|
|
type Opts struct {
|
|
Name string
|
|
Overrides bool
|
|
LikeDistros bool
|
|
}
|
|
|
|
var DefaultOpts = &Opts{
|
|
Overrides: true,
|
|
LikeDistros: true,
|
|
}
|
|
|
|
// Resolve generates a slice of possible override names in the order that they should be checked
|
|
func Resolve(info *distro.OSRelease, opts *Opts) []string {
|
|
if opts == nil {
|
|
opts = DefaultOpts
|
|
}
|
|
|
|
if !opts.Overrides {
|
|
return []string{opts.Name}
|
|
}
|
|
|
|
architectures := []string{runtime.GOARCH}
|
|
|
|
if runtime.GOARCH == "arm" {
|
|
// More specific goes first
|
|
architectures[0] = cpu.ARMVariant()
|
|
architectures = append(architectures, "arm")
|
|
}
|
|
|
|
distros := []string{info.ID}
|
|
if opts.LikeDistros {
|
|
distros = append(distros, info.Like...)
|
|
}
|
|
|
|
var out []string
|
|
for _, arch := range architectures {
|
|
for _, distro := range distros {
|
|
if opts.Name == "" {
|
|
out = append(
|
|
out,
|
|
arch+"_"+distro,
|
|
distro,
|
|
)
|
|
} else {
|
|
out = append(
|
|
out,
|
|
opts.Name+"_"+arch+"_"+distro,
|
|
opts.Name+"_"+distro,
|
|
)
|
|
}
|
|
}
|
|
if opts.Name == "" {
|
|
out = append(out, arch)
|
|
} else {
|
|
out = append(out, opts.Name+"_"+arch)
|
|
}
|
|
}
|
|
if opts.Name != "" {
|
|
out = append(out, opts.Name)
|
|
}
|
|
|
|
for index, item := range out {
|
|
out[index] = strings.ReplaceAll(item, "-", "_")
|
|
}
|
|
|
|
return out
|
|
}
|
|
|
|
func (o *Opts) WithName(name string) *Opts {
|
|
o.Name = name
|
|
return o
|
|
}
|
|
|
|
func (o *Opts) WithOverrides(v bool) *Opts {
|
|
o.Overrides = v
|
|
return o
|
|
}
|
|
|
|
func (o *Opts) WithLikeDistros(v bool) *Opts {
|
|
o.LikeDistros = v
|
|
return o
|
|
}
|