defmodule AshPaperTrail.Resource.Info do
  @moduledoc "Introspection helpers for `AshPaperTrail.Resource`"

  @spec primary_key_type(Spark.Dsl.t() | Ash.Resource.t()) :: atom
  def primary_key_type(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :primary_key_type, :uuid)
  end

  @spec attributes_as_attributes(Spark.Dsl.t() | Ash.Resource.t()) :: [atom]
  def attributes_as_attributes(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :attributes_as_attributes, [])
  end

  @spec belongs_to_actor(Spark.Dsl.t() | Ash.Resource.t()) :: [atom]
  def belongs_to_actor(resource) do
    Spark.Dsl.Extension.get_entities(resource, [:paper_trail])
    |> Enum.filter(fn
      %AshPaperTrail.Resource.BelongsToActor{} -> true
      _ -> false
    end)
  end

  @spec change_tracking_mode(Spark.Dsl.t() | Ash.Resource.t()) :: atom
  def change_tracking_mode(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :change_tracking_mode, [])
  end

  @spec ignore_attributes(Spark.Dsl.t() | Ash.Resource.t()) :: [atom]
  def ignore_attributes(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :ignore_attributes, [])
  end

  @spec ignore_actions(Spark.Dsl.t() | Ash.Resource.t()) :: [atom]
  def ignore_actions(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :ignore_actions, [])
  end

  @spec mixin(Spark.Dsl.t() | Ash.Resource.t()) :: {module, atom, list} | module | nil
  def mixin(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :mixin, nil)
  end

  @spec on_actions(Spark.Dsl.t() | Ash.Resource.t()) :: [atom]
  def on_actions(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :on_actions, nil) ||
      resource
      |> Ash.Resource.Info.actions()
      |> Enum.reject(&(&1.type == :read))
      |> Enum.map(& &1.name)
  end

  @spec reference_source?(Spark.Dsl.t() | Ash.Resource.t()) :: boolean
  def reference_source?(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :reference_source?, true)
  end

  @spec relationship_opts(Spark.Dsl.t() | Ash.Resource.t()) :: Keyword.t()
  def relationship_opts(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :relationship_opts, [])
  end

  @spec store_action_name?(Spark.Dsl.t() | Ash.Resource.t()) :: boolean
  def store_action_name?(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :store_action_name?, false)
  end

  @spec store_resource_identifier?(Spark.Dsl.t() | Ash.Resource.t()) :: boolean
  def store_resource_identifier?(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :store_resource_identifier?, false)
  end

  @spec sensitive_attributes(Spark.Dsl.t() | Ash.Resource.t()) :: :display | :ignore | :redact
  def sensitive_attributes(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :sensitive_attributes)
  end

  @spec resource_identifier(Spark.Dsl.t() | Ash.Resource.t()) :: atom
  def resource_identifier(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :resource_identifier)
  end

  @spec version_extensions(Spark.Dsl.t() | Ash.Resource.t()) :: Keyword.t()
  def version_extensions(resource) do
    Spark.Dsl.Extension.get_opt(resource, [:paper_trail], :version_extensions, [])
  end

  @spec version_resource(Spark.Dsl.t() | Ash.Resource.t()) :: Ash.Resource.t()
  def version_resource(resource) do
    if is_atom(resource) do
      Module.concat([resource, Version])
    else
      Module.concat([Spark.Dsl.Extension.get_persisted(resource, :module), Version])
    end
  end
end