Version 1

2026-06-23 18:50:37 +01:00
commit 55d9772272
133 changed files with 21515 additions and 0 deletions
@@ -0,0 +1,164 @@
+package database
+
+import (
+	"bytes"
+	"context"
+	"fmt"
+	"os"
+	"path/filepath"
+	"time"
+
+	"github.com/jackc/pgx/v5"
+)
+
+type ReseedSummary struct {
+	BackupPath      string
+	OrdersCount     int
+	OrderItemsCount int
+	InventoryCount  int
+}
+
+func (d *DB) BackupAndResetAppData(ctx context.Context) (*ReseedSummary, error) {
+	if d == nil || d.Pool == nil {
+		return nil, fmt.Errorf("database is not configured")
+	}
+
+	tx, err := d.Pool.BeginTx(ctx, pgx.TxOptions{IsoLevel: pgx.RepeatableRead})
+	if err != nil {
+		return nil, err
+	}
+
+	committed := false
+	defer func() {
+		if !committed {
+			_ = tx.Rollback(ctx)
+		}
+	}()
+
+	ordersStatements, err := collectStatements(ctx, tx, `
+		SELECT format(
+			'INSERT INTO orders (id, status, fulfillment_status, email, phone, first_name, last_name, address_line_1, address_line_2, city, region, postal_code, country, notes, currency, amount, stripe_price_id, stripe_payment_intent_id, shipping_carrier, tracking_number, fulfillment_notes, shipped_at, webhook_status, webhook_event_id, webhook_event_type, webhook_message, created_at, updated_at) VALUES (%L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L, %L);',
+			id, status, fulfillment_status, email, phone, first_name, last_name, address_line_1, address_line_2, city, region, postal_code, country, notes, currency, amount, stripe_price_id, stripe_payment_intent_id, shipping_carrier, tracking_number, fulfillment_notes, shipped_at, webhook_status, webhook_event_id, webhook_event_type, webhook_message, created_at, updated_at
+		)
+		FROM orders
+		ORDER BY created_at ASC, id ASC
+	`)
+	if err != nil {
+		return nil, err
+	}
+
+	orderItemsStatements, err := collectStatements(ctx, tx, `
+		SELECT format(
+			'INSERT INTO order_items (order_id, item_id, style, colorway_id, finish_id, quantity, unit_amount, created_at) VALUES (%L, %L, %L, %L, %L, %L, %L, %L);',
+			order_id, item_id, style, colorway_id, finish_id, quantity, unit_amount, created_at
+		)
+		FROM order_items
+		ORDER BY created_at ASC, order_id ASC
+	`)
+	if err != nil {
+		return nil, err
+	}
+
+	inventoryStatements, err := collectStatements(ctx, tx, `
+		SELECT format(
+			'INSERT INTO inventory_levels (style, colorway_id, finish_id, quantity_on_hand, notes, created_at, updated_at) VALUES (%L, %L, %L, %L, %L, %L, %L);',
+			style, colorway_id, finish_id, quantity_on_hand, notes, created_at, updated_at
+		)
+		FROM inventory_levels
+		ORDER BY style ASC, colorway_id ASC, finish_id ASC
+	`)
+	if err != nil {
+		return nil, err
+	}
+
+	backupContent := buildReseedBackupSQL(ordersStatements, orderItemsStatements, inventoryStatements)
+	backupPath, err := writeReseedBackupFile(backupContent)
+	if err != nil {
+		return nil, err
+	}
+
+	if _, err := tx.Exec(ctx, `TRUNCATE TABLE order_items, orders, inventory_levels RESTART IDENTITY CASCADE`); err != nil {
+		return nil, err
+	}
+
+	if err := tx.Commit(ctx); err != nil {
+		return nil, err
+	}
+	committed = true
+
+	return &ReseedSummary{
+		BackupPath:      backupPath,
+		OrdersCount:     len(ordersStatements),
+		OrderItemsCount: len(orderItemsStatements),
+		InventoryCount:  len(inventoryStatements),
+	}, nil
+}
+
+func collectStatements(ctx context.Context, tx pgx.Tx, query string) ([]string, error) {
+	rows, err := tx.Query(ctx, query)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	statements := make([]string, 0)
+	for rows.Next() {
+		var statement string
+		if err := rows.Scan(&statement); err != nil {
+			return nil, err
+		}
+		statements = append(statements, statement)
+	}
+
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+
+	return statements, nil
+}
+
+func buildReseedBackupSQL(ordersStatements, orderItemsStatements, inventoryStatements []string) []byte {
+	var buffer bytes.Buffer
+	timestamp := time.Now().UTC().Format(time.RFC3339)
+
+	buffer.WriteString("-- Royal Pop local reseed backup\n")
+	buffer.WriteString("-- Created at: " + timestamp + "\n")
+	buffer.WriteString(fmt.Sprintf("-- Orders: %d | Order items: %d | Inventory rows: %d\n\n", len(ordersStatements), len(orderItemsStatements), len(inventoryStatements)))
+	buffer.WriteString("BEGIN;\n")
+	buffer.WriteString("TRUNCATE TABLE order_items, orders, inventory_levels RESTART IDENTITY CASCADE;\n\n")
+
+	appendSection := func(title string, statements []string) {
+		buffer.WriteString("-- " + title + "\n")
+		if len(statements) == 0 {
+			buffer.WriteString("-- (no rows)\n\n")
+			return
+		}
+		for _, statement := range statements {
+			buffer.WriteString(statement)
+			buffer.WriteString("\n")
+		}
+		buffer.WriteString("\n")
+	}
+
+	appendSection("orders", ordersStatements)
+	appendSection("order_items", orderItemsStatements)
+	appendSection("inventory_levels", inventoryStatements)
+	buffer.WriteString("COMMIT;\n")
+
+	return buffer.Bytes()
+}
+
+func writeReseedBackupFile(content []byte) (string, error) {
+	backupDir := filepath.Join("Backups", "reseed")
+	if err := os.MkdirAll(backupDir, 0o755); err != nil {
+		return "", err
+	}
+
+	filename := fmt.Sprintf("royal-pop-reseed-%s.sql", time.Now().UTC().Format("20060102-150405"))
+	path := filepath.Join(backupDir, filename)
+	if err := os.WriteFile(path, content, 0o600); err != nil {
+		return "", err
+	}
+
+	return path, nil
+}