add colors to chatssh

This commit is contained in:
Elizabeth Hunt 2023-10-09 10:43:20 -06:00
parent 639b4ac9dc
commit f6fbffab85
Signed by: simponic
GPG Key ID: 52B3774857EB24B1
10 changed files with 82 additions and 52 deletions

View File

@ -12,7 +12,7 @@ config :chessh, RateLimits,
jail_attempt_threshold: 15, jail_attempt_threshold: 15,
max_concurrent_user_sessions: 5, max_concurrent_user_sessions: 5,
player_session_message_burst_ms: 500, player_session_message_burst_ms: 500,
player_session_message_burst_rate: 8, player_session_message_burst_rate: 10,
player_public_keys: 15, player_public_keys: 15,
create_game_ms: 60 * 1000, create_game_ms: 60 * 1000,
create_game_rate: 3, create_game_rate: 3,

View File

@ -8,5 +8,5 @@ config :chessh, RateLimits,
jail_timeout_ms: 5 * 60 * 1000, jail_timeout_ms: 5 * 60 * 1000,
jail_attempt_threshold: 15, jail_attempt_threshold: 15,
max_concurrent_user_sessions: 5, max_concurrent_user_sessions: 5,
player_session_message_burst_ms: 750, player_session_message_burst_ms: 500,
player_session_message_burst_rate: 8 player_session_message_burst_rate: 8

View File

