feat: add basic deletion finding

This commit is contained in:
2026-04-26 18:43:23 +02:00
parent 7c069c0437
commit f589ae4faf
6 changed files with 79 additions and 60 deletions
+2 -2
View File
@@ -27,7 +27,7 @@ func RegisterApiRoutes(api *gin.RouterGroup /* env runtime.Environment,*/, db *g
// Display the information on what is going on at the moment // Display the information on what is going on at the moment
api.GET("/command", func(c *gin.Context) { api.GET("/command", func(c *gin.Context) {
state, err := database.GetState(db) state, err := database.LatestState(db)
if err != nil { if err != nil {
slog.Error("unable to determine state", "error", err) slog.Error("unable to determine state", "error", err)
c.JSON(http.StatusInternalServerError, response.BasicResponse{ c.JSON(http.StatusInternalServerError, response.BasicResponse{
@@ -48,7 +48,7 @@ func RegisterApiRoutes(api *gin.RouterGroup /* env runtime.Environment,*/, db *g
// define a route to check what is registered // define a route to check what is registered
api.GET("/available", func(c *gin.Context) { api.GET("/available", func(c *gin.Context) {
files, err := database.GetFiles(db) files, err := database.ListFiles(db)
if err != nil { if err != nil {
slog.Error("failed to retrieve available files", "error", err) slog.Error("failed to retrieve available files", "error", err)
c.JSON(http.StatusInternalServerError, response.BasicResponse{ c.JSON(http.StatusInternalServerError, response.BasicResponse{
+17 -3
View File
@@ -17,8 +17,8 @@ func RegisterFileRoutes(file *gin.RouterGroup, env bootstrap.Environment, db *go
// prefix: file // prefix: file
// for example: /file/<file-name> // for example: /file/<file-name>
file.GET("/:filename", func(c *gin.Context) { file.GET("/:filename", func(c *gin.Context) {
f := c.Param("filename") fileParam := c.Param("filename")
p := filepath.Join(env.ContentDirectory, f) p := filepath.Join(env.ContentDirectory, fileParam)
c.File(p) c.File(p)
}) })
@@ -50,7 +50,7 @@ func RegisterFileRoutes(file *gin.RouterGroup, env bootstrap.Environment, db *go
return return
} }
if err := database.RegisterFile(db, fileData); err != nil { if err := database.CreateFile(db, fileData); err != nil {
if errors.Is(err, gorm.ErrDuplicatedKey) { if errors.Is(err, gorm.ErrDuplicatedKey) {
slog.Debug("discarding file since its checksum is a duplicate", "error", err) slog.Debug("discarding file since its checksum is a duplicate", "error", err)
c.JSON(http.StatusConflict, response.BasicResponse{ c.JSON(http.StatusConflict, response.BasicResponse{
@@ -80,4 +80,18 @@ func RegisterFileRoutes(file *gin.RouterGroup, env bootstrap.Environment, db *go
Data: fileData, Data: fileData,
}) })
}) })
file.DELETE("/:filename", func(c *gin.Context) {
fileParam := c.Param("filename")
f, err := database.FindFileByName(db, fileParam)
slog.Info("received a detelte request for a file", "file", f, "filename", fileParam)
if err != nil {
slog.Error("failed to filter the file database for the name")
c.JSON(http.StatusNotFound, response.BasicResponse{
Msg: response.IntErrMes,
})
}
})
} }
+5 -5
View File
@@ -25,11 +25,11 @@ func KickoffDatabase(workDir string) (*gorm.DB, error) {
// try to use GORM automigrate if the schema changes // try to use GORM automigrate if the schema changes
if err := db.AutoMigrate( if err := db.AutoMigrate(
&Command{}, &Command{}, // app state and command status
&File{}, &File{}, // files database for keeping track
&Tenant{}, &Tenant{}, // table for tenants and its data
&Group{}, &Group{}, // group table for privileges
&Device{}, &Device{}, // devices table
); err != nil { ); err != nil {
return nil, err return nil, err
} }
+14 -47
View File
@@ -1,64 +1,31 @@
package database package database
import ( import (
"fmt"
"io"
"log/slog"
"orbits-server/internal/shared/utility"
"path/filepath"
"gorm.io/gorm" "gorm.io/gorm"
) )
// it has been made more general for DRY purposes func LatestState(db *gorm.DB) (Command, error) {
// 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 := utility.CategorizeMediaType(ext)
if category == utility.Unspecified {
return File{}, fmt.Errorf("unsupported filetype")
}
checksum, err := utility.GenerateHashFromReader(r)
if err != nil {
slog.Error("failed to calculate hash of file at given path", "error", err)
return File{}, err
}
safeName := utility.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 var state Command
err := db.Last(&state).Error
if err := db.First(&state).Error; err != nil { return state, err
return Command{}, err
} }
return state, nil func ListFiles(db *gorm.DB) ([]File, error) {
}
func GetFiles(db *gorm.DB) ([]File, error) {
var files []File var files []File
err := db.Find(&files).Error
return files, db.Find(&files).Error return files, err
} }
func RegisterFile(db *gorm.DB, f File) error { func FindFileByName(db *gorm.DB, name string) (File, error) {
var file File
err := db.Where("file_name = ?", name).First(&file).Error
return file, err
}
func CreateFile(db *gorm.DB, f File) error {
return db.Create(&f).Error return db.Create(&f).Error
} }
func DeregisterFile(db *gorm.DB, f File) error { func DeleteFile(db *gorm.DB, f File) error {
return db.Delete(&f).Error return db.Delete(&f).Error
} }
+38
View File
@@ -0,0 +1,38 @@
package database
import (
"fmt"
"io"
"log/slog"
"orbits-server/internal/shared/utility"
"path/filepath"
)
// 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 := utility.CategorizeMediaType(ext)
if category == utility.Unspecified {
return File{}, fmt.Errorf("unsupported filetype")
}
checksum, err := utility.GenerateHashFromReader(r)
if err != nil {
slog.Error("failed to calculate hash of file at given path", "error", err)
return File{}, err
}
safeName := utility.GenerateSafeName(category, ext)
destPath := filepath.Join(contentDirectory, safeName)
f := File{
MediaType: category,
MetaName: origName,
FileName: safeName,
FilePath: destPath,
Checksum: checksum,
}
return f, nil
}
+3 -3
View File
@@ -30,7 +30,7 @@ func filesystemGather(env bootstrap.Environment) (map[string]struct{}, error) {
} }
func databaseGather(db *gorm.DB) (map[string]File, error) { func databaseGather(db *gorm.DB) (map[string]File, error) {
dbFiles, err := GetFiles(db) dbFiles, err := ListFiles(db)
if err != nil { if err != nil {
slog.Error("failed to retrieve the files indexed from the database", "error", err) slog.Error("failed to retrieve the files indexed from the database", "error", err)
return nil, err return nil, err
@@ -88,7 +88,7 @@ func watchdog(env bootstrap.Environment, db *gorm.DB) {
continue continue
} }
if err := RegisterFile(db, fileData); err != nil { if err := CreateFile(db, fileData); err != nil {
if errors.Is(err, gorm.ErrDuplicatedKey) { if errors.Is(err, gorm.ErrDuplicatedKey) {
slog.Debug("discarding file since its a duplicate", "error", err) slog.Debug("discarding file since its a duplicate", "error", err)
} else { } else {
@@ -128,7 +128,7 @@ func watchdog(env bootstrap.Environment, db *gorm.DB) {
if len(dbdbOrphans) > 0 { if len(dbdbOrphans) > 0 {
slog.Info("database orphans detected, engaging flow") slog.Info("database orphans detected, engaging flow")
for _, f := range dbdbOrphans { for _, f := range dbdbOrphans {
DeregisterFile(db, f) DeleteFile(db, f)
} }
} }
} }