FEAT - add csv flag

Author: fcastillo90

cmd/get.go

@@ -37,6 +37,15 @@ var getCmd = &cobra.Command{
 			return
 		}
 
+		if flags.Csv != "" && !flags.Poll {
+			printErrorProps := S.PrintErrorProps{
+				Error:   errors.New("CSV output is only supported with polling"),
+				Message: "CSV output is only supported with polling.",
+			}
+			U.PrintError(printErrorProps)
+			return
+		}
+
 		U.PollTranscription(id, flags)
 	},
 }

utils/transcribe.go

@@ -22,6 +22,7 @@ import (
 )
 
 var width int
+var Flags S.TranscribeFlags
 
 func Transcribe(params S.TranscribeParams, flags S.TranscribeFlags) {
 	Token = GetStoredToken()
@@ -223,8 +224,8 @@ func UploadFile(path string) string {
 }
 
 func PollTranscription(id string, flags S.TranscribeFlags) {
+	Flags = flags
 	fmt.Fprintln(os.Stdin, "Transcribing file with id "+id)
-
 	s := CallSpinner(" Processing time is usually 20% of the file's duration.")
 
 	for {
@@ -277,14 +278,24 @@ func PollTranscription(id string, flags S.TranscribeFlags) {
 				fmt.Println(string(print))
 				return
 			}
-			getFormattedOutput(transcript, flags)
+			if flags.Csv != "" {
+				if filepath.Ext(flags.Csv) == "" {
+					flags.Csv = flags.Csv + ".csv"
+				}
+
+				row := [][]string{}
+				row = append(row, []string{"\"" + *transcript.Text + "\""})
+				GenerateCsv(flags.Csv, []string{"text"}, row)
+			}
+
+			getFormattedOutput(transcript)
 			return
 		}
 		time.Sleep(3 * time.Second)
 	}
 }
 
-func getFormattedOutput(transcript S.TranscriptResponse, flags S.TranscribeFlags) {
+func getFormattedOutput(transcript S.TranscriptResponse) {
 	getWidth, _, err := term.GetSize(0)
 	if err != nil {
 		width = 512

utils/utils.go

@@ -468,3 +468,47 @@ func Contains(s []string, e string) bool {
 	}
 	return false
 }
+
+func GenerateCsv(filename string, headers []string, data [][]string) {
+	file, err := os.OpenFile(filename, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
+	if err != nil {
+		printErrorProps := S.PrintErrorProps{
+			Error:   err,
+			Message: "Error opening file",
+		}
+		PrintError(printErrorProps)
+	}
+	defer file.Close()
+
+	// if file is empty, write headers
+	fileInfo, err := file.Stat()
+	if err != nil {
+		printErrorProps := S.PrintErrorProps{
+			Error:   err,
+			Message: "Error getting file info",
+		}
+		PrintError(printErrorProps)
+	}
+	if fileInfo.Size() == 0 {
+		_, err := file.WriteString(strings.Join(headers, ",") + "\n")
+		if err != nil {
+			printErrorProps := S.PrintErrorProps{
+				Error:   err,
+				Message: "Error writing headers",
+			}
+			PrintError(printErrorProps)
+		}
+	}
+
+	for _, value := range data {
+		_, err := file.WriteString(strings.Join(value, ",") + "\n")
+		if err != nil {
+			printErrorProps := S.PrintErrorProps{
+				Error:   err,
+				Message: "Error writing to file",
+			}
+			PrintError(printErrorProps)
+		}
+
+	}
+}