From 81b5a844a2a155e28c497a8ce671eb5f02803e5d Mon Sep 17 00:00:00 2001
From: Alex Auvolat <alex@adnab.me>
Date: Sat, 1 Sep 2018 16:07:22 +0200
Subject: Import shardweb

---
 shardweb/lib/shard_web/channels/room_channel.ex | 68 +++++++++++++++++++++++++
 shardweb/lib/shard_web/channels/user_socket.ex  | 37 ++++++++++++++
 2 files changed, 105 insertions(+)
 create mode 100644 shardweb/lib/shard_web/channels/room_channel.ex
 create mode 100644 shardweb/lib/shard_web/channels/user_socket.ex

(limited to 'shardweb/lib/shard_web/channels')

diff --git a/shardweb/lib/shard_web/channels/room_channel.ex b/shardweb/lib/shard_web/channels/room_channel.ex
new file mode 100644
index 0000000..2dd733b
--- /dev/null
+++ b/shardweb/lib/shard_web/channels/room_channel.ex
@@ -0,0 +1,68 @@
+defmodule ShardWeb.RoomChannel do
+  use ShardWeb, :channel
+
+  require Logger
+
+  def join("room:" <> room_name, payload, socket) do
+    if authorized?(payload) do
+
+      list = for {_chid, manifest, chpid} <- :ets.tab2list(:shard_db),
+                 {:chat, chan} = manifest,
+                 do: {chan, chpid}
+      pid = case List.keyfind(list, room_name, 0) do
+        nil ->
+          {:ok, pid} = DynamicSupervisor.start_child(Shard.DynamicSupervisor, {SApp.Chat, room_name})
+          pid
+        {_, pid} ->
+          pid
+      end
+      socket = assign(socket, :pid, pid)
+
+      GenServer.cast(pid, {:subscribe, self()})
+      send(self(), :after_join)
+      
+      {:ok, socket}
+    else
+      {:error, %{reason: "unauthorized"}}
+    end
+  end
+
+  def handle_info(:after_join, socket) do
+    GenServer.call(socket.assigns.pid, {:read_history, nil, 100})
+    |> Enum.each(fn {{_ts, nick, msg}, true} -> push(socket, "shout", %{
+        name: nick,
+        message: msg,
+    }) end)
+    {:noreply, socket}
+  end
+
+  def handle_info({:chat_recv, _chan, {_ts, from, msg}}, socket) do
+    Logger.info("#{inspect self()} :chat_recv #{inspect msg}")
+    push socket, "shout", %{"name" => from, "message" => msg}
+    {:noreply, socket}
+  end
+
+  def handle_info({:chat_send, _, _}, socket) do
+    {:noreply, socket}
+  end
+
+  # Channels can be used in a request/response fashion
+  # by sending replies to requests from the client
+  def handle_in("ping", payload, socket) do
+    {:reply, {:ok, payload}, socket}
+  end
+
+  # It is also common to receive messages from the client and
+  # broadcast to everyone in the current topic (room:lobby).
+  def handle_in("shout", payload, socket) do
+    broadcast socket, "shout", payload
+    Shard.Identity.set_nickname(payload["name"])
+    GenServer.cast(socket.assigns.pid, {:chat_send, payload["message"]})
+    {:noreply, socket}
+  end
+
+  # Add authorization logic here as required.
+  defp authorized?(_payload) do
+    true
+  end
+end
diff --git a/shardweb/lib/shard_web/channels/user_socket.ex b/shardweb/lib/shard_web/channels/user_socket.ex
new file mode 100644
index 0000000..4fb76dc
--- /dev/null
+++ b/shardweb/lib/shard_web/channels/user_socket.ex
@@ -0,0 +1,37 @@
+defmodule ShardWeb.UserSocket do
+  use Phoenix.Socket
+
+  ## Channels
+  channel "room:*", ShardWeb.RoomChannel
+
+  ## Transports
+  transport :websocket, Phoenix.Transports.WebSocket
+  # transport :longpoll, Phoenix.Transports.LongPoll
+
+  # Socket params are passed from the client and can
+  # be used to verify and authenticate a user. After
+  # verification, you can put default assigns into
+  # the socket that will be set for all channels, ie
+  #
+  #     {:ok, assign(socket, :user_id, verified_user_id)}
+  #
+  # To deny connection, return `:error`.
+  #
+  # See `Phoenix.Token` documentation for examples in
+  # performing token verification on connect.
+  def connect(_params, socket) do
+    {:ok, socket}
+  end
+
+  # Socket id's are topics that allow you to identify all sockets for a given user:
+  #
+  #     def id(socket), do: "user_socket:#{socket.assigns.user_id}"
+  #
+  # Would allow you to broadcast a "disconnect" event and terminate
+  # all active sockets and channels for a given user:
+  #
+  #     ShardWeb.Endpoint.broadcast("user_socket:#{user.id}", "disconnect", %{})
+  #
+  # Returning `nil` makes this socket anonymous.
+  def id(_socket), do: nil
+end
-- 
cgit v1.2.3