Signed-off-by: Vasek - Tom C <tom.chauveau@epitech.eu>
This commit is contained in:
Vasek - Tom C 2022-01-28 15:27:32 +01:00
parent cb3a6ed0c3
commit fcd90fa78a
No known key found for this signature in database
GPG Key ID: 175D82E572427960

View File

@ -17,7 +17,7 @@ import (
var (
// FS contains the filesystem of the stdlib.
//go:embed */**/*.cue */**/**/*.cue
//go:embed */**/*.cue */**/**/*.cue */**/*.sh
FS embed.FS
)
@ -112,6 +112,17 @@ func Vendor(ctx context.Context, p string) error {
return nil
}
func isAllowedExt(ext string) bool {
// List of allowed extension to vendor in
allowedExtension := []string{".cue", ".sh"}
for _, v := range allowedExtension {
if v == ext {
return true
}
}
return false
}
func extractModules(dest string) error {
return fs.WalkDir(FS, ".", func(p string, entry fs.DirEntry, err error) error {
if err != nil {
@ -122,7 +133,7 @@ func extractModules(dest string) error {
return nil
}
if filepath.Ext(entry.Name()) != ".cue" {
if !isAllowedExt(filepath.Ext(entry.Name())) {
return nil
}