Rate limit game creation

2 files changed, 60 insertions, 30 deletions

diff --git a/lib/chessh/ssh/client/client.ex b/lib/chessh/ssh/client/client.ex
index 67aa920..461dfbe 100644
--- a/lib/chessh/ssh/client/client.ex
+++ b/lib/chessh/ssh/client/client.ex
@@ -45,17 +45,20 @@ defmodule Chessh.SSH.Client do
           screen_state_initials: screen_state_initials
         } = state
       ) do
-    {:ok, new_screen_pid} =
-      GenServer.start_link(module, [%{screen_state_initial | client_pid: self()}])
+    case GenServer.start_link(module, [%{screen_state_initial | client_pid: self()}]) do
+      {:ok, new_screen_pid} ->
+        send(new_screen_pid, {:render, width, height})
 
-    send(new_screen_pid, {:render, width, height})
+        {:noreply,
+         %State{
+           state
+           | screen_pid: new_screen_pid,
+             screen_state_initials: [{module, screen_state_initial} | screen_state_initials]
+         }}
 
-    {:noreply,
-     %State{
-       state
-       | screen_pid: new_screen_pid,
-         screen_state_initials: [{module, screen_state_initial} | screen_state_initials]
-     }}
+      _ ->
+        {:noreply, state}
+    end
   end
 
   @impl true
diff --git a/lib/chessh/ssh/client/game/game.ex b/lib/chessh/ssh/client/game/game.ex
index 3ecd2e4..4a79d05 100644
--- a/lib/chessh/ssh/client/game/game.ex
+++ b/lib/chessh/ssh/client/game/game.ex
@@ -59,32 +59,59 @@ defmodule Chessh.SSH.Client.Game do
   end
 
   def init([
-        %State{player_session: player_session, color: color, game: nil} = state
+        %State{player_session: player_session, color: color, game: nil, client_pid: client_pid} =
+          state
         | tail
       ]) do
-    # Starting a new game
-    {:ok, %Game{} = game} =
-      Game.changeset(
-        %Game{},
-        Map.merge(
-          if(color == :light,
-            do: %{light_player_id: player_session.player_id},
-            else: %{dark_player_id: player_session.player_id}
-          ),
-          %{
-            fen: @default_fen
+    [create_game_ms, create_game_rate] =
+      Application.get_env(:chessh, RateLimits)
+      |> Keyword.take([:create_game_ms, :create_game_rate])
+      |> Keyword.values()
+
+    case Hammer.check_rate_inc(
+           :redis,
+           "player-#{state.player_session.id}-create-game-rate",
+           create_game_ms,
+           create_game_rate,
+           1
+         ) do
+      {:allow, _count} ->
+        # Starting a new game
+        {:ok, %Game{} = game} =
+          Game.changeset(
+            %Game{},
+            Map.merge(
+              if(color == :light,
+                do: %{light_player_id: player_session.player_id},
+                else: %{dark_player_id: player_session.player_id}
+              ),
+              %{
+                fen: @default_fen
+              }
+            )
+          )
+          |> Repo.insert()
+
+        init([
+          %State{
+            state
+            | game: game
           }
+          | tail
+        ])
+
+      {:deny, _limit} ->
+        send(
+          client_pid,
+          {:send_to_ssh,
+           [
+             Utils.clear_codes(),
+             "You are creating too many games, and have been rate limited. Try again later.\n"
+           ]}
         )
-      )
-      |> Repo.insert()
 
-    init([
-      %State{
-        state
-        | game: game
-      }
-      | tail
-    ])
+        {:stop, :normal, state}
+    end
   end
 
   def init([