aprs.me/lib/aprs/is/is.ex

238 lines
7.4 KiB
Elixir

defmodule Aprs.Is do
use GenServer
require Logger
@aprs_timeout 30 * 1000
def start_link(opts \\ []) do
GenServer.start_link(__MODULE__, opts, name: __MODULE__)
end
@impl true
def init(_opts) do
# Trap exits so we can gracefully shut down
Process.flag(:trap_exit, true)
# Get startup parameters
server = Application.get_env(:aprs, :aprs_is_server, 'rotate.aprs2.net')
port = Application.get_env(:aprs, :aprs_is_port, 14580)
default_filter = Application.get_env(:aprs, :aprs_is_default_filter, "r/33/-96/100")
aprs_user_id = Application.get_env(:aprs, :aprs_is_login_id, "w5isp")
aprs_passcode = Application.get_env(:aprs, :aprs_is_password, "-1")
# Set up ets tables
# with {:ok, :aprs} <- :ets.file2tab(:erlang.binary_to_list("priv/aprs.ets")),
# {:ok, :aprs_messages} <- :ets.file2tab(:erlang.binary_to_list("priv/aprs_messages.ets")) do
# else
# {:error, _reason} ->
# Logger.debug("Ets files not found. Creating new ETS tables.")
# # Create new ETS tables
# :ets.new(:aprs, [:set, :protected, :named_table])
# :ets.new(:aprs_messages, [:bag, :protected, :named_table])
# # Write them to file here in case genserver is brutally terminated
# :ets.tab2file(:aprs, :erlang.binary_to_list("priv/aprs.ets"))
# :ets.tab2file(:aprs_messages, :erlang.binary_to_list("priv/aprs_messages.ets"))
# _ ->
# Logger.error("Unable to set up ETS tables for message storage.")
# end
with {:ok, socket} <- connect_to_aprs_is(server, port),
:ok <- send_login_string(socket, aprs_user_id, aprs_passcode, default_filter),
timer <- create_timer(@aprs_timeout) do
{:ok,
%{
server: server,
port: port,
socket: socket,
timer: timer
}}
else
_ ->
Logger.error("Unable to establish connection or log in to APRS-IS")
{:stop, :aprs_connection_failed}
end
end
# Client API
def stop do
Logger.info("Stopping Server")
GenServer.stop(__MODULE__, :stop)
end
def set_filter(filter_string), do: send_message("#filter #{filter_string}")
def list_active_filters, do: send_message("#filter?")
def send_message(from, to, message) do
padded_callsign = String.pad_trailing(to, 9)
send_message("#{from}>APRS,TCPIP*::#{padded_callsign}:#{message}")
end
def send_message(message) do
GenServer.call(__MODULE__, {:send_message, message})
end
# Server methods
defp connect_to_aprs_is(server, port) do
Logger.debug("Connecting to: #{server}:#{port}")
opts = [:binary, active: true]
:gen_tcp.connect(String.to_charlist(server), port, opts)
end
defp send_login_string(socket, aprs_user_id, aprs_passcode, filter) do
login_string =
"user #{aprs_user_id} pass #{aprs_passcode} vers aprs.me 0.1 filter #{filter} \n"
Logger.debug("Sending login string: #{login_string}")
:gen_tcp.send(socket, login_string)
end
defp create_timer(timeout) do
Process.send_after(self(), :aprs_no_message_timeout, timeout)
end
@impl true
def handle_call({:send_message, message}, _from, state) do
next_ack_number = :ets.update_counter(:aprs, :message_number, 1)
# Append ack number
message = message <> "{" <> to_string(next_ack_number) <> "\r"
Logger.info("Sending message: #{inspect(message)}")
:gen_tcp.send(state.socket, message)
{:reply, :ok, state}
end
@impl true
def handle_info(:aprs_no_message_timeout, state) do
Logger.error("Socket timeout detected. Killing genserver.")
{:stop, :aprs_timeout, state}
end
def handle_info({:tcp, _socket, packet}, state) do
# Cancel the previous timer
Process.cancel_timer(state.timer)
# Handle the incoming message
# Task.start(Aprs, :dispatch, [packet])
if String.contains?(packet, "\n") or String.contains?(packet, "\r") do
packet
|> String.split("\r\n")
|> Enum.each(&dispatch(String.trim(&1)))
end
dispatch(packet)
# Start a new timer
timer = Process.send_after(self(), :aprs_no_message_timeout, @aprs_timeout)
state = Map.put(state, :timer, timer)
{:noreply, state}
end
def handle_info({:tcp_closed, _socket}, state) do
Logger.info("Socket has been closed")
{:stop, :normal, state}
end
def handle_info({:tcp_error, _socket, reason}, state) do
Logger.error("Connection closed due to #{inspect(reason)}")
{:stop, :normal, state}
end
@impl true
def terminate(reason, state) do
# Do Shutdown Stuff
Logger.info("Going Down: #{inspect(reason)} - #{inspect(state)}")
# Logger.info("Attempting to write ets tables to disk.")
# :ets.tab2file(:aprs, :erlang.binary_to_list("priv/aprs.ets"))
# :ets.tab2file(:aprs_messages, :erlang.binary_to_list("priv/aprs_messages.ets"))
Logger.info("Closing socket")
:gen_tcp.close(state.socket)
:normal
end
@spec dispatch(binary) :: nil | :ok
def dispatch("#" <> comment_text) do
Logger.debug("COMMENT: " <> String.trim(comment_text))
end
def dispatch(""), do: nil
def dispatch(message) do
case Parser.parse(message) do
{:ok, parsed_message} ->
# IO.inspect(parsed_message)
# time_message_received =
# :calendar.universal_time() |> :calendar.datetime_to_gregorian_seconds()
# This doesnt work if dispatch if spawned as a task, since it does not own the table
# :ets.insert(:aprs, {parsed_message.sender, message, time_message_received})
# Get messages since last time the callsign was seen
# TODO: Get last seen timestamp and use that
# _message_count =
# case :ets.lookup(:aprs_messages, parsed_message.sender) do
# [{_callsign, _message, timestamp}] ->
# recent_messages_for(parsed_message.sender, timestamp)
# [] ->
# 0
# _ ->
# 0
# end
# Logger.debug("#{message_count} recent messages for #{parsed_message.sender}")
# Do something interesting with the message
# process(parsed_message, parsed_message.data_type)
# Publish the parsed message to all interested parties
# Registry.dispatch(Registry.PubSub, "aprs_messages", fn entries ->
# for {pid, _} <- entries, do: send(pid, {:broadcast, parsed_message})
# end)
AprsWeb.Endpoint.broadcast("aprs_messages", "packet", parsed_message)
# Phoenix.PubSub.broadcast(
# Aprs.PubSub,
# "aprs_messages",
# {:packet, parsed_message}
# )
# IO.inspect(parsed_message)
# Logger.debug("SERVER:" <> message)
{:error, error} ->
Logger.debug("PARSE ERROR: " <> error)
x ->
Logger.debug("PARSE ERROR: " <> x)
end
end
# def process(message, message_type) when message_type == :message do
# # time_message_received =
# # :calendar.universal_time() |> :calendar.datetime_to_gregorian_seconds()
# # :ets.insert(
# # :aprs_messages,
# # {message.data_extended.to, message.data_extended, time_message_received}
# # )
# end
# def process(_, _), do: nil
# def recent_messages_for(callsign, since_time) do
# callsign_guard = {:==, :"$1", {:const, callsign}}
# timestamp_guard = {:>=, :"$2", {:const, since_time}}
# total_spec = [{{:"$1", :_, :"$2"}, [{:andalso, callsign_guard, timestamp_guard}], [true]}]
# :ets.select_count(:aprs_messages, total_spec)
# end
end