Fix

drasticactions · Nov 22, 2023 · 00ba744 · 00ba744
1 parent 949f067
commit 00ba744
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 22 deletions.
diff --git a/src/FishyFlip/ATProtocol.cs b/src/FishyFlip/ATProtocol.cs
@@ -230,6 +230,32 @@ internal void OnSubscribedRepoMessageInternal(SubscribedRepoEventArgs e)
     /// <param name="session"><see cref="Session"/>.</param>
     internal void OnUserLoggedIn(Session session)
     {
+        if (this.options.UseServiceEndpointUponLogin)
+        {
+            var logger = this.options?.Logger;
+            var serviceUrl = session.DidDoc?.Service?.FirstOrDefault()?.ServiceEndpoint;
+            if (string.IsNullOrEmpty(serviceUrl))
+            {
+                logger?.LogWarning($"UseServiceEndpointUponLogin enabled, but session missing Service Endpoint.");
+            }
+            else
+            {
+                var result = Uri.TryCreate(serviceUrl, UriKind.Absolute, out Uri? uriResult);
+                if (!result || uriResult is null)
+                {
+                    logger?.LogWarning($"UseServiceEndpointUponLogin enabled, but session missing Service Endpoint.");
+                }
+                else
+                {
+                    this.Options.Url = uriResult;
+                    logger?.LogInformation($"UseServiceEndpointUponLogin enabled, switching to {uriResult}.");
+                    this.Options.Session = session;
+                    this.UpdateOptions(this.Options);
+                    return;
+                }
+            }
+        }
+
         if (this.sessionManager is null)
         {
             this.sessionManager = new SessionManager(this);

diff --git a/src/FishyFlip/SessionManager.cs b/src/FishyFlip/SessionManager.cs
@@ -93,34 +93,13 @@ internal void SetSession(Session session)
         this.UpdateBearerToken(session);
 
         this.logger?.LogDebug($"Session set, {session.Did}");
+
         if (!this.protocol.Options.AutoRenewSession)
         {
             this.logger?.LogDebug("AutoRenewSession is disabled.");
             return;
         }
 
-        if (this.protocol.Options.UseServiceEndpointUponLogin)
-        {
-            var serviceUrl = session.DidDoc?.Service?.FirstOrDefault()?.ServiceEndpoint;
-            if (!string.IsNullOrEmpty(serviceUrl))
-            {
-                this.logger?.LogWarning($"UseServiceEndpointUponLogin enabled, but session missing Service Endpoint.");
-            }
-            else
-            {
-                var result = Uri.TryCreate(serviceUrl, UriKind.Absolute, out Uri? uriResult);
-                if (!result || uriResult is null)
-                {
-                    this.logger?.LogWarning($"UseServiceEndpointUponLogin enabled, but session missing Service Endpoint.");
-                }
-                else
-                {
-                    this.protocol.Options.Url = uriResult;
-                    this.logger?.LogInformation($"UseServiceEndpointUponLogin enabled, switching to {uriResult}.");
-                }
-            }
-        }
-
         this.logger?.LogDebug("AutoRenewSession is enabled.");
         this.timer ??= new System.Timers.Timer();