X-Git-Url: http://git.squeep.com/?a=blobdiff_plain;f=restarter%2Flib%2Fpleroma.ex;h=a7186cec48b87b743ed6281ac3dc0a54651464b7;hb=08d49fba7d52e3eda5292913f1a466326f6b0cdf;hp=da714654c20864c69581f296fc5dc9d2452ae0bf;hpb=7de32e2c8b77eba414d376dedfdb8f782847ce46;p=akkoma

diff --git a/restarter/lib/pleroma.ex b/restarter/lib/pleroma.ex
index da714654c..a7186cec4 100644
--- a/restarter/lib/pleroma.ex
+++ b/restarter/lib/pleroma.ex
@@ -1,26 +1,104 @@
 defmodule Restarter.Pleroma do
   use GenServer
 
+  require Logger
+
+  @init_state %{need_reboot: false, rebooted: false, after_boot: false}
+
   def start_link(_) do
     GenServer.start_link(__MODULE__, [], name: __MODULE__)
   end
 
-  def init(_), do: {:ok, %{}}
+  def init(_), do: {:ok, @init_state}
+
+  def rebooted? do
+    GenServer.call(__MODULE__, :rebooted?)
+  end
+
+  def rebooted do
+    GenServer.cast(__MODULE__, :rebooted)
+  end
+
+  def need_reboot? do
+    GenServer.call(__MODULE__, :need_reboot?)
+  end
+
+  def need_reboot do
+    GenServer.cast(__MODULE__, :need_reboot)
+  end
+
+  def refresh do
+    GenServer.cast(__MODULE__, :refresh)
+  end
+
+  def restart(env, delay) do
+    GenServer.cast(__MODULE__, {:restart, env, delay})
+  end
+
+  def restart_after_boot(env) do
+    GenServer.cast(__MODULE__, {:after_boot, env})
+  end
+
+  def handle_call(:rebooted?, _from, state) do
+    {:reply, state[:rebooted], state}
+  end
+
+  def handle_call(:need_reboot?, _from, state) do
+    {:reply, state[:need_reboot], state}
+  end
+
+  def handle_cast(:rebooted, state) do
+    {:noreply, Map.put(state, :rebooted, true)}
+  end
+
+  def handle_cast(:need_reboot, %{need_reboot: true} = state), do: {:noreply, state}
+
+  def handle_cast(:need_reboot, state) do
+    {:noreply, Map.put(state, :need_reboot, true)}
+  end
 
-  def handle_info(:after_boot, %{after_boot: true} = state), do: {:noreply, state}
+  def handle_cast(:refresh, _state) do
+    {:noreply, @init_state}
+  end
 
-  def handle_info(:after_boot, state) do
-    restart(:pleroma)
-    {:noreply, Map.put(state, :after_boot, true)}
+  # Don't actually restart during tests.
+  # We just check if the correct call has been done.
+  # If we actually restart, we get errors during the tests like
+  #     (RuntimeError) could not lookup Ecto repo Pleroma.Repo because it was not started or
+  #      it does not exist
+  # See tests in Pleroma.Config.TransferTaskTest
+  def handle_cast({:restart, :test, _}, state) do
+    Logger.debug("pleroma manually restarted")
+    {:noreply, Map.put(state, :need_reboot, false)}
   end
 
-  def handle_info({:restart, delay}, state) do
+  def handle_cast({:restart, _, delay}, state) do
     Process.sleep(delay)
-    restart(:pleroma)
+    do_restart(:pleroma)
+    {:noreply, Map.put(state, :need_reboot, false)}
+  end
+
+  def handle_cast({:after_boot, _}, %{after_boot: true} = state), do: {:noreply, state}
+
+  # Don't actually restart during tests.
+  # We just check if the correct call has been done.
+  # If we actually restart, we get errors during the tests like
+  #     (RuntimeError) could not lookup Ecto repo Pleroma.Repo because it was not started or
+  #      it does not exist
+  # See tests in Pleroma.Config.TransferTaskTest
+  def handle_cast({:after_boot, :test}, state) do
+    Logger.debug("pleroma restarted after boot")
+    state = %{state | after_boot: true, rebooted: true}
+    {:noreply, state}
+  end
+
+  def handle_cast({:after_boot, _}, state) do
+    do_restart(:pleroma)
+    state = %{state | after_boot: true, rebooted: true}
     {:noreply, state}
   end
 
-  defp restart(app) do
+  defp do_restart(app) do
     :ok = Application.ensure_started(app)
     :ok = Application.stop(app)
     :ok = Application.start(app)