devicemngmt/action_update.go

90 lines
2.0 KiB
Go

package devicemngmt
import (
"context"
"errors"
"fmt"
"git.selly.red/Selly-Modules/logger"
"git.selly.red/Selly-Modules/mongodb"
"go.mongodb.org/mongo-driver/bson"
)
// UpdateOptions ...
type UpdateOptions struct {
UserID string
UserAgent string
AppVersion string
IP string
FCMToken string
AuthToken string
Language string
Model string
Manufacturer string
OsName string
OsVersion string
IsMobile bool
}
// UpdateByDeviceID ...
func (s Service) UpdateByDeviceID(deviceID string, payload UpdateOptions) error {
var (
ctx = context.Background()
col = s.getDeviceCollection()
cond = bson.M{
"deviceId": deviceID,
}
)
total, _ := col.CountDocuments(ctx, cond)
if total == 0 {
return errors.New("deviceId not found")
}
// Validate payload
err := payload.validate()
if err != nil {
return err
}
// Get userAgent data
osName, osVersion, isMobile := getUserAgentData(payload.UserAgent)
if osName == "" || osVersion == "" {
osName = payload.OsName
osVersion = payload.OsVersion
isMobile = payload.IsMobile
}
// Setup update data
userID, _ := mongodb.NewIDFromString(payload.UserID)
updateData := bson.M{
"$set": bson.M{
"userId": userID,
"osName": osName,
"osVersion": osVersion,
"ip": payload.IP,
"language": getLanguage(payload.Language),
"authToken": payload.AuthToken,
"fcmToken": payload.FCMToken,
"model": payload.Model,
"manufacturer": payload.Manufacturer,
"appVersion": payload.AppVersion,
"isMobile": isMobile,
"lastActivatedAt": now(),
},
}
// Update
_, err = col.UpdateOne(ctx, cond, updateData)
if err != nil {
logger.Error("devicemngmt - updateByDeviceID", logger.LogData{
Source: "devicemngmt.UpdateByDeviceID",
Message: err.Error(),
Data: deviceID,
})
return fmt.Errorf("error when update device: %s", err.Error())
}
return nil
}