asn

defmodule ASN.Matcher do
  alias ASN.Table.{IPtoAS, AStoASN}

  @external_resource Application.app_dir(:asn, "priv") <> "/ip_to_as_lookup_table.eterm"
  @external_resource Application.app_dir(:asn, "priv") <> "/as_to_asn_lookup_table.eterm"

  @ip_to_as_lookup_table Application.app_dir(:asn, "priv") <> "/ip_to_as_lookup_table.eterm"
  @as_to_asn_lookup_table Application.app_dir(:asn, "priv") <> "/as_to_asn_lookup_table.eterm"

  def start_link(opts \\ []) do
    Agent.start_link(fn ->
      {File.read!(@ip_to_as_lookup_table) |> :erlang.binary_to_term,
       File.read!(@as_to_asn_lookup_table) |> :erlang.binary_to_term}
    end, opts)
  end

  def ip_to_asn(agent, ip) when is_binary(ip),
  do: ASN.Parser.ip_to_tuple(ip) |> (&ip_to_asn(agent, &1)).()
  def ip_to_asn(agent, {_, _, _, _} = ip) do
    with {:ok, as}  <- ip_to_as(agent, ip),
         {:ok, asn} <- as_to_asn(agent, as),
         do: {:ok, asn}
  end
  def ip_to_asn(_, _), do: :error

  def ip_to_as(agent, ip) when is_binary(ip),
  do: ASN.Parser.ip_to_tuple(ip) |> (&ip_to_as(agent, &1)).()
  def ip_to_as(agent, {_, _, _, _} = ip),
  do: Agent.get(agent, fn {table, _} -> IPtoAS.lookup(table, ip) end)
  def ip_to_as(_), do: :error

  def as_to_asn(agent, as),
  do: Agent.get(agent, fn {_, table} -> AStoASN.lookup(table, as) end)

  #@doc false
  #def ip_to_as_lookup_table, do: @ip_to_as_lookup_table

  #@doc false
  #def as_to_asn_lookup_table, do: @as_to_asn_lookup_table

  #@doc false
  #def load_tables do

  #  Module.register_attribute(__MODULE__, :ip_to_as_lookup_table, accumulate: false)
  #  Module.put_attribute(__MODULE__, :ip_to_as_lookup_table,
  #    File.read!("priv/ip_to_as_lookup_table.eterm") |> :erlang.binary_to_term)

  #  Module.register_attribute(__MODULE__, :as_to_asn_lookup_table, accumulate: false)
  #  Module.put_attribute(__MODULE__, :as_to_asn_lookup_table,
  #    File.read!("db/as_to_asn_lookup_table.eterm") |> :erlang.binary_to_term)

  #  :ok
  #end
end