yep

Shogi.Api/Controllers/SessionsController.cs

@@ -77,6 +77,11 @@ public class SessionsController : ControllerBase
         return Ok(await this.queryRespository.ReadSessionPlayerCount(this.User.GetShogiUserId()));
     }
 
+    /// <summary>
+    /// Fetch the session and latest board state. Also subscribe the user to socket events for this session.
+    /// </summary>
+    /// <param name="name"></param>
+    /// <returns></returns>
     [HttpGet("{name}")]
     public async Task<ActionResult<ReadSessionResponse>> GetSession(string name)
     {
@@ -113,6 +118,7 @@ public class SessionsController : ControllerBase
             session.AddPlayer2(User.GetShogiUserId());
 
             await sessionRepository.SetPlayer2(name, User.GetShogiUserId());
+            await communicationManager.BroadcastToAll(new SessionJoinedByPlayerSocketMessage());
             return this.Ok();
         }
         return this.Conflict("This game already has two players.");
@@ -144,6 +150,8 @@ public class SessionsController : ControllerBase
             return this.Conflict(e.Message);
         }
         await sessionRepository.CreateMove(sessionName, command);
+
+        // Send socket message to both players so their clients know that new board state is available.
         await communicationManager.BroadcastToPlayers(
           new PlayerHasMovedMessage
           {

Shogi.Api/Controllers/UserController.cs

@@ -53,9 +53,13 @@ public class UserController : ControllerBase
         };
     }
 
+    /// <summary>
+    /// </summary>
+    /// <param name="returnUrl">Used by cookie authentication.</param>
+    /// <returns></returns>
     [AllowAnonymous]
     [HttpGet("LoginAsGuest")]
-	public async Task<IActionResult> GuestLogin()
+    public async Task<IActionResult> GuestLogin([FromQuery] string returnUrl)
     {
         var principal = await this.claimsTransformation.CreateClaimsFromGuestPrincipal(User);
         if (principal != null)
@@ -66,6 +70,10 @@ public class UserController : ControllerBase
                 authenticationProps
             );
         }
+        if (!string.IsNullOrWhiteSpace(returnUrl))
+        {
+            return Redirect(returnUrl);
+        }
         return Ok();
     }
 

Shogi.Api/Services/SocketService.cs

@@ -40,7 +40,7 @@ namespace Shogi.Api.Services
                 context.Response.StatusCode = (int)HttpStatusCode.Unauthorized;
                 return;
             }
-      var token = Guid.Parse(context.Request.Query["token"][0]);
+            var token = Guid.Parse(context.Request.Query["token"][0] ?? throw new InvalidOperationException("Token expected during socket connection request, but was not sent."));
             var userName = tokenManager.GetUsername(token);
 
             if (string.IsNullOrEmpty(userName))
@@ -51,7 +51,8 @@ namespace Shogi.Api.Services
             var socket = await context.WebSockets.AcceptWebSocketAsync();
 
             communicationManager.Subscribe(socket, userName);
-      while (socket.State == WebSocketState.Open)
+            // TODO: I probably don't need this while-loop anymore? Perhaps unsubscribe when a disconnect is detected instead.
+            while (socket.State.HasFlag(WebSocketState.Open))
             {
                 try
                 {
@@ -82,18 +83,22 @@ namespace Shogi.Api.Services
                     logger.LogError(ex.Message);
                 }
                 communicationManager.Unsubscribe(userName);
+
+                if (!socket.State.HasFlag(WebSocketState.Closed) && !socket.State.HasFlag(WebSocketState.Aborted))
+                {
+                    try
+                    {
+                        await socket.CloseAsync(WebSocketCloseStatus.NormalClosure,
+                            "Socket closed",
+                            CancellationToken.None);
+                    }
+                    catch (Exception ex)
+                    {
+                        Console.WriteLine($"Ignored exception during socket closing. {ex.Message}");
                     }
                 }
 
-    public async Task<bool> ValidateRequestAndReplyIfInvalid<TRequest>(WebSocket socket, IValidator<TRequest> validator, TRequest request)
+            }
-    {
-      var results = validator.Validate(request);
-      if (!results.IsValid)
-      {
-        var errors = string.Join('\n', results.Errors.Select(_ => _.ErrorMessage));
-        await socket.SendTextAsync(errors);
-      }
-      return results.IsValid;
         }
     }
 }

