ex_esdb

defmodule ExESDB.Store do
  @moduledoc """
    A GenServer wrapper around :khepri to act as a distributed event store.
  """
  use GenServer

  require Logger

  alias ExESDB.Themes, as: Themes
  alias ExESDB.StoreNaming

  defp start_khepri(opts) do
    store = opts[:store_id]
    timeout = opts[:timeout]
    data_dir = opts[:data_dir]
    :khepri.start(data_dir, store, timeout)
  end

  # Client API
  @doc """
  Get the current state of the store.
  ## Returns

      - `{:ok, state}`  if successful.
      - `{:error, reason}` if unsuccessful.

  """
  def get_state(store_id \\ nil),
    do:
      GenServer.call(
        StoreNaming.genserver_name(__MODULE__, store_id),
        {:get_state}
      )
      
  @doc """
  Get the store-specific GenServer name.
  
  This function returns the name used to register this store GenServer,
  allowing multiple stores to run on the same node.
  
  ## Parameters
  
  * `store_id` - The store identifier (optional)
  
  ## Examples
  
      iex> ExESDB.Store.store_name("my_store")
      {:ex_esdb_store, "my_store"}
      
      iex> ExESDB.Store.store_name(nil)
      ExESDB.Store
  """
  def store_name(store_id), do: StoreNaming.genserver_name(__MODULE__, store_id)

  ## CALLBACKS
  @impl true
  def handle_call({:get_state}, _from, state) do
    {:reply, {:ok, state}, state}
  end

  #### PLUMBING
  def child_spec(opts) do
    store_id = StoreNaming.extract_store_id(opts)
    
    %{
      id: StoreNaming.child_spec_id(__MODULE__, store_id),
      start: {__MODULE__, :start_link, [opts]},
      restart: :permanent,
      shutdown: 10_000,
      type: :worker
    }
  end

  def start_link(opts) do
    store_id = StoreNaming.extract_store_id(opts)
    name = StoreNaming.genserver_name(__MODULE__, store_id)
    
    GenServer.start_link(
      __MODULE__,
      opts,
      name: name
    )
  end

  # Server Callbacks
  @impl true
  def init(opts) do
    store_id = StoreNaming.extract_store_id(opts)
    Logger.info("[STORE] Initializing Store GenServer for store_id: #{inspect(store_id)}")
    Logger.info("[STORE] Store PID: #{inspect(self())}, Node: #{inspect(node())}")
    Logger.info("[STORE] Store configuration: #{inspect(opts)}")
    
    IO.puts("#{Themes.store(self(), "is UP.")}")
    Process.flag(:trap_exit, true)
    
    Logger.info("[STORE] Starting Khepri store with configuration...")
    Logger.info("[STORE] Store ID: #{inspect(opts[:store_id])}")
    Logger.info("[STORE] Data directory: #{inspect(opts[:data_dir])}")
    Logger.info("[STORE] Timeout: #{inspect(opts[:timeout])}")

    case start_khepri(opts) do
      {:ok, store} ->
        Logger.info("[STORE] ✅ Successfully started Khepri store: #{inspect(store)}")
        Logger.info("[STORE] Store is ready to handle requests")
        {:ok, [config: opts, store: store]}

      reason ->
        Logger.error("[STORE] ❌ Failed to start Khepri store. Reason: #{inspect(reason)}")
        Logger.error("[STORE] Store initialization failed - this will trigger restart")
        {:error, [config: opts, store: nil]}
    end
  end

  @impl true
  def terminate(reason, [config: opts, store: store]) do
    IO.puts("#{Themes.store(self(), "⚠️  Shutting down gracefully. Reason: #{inspect(reason)}")}")
    
    # Stop Khepri store gracefully if it was started
    if store do
      store_id = opts[:store_id]
      Logger.info("Stopping Khepri store: #{inspect(store_id)}")
      
      case :khepri.stop(store_id) do
        :ok ->
          Logger.info("Successfully stopped Khepri store: #{inspect(store_id)}")
        {:error, reason} ->
          Logger.warning("Failed to stop Khepri store #{inspect(store_id)}: #{inspect(reason)}")
        other ->
          Logger.warning("Unexpected response stopping Khepri store #{inspect(store_id)}: #{inspect(other)}")
      end
    end
    
    :ok
  end

  def terminate(reason, _state) do
    IO.puts("#{Themes.store(self(), "⚠️  Shutting down gracefully. Reason: #{inspect(reason)}")}")
    :ok
  end
end