exshopify

defmodule Shopify.RateLimiter do
  use Supervisor

  alias Shopify.RateLimiter

  #
  # client
  #

  def start_link(opts) do
    ensure_gen_stage_loaded!()

    opts = Keyword.put(opts, :name, Keyword.get(opts, :name, __MODULE__))

    Supervisor.start_link(__MODULE__, opts, name: opts[:name])
  end

  @spec make_request(binary | atom, Shopify.Request.t(), any) :: :ok
  def make_request(server, request, opts) do
    partition = Map.fetch!(URI.parse(request.url), :host)

    RateLimiter.Partition.open(server, partition)

    RateLimiter.Producer.queue_request(server, partition, request, opts)
  end

  #
  # callbacks
  #

  def init(opts) do
    Supervisor.init(children(opts), strategy: :one_for_one)
  end

  #
  # private
  #

  defp children(opts) do
    server = opts[:name]

    [
      {RateLimiter.PartitionSupervisor, server: server},
      {RateLimiter.PartitionMonitor, server: server}
    ]
  end

  defp ensure_gen_stage_loaded! do
    unless Code.ensure_loaded?(GenStage) do
      raise Code.LoadError, """
      You are trying to start Shopify.RateLimiter but GenStage is not loaded.
      Make sure you have defined gen_stage as a dependency.
      """
    end
  end
end