Implement per-bot rendering

Author: VirxEC

RLBotCS/Server/FlatBuffersSession.cs

@@ -25,13 +25,15 @@ public record DistributeBallPrediction(BallPredictionT BallPrediction) : Session
 
     public record DistributeGameState(GamePacketT GameState) : SessionMessage;
 
-    public record RendersAllowed(DebugRendering Allowed) : SessionMessage;
+    public record RendersAllowed(bool Allowed) : SessionMessage;
 
     public record StateSettingAllowed(bool Allowed) : SessionMessage;
 
     public record MatchComm(MatchCommT Message) : SessionMessage;
 
     public record StopMatch(bool Force) : SessionMessage;
+
+    public record UpdateRendering(RenderingStatus Status) : SessionMessage;
 }
 
 class FlatBuffersSession
@@ -63,7 +65,6 @@ class FlatBuffersSession
 
     private bool _stateSettingIsEnabled;
     private bool _renderingIsEnabled;
-    private DebugRendering _globalRenderingIsEnabled;
 
     private string _agentId = string.Empty;
     private uint _team = Team.Other;
@@ -96,7 +97,6 @@ bool stateSettingIsEnabled
             DebugRendering.OnByDefault => true,
             _ => false,
         };
-        _globalRenderingIsEnabled = renderingIsEnabled;
         _stateSettingIsEnabled = stateSettingIsEnabled;
 
         NetworkStream stream = _client.GetStream();
@@ -381,17 +381,7 @@ private async Task HandleInternalMessages()
 
                     break;
                 case SessionMessage.RendersAllowed m:
-                    _globalRenderingIsEnabled = m.Allowed;
-                    switch (_globalRenderingIsEnabled)
-                    {
-                        case DebugRendering.OnByDefault:
-                            _renderingIsEnabled = true;
-                            break;
-                        case DebugRendering.OffByDefault:
-                        case DebugRendering.AlwaysOff:
-                            _renderingIsEnabled = false;
-                            break;
-                    }
+                    _renderingIsEnabled = m.Allowed;
                     break;
                 case SessionMessage.StateSettingAllowed m:
                     _stateSettingIsEnabled = m.Allowed;
@@ -418,6 +408,19 @@ private async Task HandleInternalMessages()
                     when m.Force || (_connectionEstablished && _closeBetweenMatches):
                     _sessionForceClosed = m.Force;
                     return;
+                case SessionMessage.UpdateRendering m
+                    when (m.Status.IsBot && (_team == Team.Blue || _team == Team.Orange))
+                        || (!m.Status.IsBot && _team == Team.Scripts):
+
+                    foreach (var player in _playerIdPairs)
+                    {
+                        if (player.Index == m.Status.Index)
+                        {
+                            _renderingIsEnabled = m.Status.Status;
+                            break;
+                        }
+                    }
+                    break;
             }
     }
 

RLBotCS/Server/ServerMessage/StartMatch.cs

@@ -1,4 +1,4 @@
-using System.Diagnostics;
+using System.Diagnostics;
 using RLBot.Flat;
 using RLBotCS.ManagerTools;
 using RLBotCS.Server.BridgeMessage;
@@ -27,12 +27,16 @@ public ServerAction Execute(ServerContext context)
 
         BallPredictor.UpdateMode(MatchConfig);
 
+        bool defaultRendering = context.RenderingIsEnabled switch
+        {
+            DebugRendering.OnByDefault => true,
+            _ => false,
+        };
+
         // update all sessions with the new rendering and state setting settings
         foreach (var (writer, _) in context.Sessions.Values)
         {
-            SessionMessage render = new SessionMessage.RendersAllowed(
-                context.RenderingIsEnabled
-            );
+            SessionMessage render = new SessionMessage.RendersAllowed(defaultRendering);
             writer.TryWrite(render);
 
             SessionMessage stateSetting = new SessionMessage.StateSettingAllowed(

RLBotCS/Server/ServerMessage/UpdateRendering.cs

@@ -0,0 +1,24 @@
+using RLBot.Flat;
+
+namespace RLBotCS.Server.ServerMessage;
+
+record UpdateRendering(RenderingStatus Status) : IServerMessage
+{
+    public ServerAction Execute(ServerContext context)
+    {
+        // Ignore all messages if rendering is force-disabled
+        if (context.RenderingIsEnabled == DebugRendering.AlwaysOff)
+            return ServerAction.Continue;
+
+        SessionMessage.UpdateRendering message = new(Status);
+
+        // Distribute to all sessions;
+        // they will figure out on their own if rendering should be enable/disabled
+        foreach (var (id, (writer, _)) in context.Sessions)
+        {
+            writer.TryWrite(message);
+        }
+
+        return ServerAction.Continue;
+    }
+}