Paperclip/pkg/internal/services/uploader.go

151 lines
4.5 KiB
Go
Raw Normal View History

2024-05-17 07:59:51 +00:00
package services
import (
"context"
2024-08-20 09:08:40 +00:00
"errors"
2024-05-17 07:59:51 +00:00
"fmt"
"io"
"mime/multipart"
"os"
2024-05-17 07:59:51 +00:00
"path/filepath"
2024-07-29 05:22:57 +00:00
"git.solsynth.dev/hydrogen/paperclip/pkg/internal/database"
2024-06-22 04:18:54 +00:00
"git.solsynth.dev/hydrogen/paperclip/pkg/internal/models"
2024-05-17 07:59:51 +00:00
"github.com/gofiber/fiber/v2"
jsoniter "github.com/json-iterator/go"
"github.com/minio/minio-go/v7"
"github.com/minio/minio-go/v7/pkg/credentials"
"github.com/spf13/viper"
)
2024-07-28 13:03:56 +00:00
func UploadFileToTemporary(ctx *fiber.Ctx, file *multipart.FileHeader, meta models.Attachment) error {
destMap := viper.GetStringMap("destinations.temporary")
var dest models.BaseDestination
rawDest, _ := jsoniter.Marshal(destMap)
_ = jsoniter.Unmarshal(rawDest, &dest)
switch dest.Type {
case models.DestinationTypeLocal:
var destConfigured models.LocalDestination
_ = jsoniter.Unmarshal(rawDest, &destConfigured)
2024-08-20 09:08:40 +00:00
return ctx.SaveFile(file, filepath.Join(destConfigured.Path, meta.Uuid))
2024-07-28 13:03:56 +00:00
default:
return fmt.Errorf("invalid destination: unsupported protocol %s", dest.Type)
2024-05-17 07:59:51 +00:00
}
2024-07-28 13:03:56 +00:00
}
2024-08-20 09:08:40 +00:00
func UploadChunkToTemporary(ctx *fiber.Ctx, cid string, file *multipart.FileHeader, meta models.Attachment) error {
destMap := viper.GetStringMap("destinations.temporary")
var dest models.BaseDestination
rawDest, _ := jsoniter.Marshal(destMap)
_ = jsoniter.Unmarshal(rawDest, &dest)
switch dest.Type {
case models.DestinationTypeLocal:
var destConfigured models.LocalDestination
_ = jsoniter.Unmarshal(rawDest, &destConfigured)
tempPath := filepath.Join(destConfigured.Path, fmt.Sprintf("%s.part%s.partial", meta.Uuid, cid))
destPath := filepath.Join(destConfigured.Path, fmt.Sprintf("%s.part%s", meta.Uuid, cid))
if err := ctx.SaveFile(file, tempPath); err != nil {
return err
}
return os.Rename(tempPath, destPath)
2024-08-20 09:08:40 +00:00
default:
return fmt.Errorf("invalid destination: unsupported protocol %s", dest.Type)
}
}
func CheckChunkExistsInTemporary(meta models.Attachment, cid string) bool {
destMap := viper.GetStringMap("destinations.temporary")
var dest models.LocalDestination
rawDest, _ := jsoniter.Marshal(destMap)
_ = jsoniter.Unmarshal(rawDest, &dest)
path := filepath.Join(dest.Path, fmt.Sprintf("%s.part%s", meta.Uuid, cid))
2024-08-20 09:08:40 +00:00
if _, err := os.Stat(path); errors.Is(err, os.ErrNotExist) {
return false
} else {
return true
}
}
func ReUploadFileToPermanent(meta models.Attachment) error {
if meta.Destination != models.AttachmentDstTemporary {
return fmt.Errorf("attachment isn't in temporary storage, unable to process")
}
2024-07-29 05:22:57 +00:00
meta.Destination = models.AttachmentDstPermanent
2024-07-28 13:03:56 +00:00
destMap := viper.GetStringMap("destinations.permanent")
2024-05-17 07:59:51 +00:00
2024-07-28 13:03:56 +00:00
var dest models.BaseDestination
rawDest, _ := jsoniter.Marshal(destMap)
_ = jsoniter.Unmarshal(rawDest, &dest)
2024-05-17 07:59:51 +00:00
prevDestMap := viper.GetStringMap("destinations.temporary")
2024-08-20 09:08:40 +00:00
// Currently, the temporary destination only supports the local.
// So we can do this.
var prevDest models.LocalDestination
prevRawDest, _ := jsoniter.Marshal(prevDestMap)
_ = jsoniter.Unmarshal(prevRawDest, &prevDest)
inDst := filepath.Join(prevDest.Path, meta.Uuid)
2024-07-28 13:03:56 +00:00
switch dest.Type {
2024-05-17 07:59:51 +00:00
case models.DestinationTypeLocal:
var destConfigured models.LocalDestination
_ = jsoniter.Unmarshal(rawDest, &destConfigured)
in, err := os.Open(inDst)
if err != nil {
return fmt.Errorf("unable to open file in temporary storage: %v", err)
}
defer in.Close()
out, err := os.Create(filepath.Join(destConfigured.Path, meta.Uuid))
if err != nil {
return fmt.Errorf("unable to open dest file: %v", err)
}
defer out.Close()
_, err = io.Copy(out, in)
if err != nil {
return fmt.Errorf("unable to copy data to dest file: %v", err)
}
2024-07-29 05:22:57 +00:00
database.C.Save(&meta)
CacheAttachment(meta)
return nil
2024-05-17 07:59:51 +00:00
case models.DestinationTypeS3:
var destConfigured models.S3Destination
_ = jsoniter.Unmarshal(rawDest, &destConfigured)
client, err := minio.New(destConfigured.Endpoint, &minio.Options{
Creds: credentials.NewStaticV4(destConfigured.SecretID, destConfigured.SecretKey, ""),
Secure: destConfigured.EnableSSL,
})
if err != nil {
return fmt.Errorf("unable to configure s3 client: %v", err)
}
_, err = client.FPutObject(context.Background(), destConfigured.Bucket, filepath.Join(destConfigured.Path, meta.Uuid), inDst, minio.PutObjectOptions{
ContentType: meta.MimeType,
SendContentMd5: false,
DisableContentSha256: true,
})
if err != nil {
return fmt.Errorf("unable to upload file to s3: %v", err)
}
2024-07-29 05:22:57 +00:00
database.C.Save(&meta)
CacheAttachment(meta)
return nil
2024-05-17 07:59:51 +00:00
default:
2024-07-28 13:03:56 +00:00
return fmt.Errorf("invalid destination: unsupported protocol %s", dest.Type)
2024-05-17 07:59:51 +00:00
}
}