add new types to vehicle_data proto

teslamotors · Sep 6, 2024 · db650dd · db650dd
1 parent b0116cb
commit db650dd
Show file tree

Hide file tree

Showing 16 changed files with 2,989 additions and 541 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -19,8 +19,8 @@ jobs:
 
     - name: Set up protoc
       run: |
-        wget https://github.com/protocolbuffers/protobuf/releases/download/v25.1/protoc-25.1-linux-x86_64.zip
-        unzip protoc-25.1-linux-x86_64.zip
+        wget https://github.com/protocolbuffers/protobuf/releases/download/v26.1/protoc-26.1-linux-x86_64.zip
+        unzip protoc-26.1-linux-x86_64.zip
         sudo mv bin/protoc /usr/local/bin/protoc
         sudo mv include/* /usr/local/include/
 

diff --git a/datastore/simple/transformers/payload.go b/datastore/simple/transformers/payload.go
@@ -1,6 +1,7 @@
 package transformers
 
 import (
+	"fmt"
 	"time"
 
 	logrus "github.com/teslamotors/fleet-telemetry/logger"
@@ -68,9 +69,79 @@ func transformValue(value interface{}, includeTypes bool) (interface{}, bool) {
 	case *protos.Value_ShiftStateValue:
 		outputType = "shiftStateValue"
 		outputValue = v.ShiftStateValue.String()
-	case *protos.Value_ChargingValue:
-		outputType = "chargingValue"
-		outputValue = v.ChargingValue.String()
+	case *protos.Value_LaneAssistLevelValue:
+		outputType = "laneAssistLevel"
+		outputValue = v.LaneAssistLevelValue.String()
+	case *protos.Value_ScheduledChargingModeValue:
+		outputType = "scheduledChargingMode"
+		outputValue = v.ScheduledChargingModeValue.String()
+	case *protos.Value_SentryModeStateValue:
+		outputType = "sentryModeState"
+		outputValue = v.SentryModeStateValue.String()
+	case *protos.Value_SpeedAssistLevelValue:
+		outputType = "speedAssistLevel"
+		outputValue = v.SpeedAssistLevelValue.String()
+	case *protos.Value_BmsStateValue:
+		outputType = "bmsState"
+		outputValue = v.BmsStateValue.String()
+	case *protos.Value_BuckleStatusValue:
+		outputType = "buckleStatus"
+		outputValue = v.BuckleStatusValue.String()
+	case *protos.Value_CarTypeValue:
+		outputType = "carType"
+		outputValue = v.CarTypeValue.String()
+	case *protos.Value_ChargePortValue:
+		outputType = "chargePort"
+		outputValue = v.ChargePortValue.String()
+	case *protos.Value_ChargePortLatchValue:
+		outputType = "chargePortLatch"
+		outputValue = v.ChargePortLatchValue.String()
+	case *protos.Value_CruiseStateValue:
+		outputType = "cruiseState"
+		outputValue = v.CruiseStateValue.String()
+	case *protos.Value_DoorValue:
+		outputType = "doorValue"
+		outputValue = map[string]bool{
+			"DriverFront":    v.DoorValue.DriverFront,
+			"PassengerFront": v.DoorValue.PassengerFront,
+			"DriverRear":     v.DoorValue.DriverRear,
+			"PassengerRear":  v.DoorValue.PassengerRear,
+			"TrunkFront":     v.DoorValue.TrunkFront,
+			"TrunkRear":      v.DoorValue.TrunkRear,
+		}
+	case *protos.Value_DriveInverterStateValue:
+		outputType = "driveInverterState"
+		outputValue = v.DriveInverterStateValue.String()
+	case *protos.Value_HvilStatusValue:
+		outputType = "hvilStatus"
+		outputValue = v.HvilStatusValue.String()
+	case *protos.Value_WindowStateValue:
+		outputType = "windowState"
+		outputValue = v.WindowStateValue.String()
+	case *protos.Value_SeatFoldPositionValue:
+		outputType = "seatFoldPosition"
+		outputValue = v.SeatFoldPositionValue.String()
+	case *protos.Value_TractorAirStatusValue:
+		outputType = "tractorAirStatus"
+		outputValue = v.TractorAirStatusValue.String()
+	case *protos.Value_FollowDistanceValue:
+		outputType = "followDistance"
+		outputValue = v.FollowDistanceValue.String()
+	case *protos.Value_ForwardCollisionSensitivityValue:
+		outputType = "forwardCollisionSensitivity"
+		outputValue = v.ForwardCollisionSensitivityValue.String()
+	case *protos.Value_GuestModeMobileAccessValue:
+		outputType = "guestModeMobileAccess"
+		outputValue = v.GuestModeMobileAccessValue.String()
+	case *protos.Value_TrailerAirStatusValue:
+		outputType = "trailerAirStatus"
+		outputValue = v.TrailerAirStatusValue.String()
+	case *protos.Value_TimeValue:
+		outputType = "time"
+		outputValue = fmt.Sprintf("%02d:%02d:%02d", v.TimeValue.Hour, v.TimeValue.Minute, v.TimeValue.Second)
+	case *protos.Value_DetailedChargeStateValue:
+		outputType = "detailedChargeState"
+		outputValue = v.DetailedChargeStateValue.String()
 	default:
 		return nil, false
 	}

diff --git a/datastore/simple/transformers/payload_test.go b/datastore/simple/transformers/payload_test.go
@@ -162,15 +162,6 @@ var _ = Describe("Payload", func() {
 					"shiftStateValue": "ShiftStateD",
 				},
 			),
-			Entry("ChargeState with types excluded",
-				&protos.Datum{
-					Key:   protos.Field_ChargeState,
-					Value: &protos.Value{Value: &protos.Value_ChargingValue{ChargingValue: protos.ChargingState_ChargeStateCharging}},
-				},
-				excludeTypes,
-				"ChargeState",
-				"ChargeStateCharging",
-			),
 			Entry("Invalid with types excluded",
 				&protos.Datum{
 					Key:   protos.Field_BMSState,

diff --git a/protos/python/vehicle_alert_pb2.py b/protos/python/vehicle_alert_pb2.py
diff --git a/protos/python/vehicle_data_pb2.py b/protos/python/vehicle_data_pb2.py
diff --git a/protos/python/vehicle_error_pb2.py b/protos/python/vehicle_error_pb2.py
diff --git a/protos/python/vehicle_metric_pb2.py b/protos/python/vehicle_metric_pb2.py
diff --git a/protos/ruby/vehicle_alert_pb.rb b/protos/ruby/vehicle_alert_pb.rb
diff --git a/protos/ruby/vehicle_data_pb.rb b/protos/ruby/vehicle_data_pb.rb
diff --git a/protos/ruby/vehicle_error_pb.rb b/protos/ruby/vehicle_error_pb.rb
diff --git a/protos/ruby/vehicle_metric_pb.rb b/protos/ruby/vehicle_metric_pb.rb
diff --git a/protos/vehicle_alert.pb.go b/protos/vehicle_alert.pb.go