mirror of
https://github.com/roddhjav/apparmor.d.git
synced 2024-11-15 07:54:17 +01:00
refractor: use internal Intersect function.
This commit is contained in:
parent
3b82cc36ba
commit
b0f6f15a9e
@ -8,7 +8,7 @@ import (
|
|||||||
"slices"
|
"slices"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/samber/lo"
|
"github.com/roddhjav/apparmor.d/pkg/util"
|
||||||
)
|
)
|
||||||
|
|
||||||
type requirement map[string][]string
|
type requirement map[string][]string
|
||||||
@ -225,7 +225,7 @@ func (r Rules) Format() Rules {
|
|||||||
rule := r[i].(*File)
|
rule := r[i].(*File)
|
||||||
|
|
||||||
// Add padding to align with other transition rule
|
// Add padding to align with other transition rule
|
||||||
isTransition := lo.Intersect(transitions, rule.Access)
|
isTransition := util.Intersect(transitions, rule.Access)
|
||||||
if len(isTransition) > 0 {
|
if len(isTransition) > 0 {
|
||||||
ruleLen := len(rule.Path) + 1
|
ruleLen := len(rule.Path) + 1
|
||||||
paddingMaxLenght = max(ruleLen, paddingMaxLenght)
|
paddingMaxLenght = max(ruleLen, paddingMaxLenght)
|
||||||
@ -265,7 +265,7 @@ func (r Rules) Format() Rules {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Do not add new line on executable rule
|
// Do not add new line on executable rule
|
||||||
isTransition := lo.Intersect(transitions, rule.Access)
|
isTransition := util.Intersect(transitions, rule.Access)
|
||||||
if len(isTransition) > 0 {
|
if len(isTransition) > 0 {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
@ -82,6 +82,25 @@ func RemoveDuplicate[T comparable](inlist []T) []T {
|
|||||||
return list
|
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.
|
// CopyTo recursivelly copy all files from a source path to a destination path.
|
||||||
func CopyTo(src *paths.Path, dst *paths.Path) error {
|
func CopyTo(src *paths.Path, dst *paths.Path) error {
|
||||||
files, err := src.ReadDirRecursiveFiltered(nil,
|
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) {
|
func TestToRegexRepl(t *testing.T) {
|
||||||
tests := []struct {
|
tests := []struct {
|
||||||
name string
|
name string
|
||||||
|
Loading…
Reference in New Issue
Block a user