Shogi.Contracts/Socket/PlayerHasMovedMessage.cs

@@ -14,5 +14,7 @@ public class PlayerHasMovedMessage : ISocketResponse
     public PlayerHasMovedMessage()
     {
         Action = SocketAction.PieceMoved;
+        SessionName = string.Empty;
+        PlayerName = string.Empty;
     }
 }

Shogi.Contracts/Socket/SessionCreatedSocketMessage.cs

@@ -4,10 +4,5 @@ namespace Shogi.Contracts.Socket;
 
 public class SessionCreatedSocketMessage : ISocketResponse
 {
-  public SocketAction Action { get; }
+    public SocketAction Action => SocketAction.SessionCreated;
-  
-  public SessionCreatedSocketMessage()
-  {
-    Action = SocketAction.SessionCreated;
-  }
 }

Shogi.Contracts/Socket/SessionJoinedByPlayerSocketMessage.cs

@@ -0,0 +1,9 @@
+using Shogi.Contracts.Types;
+
+namespace Shogi.Contracts.Socket
+{
+    public class SessionJoinedByPlayerSocketMessage : ISocketResponse
+    {
+        public SocketAction Action => SocketAction.SessionJoined;
+    }
+}

Shogi.UI/App.razor

@@ -3,22 +3,6 @@
     <Router AppAssembly="@typeof(App).Assembly">
         <Found Context="routeData">
             <RouteView RouteData="@routeData" DefaultLayout="@typeof(MainLayout)" />
-            @*<AuthorizeRouteView RouteData="@routeData" DefaultLayout="@typeof(MainLayout)">
-                <Authorizing>
-                    Authorizing!!
-                </Authorizing>
-                <NotAuthorized>
-                    @if (context.User.Identity?.IsAuthenticated != true)
-                    {
-                        <RedirectToLogin />
-                    }
-                    else
-                    {
-                        <p role="alert">You are not authorized to access this resource.</p>
-                    }
-                </NotAuthorized>
-            </AuthorizeRouteView>*@
-            <FocusOnNavigate RouteData="@routeData" Selector="h1" />
         </Found>
         <NotFound>
             <PageTitle>Not found</PageTitle>

Shogi.UI/Pages/Home/Account/AccountManager.cs

@@ -3,7 +3,6 @@ using Microsoft.AspNetCore.Components.Authorization;
 using Microsoft.AspNetCore.Components.WebAssembly.Authentication;
 using Shogi.UI.Pages.Home.Api;
 using Shogi.UI.Shared;
-using System.Text.Json;
 
 namespace Shogi.UI.Pages.Home.Account;
 
@@ -35,21 +34,23 @@ public class AccountManager
         this.shogiSocket = shogiSocket;
     }
 
-    private User? MyUser { get => accountState.User; set => accountState.User = value; }
+    private User? MyUser => accountState.User;
+
+    private Task SetUser(User user) => accountState.SetUser(user);
+
 
     public async Task LoginWithGuestAccount()
     {
         var response = await shogiApi.GetToken(WhichAccountPlatform.Guest);
         if (response != null)
         {
-            MyUser = new User
+            await SetUser(new User
             {
                 DisplayName = response.DisplayName,
                 Id = response.UserId,
-                OneTimeSocketToken = response.OneTimeToken,
                 WhichAccountPlatform = WhichAccountPlatform.Guest
-            };
+            });
-            await shogiSocket.OpenAsync(MyUser.Value.OneTimeSocketToken.ToString());
+            await shogiSocket.OpenAsync(response.OneTimeToken.ToString());
             await localStorage.SetAccountPlatform(WhichAccountPlatform.Guest);
         }
     }
@@ -80,11 +81,12 @@ public class AccountManager
             var response = await shogiApi.GetToken(WhichAccountPlatform.Guest);
             if (response != null)
             {
-                MyUser = new User(
+                await accountState.SetUser(new User(
                     Id: response.UserId,
                     DisplayName: response.DisplayName,
-                    WhichAccountPlatform: WhichAccountPlatform.Guest,
+                    WhichAccountPlatform: WhichAccountPlatform.Guest));
-                    OneTimeSocketToken: response.OneTimeToken);
+                await shogiSocket.OpenAsync(response.OneTimeToken.ToString());
+                return true;
             }
         }
         else if (platform == WhichAccountPlatform.Microsoft)
