chore: react to new token selection when token is initally selected (#3358)

lib/pangea/toolbar/widgets/message_selection_overlay.dart

@@ -121,13 +121,6 @@ class MessageOverlayController extends State<MessageSelectionOverlay>
     WidgetsBinding.instance.addPostFrameCallback(
       (_) => widget.chatController.setSelectedEvent(event),
     );
-    newTokens = pangeaMessageEvent?.messageDisplayRepresentation?.tokens
-            ?.where((token) {
-          return token.lemma.saveVocab == true &&
-              token.vocabConstruct.uses.isEmpty &&
-              messageInUserL2;
-        }).toList() ??
-        [];
   }
 
   @override
@@ -173,6 +166,14 @@ class MessageOverlayController extends State<MessageSelectionOverlay>
               .map((token) => token.getLemmaInfo())
               .toList();
 
+      newTokens = pangeaMessageEvent?.messageDisplayRepresentation?.tokens
+              ?.where((token) {
+            return token.lemma.saveVocab == true &&
+                token.vocabConstruct.uses.isEmpty &&
+                messageInUserL2;
+          }).toList() ??
+          [];
+
       Future.wait(lemmaInfoFutures).then((resp) {
         if (mounted) {
           setState(
@@ -309,9 +310,9 @@ class MessageOverlayController extends State<MessageSelectionOverlay>
     }
 
     if (mounted) setState(() {});
-    // Future.delayed(const Duration(milliseconds: 10), () {
-    //   _showReadingAssistanceContent();
-    // });
+    if (selectedToken != null && isNewToken(selectedToken!)) {
+      _onSelectNewToken(selectedToken!);
+    }
   }
 
   void _showReadingAssistanceContent() {
@@ -570,8 +571,10 @@ class MessageOverlayController extends State<MessageSelectionOverlay>
     }
 
     updateSelectedSpan(token.text);
+  }
 
-    if (isNewToken(token)) {
+  void _onSelectNewToken(PangeaToken token) {
+    if (!isNewToken(token)) return;
     Future.delayed(const Duration(milliseconds: 1700), () {
       MatrixState.pangeaController.putAnalytics.setState(
         AnalyticsStream(
@@ -608,7 +611,6 @@ class MessageOverlayController extends State<MessageSelectionOverlay>
       }
     });
   }
-  }
 
   PracticeTarget? practiceTargetForToken(PangeaToken token) {
     if (toolbarMode.associatedActivityType == null) return null;