From 6063508f2df1a5623f113cda34dcd59a1f4b2ac9 Mon Sep 17 00:00:00 2001
From: Shane Smiskol <shane@smiskol.com>
Date: Wed, 13 Mar 2024 00:39:22 -0700
Subject: [PATCH] migrate controlsState

---
 selfdrive/test/process_replay/migration.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/selfdrive/test/process_replay/migration.py b/selfdrive/test/process_replay/migration.py
index afcf705ff94a4e..3c77dc88353378 100644
--- a/selfdrive/test/process_replay/migration.py
+++ b/selfdrive/test/process_replay/migration.py
@@ -13,6 +13,7 @@ def migrate_all(lr, old_logtime=False, manager_states=False, panda_states=False,
   msgs = migrate_carParams(msgs, old_logtime)
   msgs = migrate_gpsLocation(msgs)
   msgs = migrate_deviceState(msgs)
+  msgs = migrate_controlsState(msgs)
   if manager_states:
     msgs = migrate_managerState(msgs)
   if panda_states:
@@ -68,6 +69,22 @@ def migrate_deviceState(lr):
   return all_msgs
 
 
+def migrate_controlsState(lr):
+  all_msgs = []
+  personality = None
+  for msg in lr:
+    if msg.which() == 'longitudinalPlan':
+      personality = msg.longitudinalPlan.personalityDEPRECATED
+    # Detect old logs with personality in longitidunalPlan
+    if msg.which() == 'controlsState' and msg.controlsState.personality == 0 and personality is not None:
+      n = msg.as_builder()
+      n.controlsState.personality = personality
+      all_msgs.append(n.as_reader())
+    else:
+      all_msgs.append(msg)
+  return all_msgs
+
+
 def migrate_pandaStates(lr):
   all_msgs = []
   # TODO: safety param migration should be handled automatically