2021-09-04 21:53:48 +03:00
|
|
|
package utils
|
|
|
|
|
|
|
|
import (
|
|
|
|
"errors"
|
2021-09-12 14:06:52 +03:00
|
|
|
"image"
|
|
|
|
"image/jpeg"
|
|
|
|
"image/png"
|
2021-10-10 10:41:36 +03:00
|
|
|
"mime/multipart"
|
2021-09-04 21:53:48 +03:00
|
|
|
"os"
|
|
|
|
"path/filepath"
|
|
|
|
"strconv"
|
2021-09-12 14:06:52 +03:00
|
|
|
|
|
|
|
"github.com/disintegration/imaging"
|
2021-09-04 21:53:48 +03:00
|
|
|
)
|
|
|
|
|
2021-10-10 10:42:28 +03:00
|
|
|
const UPLOADS_DIR_PATH = "uploads"
|
|
|
|
const FILE_SIZE_IN_BYTES = 3145728 // 3MB
|
2021-09-04 21:53:48 +03:00
|
|
|
|
|
|
|
// Check dir existence.
|
|
|
|
func CheckIfFolderExists(path string) bool {
|
|
|
|
_, err := os.Stat(path)
|
|
|
|
return !os.IsNotExist(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Creates folder for uploads.
|
|
|
|
func CreateUploadsFolder() error {
|
|
|
|
isExists := CheckIfFolderExists(UPLOADS_DIR_PATH)
|
|
|
|
if isExists {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
err := os.Mkdir(UPLOADS_DIR_PATH, 0755)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Creates folder for thread.
|
|
|
|
func CreateThreadFolder(postID int) error {
|
|
|
|
|
|
|
|
threadDirPath := filepath.Join(UPLOADS_DIR_PATH, strconv.Itoa(postID))
|
|
|
|
isExists := CheckIfFolderExists(threadDirPath)
|
|
|
|
if isExists {
|
|
|
|
return errors.New("folder already exists")
|
|
|
|
}
|
|
|
|
|
|
|
|
err := os.Mkdir(threadDirPath, 0755)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2021-09-12 11:31:43 +03:00
|
|
|
originalsFolder := filepath.Join(threadDirPath, "o")
|
|
|
|
err = os.Mkdir(originalsFolder, 0755)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
thumbnailsFolder := filepath.Join(threadDirPath, "t")
|
|
|
|
err = os.Mkdir(thumbnailsFolder, 0755)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2021-09-04 21:53:48 +03:00
|
|
|
return nil
|
|
|
|
}
|
2021-09-11 16:56:22 +03:00
|
|
|
|
2021-10-10 10:41:36 +03:00
|
|
|
// TODO: add files length check
|
2021-09-11 16:56:22 +03:00
|
|
|
func ValidatePost(title, text string) bool {
|
2021-09-12 00:30:48 +03:00
|
|
|
return (title == "" && text != "") ||
|
|
|
|
(title != "" && text == "") ||
|
|
|
|
(title != "" && text != "")
|
2021-09-11 16:56:22 +03:00
|
|
|
}
|
2021-09-12 14:06:52 +03:00
|
|
|
|
2021-10-10 10:41:36 +03:00
|
|
|
func CheckFilesSize(files []*multipart.FileHeader) bool {
|
|
|
|
for _, file := range files {
|
|
|
|
if file.Size > int64(FILE_SIZE_IN_BYTES) {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
// func CheckFilesExt(){
|
|
|
|
|
|
|
|
// }
|
|
|
|
|
2021-09-12 14:06:52 +03:00
|
|
|
func MakeImageThumbnail(originalPath, ext string, threadID, fileID int) (*image.NRGBA, error) {
|
|
|
|
img, err := imaging.Open(originalPath, imaging.AutoOrientation(true))
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
dstImage := imaging.Resize(img, 0, 150, imaging.NearestNeighbor)
|
|
|
|
|
|
|
|
return dstImage, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func SaveImageThumbnail(img *image.NRGBA, threadID, fileID int, ext string) error {
|
|
|
|
thumbnailPath := filepath.Join(
|
|
|
|
UPLOADS_DIR_PATH,
|
|
|
|
strconv.Itoa(threadID),
|
|
|
|
"t",
|
|
|
|
strconv.Itoa(fileID)+"."+ext,
|
|
|
|
)
|
|
|
|
|
|
|
|
f, err := os.Create(thumbnailPath)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
switch ext {
|
|
|
|
case "png":
|
|
|
|
err = png.Encode(f, img)
|
|
|
|
case "jpeg":
|
|
|
|
err = jpeg.Encode(f, img, nil)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|