Show stored images to the user

[onechat] / telegram_fake_helper_test.go

diff --git a/telegram_fake_helper_test.go b/telegram_fake_helper_test.go
index f6ccd8425d41e36372fc51fb8e8a5afc7f2b2216..b71eee038ebf084da3481bd63c3643cda8600c0d 100644 (file)
--- a/telegram_fake_helper_test.go
+++ b/telegram_fake_helper_test.go
@@ -2,8 +2,10 @@ package main
 
 import (
        "encoding/json"
+       "errors"
        "fmt"
        "io"
+       "mime/multipart"
        "net/http"
        "net/http/httptest"
        "strings"
@@ -139,7 +141,38 @@ func (f *TelegramFake) serve(w http.ResponseWriter, r *http.Request) {
 
        var params map[string]any
 
-       err := json.NewDecoder(r.Body).Decode(&params)
+       var err error
+
+       if strings.HasPrefix(r.Header.Get("Content-Type"), "multipart/form-data;") {
+               err = r.ParseMultipartForm(imageMaxBytes + (1 << 20))
+               if err == nil {
+                       params = make(map[string]any)
+                       for key, values := range r.MultipartForm.Value {
+                               params[key] = values[0]
+                       }
+
+                       for key, files := range r.MultipartForm.File {
+                               var file multipart.File
+
+                               file, err = files[0].Open()
+                               if err != nil {
+                                       break
+                               }
+
+                               params[key], err = io.ReadAll(file)
+                               err = errors.Join(err, file.Close())
+                               params[key+"_name"] = files[0].Filename
+                               if err != nil {
+                                       break
+                               }
+                       }
+
+                       err = errors.Join(err, r.MultipartForm.RemoveAll())
+               }
+       } else {
+               err = json.NewDecoder(r.Body).Decode(&params)
+       }
+
        if err != nil {
                http.Error(w, err.Error(), http.StatusBadRequest)