mirror of
https://github.com/StarFleetCPTN/GoMFT.git
synced 2026-09-08 15:41:20 +02:00
StarFleetCPTN
1d43a7582d
- Remove renaming logic for job_histories and password_histories tables, ensuring consistency with plural naming. - Adjust CREATE TABLE statements to reflect the correct plural forms for job_histories and password_histories. - Update drop table order to match the new naming conventions, maintaining foreign key constraints.
256 lines
7.6 KiB
Go
256 lines
7.6 KiB
Go
package migrations
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/go-gormigrate/gormigrate/v2"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func InitialSchema() *gormigrate.Migration {
|
|
return &gormigrate.Migration{
|
|
ID: "001_initial_schema",
|
|
Migrate: func(tx *gorm.DB) error {
|
|
// Check if any tables exist (indicating an existing database)
|
|
var count int64
|
|
if err := tx.Raw("SELECT count(*) FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%'").Scan(&count).Error; err != nil {
|
|
return fmt.Errorf("failed to check for existing tables: %v", err)
|
|
}
|
|
|
|
// If tables exist, create a backup
|
|
if count > 0 {
|
|
// Get the database path
|
|
sqlDB, err := tx.DB()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get underlying database: %v", err)
|
|
}
|
|
|
|
var seq int
|
|
var name, dbPath string
|
|
if err := sqlDB.QueryRow("PRAGMA database_list").Scan(&seq, &name, &dbPath); err != nil {
|
|
return fmt.Errorf("failed to get database path: %v", err)
|
|
}
|
|
|
|
// Create backup file with timestamp
|
|
backupFile := fmt.Sprintf("%s.backup.%s", dbPath, time.Now().Format("20060102_150405"))
|
|
|
|
// Read original database
|
|
data, err := os.ReadFile(dbPath)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to read database for backup: %v", err)
|
|
}
|
|
|
|
// Write backup
|
|
if err := os.WriteFile(backupFile, data, 0600); err != nil {
|
|
return fmt.Errorf("failed to create database backup: %v", err)
|
|
}
|
|
|
|
fmt.Printf("Created database backup at: %s\n", backupFile)
|
|
}
|
|
|
|
// Disable foreign key constraints while creating tables
|
|
if err := tx.Exec("PRAGMA foreign_keys = OFF").Error; err != nil {
|
|
return fmt.Errorf("failed to disable foreign key constraints: %v", err)
|
|
}
|
|
defer func() {
|
|
if err := tx.Exec("PRAGMA foreign_keys = ON").Error; err != nil {
|
|
fmt.Printf("Warning: failed to re-enable foreign key constraints: %v\n", err)
|
|
}
|
|
}()
|
|
|
|
// Create Users table
|
|
if err := tx.Exec(`CREATE TABLE IF NOT EXISTS users (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
email VARCHAR(255) NOT NULL UNIQUE,
|
|
password_hash VARCHAR(255) NOT NULL,
|
|
is_admin BOOLEAN DEFAULT FALSE,
|
|
last_password_change DATETIME,
|
|
failed_login_attempts INTEGER DEFAULT 0,
|
|
account_locked BOOLEAN DEFAULT FALSE,
|
|
lockout_until DATETIME,
|
|
theme VARCHAR(255) DEFAULT 'light',
|
|
created_at DATETIME,
|
|
updated_at DATETIME
|
|
)`).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
// Create PasswordHistory table
|
|
if err := tx.Exec(`CREATE TABLE IF NOT EXISTS password_histories (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
user_id INTEGER NOT NULL,
|
|
password_hash VARCHAR(255) NOT NULL,
|
|
created_at DATETIME,
|
|
FOREIGN KEY (user_id) REFERENCES users(id)
|
|
)`).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
// Create PasswordResetToken table
|
|
if err := tx.Exec(`CREATE TABLE IF NOT EXISTS password_reset_tokens (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
user_id INTEGER NOT NULL,
|
|
token VARCHAR(255) NOT NULL,
|
|
expires_at DATETIME NOT NULL,
|
|
used BOOLEAN DEFAULT FALSE,
|
|
created_at DATETIME,
|
|
updated_at DATETIME,
|
|
FOREIGN KEY (user_id) REFERENCES users(id)
|
|
)`).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
// Create TransferConfigs table
|
|
if err := tx.Exec(`CREATE TABLE IF NOT EXISTS transfer_configs (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
name VARCHAR(255) NOT NULL,
|
|
source_type VARCHAR(255) NOT NULL,
|
|
source_path TEXT NOT NULL,
|
|
source_host VARCHAR(255),
|
|
source_port INTEGER DEFAULT 22,
|
|
source_user VARCHAR(255),
|
|
source_key_file TEXT,
|
|
source_bucket VARCHAR(255),
|
|
source_region VARCHAR(255),
|
|
source_access_key VARCHAR(255),
|
|
source_endpoint VARCHAR(255),
|
|
source_share VARCHAR(255),
|
|
source_domain VARCHAR(255),
|
|
source_passive_mode BOOLEAN DEFAULT TRUE,
|
|
source_client_id VARCHAR(255),
|
|
source_drive_id VARCHAR(255),
|
|
source_team_drive VARCHAR(255),
|
|
source_read_only BOOLEAN,
|
|
source_start_year INTEGER,
|
|
source_include_archived BOOLEAN,
|
|
file_pattern VARCHAR(255) DEFAULT '*',
|
|
output_pattern TEXT,
|
|
destination_type VARCHAR(255) NOT NULL,
|
|
destination_path TEXT NOT NULL,
|
|
dest_host VARCHAR(255),
|
|
dest_port INTEGER DEFAULT 22,
|
|
dest_user VARCHAR(255),
|
|
dest_key_file TEXT,
|
|
dest_bucket VARCHAR(255),
|
|
dest_region VARCHAR(255),
|
|
dest_access_key VARCHAR(255),
|
|
dest_endpoint VARCHAR(255),
|
|
dest_share VARCHAR(255),
|
|
dest_domain VARCHAR(255),
|
|
dest_passive_mode BOOLEAN DEFAULT TRUE,
|
|
dest_client_id VARCHAR(255),
|
|
dest_drive_id VARCHAR(255),
|
|
dest_team_drive VARCHAR(255),
|
|
dest_read_only BOOLEAN,
|
|
dest_start_year INTEGER,
|
|
dest_include_archived BOOLEAN,
|
|
use_builtin_auth_source BOOLEAN DEFAULT TRUE,
|
|
use_builtin_auth_dest BOOLEAN DEFAULT TRUE,
|
|
google_drive_authenticated BOOLEAN,
|
|
archive_path TEXT,
|
|
archive_enabled BOOLEAN DEFAULT FALSE,
|
|
rclone_flags TEXT,
|
|
delete_after_transfer BOOLEAN DEFAULT FALSE,
|
|
skip_processed_files BOOLEAN DEFAULT TRUE,
|
|
max_concurrent_transfers INTEGER DEFAULT 4,
|
|
created_by INTEGER,
|
|
created_at DATETIME,
|
|
updated_at DATETIME,
|
|
FOREIGN KEY (created_by) REFERENCES users(id)
|
|
)`).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
// Create Jobs table
|
|
if err := tx.Exec(`CREATE TABLE IF NOT EXISTS jobs (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
name VARCHAR(255),
|
|
config_id INTEGER NOT NULL,
|
|
config_ids TEXT,
|
|
schedule VARCHAR(255) NOT NULL,
|
|
enabled BOOLEAN DEFAULT TRUE,
|
|
last_run DATETIME,
|
|
next_run DATETIME,
|
|
webhook_enabled BOOLEAN DEFAULT FALSE,
|
|
webhook_url TEXT,
|
|
webhook_secret TEXT,
|
|
webhook_headers TEXT,
|
|
notify_on_success BOOLEAN DEFAULT TRUE,
|
|
notify_on_failure BOOLEAN DEFAULT TRUE,
|
|
created_by INTEGER,
|
|
created_at DATETIME,
|
|
updated_at DATETIME,
|
|
FOREIGN KEY (config_id) REFERENCES transfer_configs(id),
|
|
FOREIGN KEY (created_by) REFERENCES users(id)
|
|
)`).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
// Create JobHistory table
|
|
if err := tx.Exec(`CREATE TABLE IF NOT EXISTS job_histories (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
job_id INTEGER NOT NULL,
|
|
config_id INTEGER DEFAULT 0,
|
|
start_time DATETIME NOT NULL,
|
|
end_time DATETIME,
|
|
status VARCHAR(255) NOT NULL,
|
|
bytes_transferred INTEGER,
|
|
files_transferred INTEGER,
|
|
error_message TEXT,
|
|
FOREIGN KEY (job_id) REFERENCES jobs(id)
|
|
)`).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
// Create FileMetadata table
|
|
if err := tx.Exec(`CREATE TABLE IF NOT EXISTS file_metadata (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
job_id INTEGER NOT NULL,
|
|
config_id INTEGER DEFAULT 0,
|
|
file_name VARCHAR(255) NOT NULL,
|
|
original_path TEXT NOT NULL,
|
|
file_size INTEGER NOT NULL,
|
|
file_hash VARCHAR(255),
|
|
creation_time DATETIME,
|
|
mod_time DATETIME,
|
|
processed_time DATETIME NOT NULL,
|
|
destination_path TEXT NOT NULL,
|
|
status VARCHAR(255) NOT NULL,
|
|
error_message TEXT,
|
|
created_at DATETIME,
|
|
updated_at DATETIME,
|
|
FOREIGN KEY (job_id) REFERENCES jobs(id)
|
|
)`).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
// Re-enable foreign key constraints and verify integrity
|
|
if err := tx.Exec("PRAGMA foreign_key_check").Error; err != nil {
|
|
return fmt.Errorf("foreign key integrity check failed: %v", err)
|
|
}
|
|
|
|
return nil
|
|
},
|
|
Rollback: func(tx *gorm.DB) error {
|
|
// Drop tables in reverse order to handle foreign key constraints
|
|
tables := []string{
|
|
"file_metadata",
|
|
"job_histories",
|
|
"jobs",
|
|
"transfer_configs",
|
|
"password_reset_tokens",
|
|
"password_histories",
|
|
"users",
|
|
}
|
|
for _, table := range tables {
|
|
if err := tx.Exec(fmt.Sprintf("DROP TABLE IF EXISTS %s", table)).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
}
|