mirror of
https://github.com/icereed/paperless-gpt.git
synced 2025-03-12 12:58:02 -05:00
101 lines
3.5 KiB
Go
101 lines
3.5 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// ModificationHistory represents the schema of the modification_history table
|
|
type ModificationHistory struct {
|
|
ID uint `gorm:"primaryKey"` // Auto-incrementing primary key
|
|
DocumentID uint `gorm:"not null"` // Foreign key to documents table (if applicable)
|
|
DateChanged string `gorm:"not null"` // Date and time of modification
|
|
ModField string `gorm:"size:255;not null"` // Field being modified
|
|
PreviousValue string `gorm:"size:1048576"` // Previous value of the field
|
|
NewValue string `gorm:"size:1048576"` // New value of the field
|
|
Undone bool `gorm:"not null;default:false"` // Whether the modification has been undone
|
|
UndoneDate string `gorm:"default:null"` // Date and time of undoing the modification
|
|
}
|
|
|
|
// InitializeDB initializes the SQLite database and migrates the schema
|
|
func InitializeDB() *gorm.DB {
|
|
// Ensure db directory exists
|
|
dbDir := "db"
|
|
if err := os.MkdirAll(dbDir, os.ModePerm); err != nil {
|
|
log.Fatalf("Failed to create db directory: %v", err)
|
|
}
|
|
|
|
dbPath := filepath.Join(dbDir, "modification_history.db")
|
|
|
|
// Connect to SQLite database
|
|
db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
|
|
if err != nil {
|
|
log.Fatalf("Failed to connect to database: %v", err)
|
|
}
|
|
|
|
// Migrate the schema (create the table if it doesn't exist)
|
|
err = db.AutoMigrate(&ModificationHistory{})
|
|
if err != nil {
|
|
log.Fatalf("Failed to migrate database schema: %v", err)
|
|
}
|
|
|
|
return db
|
|
}
|
|
|
|
// InsertModification inserts a new modification record into the database
|
|
func InsertModification(db *gorm.DB, record *ModificationHistory) error {
|
|
log.Debugf("Passed modification record: %+v", record)
|
|
record.DateChanged = time.Now().Format(time.RFC3339) // Set the DateChanged field to the current time
|
|
log.Debugf("Inserting modification record: %+v", record)
|
|
result := db.Create(&record) // GORM's Create method
|
|
log.Debugf("Insertion result: %+v", result)
|
|
return result.Error
|
|
}
|
|
|
|
// GetModification retrieves a modification record by its ID
|
|
func GetModification(db *gorm.DB, id uint) (*ModificationHistory, error) {
|
|
var record ModificationHistory
|
|
result := db.First(&record, id) // GORM's First method retrieves the first record matching the ID
|
|
return &record, result.Error
|
|
}
|
|
|
|
// GetAllModifications retrieves all modification records from the database (deprecated - use GetPaginatedModifications instead)
|
|
func GetAllModifications(db *gorm.DB) ([]ModificationHistory, error) {
|
|
var records []ModificationHistory
|
|
result := db.Order("date_changed DESC").Find(&records)
|
|
return records, result.Error
|
|
}
|
|
|
|
// GetPaginatedModifications retrieves a page of modification records with total count
|
|
func GetPaginatedModifications(db *gorm.DB, page int, pageSize int) ([]ModificationHistory, int64, error) {
|
|
var records []ModificationHistory
|
|
var total int64
|
|
|
|
// Get total count
|
|
if err := db.Model(&ModificationHistory{}).Count(&total).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
// Calculate offset
|
|
offset := (page - 1) * pageSize
|
|
|
|
// Get paginated records
|
|
result := db.Order("date_changed DESC").
|
|
Offset(offset).
|
|
Limit(pageSize).
|
|
Find(&records)
|
|
|
|
return records, total, result.Error
|
|
}
|
|
|
|
// UndoModification marks a modification record as undone and sets the undo date
|
|
func SetModificationUndone(db *gorm.DB, record *ModificationHistory) error {
|
|
record.Undone = true
|
|
record.UndoneDate = time.Now().Format(time.RFC3339)
|
|
result := db.Save(&record) // GORM's Save method
|
|
return result.Error
|
|
}
|