From 8f185c8e8508083b8412dd2b40e3bf56883d11fc Mon Sep 17 00:00:00 2001
From: Victor Lacasse-Beaudoin <vlbeaudoin@agecem.org>
Date: Tue, 19 Dec 2023 19:42:08 -0500
Subject: [PATCH] feat(api): ajouter `UploadDocuments` pour `V1DocumentsPOST`

---
 api/api.go | 74 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)

diff --git a/api/api.go b/api/api.go
index c1e49e3..06bda1a 100644
--- a/api/api.go
+++ b/api/api.go
@@ -36,6 +36,80 @@ func New(client *http.Client, host, key string, port int, protocol string) (*API
 	return &API{Voki: voki.New(client, host, key, port, protocol)}, nil
 }
 
+func (a *API) UploadDocuments(bucketName string, fileHeaders ...*multipart.FileHeader) (response apiresponse.V1DocumentsPOST, err error) {
+	if count := len(fileHeaders); count == 0 {
+		err = fmt.Errorf("api.(*API).UploadDocuments requiert au moins 1 fichier")
+		return
+	}
+
+	endpoint := fmt.Sprintf("%s://%s:%d/v1/bucket/%s/many",
+		a.Voki.Protocol,
+		a.Voki.Host,
+		a.Voki.Port,
+		bucketName,
+	)
+
+	// Create new multipart writer
+	body := &bytes.Buffer{}
+	writer := multipart.NewWriter(body)
+
+	// Add files to the request
+	for i, fileHeader := range fileHeaders {
+		if fileHeader == nil {
+			return response, fmt.Errorf("Fichier %d pointe vers une addresse mémoire nulle", i)
+		}
+
+		file, err := fileHeader.Open()
+		if err != nil {
+			return response, fmt.Errorf("Impossible de lire le contenu du fichier %d '%s': %s", i, fileHeader.Filename, err)
+		}
+		defer file.Close()
+
+		fileName, err := url.QueryUnescape(fileHeader.Filename)
+		if err != nil {
+			return response, fmt.Errorf("Fichier %d '%s' a un nom invalide et impossible à convertir: %s", i, fileHeader.Filename, err)
+		}
+
+		part, err := writer.CreatePart(fileHeader.Header)
+		if err != nil {
+			return response, fmt.Errorf("Impossible d'ajouter %d '%s' au formulaire de téléversement: %s", i, fileName, err)
+		}
+
+		_, err = io.Copy(part, file)
+		if err != nil {
+			return response, fmt.Errorf("Impossible d'ajouter le contenu de %d '%s' au formulaire de téléversement: %s", i, fileName, err)
+		}
+	}
+
+	if err := writer.Close(); err != nil {
+		return response, fmt.Errorf("Impossible de fermer le io.Writer: %s", err)
+	}
+
+	req, err := http.NewRequest(http.MethodPost, endpoint, body)
+	if err != nil {
+		return response, fmt.Errorf("Impossible de produire une requête: %s", err)
+	}
+
+	if err := req.ParseForm(); err != nil {
+		return response, fmt.Errorf("Impossible de parse le formulaire: %s", err)
+	}
+
+	req.Header.Set("Content-Type", writer.FormDataContentType())
+
+	if a.Voki.Key != "" {
+		req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", a.Voki.Key))
+	}
+
+	// Send the HTTP request
+	httpResponse, err := a.Voki.Client.Do(req)
+	if err != nil {
+		return response, fmt.Errorf("Impossible d'exécuter la requête http: %s", err)
+	}
+	defer httpResponse.Body.Close()
+
+	return response, json.NewDecoder(httpResponse.Body).Decode(&response)
+}
+
 func (a *API) UploadDocument(bucket string, file_header *multipart.FileHeader) (apiresponse.V1DocumentPOST, error) {
 	var response apiresponse.V1DocumentPOST
 	endpoint := fmt.Sprintf("%s://%s:%d",