@@ -101,26 +103,21 @@ public class AccountManager
                 }
                 var id = state.User.Claims.Single(claim => claim.Type == "oid").Value;
                 var displayName = state.User.Identity.Name;
-                MyUser = new User(
+                await accountState.SetUser(new User(
                     Id: id,
                     DisplayName: displayName,
-                    WhichAccountPlatform: WhichAccountPlatform.Microsoft,
+                    WhichAccountPlatform: WhichAccountPlatform.Microsoft));
-                    OneTimeSocketToken: response.OneTimeToken);
+                await shogiSocket.OpenAsync(response.OneTimeToken.ToString());
-            }
-        }
-
-        if (MyUser != null)
-        {
-            await shogiSocket.OpenAsync(MyUser.Value.OneTimeSocketToken.ToString());
                 return true;
             }
+        }
 
         return false;
     }
 
     public async Task LogoutAsync()
     {
-        MyUser = null;
+        await accountState.SetUser(null);
         var platform = await localStorage.GetAccountPlatform();
         await localStorage.DeleteAccountPlatform();
 

Shogi.UI/Pages/Home/Account/AccountState.cs

@@ -1,28 +1,27 @@
-namespace Shogi.UI.Pages.Home.Account;
+using static Shogi.UI.Shared.Events;
+
+namespace Shogi.UI.Pages.Home.Account;
 
 public class AccountState
 {
-    public event EventHandler<LoginEventArgs>? LoginChangedEvent;
+    public event AsyncEventHandler<LoginEventArgs>? LoginChangedEvent;
 
-    private User? user;
+    public User? User { get; private set; }
-    public User? User
+
+    public Task SetUser(User? user)
     {
-        get => user;
+        User = user;
-        set
+        return EmitLoginChangedEvent();
-        {
-            if (user != value)
-            {
-                user = value;
-                EmitLoginChangedEvent();
-            }
-        }
     }
 
-    private void EmitLoginChangedEvent()
+    private async Task EmitLoginChangedEvent()
     {
-        LoginChangedEvent?.Invoke(this, new LoginEventArgs
+        if (LoginChangedEvent is not null)
+        {
+            await LoginChangedEvent.Invoke(new LoginEventArgs
             {
                 User = User
             });
         }
     }
+}

Shogi.UI/Pages/Home/Account/User.cs

@@ -3,8 +3,7 @@
     public readonly record struct User(
         string Id,
         string DisplayName,
-        WhichAccountPlatform WhichAccountPlatform,
+        WhichAccountPlatform WhichAccountPlatform)
-        Guid OneTimeSocketToken)
     {
     }
 }

Shogi.UI/Pages/Home/Api/IShogiApi.cs

@@ -12,6 +12,6 @@ public interface IShogiApi
 	Task<CreateTokenResponse?> GetToken(WhichAccountPlatform whichAccountPlatform);
 	Task GuestLogout();
 	Task Move(string sessionName, MovePieceCommand move);
-    Task<HttpStatusCode> PatchJoinGame(string name);
+    Task<HttpResponseMessage> PatchJoinGame(string name);
     Task<HttpStatusCode> PostSession(string name, bool isPrivate);
 }

Shogi.UI/Pages/Home/Api/ShogiApi.cs

@@ -65,8 +65,16 @@ namespace Shogi.UI.Pages.Home.Api
             var httpClient = whichAccountPlatform == WhichAccountPlatform.Microsoft
                 ? clientFactory.CreateClient(MsalClientName)
                 : clientFactory.CreateClient(GuestClientName);
-            var response = await httpClient.GetFromJsonAsync<CreateTokenResponse>(RelativeUri("User/Token"), serializerOptions);
+            var response = await httpClient.GetAsync(RelativeUri("User/Token"));
-            return response;
+            if (response.IsSuccessStatusCode)
+            {
+                var content = await response.Content.ReadAsStringAsync();
+                if (!string.IsNullOrEmpty(content))
+                {
+                    return await response.Content.ReadFromJsonAsync<CreateTokenResponse>(serializerOptions);
+                }
+            }
+            return null;
         }
 
         public async Task Move(string sessionName, MovePieceCommand command)
