From 3497a2fde64c73c7e62e483d96343e7c6b758dc0 Mon Sep 17 00:00:00 2001
From: James Reategui <JamesReate@users.noreply.github.com>
Date: Mon, 23 Jun 2025 14:40:30 -0400
Subject: [PATCH] add powertrain type to vin profile using existing logic

---
 internal/api/api.go                      |  2 +-
 internal/core/queries/get_vin_profile.go | 43 +++++++++++++++++-------
 2 files changed, 32 insertions(+), 13 deletions(-)

diff --git a/internal/api/api.go b/internal/api/api.go
index 12932663..72e7f4fc 100644
--- a/internal/api/api.go
+++ b/internal/api/api.go
@@ -122,7 +122,7 @@ func Run(ctx context.Context, logger zerolog.Logger, settings *config.Settings,
 		mediator.WithHandler(&queries.GetAllDeviceDefinitionByAutocompleteQuery{}, queries.NewGetAllDeviceDefinitionByAutocompleteQueryHandler(searchService)),
 		mediator.WithHandler(&queries.GetCompatibilityR1SheetQuery{}, queries.NewCompatibilityR1SheetQueryHandler(settings)),
 		mediator.WithHandler(&queries.GetDeviceDefinitionByIDQueryV2{}, queries.NewGetDeviceDefinitionByIDQueryV2Handler(ddOnChainService, pdb.DBS)),
-		mediator.WithHandler(&queries.GetVINProfileQuery{}, queries.NewGetVINProfileQueryHandler(pdb.DBS, &logger)),
+		mediator.WithHandler(&queries.GetVINProfileQuery{}, queries.NewGetVINProfileQueryHandler(pdb.DBS, &logger, powerTrainTypeService)),
 
 		mediator.WithHandler(&queries.DecodeVINQuery{}, decodeVINHandler),
 		mediator.WithHandler(&queries.UpsertDecodingQuery{}, upsertVINHandler),
diff --git a/internal/core/queries/get_vin_profile.go b/internal/core/queries/get_vin_profile.go
index 33d76b96..def3362d 100644
--- a/internal/core/queries/get_vin_profile.go
+++ b/internal/core/queries/get_vin_profile.go
@@ -5,6 +5,10 @@ import (
 	"context"
 	"database/sql"
 	"fmt"
+	coremodels "github.com/DIMO-Network/device-definitions-api/internal/core/models"
+	"github.com/DIMO-Network/device-definitions-api/internal/core/services"
+	"github.com/DIMO-Network/shared/pkg/logfields"
+	"strings"
 
 	"github.com/DIMO-Network/device-definitions-api/internal/core/mediator"
 	"github.com/DIMO-Network/device-definitions-api/internal/infrastructure/db/models"
@@ -15,8 +19,9 @@ import (
 )
 
 type GetVINProfileQueryHandler struct {
-	dbs    func() *db.ReaderWriter
-	logger *zerolog.Logger
+	dbs           func() *db.ReaderWriter
+	logger        *zerolog.Logger
+	powertrainSvc services.PowerTrainTypeService
 }
 
 type GetVINProfileQuery struct {
@@ -24,17 +29,19 @@ type GetVINProfileQuery struct {
 }
 
 type GetVINProfileResponse struct {
-	VIN        string `json:"vin"`
-	ProfileRaw []byte `json:"profileRaw"`
-	Vendor     string `json:"vendor"`
+	VIN            string `json:"vin"`
+	ProfileRaw     []byte `json:"profileRaw"`
+	Vendor         string `json:"vendor"`
+	PowertrainType string `json:"powertrainType,omitempty"`
 }
 
 func (*GetVINProfileQuery) Key() string { return "GetVINProfileQuery" }
 
-func NewGetVINProfileQueryHandler(dbs func() *db.ReaderWriter, logger *zerolog.Logger) GetVINProfileQueryHandler {
+func NewGetVINProfileQueryHandler(dbs func() *db.ReaderWriter, logger *zerolog.Logger, powertrainSvc services.PowerTrainTypeService) GetVINProfileQueryHandler {
 	return GetVINProfileQueryHandler{
-		dbs:    dbs,
-		logger: logger,
+		dbs:           dbs,
+		logger:        logger,
+		powertrainSvc: powertrainSvc,
 	}
 }
 
@@ -44,7 +51,7 @@ func (dc GetVINProfileQueryHandler) Handle(ctx context.Context, query mediator.M
 		return nil, &exceptions.ValidationError{Err: fmt.Errorf("invalid vin %s", qry.VIN)}
 	}
 
-	vinNumber, err := models.VinNumbers(models.VinNumberWhere.Vin.EQ(qry.VIN), models.VinNumberWhere.DrivlyData.IsNotNull()).One(ctx, dc.dbs().Reader)
+	vinNumber, err := models.VinNumbers(models.VinNumberWhere.Vin.EQ(qry.VIN)).One(ctx, dc.dbs().Reader)
 
 	if err != nil {
 		if errors.Is(err, sql.ErrNoRows) {
@@ -52,10 +59,22 @@ func (dc GetVINProfileQueryHandler) Handle(ctx context.Context, query mediator.M
 		}
 		return nil, &exceptions.InternalError{Err: fmt.Errorf("failed to get vin %s", qry.VIN)}
 	}
+	split := strings.Split(vinNumber.DefinitionID, "_")
+	powertrain := ""
+	if len(split) == 3 {
+		powertrain, err = dc.powertrainSvc.ResolvePowerTrainType(split[0], split[1], vinNumber.DrivlyData, vinNumber.VincarioData)
+		if err != nil {
+			dc.logger.Error().Err(err).Str(logfields.VIN, qry.VIN).Msg("failed to resolve powertrain type for vin, continuing with ICE")
+		}
+	}
+	if powertrain == "" {
+		powertrain = string(coremodels.ICE)
+	}
 
 	return &GetVINProfileResponse{
-		VIN:        qry.VIN,
-		ProfileRaw: vinNumber.DrivlyData.JSON,
-		Vendor:     "drivly",
+		VIN:            qry.VIN,
+		ProfileRaw:     vinNumber.DrivlyData.JSON,
+		Vendor:         vinNumber.DecodeProvider.String,
+		PowertrainType: powertrain,
 	}, nil
 }