feat: add basic deletion finding

internal/server/database/database.go

@@ -25,11 +25,11 @@ func KickoffDatabase(workDir string) (*gorm.DB, error) {
 
 	// try to use GORM automigrate if the schema changes
 	if err := db.AutoMigrate(
-		&Command{},
-		&File{},
-		&Tenant{},
-		&Group{},
-		&Device{},
+		&Command{}, // app state and command status
+		&File{},    // files database for keeping track
+		&Tenant{},  // table for tenants and its data
+		&Group{},   // group table for privileges
+		&Device{},  // devices table
 	); err != nil {
 		return nil, err
 	}

internal/server/database/functions.go

@@ -1,64 +1,31 @@
 package database
 
 import (
-	"fmt"
-	"io"
-	"log/slog"
-	"orbits-server/internal/shared/utility"
-	"path/filepath"
-
 	"gorm.io/gorm"
 )
 
-// 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)
-
-	fData := File{
-		MediaType: category,
-		MetaName:  origName,
-		FileName:  safeName,
-		FilePath:  destPath,
-		Checksum:  checksum,
-	}
-
-	return fData, nil
-}
-
-func GetState(db *gorm.DB) (Command, error) {
+func LatestState(db *gorm.DB) (Command, error) {
 	var state Command
-
-	if err := db.First(&state).Error; err != nil {
-		return Command{}, err
-	}
-
-	return state, nil
+	err := db.Last(&state).Error
+	return state, err
 }
 
-func GetFiles(db *gorm.DB) ([]File, error) {
+func ListFiles(db *gorm.DB) ([]File, error) {
 	var files []File
-
-	return files, db.Find(&files).Error
+	err := 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
 }
 
-func DeregisterFile(db *gorm.DB, f File) error {
+func DeleteFile(db *gorm.DB, f File) error {
 	return db.Delete(&f).Error
 }

internal/server/database/utility.go

@@ -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
+}

internal/server/database/watchdog.go

@@ -30,7 +30,7 @@ func filesystemGather(env bootstrap.Environment) (map[string]struct{}, error) {
 }
 
 func databaseGather(db *gorm.DB) (map[string]File, error) {
-	dbFiles, err := GetFiles(db)
+	dbFiles, err := ListFiles(db)
 	if err != nil {
 		slog.Error("failed to retrieve the files indexed from the database", "error", err)
 		return nil, err
@@ -88,7 +88,7 @@ func watchdog(env bootstrap.Environment, db *gorm.DB) {
 					continue
 				}
 
-				if err := RegisterFile(db, fileData); err != nil {
+				if err := CreateFile(db, fileData); err != nil {
 					if errors.Is(err, gorm.ErrDuplicatedKey) {
 						slog.Debug("discarding file since its a duplicate", "error", err)
 					} else {
@@ -128,7 +128,7 @@ func watchdog(env bootstrap.Environment, db *gorm.DB) {
 	if len(dbdbOrphans) > 0 {
 		slog.Info("database orphans detected, engaging flow")
 		for _, f := range dbdbOrphans {
-			DeregisterFile(db, f)
+			DeleteFile(db, f)
 		}
 	}
 }