refractor(build): move builder tasks to the builder sub package. Add tests.

This commit is contained in:
Alexandre Pujol 2024-03-25 23:16:00 +00:00
parent 08d4110c2a
commit 16f00ebfc7
No known key found for this signature in database
GPG key ID: C5469996F0DF68EC
7 changed files with 516 additions and 0 deletions

View file

@ -0,0 +1,47 @@
// apparmor.d - Full set of apparmor profiles
// Copyright (C) 2021-2024 Alexandre Pujol <alexandre@pujol.io>
// SPDX-License-Identifier: GPL-2.0-only
package builder
import (
"slices"
"strings"
"github.com/roddhjav/apparmor.d/pkg/prebuild/cfg"
)
type Enforce struct {
cfg.Base
}
func init() {
RegisterBuilder(&Enforce{
Base: cfg.Base{
Keyword: "enforce",
Msg: "All profiles have been enforced",
},
})
}
func (b Enforce) Apply(profile string) string {
matches := regFlags.FindStringSubmatch(profile)
if len(matches) == 0 {
return profile
}
flags := strings.Split(matches[1], ",")
idx := slices.Index(flags, "complain")
if idx == -1 {
return profile
}
flags = slices.Delete(flags, idx, idx+1)
strFlags := "{"
if len(flags) >= 1 {
strFlags = " flags=(" + strings.Join(flags, ",") + ") {"
}
// Remove all flags definition, then set new flags
profile = regFlags.ReplaceAllLiteralString(profile, "")
return regProfileHeader.ReplaceAllLiteralString(profile, strFlags)
}