Document

Author: sungwoncho

README.md

@@ -2,6 +2,12 @@
 
 Migration tool from PostgreSQL to SQLite for Dnote.
 
+This tool migrates data from **Dnote server v2.x** (PostgreSQL) to **Dnote server v3** (SQLite).
+
+## Prerequisites
+
+- You must be running **Dnote server v2.x**
+
 ## Installation
 
 ```bash
@@ -17,9 +23,13 @@ dnote-pg2sqlite \
   --pg-database dnote \
   --pg-user dnote \
   --pg-password yourpassword \
-  --sqlite-path /path/to/server.db
+  --sqlite-path ~/.local/share/dnote/server.db
 ```
 
+**Note**: Dnote v3 uses XDG directories. The default SQLite database path is `~/.local/share/dnote/server.db` (or `$XDG_DATA_HOME/dnote/server.db`).
+
+**Safety**: The migration tool will refuse to run if the SQLite file already exists, preventing accidental overwrites. Remove the existing file if you need to re-run the migration.
+
 ## Backup First
 
 **Always backup PostgreSQL before migrating:**
@@ -36,10 +46,19 @@ pg_dump -h localhost -U dnote -d dnote > dnote_backup.sql
 
 Full-text search index rebuilds automatically.
 
+## Migration Workflow
+
+1. **Ensure you're on v2.x**: Upgrade to Dnote server v2.x if needed
+2. **Stop your Dnote server** to ensure data consistency during migration
+3. **Backup your PostgreSQL database** (see above)
+4. **Run the migration tool** with your PostgreSQL credentials
+5. **Verify the migration** succeeded (check record counts)
+6. **Upgrade to Dnote v3** and configure it to use the new SQLite database
+
 ## Verification
 
 After migration:
 
 1. Check the tool's output for record counts
-2. Start Dnote and verify login works
+2. Start Dnote v3 and verify login works
 3. Verify notes and search work correctly

main.go

@@ -6,9 +6,12 @@ import (
 	"fmt"
 	"log"
 	"os"
+	"path/filepath"
 
 	_ "github.com/lib/pq"
 	_ "github.com/mattn/go-sqlite3"
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
 )
 
 type Config struct {
@@ -61,6 +64,19 @@ func validate(c Config) error {
 }
 
 func run(config Config) error {
+	// Create directory if it doesn't exist
+	dir := filepath.Dir(config.SqlitePath)
+	if err := os.MkdirAll(dir, 0755); err != nil {
+		return fmt.Errorf("creating database directory at %s: %w", dir, err)
+	}
+
+	// Check if SQLite file already exists
+	if _, err := os.Stat(config.SqlitePath); err == nil {
+		return fmt.Errorf("SQLite database already exists at %s - refusing to overwrite. Please remove the file or choose a different path", config.SqlitePath)
+	} else if !os.IsNotExist(err) {
+		return fmt.Errorf("checking if SQLite file exists: %w", err)
+	}
+
 	// Connect to PostgreSQL
 	pgDSN := fmt.Sprintf("host=%s port=%s dbname=%s user=%s password=%s sslmode=disable",
 		config.PgHost, config.PgPort, config.PgDatabase, config.PgUser, config.PgPassword)
@@ -77,7 +93,7 @@ func run(config Config) error {
 
 	fmt.Println("Connected to PostgreSQL")
 
-	// Connect to SQLite
+	// Connect to SQLite with GORM
 	sqliteDB, err := sql.Open("sqlite3", config.SqlitePath)
 	if err != nil {
 		return fmt.Errorf("opening SQLite: %w", err)
@@ -90,6 +106,33 @@ func run(config Config) error {
 
 	fmt.Println("Connected to SQLite")
 
+	// Initialize SQLite schema using GORM
+	fmt.Println("Creating SQLite schema...")
+	if err := initSQLiteSchema(config.SqlitePath); err != nil {
+		return fmt.Errorf("initializing SQLite schema: %w", err)
+	}
+
 	// Run migration
 	return migrate(pgDB, sqliteDB)
 }
+
+func initSQLiteSchema(sqlitePath string) error {
+	db, err := gorm.Open(sqlite.Open(sqlitePath), &gorm.Config{})
+	if err != nil {
+		return fmt.Errorf("opening SQLite with GORM: %w", err)
+	}
+
+	// AutoMigrate SQLite models
+	if err := db.AutoMigrate(
+		&SqliteUser{},
+		&SqliteAccount{},
+		&SqliteBook{},
+		&SqliteNote{},
+		&SqliteToken{},
+		&SqliteSession{},
+	); err != nil {
+		return fmt.Errorf("running AutoMigrate: %w", err)
+	}
+
+	return nil
+}

migrate_test.go

@@ -185,25 +185,7 @@ func TestMigration(t *testing.T) {
 	sqlitePath := "/tmp/dnote_test.db"
 	os.Remove(sqlitePath) // Clean up any existing test db
 
-	// Create SQLite schema using GORM AutoMigrate
-	sqliteDB, err := gorm.Open(sqlite.Open(sqlitePath), &gorm.Config{})
-	if err != nil {
-		t.Fatalf("Failed to open SQLite: %v", err)
-	}
-
-	// AutoMigrate SQLite models
-	if err := sqliteDB.AutoMigrate(
-		&SqliteUser{},
-		&SqliteAccount{},
-		&SqliteBook{},
-		&SqliteNote{},
-		&SqliteToken{},
-		&SqliteSession{},
-	); err != nil {
-		t.Fatalf("Failed to migrate SQLite schema: %v", err)
-	}
-
-	// Run migration
+	// Run migration (will create schema automatically)
 	config := Config{
 		PgHost:     pgHost,
 		PgPort:     pgPort,
@@ -217,6 +199,12 @@ func TestMigration(t *testing.T) {
 		t.Fatalf("Migration failed: %v", err)
 	}
 
+	// Open SQLite with GORM for verification
+	sqliteDB, err := gorm.Open(sqlite.Open(sqlitePath), &gorm.Config{})
+	if err != nil {
+		t.Fatalf("Failed to open SQLite for verification: %v", err)
+	}
+
 	// Verify SQLite data using GORM
 	// Verify user1
 	var sqliteUser1 SqliteUser