system_registry

defmodule SystemRegistry.Local do
  @moduledoc false

  use GenServer

  alias SystemRegistry.{Transaction, Processor, Registration}
  alias SystemRegistry.Storage.State, as: S

  import SystemRegistry.Utils
  #import SystemRegistry.Transaction

  def start_link(opts \\ []) do
    GenServer.start_link(__MODULE__, [opts], name: __MODULE__)
  end

  def register_processor({mod, pid}) do
    GenServer.call(__MODULE__, {:register_processor, {mod, pid}})
  end

  # GenServer API

  def init(_opts) do
    {:ok, %{
      processors: [],
      monitors: []
    }}
  end

  def handle_call({:register_processor, {mod, pid}}, _from, s) do
    {:reply, :ok, %{s | processors: [{mod, pid} | s.processors]}}
  end

  def handle_call({:commit, transaction}, _, s) do
    {reply, s} = commit(transaction, s)
    {:reply, reply, s}
  end

  def handle_call({:update_in, t, scope, fun}, _, s) do
    value =
      Registry.lookup(S, t.key)
      |> strip()
      |> get_in(scope)
    value = fun.(value)
      t = Transaction.update(t, scope, value)
    {reply, s} = commit(t, s)
    {:reply, reply, s}
  end

  def handle_call({:delete_all, pid}, _, s) do
    {reply, s} = delete_all(pid, s)
    {:reply, reply, s}
  end

  def handle_info({:DOWN, _, _, pid, _}, s) do
    Registration.unregister_all(pid)
    {_, s} = delete_all(pid, s)
    s = demonitor(pid, s)
    {:noreply, s}
  end

  defp delete_all(pid, s) do
    frag =
    Registry.lookup(S, pid)
    |> strip()

    t = Transaction.begin()

    %{t | pid: pid, key: pid}
    |> Transaction.delete(frag)
    |> commit(s)
  end

  defp commit(%Transaction{pid: pid} = t, s) do
    with                    t <- Transaction.prepare(t),
                          :ok <- Processor.call(s.processors, :validate, [t]),
   {:ok, {new, _old} = delta} <- Transaction.commit(t),
                            s <- monitor(pid, s),
                          :ok <- Registration.notify(pid, new),
                          :ok <- Processor.call(s.processors, :commit, [t]) do


      {{:ok, delta}, s}
    else
      error ->
        {error, s}
    end
  end

  defp monitor(pid, s) do
    if Enum.any?(s.monitors, fn({m_pid, _ref}) -> m_pid == pid end) do
      s
    else
      monitors = [{pid, Process.monitor(pid)} | s.monitors]
      %{s | monitors: monitors}
    end
  end

  defp demonitor(pid, s) do
    {[{_, ref}], monitors} =
      Enum.split_with(s.monitors, fn({m_pid, _}) -> m_pid == pid end)
    Process.demonitor(ref)
    %{s | monitors: monitors}
  end
end