@@ -83,10 +91,9 @@ namespace Shogi.UI.Pages.Home.Api
             return response.StatusCode;
         }
 
-        public async Task<HttpStatusCode> PatchJoinGame(string name)
+        public Task<HttpResponseMessage> PatchJoinGame(string name)
         {
-            var response = await HttpClient.PatchAsync(RelativeUri($"Sessions/{name}/Join"), null);
+            return HttpClient.PatchAsync(RelativeUri($"Sessions/{name}/Join"), null);
-            return response.StatusCode;
         }
 
         private static Uri RelativeUri(string path) => new Uri(path, UriKind.Relative);

Shogi.UI/Pages/Home/GameBoard/EmptyGameBoard.razor

@@ -3,9 +3,4 @@
 <GameBoardPresentation Perspective="WhichPlayer.Player1" />
 
 @code {
-    protected override void OnInitialized()
-    {
-        base.OnInitialized();
-        Console.WriteLine("Empty Game Board.");
-    }
 }

Shogi.UI/Pages/Home/GameBoard/GameBoard.razor

@@ -1,9 +1,11 @@
 @using Shogi.Contracts.Api
+@using Shogi.Contracts.Socket;
 @using Shogi.Contracts.Types;
 @using System.Text.RegularExpressions;
 @inject IShogiApi ShogiApi
 @inject AccountState Account;
 @inject PromotePrompt PromotePrompt;
+@inject ShogiSocket ShogiSocket;
 
 @if (session == null)
 {
@@ -11,11 +13,11 @@
 }
 else if (isSpectating)
 {
-    <SpectatorGameBoard Session="session" />
+    <SpectatorGameBoard Session="session" OnRefetchSession="RefetchSession" />
 }
 else
 {
-    <SeatedGameBoard Perspective="perspective" Session="session" OnRefetchSession="RefetchSession" />
+    <SeatedGameBoard Perspective="perspective" Session="session" />
 }
 
 
@@ -27,6 +29,13 @@ else
     private WhichPlayer perspective;
     private bool isSpectating;
 
+    protected override void OnInitialized()
+    {
+        base.OnInitialized();
+        ShogiSocket.OnPlayerMoved += OnPlayerMoved_RefetchSession;
+        ShogiSocket.OnSessionJoined += 
+    }
+
     protected override async Task OnParametersSetAsync()
     {
         await RefetchSession();
@@ -44,9 +53,18 @@ else
                 this.perspective = accountId == session.Player2 ? WhichPlayer.Player2 : WhichPlayer.Player1;
                 this.isSpectating = !(accountId == this.session.Player1 || accountId == this.session.Player2);
                 Console.WriteLine($"IsSpectating - {isSpectating}. AccountId - {accountId}. Player1 - {this.session.Player1}. Player2 - {this.session.Player2}");
+
             }
+            StateHasChanged();
         }
     }
 
-
+    Task OnPlayerMoved_RefetchSession(PlayerHasMovedMessage args)
+    {
+        if (args.SessionName == SessionName)
+        {
+            return RefetchSession();
+        }
+        return Task.CompletedTask;
+    }
 }

Shogi.UI/Pages/Home/GameBoard/GameBoardPresentation.razor

@@ -68,10 +68,10 @@
         <aside class="side-board">
             <div class="player-area">
                 <div class="hand">
