add money to wallet if correct

server/controllers/GPS.InstalledDevice.js

@@ -3,35 +3,73 @@ const { checkValidations } = require('../plugins/controllersHelperFunctions')
 const User = require('../models/GPS.User')
 const InstalledDevice = require('../models/GPS.InstalledDevice')
 const Device = require('../models/GPS.Device')
+const userGPS = require('../models/GPS.User')
 const Score = require('../models/GPS.Score')
 const { _sr } = require('../plugins/serverResponses')
 const { notifyAdmin } = require('../WebSocket/controllers/admin_EventsController')
+const { isDeviceRegisteredOnKoja } = require('../kojaWebService')
 
 module.exports.deviceRegistration = [
   [body('IMEI').notEmpty().isString().withMessage('لطفا IMEI را وارد یا تصحیح کنید')],
   checkValidations(validationResult),
   async (req, res) => {
-    const device = await Device.findOne({ IMEI: req.body.IMEI })
-    if (!device) {
-      throw res.status(404).json({ msg: 'این IMEI وجود ندارد' })
-      // eslint-disable-next-line eqeqeq
-    } else if (device.status == 1) {
-      throw res.status(404).json({ msg: 'این IMEI قبلا ثبت شده' })
-    } else {
-      const data = {
-        deviceId: device.id,
-        status: 0,
-        registerDate: new Date(),
-        userId: req.user_id,
-        IMEI: req.body.IMEI
-      }
-      const newRequest = await InstalledDevice.create(data)
-      device.status = 1
-      device.save()
-      notifyAdmin('newInstallDeviceRequest')
+    const IMEI = req.body.IMEI
 
-      // res.status(201).json({ msg: _sr.fa.response.success_save, data: newRequest })
-      res.status(201).json(newRequest)
+    try {
+      const device = await Device.findOne({ IMEI })
+      if (!device) {
+        return res.status(404).json({ msg: 'این IMEI وجود ندارد' })
+      }
+      if (device.status === 1) {
+        return res.status(400).json({ msg: 'این IMEI قبلا ثبت شده ' })
+      }
+      const isRegistered = await isDeviceRegisteredOnKoja(IMEI)
+
+      if (isRegistered) {
+        let installedDeviceId = null
+        try {
+          const data = {
+            deviceId: device.id,
+            status: 0,
+            registerDate: new Date(),
+            userId: req.user_id,
+            IMEI: req.body.IMEI
+          }
+          // 1.add to installed device
+          const installedDevice = await InstalledDevice.create(data)
+          installedDeviceId = installedDevice._id
+          // 2. change status
+          await Device.findOneAndUpdate({ IMEI }, { status: 1 })
+
+          // 3. add to wallet
+          await userGPS.findByIdAndUpdate(req.user_id, { $inc: { walletBalance: 100_000 } })
+
+          notifyAdmin('newInstallDeviceRequest')
+          res.status(201).json({ msg: 'مبلغ صد هزار تومان به کیف پول شما اضافه شد' })
+        } catch (err) {
+          console.log('Error in Db Transactions', err?.message)
+          if (installedDeviceId) {
+            try {
+              await InstalledDevice.deleteOne({ _id: installedDeviceId })
+              console.log('Rollback 1/2: InstalledDevice deleted.')
+              await Device.findOneAndUpdate({ IMEI }, { status: 0 })
+              console.log('Rollback 2/2: Device status reset to 0.')
+            } catch (error) {
+              console.error('CRITICAL: Rollback failed for some changes. MANUAL INTERVENTION REQUIRED.', error)
+            }
+          }
+          return res.status(err?.status || 500).json({ msg: err?.msg || 'خطا در تراکنش های دیتابیس ' })
+        }
+      } else {
+        res.status(400).json({ msg: 'هنوز در سامانه ردیاب ثبت نشده است.' })
+      }
+    } catch (err) {
+      console.log('deviceRegistration Error: ', err?.message)
+      if (err.response) {
+        const { status } = err.response
+        return res.status(status || 500).json({ msg: err.message || 'خطا!' })
+      }
+      return res.status(err.status || 500).json({ msg: err.msg || 'خطا در ثبت دستگاه' })
     }
   }
 ]
@@ -134,7 +172,7 @@ module.exports.updateInstalled = [
   async (req, res) => {
     const installedDevice = await InstalledDevice.findById(req.params.id)
     if (!installedDevice) {
-      throw res.status(404).json({ msg: 'این ایدی وجود ندارد' })
+      return res.status(404).json({ msg: 'این ایدی وجود ندارد' })
     } else {
       const user = await User.findById(installedDevice.userId).select('walletBalance score totalScore')
       const device = await Device.findById(installedDevice.deviceId)