mirror of
https://github.com/mudler/LocalAI.git
synced 2024-12-19 12:47:54 +00:00
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/base64"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
var base64DownloadClient http.Client = http.Client{
|
||
|
|
Timeout: 30 * time.Second,
|
||
|
|
}
|
||
|
|
|
||
|
|
// this function check if the string is an URL, if it's an URL downloads the image in memory
|
||
|
|
// encodes it in base64 and returns the base64 string
|
||
|
|
|
||
|
|
// This may look weird down in pkg/utils while it is currently only used in core/config
|
||
|
|
//
|
||
|
|
// but I believe it may be useful for MQTT as well in the near future, so I'm
|
||
|
|
// extracting it while I'm thinking of it.
|
||
|
|
func GetImageURLAsBase64(s string) (string, error) {
|
||
|
|
if strings.HasPrefix(s, "http") {
|
||
|
|
// download the image
|
||
|
|
resp, err := base64DownloadClient.Get(s)
|
||
|
|
if err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
|
||
|
|
// read the image data into memory
|
||
|
|
data, err := io.ReadAll(resp.Body)
|
||
|
|
if err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
|
||
|
|
// encode the image data in base64
|
||
|
|
encoded := base64.StdEncoding.EncodeToString(data)
|
||
|
|
|
||
|
|
// return the base64 string
|
||
|
|
return encoded, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// if the string instead is prefixed with "data:image/jpeg;base64,", drop it
|
||
|
|
if strings.HasPrefix(s, "data:image/jpeg;base64,") {
|
||
|
|
return strings.ReplaceAll(s, "data:image/jpeg;base64,", ""), nil
|
||
|
|
}
|
||
|
|
return "", fmt.Errorf("not valid string")
|
||
|
|
}
|