oaskit

defmodule Oaskit do
  alias Oaskit.Internal.Normalizer
  alias Oaskit.Internal.SpecBuilder

  @moduledoc """
  The main API to work with OpenAPI specifications.

  This module can be used to define a specification module that will then be
  used in your Phoenix router and controllers.

  ### Example

  ```elixir
  defmodule MyAppWeb.OpenAPISpec do
    alias Oaskit.Spec.Paths
    alias Oaskit.Spec.Server
    use Oaskit

    @impl true
    def spec do
      %{
        openapi: "3.1.1",
        info: %{title: "My App API", version: "1.0.0"},
        servers: [Server.from_config(:my_app, MyAppWeb.Endpoint)],
        paths: Paths.from_router(MyAppWeb.Router, filter: &String.starts_with?(&1.path, "/api/"))
      }
    end
  end
  ```
  """

  @typedoc """
  The cache keys used by this module when calling `c:cache/1`.
  """
  @type cache_key :: {:oaskit_cache, module, responses? :: boolean, variant :: term}

  @typedoc """
  A cache action given to `c:cache/1`. `#{inspect(__MODULE__)}` will use keys of
  type `t:cache_key/0` when calling the cache.
  """
  @type cache_action ::
          {:get, key :: cache_key | term} | {:put, key :: cache_key | term, value :: term}

  @doc """
  This function should return the OpenAPI specification for your application.

  It can be returned as an `%#{OpenAPI}{}` struct, or a bare map with atoms or
  binary keys (for instance by reading from a JSON file at compile time).

  The returned value will be normalized, any extra data not defined in the
  `#{inspect(Oaskit.Spec)}...` namespace will be lost.
  """
  @callback spec :: map

  @doc """
  This callback is used to cache the built version of the OpenAPI specification,
  with JSV schemas turned into validators.

  * The callback will be called with `:get` to retrieve a cached build, in which
    case the callback should return `{:ok, cached}` or `:error`.
  * It will be called with `{:put, value}` to set the cache, in which case it
    must return `:ok`.

  Caching is very important, otherwise the spec will be built for each request
  calling a controller that uses `#{inspect(ValidateRequest)}`. An efficient
  default implementation using `:persistent_term` is automatically generated.
  Override this callback if you need more control over the cache.
  """
  @callback cache(cache_action) :: :ok | {:ok, term} | :error

  @doc """
  Returns the options that will be passed to `JSV.build/2` when building the
  spec for the implementation module.

  The default implementation delegates to `Oaskit.default_jsv_opts/0`.
  """
  @callback jsv_opts :: [JSV.build_opt()]

  @doc """
  This function is intended to change cache keys at runtime. The variant is any
  term used as the last element of a `t:cache_key/0`.

  This is useful if you need to rebuild the OpenAPI specification and its
  validators at runtime, when the used schemas or even routes depend on current
  application state. For instance, if a schema for a given entity is fetched
  regularly from a remote source and changes over time.

  The default implementation returns `nil`.

  > #### Stale cache entries are not purged automatically {: .warning}
  >
  > If you return a new variant from this callback, cache entries stored with
  > previous variants in the key are not automatically cleaned. You will need to
  > take care of that. See `c:cache/1` to implement a cache mechanism that you
  > can control.
  """
  @callback cache_variant :: term

  defmacro __using__(_) do
    quote do
      @behaviour unquote(__MODULE__)

      @impl true
      def cache({:get, key}) do
        case(:persistent_term.get(key, :__undef__)) do
          :__undef__ -> :error
          cached -> {:ok, cached}
        end
      end

      def cache({:put, key, cacheable}) do
        :ok = :persistent_term.put(key, cacheable)
      end

      @impl true
      def jsv_opts do
        unquote(__MODULE__).default_jsv_opts()
      end

      @impl true
      def cache_variant do
        nil
      end

      defoverridable unquote(__MODULE__)
    end
  end

  @default_jsv_opts (quote do
                       [
                         default_meta: JSV.default_meta(),
                         formats: [
                           Oaskit.JsonSchema.Formats | JSV.default_format_validator_modules()
                         ]
                       ]
                     end)

  @doc """
  Default options used for the `JSV.build/2` function when building schemas:

  ```
  #{Macro.to_string(@default_jsv_opts)}
  ```
  """
  def default_jsv_opts do
    unquote(@default_jsv_opts)
  end

  @doc false
  # opt :cache defaults to true
  # opt :responses defaults to false
  def build_spec!(spec_module, opts \\ []) do
    cache? = Keyword.get(opts, :cache, true)

    if cache? do
      cache_key = cache_key(spec_module, opts)
      cached(spec_module, cache_key, fn -> do_build_spec!(spec_module, opts) end)
    else
      do_build_spec!(spec_module, opts)
    end
  end

  @spec cache_key(module, keyword) :: cache_key
  defp cache_key(spec_module, opts) do
    {:oaskit_cache, spec_module, !!opts[:responses], spec_module.cache_variant()}
  end

  @doc """
  Retrieves a cached from the implementation module.

  If the value is not in catche, the `generator` is called and the generated
  value is put in cache before being returned.
  """
  def cached(spec_module, cache_key, generator) do
    case spec_module.cache({:get, cache_key}) do
      {:ok, value} ->
        value

      :error ->
        value = generator.()
        :ok = spec_module.cache({:put, cache_key, value})
        value
    end
  end

  defp do_build_spec!(spec_module, opts) do
    opts = build_opts(spec_module, opts)

    spec_module.spec()
    |> Normalizer.normalize!()
    |> SpecBuilder.build_operations(opts)
  end

  defp build_opts(spec_module, opts) do
    opts
    |> Keyword.delete(:cache)
    |> Keyword.put_new_lazy(:jsv_opts, fn -> spec_module.jsv_opts() end)
    |> Keyword.put_new(:responses, false)
    |> Map.new()
  end

  @doc """
  Normalizes OpenAPI specification data.

  Takes specification data (raw maps or structs) and normalizes it to a
  JSON-compatible version (with binary keys).
  """
  def normalize_spec!(data) do
    Normalizer.normalize!(data)
  end

  @doc """
  Validates the given OpenAPI specification module returns a representation of
  the specification using structs such as `#{inspect(Oaskit.Spec.OpenAPI)}`,
  `#{inspect(Oaskit.Spec.Response)}`, _etc_.
  """
  def cast!(module) do
    module.spec()
    |> normalize_spec!()
    |> Oaskit.SpecValidator.validate!()
  end

  @doc """
  Returns a JSON representation of the given OpenAPI specification module.

  See `Oaskit.SpecDumper.to_json!/2` for options.
  """
  @spec to_json!(module, keyword | map) :: String.t()
  def to_json!(module, opts \\ []) do
    module.spec()
    |> normalize_spec!()
    |> Oaskit.SpecDumper.to_json!(opts)
  end
end