mirror of
https://github.com/roddhjav/apparmor.d.git
synced 2024-11-14 23:43:56 +01:00
refractor: use internal Intersect function.
This commit is contained in:
parent
3b82cc36ba
commit
b0f6f15a9e
@ -8,7 +8,7 @@ import (
|
||||
"slices"
|
||||
"strings"
|
||||
|
||||
"github.com/samber/lo"
|
||||
"github.com/roddhjav/apparmor.d/pkg/util"
|
||||
)
|
||||
|
||||
type requirement map[string][]string
|
||||
@ -225,7 +225,7 @@ func (r Rules) Format() Rules {
|
||||
rule := r[i].(*File)
|
||||
|
||||
// Add padding to align with other transition rule
|
||||
isTransition := lo.Intersect(transitions, rule.Access)
|
||||
isTransition := util.Intersect(transitions, rule.Access)
|
||||
if len(isTransition) > 0 {
|
||||
ruleLen := len(rule.Path) + 1
|
||||
paddingMaxLenght = max(ruleLen, paddingMaxLenght)
|
||||
@ -265,7 +265,7 @@ func (r Rules) Format() Rules {
|
||||
}
|
||||
|
||||
// Do not add new line on executable rule
|
||||
isTransition := lo.Intersect(transitions, rule.Access)
|
||||
isTransition := util.Intersect(transitions, rule.Access)
|
||||
if len(isTransition) > 0 {
|
||||
continue
|
||||
}
|
||||
|
@ -82,6 +82,25 @@ func RemoveDuplicate[T comparable](inlist []T) []T {
|
||||
return list
|
||||
}
|
||||
|
||||
// Intersect returns the intersection between two collections.
|
||||
// From https://github.com/samber/lo
|
||||
func Intersect[T comparable](list1 []T, list2 []T) []T {
|
||||
result := []T{}
|
||||
seen := map[T]struct{}{}
|
||||
|
||||
for _, elem := range list1 {
|
||||
seen[elem] = struct{}{}
|
||||
}
|
||||
|
||||
for _, elem := range list2 {
|
||||
if _, ok := seen[elem]; ok {
|
||||
result = append(result, elem)
|
||||
}
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
// CopyTo recursivelly copy all files from a source path to a destination path.
|
||||
func CopyTo(src *paths.Path, dst *paths.Path) error {
|
||||
files, err := src.ReadDirRecursiveFiltered(nil,
|
||||
|
@ -59,6 +59,53 @@ func TestRemoveDuplicate(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestIntersect(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
list1 []int
|
||||
list2 []int
|
||||
want []int
|
||||
}{
|
||||
{
|
||||
name: "1",
|
||||
list1: []int{0, 1, 2, 3, 4, 5},
|
||||
list2: []int{0, 2},
|
||||
want: []int{0, 2},
|
||||
},
|
||||
{
|
||||
name: "2",
|
||||
list1: []int{0, 1, 2, 3, 4, 5},
|
||||
list2: []int{0, 6},
|
||||
want: []int{0},
|
||||
},
|
||||
{
|
||||
name: "3",
|
||||
list1: []int{0, 1, 2, 3, 4, 5},
|
||||
list2: []int{-1, 6},
|
||||
want: []int{},
|
||||
},
|
||||
{
|
||||
name: "4",
|
||||
list1: []int{0, 6},
|
||||
list2: []int{0, 1, 2, 3, 4, 5},
|
||||
want: []int{0},
|
||||
},
|
||||
{
|
||||
name: "5",
|
||||
list1: []int{0, 6, 0},
|
||||
list2: []int{0, 1, 2, 3, 4, 5},
|
||||
want: []int{0},
|
||||
},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
if got := Intersect(tt.list1, tt.list2); !reflect.DeepEqual(got, tt.want) {
|
||||
t.Errorf("Intersect() = %v, want %v", got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestToRegexRepl(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
|
Loading…
Reference in New Issue
Block a user