-                    @if (OpponentHand.Any())
+                    @if (opponentHand.Any())
                     {
 
-                        @foreach (var piece in OpponentHand)
+                        @foreach (var piece in opponentHand)
                         {
                             <div class="tile">
                                 <GamePiece Piece="piece" Perspective="Perspective" />
@@ -86,6 +86,8 @@
             </div>
 
             <div class="spacer place-self-center">
+                <p>@opponentName</p>
+                <p>@userName</p>
             </div>
 
             <div class="player-area">
@@ -93,15 +95,15 @@
                 {
                     <div class="place-self-center">
                         <p>Seat is Empty</p>
-                        <button @onclick="OnClickJoinGameInternal()">Join Game</button>
+                        <button @onclick="OnClickJoinGameInternal">Join Game</button>
                     </div>
                 }
                 else
                 {
                     <div class="hand">
-                        @if (UserHand.Any())
+                        @if (userHand.Any())
                         {
-                            @foreach (var piece in UserHand)
+                            @foreach (var piece in userHand)
                             {
                                 <div class="title" @onclick="OnClickHandInternal(piece)">
                                     <GamePiece Piece="piece" Perspective="Perspective" />
@@ -121,6 +123,11 @@
 </article>
 
 @code {
+    static readonly string[] Files = new[] { "A", "B", "C", "D", "E", "F", "G", "H", "I" };
+
+    /// <summary>
+    /// When true, an icon is displayed indicating that the user is spectating.
+    /// </summary>
     [Parameter] public bool IsSpectating { get; set; } = false;
     [Parameter] public WhichPlayer Perspective { get; set; }
     [Parameter] public Session? Session { get; set; }
@@ -130,32 +137,48 @@
     [Parameter] public Func<Piece, Task>? OnClickHand { get; set; }
     [Parameter] public Func<Task>? OnClickJoinGame { get; set; }
 
-    static readonly string[] Files = new[] { "A", "B", "C", "D", "E", "F", "G", "H", "I" };
+    private IReadOnlyCollection<Piece> opponentHand;
+    private IReadOnlyCollection<Piece> userHand;
+    private string? userName;
+    private string? opponentName;
 
-    private IReadOnlyCollection<Piece> OpponentHand
+    public GameBoardPresentation()
     {
-        get
+        opponentHand = Array.Empty<Piece>();
-        {
+        userHand = Array.Empty<Piece>();
-            if (this.Session == null) return Array.Empty<Piece>();
+        userName = string.Empty;
+        opponentName = string.Empty;
+    }
 
-            return Perspective == WhichPlayer.Player1
+    protected override void OnParametersSet()
-                ? this.Session.BoardState.Player1Hand
+    {
-                : this.Session.BoardState.Player2Hand;
+        Console.WriteLine("Params changed. SelectedPosition = {0}", SelectedPosition);
-        }
+        base.OnParametersSet();
-    }
+        if (Session == null)
-    IReadOnlyCollection<Piece> UserHand
+        {
-    {
+            opponentHand = Array.Empty<Piece>();
-        get
+            userHand = Array.Empty<Piece>();
-        {
+            userName = string.Empty;
-            if (this.Session == null) return Array.Empty<Piece>();
+            opponentName = string.Empty;
-
+        }
-            return Perspective == WhichPlayer.Player1
+        else
+        {
+            opponentHand = Perspective == WhichPlayer.Player1
+                ? this.Session.BoardState.Player2Hand
+                : this.Session.BoardState.Player1Hand;
+            userHand = Perspective == WhichPlayer.Player1
                 ? this.Session.BoardState.Player1Hand
                 : this.Session.BoardState.Player2Hand;
+            userName = Perspective == WhichPlayer.Player1
+                ? this.Session.Player1
+                : this.Session.Player2;
+            opponentName = Perspective == WhichPlayer.Player1
+                ? this.Session.Player2
+                : this.Session.Player1;
         }
     }
 
     private Action OnClickTileInternal(Piece? piece, string position) => () => OnClickTile?.Invoke(piece, position);
     private Action OnClickHandInternal(Piece piece) => () => OnClickHand?.Invoke(piece);
-    private Action OnClickJoinGameInternal() => () => OnClickJoinGame?.Invoke();
+    private void OnClickJoinGameInternal() => OnClickJoinGame?.Invoke();
 }

Shogi.UI/Pages/Home/GameBoard/SeatedGameBoard.razor

@@ -9,18 +9,26 @@
                        Perspective="Perspective"
                        OnClickHand="OnClickHand"
                        OnClickTile="OnClickTile"
-                       OnClickJoinGame="OnClickJoinGame" />
+                       SelectedPosition="@selectedBoardPosition" />
 
 @code {
     [Parameter, EditorRequired]
     public WhichPlayer Perspective { get; set; }
     [Parameter, EditorRequired]
     public Session Session { get; set; }
-    [Parameter] public Func<Task>? OnRefetchSession { get; set; }
     private bool IsMyTurn => Session?.BoardState.WhoseTurn == Perspective;
     private string? selectedBoardPosition;
     private WhichPiece? selectedPieceFromHand;
 
+    protected override void OnParametersSet()
+    {
+        base.OnParametersSet();
+        if (Session == null)
+        {
+            throw new ArgumentException($"{nameof(Session)} cannot be null.", nameof(Session));
+        }
+    }
+
     bool ShouldPromptForPromotion(string position)
     {
         if (Perspective == WhichPlayer.Player1 && Regex.IsMatch(position, ".[7-9]"))
@@ -36,12 +44,16 @@
 
     async Task OnClickTile(Piece? piece, string position)
     {
+        Console.WriteLine("Is my turn?");
+        Console.WriteLine(true);
         if (!IsMyTurn) return;
 
         if (selectedBoardPosition == null || piece?.Owner == Perspective)
         {
             // Select a position.
+            Console.WriteLine("Position {0}", position);
             selectedBoardPosition = position;
+            StateHasChanged();
             return;
         }
         if (selectedBoardPosition == position)
@@ -77,16 +89,4 @@
         selectedPieceFromHand = piece.WhichPiece;
         await Task.CompletedTask;
     }
-
-    async Task OnClickJoinGame()
-    {
-        if (Session != null && OnRefetchSession != null)
-        {
-            var status = await ShogiApi.PatchJoinGame(Session.SessionName);
-            if (status == HttpStatusCode.OK)
-            {
-                await OnRefetchSession.Invoke();
-            }
-        }
-    }
 }

Shogi.UI/Pages/Home/GameBoard/SpectatorGameBoard.razor

@@ -1,7 +1,27 @@
 @using Contracts.Types;
+@using System.Net;
+@inject IShogiApi ShogiApi;
 
-<GameBoardPresentation IsSpectating="true" Perspective="WhichPlayer.Player1" Session="Session" />
+<GameBoardPresentation IsSpectating="true"
+                       Perspective="WhichPlayer.Player1"
+                       Session="Session"
+                       OnClickJoinGame="OnClickJoinGame" />
 
 @code {
     [Parameter] public Session Session { get; set; }
+
+    protected override void OnParametersSet()
+    {
+        base.OnParametersSet();
+        if (Session == null)
+        {
+            throw new ArgumentException($"{nameof(Session)} cannot be null.", nameof(Session));
+        }
+    }
+
+    async Task OnClickJoinGame()
+    {
+        var response = await ShogiApi.PatchJoinGame(Session.SessionName);
+        response.EnsureSuccessStatusCode();
+    }
 }

Shogi.UI/Pages/Home/GameBrowser.razor

@@ -1,7 +1,10 @@
-@using Shogi.Contracts.Types;
+@implements IDisposable;
+
+@using Shogi.Contracts.Types;
 @using System.ComponentModel.DataAnnotations;
 @using System.Net;
 @using System.Text.Json;
+
 @inject IShogiApi ShogiApi;
 @inject ShogiSocket ShogiSocket;
 @inject AccountState Account;
@@ -76,7 +79,6 @@
                         The name you chose is taken; choose another.
                     </div>
                 }
-
             </EditForm>
         </div>
     </div>
@@ -92,17 +94,12 @@
     private SessionMetadata? activeSession;
     private HttpStatusCode? createSessionStatusCode;
 
-    protected override async Task OnInitializedAsync()
+    protected override void OnInitialized()
     {
-        ShogiSocket.OnCreateGameMessage += async (sender, message) => await FetchSessions();
+        base.OnInitialized();
-        Account.LoginChangedEvent += async (sender, message) =>
+        ShogiSocket.OnSessionCreated += FetchSessions;
-        {
+        ShogiSocket.OnSessionJoined += FetchSessions;
-            Console.WriteLine($"LoginEvent. Message={JsonSerializer.Serialize(message)}.");
+        Account.LoginChangedEvent += LoginChangedEvent_FetchSessions;
-            if (message.User != null)
-            {
-                await FetchSessions();
-            }
-        };
     }
 
     string ActiveCss(SessionMetadata s) => s == activeSession ? "active" : string.Empty;
@@ -112,6 +109,14 @@
         activeSession = s;
         ActiveSessionChanged?.Invoke(s);
     }
+    Task LoginChangedEvent_FetchSessions(LoginEventArgs args)
+    {
+        if (args.User != null)
+        {
+            return FetchSessions();
+        }
+        return Task.CompletedTask;
+    }
 
     async Task FetchSessions()
     {
@@ -129,6 +134,13 @@
         createSessionStatusCode = await ShogiApi.PostSession(createForm.Name, createForm.IsPrivate);
     }
 
+    public void Dispose()
+    {
+        ShogiSocket.OnSessionCreated -= FetchSessions;
+        ShogiSocket.OnSessionJoined -= FetchSessions;
+        Account.LoginChangedEvent -= LoginChangedEvent_FetchSessions;
+    }
+
     private class CreateForm
     {
         [Required]

Shogi.UI/Pages/Home/Home.razor

@@ -26,9 +26,16 @@
 </main>
 
 @code {
-    bool welcomeModalIsVisible = false;
+    private bool welcomeModalIsVisible;
-    string activeSessionName = string.Empty;
+    private string activeSessionName;
-    ClientWebSocket socket = new ClientWebSocket();
+    private ClientWebSocket socket;
+    
+    public Home()
+    {
+        welcomeModalIsVisible = false;
+        activeSessionName = string.Empty;
+        socket = new ClientWebSocket();
+    }
 
     protected override async Task OnInitializedAsync()
     {
@@ -40,10 +47,11 @@
         }
     }
 
-    private void OnLoginChanged(object? sender, LoginEventArgs args)
+    private Task OnLoginChanged(LoginEventArgs args)
     {
         welcomeModalIsVisible = args.User == null;
         StateHasChanged();
+        return Task.CompletedTask;
     }
     private void OnChangeSession(SessionMetadata s)
     {

Shogi.UI/Pages/Home/PageHeader.razor

@@ -21,11 +21,12 @@
         Account.LoginChangedEvent += OnLoginChange;
     }
 
-    private void OnLoginChange(object? sender, LoginEventArgs args)
+    private Task OnLoginChange(LoginEventArgs args)
     {
         if (args == null)
             throw new ArgumentException(nameof(args));
         user = args.User;
         StateHasChanged();
+        return Task.CompletedTask;
     }
 }

Shogi.UI/Program.cs

@@ -21,7 +21,13 @@ static void ConfigureDependencies(IServiceCollection services, IConfiguration co
 	 * Why two HTTP clients?
 	 * See qhttps://docs.microsoft.com/en-us/aspnet/core/blazor/security/webassembly/additional-scenarios?source=recommendations&view=aspnetcore-6.0#unauthenticated-or-unauthorized-web-api-requests-in-an-app-with-a-secure-default-client
 	 */
-	var shogiApiUrl = new Uri(configuration["ShogiApiUrl"], UriKind.Absolute);
+    var baseUrl = configuration["ShogiApiUrl"];
+    if (string.IsNullOrWhiteSpace(baseUrl))
+    {
+        throw new InvalidOperationException("ShogiApiUrl configuration is missing.");
+    }
+
+    var shogiApiUrl = new Uri(baseUrl, UriKind.Absolute);
     services
         .AddHttpClient(ShogiApi.MsalClientName, client => client.BaseAddress = shogiApiUrl)
         .AddHttpMessageHandler<MsalMessageHandler>();

Shogi.UI/Shared/Events.cs

@@ -0,0 +1,8 @@
+namespace Shogi.UI.Shared
+{
+    public static class Events
+    {
+        public delegate Task AsyncEventHandler();
+        public delegate Task AsyncEventHandler<TArgs>(TArgs args);
+    }
+}

Shogi.UI/Shared/LoginDisplay.razor

@@ -1,24 +0,0 @@
-@*@using Microsoft.AspNetCore.Components.Authorization
-@using Microsoft.AspNetCore.Components.WebAssembly.Authentication*@
-
-@*@inject NavigationManager Navigation
-@inject SignOutSessionStateManager SignOutManager*@
-
-@*<AuthorizeView>
-    <Authorized>
-        Hello, @context.User.Identity?.Name!
-        <button class="nav-link btn btn-link" @onclick="BeginSignOut">Log out</button>
-    </Authorized>
-    <NotAuthorized>
-        <a href="authentication/login">Log in</a>
-    </NotAuthorized>
-</AuthorizeView>*@
-
-@code{
-    // https://docs.microsoft.com/en-us/aspnet/core/blazor/security/webassembly/additional-scenarios?view=aspnetcore-6.0#customize-the-authentication-user-interface
-    //private async Task BeginSignOut(MouseEventArgs args)
-    //{
-    //    await SignOutManager.SetSignOutState();
-    //    Navigation.NavigateTo("authentication/logout");
-    //}
-}

Shogi.UI/Shared/MyNotAuthorized.razor

@@ -1,5 +0,0 @@
-<h3>MyNotAuthorized</h3>
-
-@code {
-
-}

Shogi.UI/Shared/RedirectToLogin.razor

@@ -1,14 +0,0 @@
-@inject NavigationManager Navigation
-@inject ModalService ModalService
-@inject AccountManager ShogiService
-
-@*<button @onclick="() => ShogiService.ConnectAsync(asGuest: true)">Guest Login</button>*@
-<div>Not implemented!</div>
-
-@code {
-    protected override void OnInitialized()
-    {
-        //ModalService.ShowLoginModal();
-        Navigation.NavigateTo($"authentication/login?returnUrl={Uri.EscapeDataString(Navigation.Uri)}");
-    }
-}

Shogi.UI/Shared/ShogiSocket.cs

@@ -4,12 +4,15 @@ using Shogi.Contracts.Types;
 using System.Buffers;
 using System.Net.WebSockets;
 using System.Text.Json;
+using static Shogi.UI.Shared.Events;
 
 namespace Shogi.UI.Shared;
 
 public class ShogiSocket : IDisposable
 {
-    public event EventHandler<SessionCreatedSocketMessage>? OnCreateGameMessage;
+    public event AsyncEventHandler? OnSessionCreated;
+    public event AsyncEventHandler? OnSessionJoined;
+    public event AsyncEventHandler<PlayerHasMovedMessage>? OnPlayerMoved;
 
     private readonly ClientWebSocket socket;
     private readonly JsonSerializerOptions serializerOptions;
@@ -37,8 +40,11 @@ public class ShogiSocket : IDisposable
         await socket.ConnectAsync(this.uriBuilder.Uri, cancelToken.Token);
         Console.WriteLine("Socket Connected");
         // Fire and forget! I'm way too lazy to write my own javascript interop to a web worker. Nooo thanks.
-        var listening = Listen().ContinueWith(antecedent =>
+        _ = Listen().ContinueWith(async antecedent =>
         {
+            Console.WriteLine($"Socket fault. {antecedent.Exception}");
+            this.cancelToken.Cancel();
+            await this.socket.CloseAsync(WebSocketCloseStatus.NormalClosure, "Page was probably closed or refresh.", CancellationToken.None);
             if (antecedent.Exception != null)
             {
                 throw antecedent.Exception;
@@ -58,16 +64,33 @@ public class ShogiSocket : IDisposable
               .GetProperty(nameof(ISocketResponse.Action))
               .Deserialize<SocketAction>();
 
+            Console.WriteLine($"Socket action: {action}");
             switch (action)
             {
                 case SocketAction.SessionCreated:
-                    Console.WriteLine("Session created event.");
+                    if (this.OnSessionCreated is not null)
-                    this.OnCreateGameMessage?.Invoke(this, JsonSerializer.Deserialize<SessionCreatedSocketMessage>(memory, this.serializerOptions)!);
+                    {
+                        await this.OnSessionCreated();
+                    }
+                    break;
+                case SocketAction.SessionJoined:
+                    if (this.OnSessionJoined is not null)
+                    {
+                        await this.OnSessionJoined();
+                    }
+                    break;
+                case SocketAction.PieceMoved:
+                    if (this.OnPlayerMoved is not null)
+                    {
+                        var args = JsonSerializer.Deserialize<PlayerHasMovedMessage>(memory[..result.Count], serializerOptions);
+                        await this.OnPlayerMoved(args!);
+                    }
                     break;
                 default:
-                    break;
+                    throw new NotImplementedException($"Socket message for action:{action} is not implemented.");
             }
         }
+        await socket.CloseAsync(WebSocketCloseStatus.NormalClosure, "Socket closed because cancellation token was cancelled.", CancellationToken.None);
         if (!cancelToken.IsCancellationRequested)
         {
             throw new InvalidOperationException("Stopped socket listening without cancelling.");