SM-5 - persist protocol (#212)

* persist protocol * Fix comment --------- Co-authored-by: ricardo <[email protected]>
DIMO-Network · Oct 11, 2023 · 13e4173 · 13e4173
1 parent bf14809
commit 13e4173
Showing 1 changed file with 39 additions and 1 deletion.
diff --git a/internal/services/fingerprint/consumer.go b/internal/services/fingerprint/consumer.go
@@ -5,6 +5,7 @@ import (
 	"database/sql"
 	"encoding/json"
 	"fmt"
+	"github.com/DIMO-Network/devices-api/internal/services"
 	"regexp"
 	"strings"
 	"time"
@@ -122,6 +123,32 @@ func (c *Consumer) HandleDeviceFingerprint(ctx context.Context, event *Event) er
 		return err
 	}
 
+	// Save Protocol
+	if ad.R.VehicleToken.R.UserDevice != nil {
+		md := services.UserDeviceMetadata{}
+		if err = ad.R.VehicleToken.R.UserDevice.Metadata.Unmarshal(&md); err != nil {
+			c.logger.Error().Msgf("Could not unmarshal userdevice metadata for device: %s", ad.R.VehicleToken.R.UserDevice.ID)
+			return err
+		}
+
+		if md.CANProtocol == nil {
+			protocol, err := ExtractProtocol(event.Data)
+			if err != nil {
+				c.logger.Error().Err(err)
+			}
+
+			if err == nil && protocol != nil {
+				md.CANProtocol = protocol
+			}
+		}
+
+		err = ad.R.VehicleToken.R.UserDevice.Metadata.Marshal(&md)
+		if err != nil {
+			c.logger.Error().Msgf("could not marshal userdevice metadata for device: %s", ad.R.VehicleToken.R.UserDevice.ID)
+			return err
+		}
+	}
+
 	c.logger.Info().Str("device-addr", event.Subject).Msg("issued vin credential")
 
 	return nil
@@ -192,7 +219,6 @@ func NumToWeekEnd(n int) time.Time {
 }
 
 var ErrNoVIN = errors.New("no VIN field")
-
 var basicVINExp = regexp.MustCompile(`^[A-Z0-9]{17}$`)
 
 // ExtractVIN extracts the vin field from a status update's data object.
@@ -221,3 +247,15 @@ func ExtractVIN(data []byte) (string, error) {
 
 	return vin, nil
 }
+
+func ExtractProtocol(data []byte) (*string, error) {
+	partialData := new(struct {
+		Protocol *string `json:"protocol"`
+	})
+
+	if err := json.Unmarshal(data, partialData); err != nil {
+		return nil, fmt.Errorf("failed parsing data field: %w", err)
+	}
+
+	return partialData.Protocol, nil
+}