41 lines
883 B
Go
41 lines
883 B
Go
|
package services
|
||
|
|
|
||
|
|
import (
|
||
|
|
"code.smartsheep.studio/hydrogen/interactive/pkg/database"
|
||
|
|
"code.smartsheep.studio/hydrogen/interactive/pkg/models"
|
||
|
|
"github.com/google/uuid"
|
||
|
|
"mime/multipart"
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewAttachment(user models.Account, header *multipart.FileHeader) (models.Attachment, error) {
|
||
|
|
attachment := models.Attachment{
|
||
|
|
FileID: uuid.NewString(),
|
||
|
|
Filesize: header.Size,
|
||
|
|
Filename: header.Filename,
|
||
|
|
Mimetype: "unknown/unknown",
|
||
|
|
PostID: nil,
|
||
|
|
AuthorID: user.ID,
|
||
|
|
}
|
||
|
|
|
||
|
|
// Open file
|
||
|
|
file, err := header.Open()
|
||
|
|
if err != nil {
|
||
|
|
return attachment, err
|
||
|
|
}
|
||
|
|
defer file.Close()
|
||
|
|
|
||
|
|
// Detect mimetype
|
||
|
|
fileHeader := make([]byte, 512)
|
||
|
|
_, err = file.Read(fileHeader)
|
||
|
|
if err != nil {
|
||
|
|
return attachment, err
|
||
|
|
}
|
||
|
|
attachment.Mimetype = http.DetectContentType(fileHeader)
|
||
|
|
|
||
|
|
// Save into database
|
||
|
|
err = database.C.Save(&attachment).Error
|
||
|
|
|
||
|
|
return attachment, err
|
||
|
|
}
|