Add filename regex

italiaremote · Nov 12, 2022 · 8291e4f · 8291e4f
1 parent dc601b0
commit 8291e4f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/cmd/formatter/main.go b/src/cmd/formatter/main.go
@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"log"
 	"os"
+	"regexp"
 
 	"github.com/italiaremote/awesome-italia-remote/pkg/cmp"
 )
@@ -16,6 +17,7 @@ func main() {
 	if err != nil {
 		log.Fatalln(err)
 	}
+	fileNameRegex := regexp.MustCompile(`([a-zA-Z0-9\_\-]+).json$`)
 
 	for _, companyFiles := range companiesFile {
 		fileName := dataPath + companyFiles.Name()
@@ -25,6 +27,10 @@ func main() {
 		}
 		var company cmp.Company
 
+		if !fileNameRegex.MatchString(fileName) {
+			log.Fatalln("Invalid filename, use only letters. numbers and - _")
+		}
+
 		err = json.Unmarshal(file, &company)
 		if err != nil {
 			log.Fatalln(err)