Added DB

local_db.go

@@ -0,0 +1,57 @@
+package main
+
+import (
+	"os"
+	"path/filepath"
+
+	"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)
+	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
+}
+
+// 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 {
+	result := db.Create(&record) // GORM's Create method
+	return result.Error
+}
+
+// GetAllModifications retrieves all modification records from the database
+func GetAllModifications(db *gorm.DB) ([]ModificationHistory, error) {
+	var records []ModificationHistory
+	result := db.Find(&records) // GORM's Find method retrieves all records
+	return records, result.Error
+}