kuvia2/files.go

121 lines
2.9 KiB
Go
Raw Normal View History

2022-01-14 23:09:03 +00:00
package main
import (
"bytes"
"context"
"fmt"
"github.com/flamego/flamego"
"github.com/flamego/session"
"github.com/flamego/template"
"github.com/minio/minio-go/v7"
"github.com/nfnt/resize"
"github.com/rs/zerolog/log"
"image"
"image/jpeg"
"math"
"net/http"
"os"
"strconv"
"time"
)
type File struct {
ID int `db:"id"`
GalleryId int `db:"gallery_id"`
Path string `db:"path"`
PathOrginalSize string `db:"path_orginal"`
}
func uploadImageForm(t template.Template, data template.Data , s session.Session, c flamego.Context) {
id := c.Request().FormValue("id")
if id == "" {
return
}
data["id"] = id
2022-01-15 12:07:08 +00:00
data["active"] = "upload"
2022-01-14 23:09:03 +00:00
t.HTML(http.StatusOK, "file_upload")
}
/*
Something foobar das gibt es doch nicht
*/
func uploadImage(c flamego.Context, s session.Session) {
//todo: check if loggedin
//todo: check if gllery is owned by uploader
file, fileHeader, err := c.Request().FormFile("file")
if err != nil {
panic(err)
}
defer file.Close()
galleryIdString := c.Request().FormValue("id")
galleryId, err := strconv.Atoi(galleryIdString)
if err != nil {
panic(err)
}
path := fmt.Sprintf("/gallery/%v/%v/data/%v-%v", s.Get("user_id"), galleryId, time.Now().UnixNano(), fileHeader.Filename)
pathOrignal := fmt.Sprintf("/gallery/%v/%v/data/%v-%v", s.Get("user_id"), galleryId, time.Now().UnixNano(), fileHeader.Filename)
imgO, _ , err := image.Decode(file)
if err != nil {
panic(err)
}
newImage := imgO
x := math.Max(float64(imgO.Bounds().Dx()), float64(imgO.Bounds().Dy()))
if x > 1500 {
if imgO.Bounds().Dx() > imgO.Bounds().Dy() {
newImage = resize.Resize(1200, 0, imgO, resize.Lanczos3)
} else {
newImage = resize.Resize(0, 1200, imgO, resize.Lanczos3)
}
pathOrignal = fmt.Sprintf("/gallery/%v/%v/data/%v-o-%v", s.Get("user_id"), galleryId, time.Now().UnixNano(), fileHeader.Filename)
uploadImageToS3(imgO, pathOrignal)
}
info, err := uploadImageToS3(newImage, path)
if err != nil {
panic(err)
}
_, err = sqlConnection.NamedExec("INSERT INTO `images` (gallery_id, path, path_orginal) VALUES (:gallery, :path, :path_orginal)", map[string]interface{}{
"gallery": galleryId,
"path": info.Key,
"path_orginal": pathOrignal,
})
if err != nil {
err2 := s3.RemoveObject(context.Background(), os.Getenv("S3_BUCKET"), info.Key, minio.RemoveObjectOptions{})
if err2 != nil {
log.Warn().Err(err).Msg("Cant remove Unused File from s3 stroage")
}
panic(err)
}
c.ResponseWriter().Write([]byte("ok"))
}
func uploadImageToS3(img image.Image, path string) (minio.UploadInfo, error) {
buf := new(bytes.Buffer)
//err := png.Encode(buf, img)
o := jpeg.Options{
Quality: 100,
}
err := jpeg.Encode(buf, img, &o)
if err != nil {
return minio.UploadInfo{}, err
}
//send_s3 := buf.Bytes()
info, err := s3.PutObject(context.Background(), os.Getenv("S3_BUCKET"), path, buf, -1, minio.PutObjectOptions{})
return info, err
}