Initial cloud-services repo - gateway service + pkg modules

pkg/manifestsender/tbox_config_manifest_sender.go

@@ -0,0 +1,143 @@
+package manifestsender
+
+import (
+	"fmt"
+	"time"
+
+	"errors"
+
+	"fiskerinc.com/modules/carcommand"
+	"fiskerinc.com/modules/common"
+	"fiskerinc.com/modules/common/carupdatestatus"
+	"fiskerinc.com/modules/db/queries"
+	"fiskerinc.com/modules/logger"
+	"fiskerinc.com/modules/utils/whereami"
+)
+
+// ProcessConfigUpdate handles all steps on the config only update
+// It will create a car_update record, manifest record, car_update_status record
+type ProcessConfigUpdateStruct struct {
+	VIN                     string // The vin of the car that configuration we are generating
+	Username                string // The username of who is generating this update
+	Name                    string // The name of the generated manifest, leave blank for auto generated. name,version is enforced unique
+	SendToCar               bool   // True to send the manifest to the car
+	DontCreateDatabaseEntry bool   // False to create database entry, by default. Cannot be true with SendToCar being true
+	Forced                  bool
+}
+
+var ERR_SEND_AND_DATABASE_MISMATCH = errors.New("cannot send to car with dont create database entry")
+
+// If called from aftersales, this will be marked as an aftersales update, which are not sent to the car
+// and they are automatically cancelled when they are hit cancel on the portal
+func (t *TBOXManifestSender) ProcessConfigUpdate(input ProcessConfigUpdateStruct, ccd queries.CarConfigDataInterface) (ucm common.UpdateConfigManifest, err error) {
+	if input.DontCreateDatabaseEntry && input.SendToCar {
+		return ucm, ERR_SEND_AND_DATABASE_MISMATCH
+	}
+
+	// We have the send to car, so that should be all the info we need to make changes based on the system :()()
+	// Get the VOD and CDSs for the car
+	cds, err := t.getVODCDS(input.VIN, ccd)
+	if err != nil {
+		return
+	}
+
+	manifest := common.UpdateManifest{}
+	if input.Name != "" {
+		manifest.Name = input.Name
+	} else {
+		manifest.Name = fmt.Sprintf("configuration update %s %s", input.VIN, time.Now().Format("2006-01-02"))
+	}
+
+	manifest.Version = time.Now().Format("2006-01-02 15:04:05")
+	manifest.Description = fmt.Sprintf("configuration for %s by %s", input.VIN, input.Username)
+	manifest.ManifestType = common.ConfigUpdateType
+
+	if input.Forced {
+		manifest.Type = common.ManifestTypeForced
+	} else {
+		manifest.Type = common.ManifestTypeStandard
+	}
+
+	manifestDB := t.db.GetUpdateManifests()
+	if !input.DontCreateDatabaseEntry {
+		_, err = manifestDB.Insert(&manifest)
+		if err != nil {
+			return
+		}
+	}
+
+	// AddCDSToECUs calls AddVOD
+	// AddECUsFromCDSList requires the manifest having been inserted so that they
+	// have the manifestID
+	manifest.AddECUsFromCDSList(cds)
+	manifest.SortECUs()
+	manifest.FillECUList()
+
+	//  Create a car update for this manifest ot be sent as
+	updatesDB := t.db.GetCarUpdates()
+	update := common.CarUpdate{
+		VIN:              input.VIN,
+		UpdateManifestID: manifest.ID,
+		UpdateManifest:   &manifest,
+		Username:         input.Username,
+		Status:           carupdatestatus.Pending,
+	}
+
+	if whereami.Service == whereami.AFTERSALES {
+		input.SendToCar = false
+		update.UpdateSource = common.UPDATE_SOURCE_AFTERSALES
+	} else {
+		update.UpdateSource = common.UPDATE_SOURCE_OTA
+	}
+
+	err = insertUpdateDatabase(input, updatesDB, manifestDB, manifest, &update)
+	if err != nil {
+		return
+	}
+
+	manifest.CarUpdateID = update.ID
+
+	manifest.Scrub(common.TRex)
+	ucm = manifest.ToUpdateConfigManifest()
+
+	if !input.SendToCar {
+		return
+	}
+	return ucm, t.SendConfig(input.VIN, ucm)
+}
+
+func insertUpdateDatabase(input ProcessConfigUpdateStruct, updatesDB queries.CarUpdatesInterface, manifestDB queries.UpdateManifestsInterface, manifest common.UpdateManifest, update *common.CarUpdate) (err error) {
+	if !input.DontCreateDatabaseEntry {
+		_, err = updatesDB.InsertAndCreateStatus(update)
+		if err != nil {
+			_, err2 := manifestDB.Delete(&manifest)
+			if err2 != nil {
+				err = errors.Join(err, err2)
+			}
+			return
+		}
+	}
+	return
+}
+
+func (t *TBOXManifestSender) SendConfig(vin string, manifest common.UpdateConfigManifest) (err error) {
+	err = t.Redis.SafeQueueMessage(common.TRex.Key(vin), common.Message{
+		Handler: "config_update",
+		Data:    manifest,
+	})
+	if err != nil {
+		return err
+	}
+
+	logger.At(logger.Info(), common.TRex.Key(vin), "update").Msgf("TBOXManifestSender sent %v %s %d", common.TRex, vin, manifest.CarUpdateID)
+
+	// For now, do not check on config update if the car wakes up correctly
+	if manifest.Type == common.ManifestTypeForced && t.sms != nil{
+		_, err = carcommand.QueueSMSWakeUp(vin, false, t.Redis, t.db.GetCars(), t.sms)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}