Add region to TeslaCredentials published to kafka

DIMO-Network · Mar 4, 2024 · ccba2f2 · ccba2f2
1 parent 2a4d219
commit ccba2f2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/internal/controllers/user_integrations_controller.go b/internal/controllers/user_integrations_controller.go
@@ -1941,7 +1941,7 @@ func (udc *UserDevicesController) registerDeviceTesla(c *fiber.Ctx, logger *zero
 		udc.requestInstantOffer(userDeviceID, tokenID)
 	}
 
-	if err := udc.teslaTaskService.StartPoll(v, &integration, apiVersion); err != nil {
+	if err := udc.teslaTaskService.StartPoll(v, &integration, apiVersion, region); err != nil {
 		return err
 	}
 

diff --git a/internal/services/tesla_task_service.go b/internal/services/tesla_task_service.go
@@ -16,7 +16,7 @@ import (
 //go:generate mockgen -source tesla_task_service.go -destination mocks/tesla_task_service_mock.go
 
 type TeslaTaskService interface {
-	StartPoll(vehicle *TeslaVehicle, udai *models.UserDeviceAPIIntegration, apiVersion int) error
+	StartPoll(vehicle *TeslaVehicle, udai *models.UserDeviceAPIIntegration, apiVersion int, region string) error
 	StopPoll(udai *models.UserDeviceAPIIntegration) error
 	UnlockDoors(udai *models.UserDeviceAPIIntegration) (string, error)
 	LockDoors(udai *models.UserDeviceAPIIntegration) (string, error)
@@ -52,6 +52,7 @@ type TeslaCredentialsV2 struct {
 	Expiry        time.Time `json:"expiry"`
 	RefreshToken  string    `json:"refreshToken"`
 	Version       int       `json:"version"`
+	Region        string    `json:"region"`
 }
 
 type TeslaTask struct {
@@ -82,7 +83,7 @@ type TeslaCredentialsCloudEventV2 struct {
 	Data TeslaCredentialsV2 `json:"data"`
 }
 
-func (t *teslaTaskService) StartPoll(vehicle *TeslaVehicle, udai *models.UserDeviceAPIIntegration, version int) error {
+func (t *teslaTaskService) StartPoll(vehicle *TeslaVehicle, udai *models.UserDeviceAPIIntegration, version int, region string) error {
 	tt := TeslaTaskCloudEvent{
 		CloudEventHeaders: CloudEventHeaders{
 			ID:          ksuid.New().String(),
@@ -121,6 +122,7 @@ func (t *teslaTaskService) StartPoll(vehicle *TeslaVehicle, udai *models.UserDev
 			Expiry:        udai.AccessExpiresAt.Time,
 			RefreshToken:  udai.RefreshToken.String,
 			Version:       version,
+			Region:        region,
 		},
 	}