personal-website/libs/imglib.go

package libs

import (
	"errors"
	"net/http"
	"personalwebsite/webfilesystem"

	"github.com/gin-gonic/gin"
	"go.mongodb.org/mongo-driver/bson/primitive"
)

type ImagLib struct {
	fs *webfilesystem.WebFileSystem
}

func NewImgLib(webfs *webfilesystem.WebFileSystem) *ImagLib {
	return &ImagLib{
		fs: webfs,
	}
}

func ReadImage(img *webfilesystem.WebFSFile) (*Img, error) {
	data, ok := img.Data.(primitive.D).Map()["srcdata"]
	if !ok {
		return nil, errors.New("error in file decoding")
	}
	bin := data.(primitive.Binary).Data

	min32Data, ok := img.Data.(primitive.D).Map()["min32data"]
	if !ok {
		return nil, errors.New("error in file decoding")
	}
	min32Bin := min32Data.(primitive.Binary).Data
	return &Img{
		// Header: header,
		Data:        bin,
		Miniature32: min32Bin,
	}, nil
}

func (l *ImagLib) Route(route *gin.RouterGroup) {
	route.GET("get", func(ctx *gin.Context) {
		path := ctx.Query("path")
		if path == "" {
			ctx.JSON(http.StatusBadRequest, "TODO") //TODO json error struct
			return
		}

		file, err := l.fs.NewRead(path)
		if err != nil {
			ctx.String(http.StatusInternalServerError, "TODO") //TODO
		}
		data := file.Data.(primitive.Binary)

		ctx.Data(http.StatusOK, "image/jpeg", data.Data)
	})

}

func GetBase64Image(img *Base64Img, min string) (string, error) {
	imgString := ""
	switch min {
	case "min32":
		imgString = "data:" + img.Header + ", " + img.Base64Miniature32
	default:
		imgString = "data:" + img.Header + ", " + img.Base64
	}

	return imgString, nil
}

type Base64Img struct {
	Header            string `bson:"header"`
	Base64            string `bson:"base64"`
	Base64Miniature32 string `bson:"base64min32"`
}

type Img struct {
	Header      string `bson:"header"`
	Data        []byte `bson:"srcdata"`
	Miniature32 []byte `bson:"min32data"`
}