From fb0cb0fcdcf045f7fde2398f1713fcf0527123a7 Mon Sep 17 00:00:00 2001
From: Aidan-Z <aidanzamfir@gmail.com>
Date: Mon, 10 Jun 2024 10:35:22 -0300
Subject: [PATCH] flag working in nm/st

---
 cli/normalize.go   | 31 ++++++++++++-------------------
 cli/standardise.go | 21 ++++++++++++++-------
 2 files changed, 26 insertions(+), 26 deletions(-)

diff --git a/cli/normalize.go b/cli/normalize.go
index 69d2b33..c33ee41 100644
--- a/cli/normalize.go
+++ b/cli/normalize.go
@@ -28,32 +28,25 @@ var normalizeCmd = &cobra.Command{
 			cols = append(cols, result)
 		}
 
-		//var exportFilePath string
-		//log.Println(exportFilePath, "path")
-		//if exportFilePath == " " {
-		//	log.Println("was nil")
-		//	err = csvData.ToCSV(head, cols, "./DataNM.csv")
-		//	if err != nil {
-		//		log.Fatal(err)
-		//	}
-		//}
-		//log.Println("Not empty???")
-		//err = csvData.ToCSV(head, cols, exportFilePath)
-		//if err != nil {
-		//	log.Fatal(err)
-		//}
-
-		err = csvData.ToCSV(head, cols, "./DataNM.csv")
-		if err != nil {
-			log.Fatal(err)
+		if exportFilePath != "" {
+			err = csvData.ToCSV(head, cols, exportFilePath)
+			if err != nil {
+				log.Fatal(err)
+			}
+		} else {
+			err = csvData.ToCSV(head, cols, "./DataNM.csv")
+			if err != nil {
+				log.Fatal(err)
+			}
 		}
 
 	},
 }
 
 func init() {
-	var exportFilePath string
 
 	rootCmd.AddCommand(normalizeCmd)
 	normalizeCmd.Flags().StringVarP(&exportFilePath, "export", "e", "", "Export to filepath -> (provide path) | if empty -> ./DataNM.csv")
 }
+
+// go run . nm testdata.csv -e ../../Desktop/test.csv
diff --git a/cli/standardise.go b/cli/standardise.go
index baf34b7..ed966fc 100644
--- a/cli/standardise.go
+++ b/cli/standardise.go
@@ -7,6 +7,8 @@ import (
 	"log"
 )
 
+var exportFilePath string
+
 var standardiseCmd = &cobra.Command{
 	Use:   "stand",
 	Short: "Will return standardised values as .csv file", //decide
@@ -26,18 +28,23 @@ var standardiseCmd = &cobra.Command{
 			result := <-c
 			cols = append(cols, result)
 		}
-
-		err = csvData.ToCSV(head, cols, "./DataST.csv")
-		if err != nil {
-			log.Fatal(err)
+		if exportFilePath != "" {
+			err = csvData.ToCSV(head, cols, exportFilePath)
+			if err != nil {
+				log.Fatal(err)
+			}
+		} else {
+			err = csvData.ToCSV(head, cols, "./DataST.csv")
+			if err != nil {
+				log.Fatal(err)
+			}
 		}
-
 	},
 }
 
 func init() {
-	var exportFilePath string
-
 	rootCmd.AddCommand(standardiseCmd)
 	standardiseCmd.Flags().StringVarP(&exportFilePath, "export", "e", "", "Export to filepath -> (provide path) | if empty -> ./DataST.csv")
 }
+
+// go run . stand testdata.csv -e ../../Desktop/test.csv