@ -94,8 +94,6 @@ defmodule Chessh.SSH.Client do
def handle( def handle(
{:data, data}, {:data, data},
%State{ %State{
width: width,
height: height,
screen_pid: screen_pid, screen_pid: screen_pid,
player_session: player_session player_session: player_session
} = state } = state
@ -111,7 +109,7 @@ defmodule Chessh.SSH.Client do
{:noreply, state} {:noreply, state}
action -> action ->
send(screen_pid, {:input, width, height, action, data}) send(screen_pid, {:input, action, data})
{:noreply, state} {:noreply, state}
end end
end end

View File

@ -218,8 +218,6 @@ defmodule Chessh.SSH.Client.Game do
end end
def input( def input(
_width,
_height,
action, action,
_data, _data,
%State{ %State{

View File

@ -54,8 +54,6 @@ defmodule Chessh.SSH.Client.PreviousGame do
end end
def input( def input(
_width,
_height,
action, action,
_data, _data,
%State{ %State{

View File

@ -45,9 +45,7 @@ defmodule Chessh.SSH.Client.Game.PromotionScreen do
end end
def input( def input(
_, _action,
_,
_,
data, data,
%State{client_pid: client_pid, game_pid: game_pid, game_state: %Game.State{} = game_state} = %State{client_pid: client_pid, game_pid: game_pid, game_state: %Game.State{} = game_state} =
state state

View File

@ -35,7 +35,7 @@ defmodule Chessh.SSH.Client.MainMenu do
{"Previous Games", {"Previous Games",
{Chessh.SSH.Client.SelectPreviousGame, {Chessh.SSH.Client.SelectPreviousGame,
%Chessh.SSH.Client.SelectPaginatePoller.State{player_session: player_session}}}, %Chessh.SSH.Client.SelectPaginatePoller.State{player_session: player_session}}},
{"TrongleChat", {"ChatsSH",
{Chessh.SSH.Client.TrongleChat, {Chessh.SSH.Client.TrongleChat,
%Chessh.SSH.Client.TrongleChat.State{player_session: player_session}}} %Chessh.SSH.Client.TrongleChat.State{player_session: player_session}}}
] ]

View File

@ -27,6 +27,8 @@ defmodule Chessh.SSH.Client.SelectPaginatePoller do
options: [], options: [],
tick: 0, tick: 0,
cursor: nil, cursor: nil,
width: 0,
height: 0,
extra_info: %{} extra_info: %{}
end end
@ -112,15 +114,15 @@ defmodule Chessh.SSH.Client.SelectPaginatePoller do
{:send_to_ssh, render_state(width, height, state)} {:send_to_ssh, render_state(width, height, state)}
) )
state %State{state | width: width, height: height}
end end
def input( def input(
width,
height,
action, action,
_data, _data,
%State{ %State{
width: width,
height: height,
client_pid: client_pid, client_pid: client_pid,
options: options, options: options,
selected_option_idx: selected_option_idx selected_option_idx: selected_option_idx

View File

@ -1,8 +1,6 @@
defmodule Chessh.SSH.Client.Screen do defmodule Chessh.SSH.Client.Screen do
@callback render(width :: integer(), height :: integer(), state :: any()) :: any() @callback render(width :: integer(), height :: integer(), state :: any()) :: any()
@callback input( @callback input(
width :: integer(),
height :: integer(),
action :: any(), action :: any(),
data :: String.t(), data :: String.t(),
state :: any() state :: any()
@ -17,8 +15,8 @@ defmodule Chessh.SSH.Client.Screen do
def handle_info({:render, width, height}, state), def handle_info({:render, width, height}, state),
do: {:noreply, render(width, height, state)} do: {:noreply, render(width, height, state)}
def handle_info({:input, width, height, action, data}, state), def handle_info({:input, action, data}, state),
do: {:noreply, input(width, height, action, data, state)} do: {:noreply, input(action, data, state)}
end end
end end
end end

View File

@ -3,6 +3,16 @@ defmodule Chessh.SSH.Client.TrongleChat do
alias Chessh.{Player, Chat, Utils, Repo, PlayerSession} alias Chessh.{Player, Chat, Utils, Repo, PlayerSession}
import Ecto.Query import Ecto.Query
@colors [
IO.ANSI.light_blue(),
IO.ANSI.light_red(),
IO.ANSI.green(),
IO.ANSI.light_magenta(),
IO.ANSI.cyan(),
IO.ANSI.blue(),
IO.ANSI.yellow()
]
defmodule State do defmodule State do
defstruct client_pid: nil, defstruct client_pid: nil,
message: "", message: "",
@ -26,36 +36,28 @@ defmodule Chessh.SSH.Client.TrongleChat do
{:noreply, render(width, height, new_state)} {:noreply, render(width, height, new_state)}
end end
defp get_initial_chats() do def init([
from(c in Chat, %State{
order_by: [desc: c.id], client_pid: client_pid,
limit: 100 player_session: %PlayerSession{player_id: player_id} = player_session
) } = state
|> Repo.all() ]) do
|> Repo.preload([:chatter])
end
defp get_player(%PlayerSession{player_id: player_id}), do: Repo.get!(Player, player_id)
def init([%State{client_pid: client_pid, player_session: player_session} = state]) do
:syn.add_node_to_scopes([:chat]) :syn.add_node_to_scopes([:chat])
:ok = :syn.join(:chat, {:tronglechat}, self()) :ok = :syn.join(:chat, {:tronglechat}, self())
send(client_pid, {:send_to_ssh, Utils.clear_codes()}) send(client_pid, {:send_to_ssh, Utils.clear_codes()})
chats = get_initial_chats()
{:ok, {:ok,
%State{ %State{
state state
| chats: chats, | chats: get_initial_chats(),
player_session: %PlayerSession{player_session | player: get_player(player_session)} player_session: %PlayerSession{player_session | player: Repo.get!(Player, player_id)}
}} }}
end end
def render( def render(
_width, width,
_height, height,
%State{ %State{
client_pid: client_pid, client_pid: client_pid,
chats: chats, chats: chats,
@ -64,38 +66,39 @@ defmodule Chessh.SSH.Client.TrongleChat do
} = state } = state
) do ) do
chat_msgs = chat_msgs =
Enum.map(chats, fn %Chat{message: message, chatter: %Player{username: chat_username}} = chats
_chat -> |> Enum.slice(0, height - 1)
chat_username <> "> " <> message |> Enum.map(&format_chat/1)
end)
|> Enum.join("\r\n") |> Enum.join("\r\n")
prompt = username <> "> " <> message {prompt, prompt_len} = format_prompt(username, message)
send( send(
client_pid, client_pid,
{:send_to_ssh, {:send_to_ssh,
[ [
Utils.clear_codes(), Utils.clear_codes(),
prompt <> "\r\n" <> chat_msgs <> IO.ANSI.cursor(0, String.length(prompt)) prompt <>
"\r\n" <> chat_msgs <> IO.ANSI.cursor(0, prompt_len + 1)
]} ]}
) )
state %State{state | width: width, height: height}
end end
def input( def input(
width,
height,
action, action,
data, data,
%State{ %State{
player_session: %PlayerSession{player: player}, player_session: %PlayerSession{player: player},
chats: chats, message: message,
message: message width: width,
height: height
} = state } = state
) do ) do
appended_message = safe_char_regex = ~r/[A-Za-z0-9._~()'!*:@,;+?-]/
appended_message_state =
case action do case action do
:backspace -> :backspace ->
%State{state | message: String.slice(message, 0..-2)} %State{state | message: String.slice(message, 0..-2)}
@ -110,13 +113,48 @@ defmodule Chessh.SSH.Client.TrongleChat do
end end
_ -> _ ->
if String.match?(data, ~r/[a-zA-Z \.!-]/) do if String.match?(data, safe_char_regex) do
%State{state | message: message <> data} %State{state | message: message <> data}
else else
state state
end end
end end
render(width, height, appended_message) render(width, height, appended_message_state)
end
defp get_initial_chats() do
from(c in Chat,
order_by: [desc: c.id],
limit: 100
)
|> Repo.all()
|> Repo.preload([:chatter])
end
defp username_color(username, colors \\ @colors) do
ind =
String.to_charlist(username)
|> Enum.sum()
|> rem(length(colors))
Enum.at(colors, ind)
end
defp format_prompt(username, message) do
{
[
IO.ANSI.format_fragment([username_color(username), username, IO.ANSI.default_color()]),
"> ",
message
]
|> Enum.join(""),
String.length(username) + String.length(message) + 2
}
end
defp format_chat(%Chat{chatter: %Player{username: username}, message: message}) do
{prompt, _} = format_prompt(username, message)
prompt
end end
end end