defmodule Exograph.DSL.Sources do
  @moduledoc false

  alias Exograph.Postgres.{CallEdgeRecord, DefinitionRecord, Options, ReferenceRecord}

  @symbol_fact_fields MapSet.new([
                        :id,
                        :package_id,
                        :package_version_id,
                        :file_id,
                        :fragment_id,
                        :kind,
                        :module,
                        :name,
                        :arity,
                        :qualified_name,
                        :line,
                        :column
                      ])

  @fragment_fields MapSet.new([
                     :id,
                     :package_id,
                     :package_version_id,
                     :file_id,
                     :kind,
                     :module,
                     :name,
                     :arity,
                     :line,
                     :end_line,
                     :mass
                   ])

  @call_edge_fields MapSet.new([
                      :id,
                      :package_id,
                      :package_version_id,
                      :file_id,
                      :caller_node_id,
                      :callee_node_id,
                      :call_site_fragment_id,
                      :caller_qualified_name,
                      :callee_qualified_name,
                      :line,
                      :column
                    ])

  def source(:definition, prefix), do: Options.definitions_source(prefix)
  def source(:reference, prefix), do: Options.references_source(prefix)
  def source(:call_edge, prefix), do: Options.call_edges_source(prefix)

  def source_record(:definition), do: DefinitionRecord
  def source_record(:reference), do: ReferenceRecord
  def source_record(:call_edge), do: CallEdgeRecord

  def primary_source(:definitions, prefix), do: Options.definitions_source(prefix)
  def primary_source(:references, prefix), do: Options.references_source(prefix)
  def primary_source(:calls, prefix), do: Options.call_edges_source(prefix)

  def join_source(:definitions, prefix), do: Options.definitions_source(prefix)
  def join_source(:references, prefix), do: Options.references_source(prefix)
  def join_source(:calls, prefix), do: Options.call_edges_source(prefix)

  def fields(:fragment), do: @fragment_fields
  def fields(:definition), do: @symbol_fact_fields
  def fields(:reference), do: @symbol_fact_fields
  def fields(:definitions), do: @symbol_fact_fields
  def fields(:references), do: @symbol_fact_fields
  def fields(:call_edge), do: @call_edge_fields
  def fields(:calls), do: @call_edge_fields

  def assert_field!(source, field) do
    unless MapSet.member?(fields(source), field) do
      raise ArgumentError, "unsupported #{source} field in Exograph DSL: #{field}"
    end
  end
end