Add support for when a device disconnects

Author: joshk

lib/nerves_hub/rpc/device_auth.ex

@@ -52,6 +52,28 @@ defmodule NervesHub.RPC.DeviceAuth do
     end
   end
 
+  def disconnect_device({:error, {:shutdown, :disconnected}}, device, reference_id) do
+    :telemetry.execute([:nerves_hub, :devices, :duplicate_connection], %{count: 1}, %{
+      ref_id: reference_id,
+      device: device
+    })
+
+    disconnect_device(:ok, device, reference_id)
+  end
+
+  def disconnect_device(_, device, reference_id) do
+    :telemetry.execute([:nerves_hub, :devices, :disconnect], %{count: 1}, %{
+      ref_id: reference_id,
+      identifier: device.identifier
+    })
+
+    {:ok, _device_connection} = Connections.device_disconnected(reference_id)
+
+    Tracker.offline(device)
+
+    :ok
+  end
+
   defp on_connect(%Device{status: :registered} = device) do
     Devices.set_as_provisioned!(device)
     |> on_connect()

lib/nerves_hub_web/channels/device_socket.ex

@@ -6,8 +6,6 @@ defmodule NervesHubWeb.DeviceSocket do
 
   alias NervesHub.Devices.Connections
 
-  alias NervesHub.Tracker
-
   alias NervesHub.RPC.DeviceAuth
 
   channel("console", NervesHubWeb.ConsoleChannel)
@@ -19,7 +17,8 @@ defmodule NervesHubWeb.DeviceSocket do
   @impl Phoenix.Socket.Transport
   @decorate with_span("Channels.DeviceSocket.terminate")
   def terminate(reason, {_channels_info, socket} = state) do
-    on_disconnect(reason, socket)
+    %{assigns: %{device: device, reference_id: reference_id}} = socket
+    DeviceAuth.disconnect_device(reason, device, reference_id)
     super(reason, state)
   end
 
@@ -114,6 +113,7 @@ defmodule NervesHubWeb.DeviceSocket do
   @decorate with_span("Channels.DeviceSocket.on_disconnect")
   defp on_disconnect(exit_reason, socket)
 
+  @decorate with_span("Channels.DeviceSocket.on_disconnect")
   defp on_disconnect({:error, reason}, %{
          assigns: %{
            device: device,
@@ -132,6 +132,7 @@ defmodule NervesHubWeb.DeviceSocket do
     :ok
   end
 
+  @decorate with_span("Channels.DeviceSocket.on_disconnect")
   defp on_disconnect(_, %{
          assigns: %{
            device: device,