From dcde30ba867388373e19782cf3401fb2dc1bac86 Mon Sep 17 00:00:00 2001
From: Claudio Ortolina <cloud8421@gmail.com>
Date: Sun, 6 Oct 2024 16:16:23 +0100
Subject: [PATCH] Extend script to restore remote data to local

---
 lib/mix/tasks/music_library/db_pull.ex | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/lib/mix/tasks/music_library/db_pull.ex b/lib/mix/tasks/music_library/db_pull.ex
index 9090a28b..c90269b1 100644
--- a/lib/mix/tasks/music_library/db_pull.ex
+++ b/lib/mix/tasks/music_library/db_pull.ex
@@ -16,5 +16,14 @@ defmodule Mix.Tasks.MusicLibrary.DbPull do
     local_db = "music_library_prod_#{DateTime.to_unix(current_time)}.db"
 
     System.cmd("flyctl", ["ssh", "sftp", "get", remote_db, local_db], into: IO.stream())
+
+    IO.puts("Database pulled successfully")
+
+    IO.puts("Restoring as local dev database")
+
+    Path.wildcard("music_library_dev.db*")
+    |> Enum.each(&File.rm!/1)
+
+    File.cp!(local_db, "music_library_dev.db")
   end
 end