refractor(build): move os check as an internal build pkg.
This commit is contained in:
Alexandre Pujol
parent
ac935ce81c
commit
b6aed5cd8d
2 changed files with 4 additions and 4 deletions
90
pkg/prebuild/cfg/os.go
Normal file
90
pkg/prebuild/cfg/os.go
Normal file
|
|
@ -0,0 +1,90 @@
|
|||
// apparmor.d - Full set of apparmor profiles
|
||||
// Copyright (C) 2021-2024 Alexandre Pujol <alexandre@pujol.io>
|
||||
// SPDX-License-Identifier: GPL-2.0-only
|
||||
|
||||
package cfg
|
||||
|
||||
import (
|
||||
"os"
|
||||
"strings"
|
||||
|
||||
"github.com/arduino/go-paths-helper"
|
||||
"golang.org/x/exp/slices"
|
||||
)
|
||||
|
||||
var (
|
||||
Distribution = getDistribution()
|
||||
Release = getOSRelease()
|
||||
Family = getFamily()
|
||||
)
|
||||
|
||||
var (
|
||||
osReleaseFile = "/etc/os-release"
|
||||
supportedDists = map[string][]string{
|
||||
"arch": {},
|
||||
"debian": {},
|
||||
"ubuntu": {},
|
||||
"opensuse": {"suse", "opensuse-tumbleweed"},
|
||||
"whonix": {},
|
||||
}
|
||||
famillyDists = map[string][]string{
|
||||
"apt": {"debian", "ubuntu", "whonix"},
|
||||
"pacman": {"arch"},
|
||||
"zypper": {"opensuse"},
|
||||
}
|
||||
)
|
||||
|
||||
func getOSRelease() map[string]string {
|
||||
var lines []string
|
||||
var err error
|
||||
for _, name := range []string{osReleaseFile, "/usr/lib/os-release"} {
|
||||
path := paths.New(name)
|
||||
if path.Exist() {
|
||||
lines, err = path.ReadFileAsLines()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
break
|
||||
}
|
||||
}
|
||||
os := map[string]string{}
|
||||
for _, line := range lines {
|
||||
item := strings.Split(line, "=")
|
||||
if len(item) == 2 {
|
||||
os[item[0]] = strings.Trim(item[1], "\"")
|
||||
}
|
||||
}
|
||||
return os
|
||||
}
|
||||
|
||||
func getDistribution() string {
|
||||
dist, present := os.LookupEnv("DISTRIBUTION")
|
||||
if present {
|
||||
return dist
|
||||
}
|
||||
|
||||
id := Release["ID"]
|
||||
if id == "ubuntu" {
|
||||
return id
|
||||
}
|
||||
id_like := Release["ID_LIKE"]
|
||||
for main, based := range supportedDists {
|
||||
if main == id || main == id_like {
|
||||
return main
|
||||
} else if slices.Contains(based, id) {
|
||||
return main
|
||||
} else if slices.Contains(based, id_like) {
|
||||
return main
|
||||
}
|
||||
}
|
||||
return id
|
||||
}
|
||||
|
||||
func getFamily() string {
|
||||
for familly, dist := range famillyDists {
|
||||
if slices.Contains(dist, Distribution) {
|
||||
return familly
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
223
pkg/prebuild/cfg/os_test.go
Normal file
223
pkg/prebuild/cfg/os_test.go
Normal file
|
|
@ -0,0 +1,223 @@
|
|||
// apparmor.d - Full set of apparmor profiles
|
||||
// Copyright (C) 2021-2024 Alexandre Pujol <alexandre@pujol.io>
|
||||
// SPDX-License-Identifier: GPL-2.0-only
|
||||
|
||||
package cfg
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
"testing"
|
||||
|
||||
"github.com/arduino/go-paths-helper"
|
||||
)
|
||||
|
||||
const (
|
||||
Archlinux = `NAME="Arch Linux"
|
||||
PRETTY_NAME="Arch Linux"
|
||||
ID=arch
|
||||
BUILD_ID=rolling
|
||||
ANSI_COLOR="38;2;23;147;209"
|
||||
HOME_URL="https://archlinux.org/"
|
||||
DOCUMENTATION_URL="https://wiki.archlinux.org/"
|
||||
SUPPORT_URL="https://bbs.archlinux.org/"
|
||||
BUG_REPORT_URL="https://bugs.archlinux.org/"
|
||||
PRIVACY_POLICY_URL="https://terms.archlinux.org/docs/privacy-policy/"
|
||||
LOGO=archlinux-logo`
|
||||
|
||||
Ubuntu = `PRETTY_NAME="Ubuntu 22.04.2 LTS"
|
||||
NAME="Ubuntu"
|
||||
VERSION_ID="22.04"
|
||||
VERSION="22.04.2 LTS (Jammy Jellyfish)"
|
||||
VERSION_CODENAME=jammy
|
||||
ID=ubuntu
|
||||
ID_LIKE=debian
|
||||
HOME_URL="https://www.ubuntu.com/"
|
||||
SUPPORT_URL="https://help.ubuntu.com/"
|
||||
BUG_REPORT_URL="https://bugs.launchpad.net/ubuntu/"
|
||||
PRIVACY_POLICY_URL="https://www.ubuntu.com/legal/terms-and-policies/privacy-policy"
|
||||
UBUNTU_CODENAME=jammy`
|
||||
|
||||
Debian = `PRETTY_NAME="Debian GNU/Linux 11 (bullseye)"
|
||||
NAME="Debian GNU/Linux"
|
||||
VERSION_ID="11"
|
||||
VERSION="11 (bullseye)"
|
||||
VERSION_CODENAME=bullseye
|
||||
ID=debian
|
||||
HOME_URL="https://www.debian.org/"
|
||||
SUPPORT_URL="https://www.debian.org/support"
|
||||
BUG_REPORT_URL="https://bugs.debian.org/"`
|
||||
|
||||
OpenSUSETumbleweed = `ID="opensuse-tumbleweed"
|
||||
ID_LIKE="opensuse suse"
|
||||
VERSION_ID="20230404"
|
||||
PRETTY_NAME="openSUSE Tumbleweed"
|
||||
ANSI_COLOR="0;32"
|
||||
CPE_NAME="cpe:/o:opensuse:tumbleweed:20230404"
|
||||
BUG_REPORT_URL="https://bugs.opensuse.org"
|
||||
HOME_URL="https://www.opensuse.org/"
|
||||
DOCUMENTATION_URL="https://en.opensuse.org/Portal:Tumbleweed"
|
||||
LOGO="distributor-logo-Tumbleweed"`
|
||||
|
||||
ArcoLinux = `NAME=ArcoLinux
|
||||
ID=arcolinux
|
||||
ID_LIKE=arch
|
||||
BUILD_ID=rolling
|
||||
ANSI_COLOR="0;36"
|
||||
HOME_URL="https://arcolinux.info/"
|
||||
SUPPORT_URL="https://arcolinuxforum.com/"
|
||||
BUG_REPORT_URL="https://github.com/arcolinux"
|
||||
LOGO=arcolinux-hello`
|
||||
|
||||
Fedora = `NAME="Fedora Linux"
|
||||
VERSION="37 (Workstation Edition)"
|
||||
ID=fedora
|
||||
VERSION_ID=37
|
||||
VERSION_CODENAME=""
|
||||
PLATFORM_ID="platform:f37"
|
||||
PRETTY_NAME="Fedora Linux 37 (Workstation Edition)"
|
||||
ANSI_COLOR="0;38;2;60;110;180"
|
||||
LOGO=fedora-logo-icon`
|
||||
)
|
||||
|
||||
func Test_getOSRelease(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
osRelease string
|
||||
want map[string]string
|
||||
}{
|
||||
{
|
||||
name: "Archlinux",
|
||||
osRelease: Archlinux,
|
||||
want: map[string]string{
|
||||
"NAME": "Arch Linux",
|
||||
"PRETTY_NAME": "Arch Linux",
|
||||
"ID": "arch",
|
||||
"BUILD_ID": "rolling",
|
||||
"ANSI_COLOR": "38;2;23;147;209",
|
||||
"HOME_URL": "https://archlinux.org/",
|
||||
"DOCUMENTATION_URL": "https://wiki.archlinux.org/",
|
||||
"SUPPORT_URL": "https://bbs.archlinux.org/",
|
||||
"BUG_REPORT_URL": "https://bugs.archlinux.org/",
|
||||
"PRIVACY_POLICY_URL": "https://terms.archlinux.org/docs/privacy-policy/",
|
||||
"LOGO": "archlinux-logo",
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Ubuntu",
|
||||
osRelease: Ubuntu,
|
||||
want: map[string]string{
|
||||
"PRETTY_NAME": "Ubuntu 22.04.2 LTS",
|
||||
"NAME": "Ubuntu",
|
||||
"VERSION_ID": "22.04",
|
||||
"VERSION": "22.04.2 LTS (Jammy Jellyfish)",
|
||||
"VERSION_CODENAME": "jammy",
|
||||
"ID": "ubuntu",
|
||||
"ID_LIKE": "debian",
|
||||
"HOME_URL": "https://www.ubuntu.com/",
|
||||
"SUPPORT_URL": "https://help.ubuntu.com/",
|
||||
"BUG_REPORT_URL": "https://bugs.launchpad.net/ubuntu/",
|
||||
"PRIVACY_POLICY_URL": "https://www.ubuntu.com/legal/terms-and-policies/privacy-policy",
|
||||
"UBUNTU_CODENAME": "jammy",
|
||||
},
|
||||
},
|
||||
}
|
||||
osReleaseFile = "/tmp/os-release"
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
err := paths.New(osReleaseFile).WriteFile([]byte(tt.osRelease))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if got := getOSRelease(); !reflect.DeepEqual(got, tt.want) {
|
||||
t.Errorf("getOSRelease() = %v, want %v", got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func Test_getDistribution(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
osRelease string
|
||||
want string
|
||||
}{
|
||||
{
|
||||
name: "Archlinux",
|
||||
osRelease: Archlinux,
|
||||
want: "arch",
|
||||
},
|
||||
{
|
||||
name: "Ubuntu",
|
||||
osRelease: Ubuntu,
|
||||
want: "ubuntu",
|
||||
},
|
||||
{
|
||||
name: "Debian",
|
||||
osRelease: Debian,
|
||||
want: "debian",
|
||||
},
|
||||
{
|
||||
name: "OpenSUSE Tumbleweed",
|
||||
osRelease: OpenSUSETumbleweed,
|
||||
want: "opensuse",
|
||||
},
|
||||
{
|
||||
name: "Fedora",
|
||||
osRelease: Fedora,
|
||||
want: "fedora",
|
||||
},
|
||||
}
|
||||
|
||||
osReleaseFile = "/tmp/os-release"
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
err := paths.New(osReleaseFile).WriteFile([]byte(tt.osRelease))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
Release = getOSRelease()
|
||||
got := getDistribution()
|
||||
if got != tt.want {
|
||||
t.Errorf("getDistribution() = %v, want %v", got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func Test_getFamily(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
dist string
|
||||
want string
|
||||
}{
|
||||
{
|
||||
name: "Archlinux",
|
||||
dist: "arch",
|
||||
want: "pacman",
|
||||
},
|
||||
{
|
||||
name: "Ubuntu",
|
||||
dist: "ubuntu",
|
||||
want: "apt",
|
||||
},
|
||||
{
|
||||
name: "Debian",
|
||||
dist: "debian",
|
||||
want: "apt",
|
||||
},
|
||||
{
|
||||
name: "OpenSUSE Tumbleweed",
|
||||
dist: "opensuse",
|
||||
want: "zypper",
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
Distribution = tt.dist
|
||||
if got := getFamily(); got != tt.want {
|
||||
t.Errorf("getFamily() = %v, want %v", got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue