DaanSelen
87 lines
1.8 KiB
Go
87 lines
1.8 KiB
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"log/slog"
|
|
"orbits-server/internal/shared/utility"
|
|
"path/filepath"
|
|
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// 0: unspecified
|
|
// 1: video
|
|
// 2: presentation
|
|
// 3: internet URL
|
|
func CategorizeMediaType(ext string) (MediaType, bool) {
|
|
|
|
switch ext {
|
|
case ".mp4", ".mov", ".avi", ".mkv", ".webm", ".m4a":
|
|
return Video, true
|
|
case ".pptx", ".ppt", ".key", ".odp":
|
|
return Presentation, true
|
|
default:
|
|
slog.Debug("marking file as invalid due to its undefined extension")
|
|
return "", false
|
|
}
|
|
}
|
|
|
|
func GenerateSafeName(category MediaType, ext string) string {
|
|
return uuid.New().String() + "_" + string(category) + ext
|
|
}
|
|
|
|
// it has been made more general for DRY purposes
|
|
// this function should only be called after manually checking the filetype
|
|
func BuildFileRecord(r io.Reader, origName string, contentDirectory string) (File, error) {
|
|
ext := filepath.Ext(origName)
|
|
category, ok := CategorizeMediaType(ext)
|
|
if !ok {
|
|
return File{}, fmt.Errorf("unsupported filetype")
|
|
}
|
|
|
|
checksum, err := utility.HashReader(r)
|
|
if err != nil {
|
|
slog.Error("failed to calculate hash of file at given path", "error", err)
|
|
return File{}, err
|
|
}
|
|
|
|
safeName := GenerateSafeName(category, ext)
|
|
destPath := filepath.Join(contentDirectory, safeName)
|
|
|
|
fData := File{
|
|
MediaType: category,
|
|
MetaName: origName,
|
|
FileName: safeName,
|
|
FilePath: destPath,
|
|
Checksum: checksum,
|
|
}
|
|
|
|
return fData, nil
|
|
}
|
|
|
|
func GetState(db *gorm.DB) (Command, error) {
|
|
var state Command
|
|
|
|
if err := db.First(&state).Error; err != nil {
|
|
return Command{}, err
|
|
}
|
|
|
|
return state, nil
|
|
}
|
|
|
|
func GetFiles(db *gorm.DB) ([]File, error) {
|
|
var files []File
|
|
|
|
return files, db.Find(&files).Error
|
|
}
|
|
|
|
func RegisterFile(db *gorm.DB, f File) error {
|
|
return db.Create(&f).Error
|
|
}
|
|
|
|
func DeregisterFile(db *gorm.DB, f File) error {
|
|
return db.Delete(&f).Error
|
|
}
|