format mapper added

localeum · May 18, 2021 · 4bb5dc0 · 4bb5dc0
1 parent 5eba728
commit 4bb5dc0
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/cmd/pull.go b/cmd/pull.go
@@ -75,7 +75,7 @@ func fetchFile(lang string, wg *sync.WaitGroup) {
 		return
 	}
 
-	filename := viper.GetString(DirectoryFlag) + "/" + lang + "." + format
+	filename := viper.GetString(DirectoryFlag) + "/" + lang + formatToExtensionMapper(format)
 	file, err := os.Create(filename)
 	if err != nil {
 		fmt.Println(err)
@@ -93,3 +93,18 @@ func fetchFile(lang string, wg *sync.WaitGroup) {
 	fmt.Println("File -", filename, "downloaded")
 	wg.Done()
 }
+
+func formatToExtensionMapper(format string) string {
+	switch format {
+	case "json":
+		return ".json"
+	case "json_nested":
+		return ".json"
+	case "csv":
+		return ".csv"
+	case "arb":
+		return ".arb"
+	default:
+		return ""
+	}
+}