From 3fe2569fd99427aaf36404651af26280ca4df489 Mon Sep 17 00:00:00 2001
From: Davidson Gomes <davidsongviolao@gmail.com>
Date: Tue, 22 Oct 2024 08:58:48 -0300
Subject: [PATCH] feat: convert to mp3

---
 main.go | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index 5a887bd..e48ead8 100644
--- a/main.go
+++ b/main.go
@@ -69,9 +69,14 @@ func validateAPIKey(c *gin.Context) bool {
 	return true
 }
 
-func convertAudioToOpusWithDuration(inputData []byte) ([]byte, int, error) {
-	cmd := exec.Command("ffmpeg", "-i", "pipe:0", "-ac", "1", "-ar", "16000", "-c:a", "libopus", "-f", "ogg", "pipe:1")
-
+func convertAudio(inputData []byte, format string) ([]byte, int, error) {
+	var cmd *exec.Cmd
+	switch format {
+	case "mp3":
+		cmd = exec.Command("ffmpeg", "-i", "pipe:0", "-f", "mp3", "pipe:1")
+	default:
+		cmd = exec.Command("ffmpeg", "-i", "pipe:0", "-ac", "1", "-ar", "16000", "-c:a", "libopus", "-f", "ogg", "pipe:1")
+	}
 	outBuffer := bufferPool.Get().(*bytes.Buffer)
 	errBuffer := bufferPool.Get().(*bytes.Buffer)
 	defer bufferPool.Put(outBuffer)
@@ -151,7 +156,9 @@ func processAudio(c *gin.Context) {
 		return
 	}
 
-	convertedData, duration, err := convertAudioToOpusWithDuration(inputData)
+	format := c.DefaultPostForm("format", "ogg")
+
+	convertedData, duration, err := convertAudio(inputData, format)
 	if err != nil {
 		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
 		return
@@ -160,6 +167,7 @@ func processAudio(c *gin.Context) {
 	c.JSON(http.StatusOK, gin.H{
 		"duration": duration,
 		"audio":    base64.StdEncoding.EncodeToString(convertedData),
+		"format":   format,
 	})
 }