From da577bc6e6400e2ebc29629ae122d947497540ff Mon Sep 17 00:00:00 2001
From: Justin Edmund <justin@jedmund.com>
Date: Thu, 27 Feb 2025 21:48:54 -0800
Subject: [PATCH] Add rake task for backing up/restoring prod db

---
 lib/tasks/database.rake | 42 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 lib/tasks/database.rake

diff --git a/lib/tasks/database.rake b/lib/tasks/database.rake
new file mode 100644
index 0000000..227fb16
--- /dev/null
+++ b/lib/tasks/database.rake
@@ -0,0 +1,42 @@
+namespace :db do
+  desc 'Backup remote PostgreSQL database'
+  task :backup do
+    remote_host = ENV.fetch('REMOTE_DB_HOST', 'roundhouse.proxy.rlwy.net')
+    remote_port = ENV.fetch('REMOTE_DB_PORT', '54629')
+    remote_user = ENV.fetch('REMOTE_DB_USER', 'postgres')
+    remote_db = ENV.fetch('REMOTE_DB_NAME', 'railway')
+    password = ENV.fetch('REMOTE_DB_PASSWORD') { raise 'Please set REMOTE_DB_PASSWORD' }
+
+    backup_dir = File.expand_path('backups')
+    FileUtils.mkdir_p(backup_dir)
+    backup_file = File.join(backup_dir, "#{Time.now.strftime('%Y%m%d_%H%M%S')}-prod-backup.tar")
+
+    cmd = %W[
+      pg_dump -h #{remote_host} -p #{remote_port} -U #{remote_user} -d #{remote_db} -F t
+      --no-owner --exclude-extension=timescaledb --exclude-extension=timescaledb_toolkit
+    ].join(' ')
+
+    puts "Backing up remote database to #{backup_file}..."
+    system({ 'PGPASSWORD' => password }, "#{cmd} > #{backup_file}")
+    puts 'Backup completed!'
+  end
+
+  desc 'Restore PostgreSQL database from backup'
+  task :restore, [:backup_file] => [:environment] do |_, args|
+    local_user = ENV.fetch('LOCAL_DB_USER', 'justin')
+    local_db = ENV.fetch('LOCAL_DB_NAME', 'hensei_dev')
+
+    # Use the specified backup file or find the most recent one
+    backup_dir = File.expand_path('backups')
+    backup_file = args[:backup_file] || Dir.glob("#{backup_dir}/*-prod-backup.tar").max
+
+    raise 'Backup file not found. Please specify a valid backup file.' unless backup_file && File.exist?(backup_file)
+
+    puts "Restoring database from #{backup_file}..."
+    system("pg_restore --no-owner --role=#{local_user} --disable-triggers -U #{local_user} -d #{local_db} #{backup_file}")
+    puts 'Restore completed!'
+  end
+
+  desc 'Backup remote database and restore locally'
+  task backup_and_restore: %i[backup restore]
+end