FS-IOT/code/VideoAccess-VCMP/api/app/lib/controllers/nvr/index.js

'use strict';
const moment = require('moment')

async function edit (ctx, next) {
   const transaction = await ctx.fs.dc.orm.transaction();
   try {
      const models = ctx.fs.dc.models;
      const { userId } = ctx.fs.api
      const data = ctx.request.body;

      // 或取其他服务信息
      const nvrData = {
         channelCount: 8,
         port: 8080,
      }

      if (data.id) {
         // 修改
         const storageData = Object.assign({}, data, nvrData)
         await models.Nvr.update(storageData, {
            where: {
               id: data.id
            },
            transaction
         })
      } else {
         // 添加
         const storageData = Object.assign({}, data, nvrData, {
            createTime: moment().format(),
            createUserId: userId,
            delete: false,
         })
         await models.Nvr.create(storageData, { transaction })
      }

      await transaction.commit();
      ctx.status = 204;
   } catch (error) {
      await transaction.rollback();
      ctx.fs.logger.error(`path: ${ctx.path}, error: ${error}`);
      ctx.status = 400;
      ctx.body = {}
   }
}

async function get (ctx) {
   try {
      const models = ctx.fs.dc.models;
      const { userId, token } = ctx.fs.api
      const { limit, page, orderBy, orderDirection, keyword, venderId } = ctx.query
      let findOption = {
         attributes: { exclude: ['delete'] },
         where: {
            createUserId: userId,
            delete: false,
         },
         order: [
            [orderBy || 'id', orderDirection || 'DESC']
         ]
      }
      if (limit) {
         findOption.limit = limit
      }
      if (page && limit) {
         findOption.offset = page * limit
      }
      if (keyword) {
         findOption.where.name = { $like: `%${keyword}%` }
      }
      if (venderId) {
         findOption.where.venderId = venderId
      }

      const nvrRes = await models.Nvr.findAll(findOption)
      const total = await models.Nvr.count({
         where: findOption.where
      })

      const nvrIds = nvrRes.map(r => r.id)
      const cameraRes = await models.Camera.findAll({
         where: {
            nvrId: { $in: nvrIds }
         }
      })
      const cameraIds = cameraRes.map(c => c.id)

      // 查在安心云绑定的数据
      const axbindCameraRes = cameraIds.length ?
         await ctx.app.fs.axyRequest.get('vcmp/camera/project', { query: { token, cameraId: cameraIds.join(',') } })
         : []

      for (let { dataValues: n } of nvrRes) {
         const corCameras = cameraRes.filter(c => c.nvrId == n.id)
         const corBind = axbindCameraRes.filter(b => corCameras.some(c => c.id == b.cameraId))

         if (corBind.length) {
            n.station = []
            for (let c of corBind) {
               n.station = n.station.concat(c.stations)
            }
         } else {
            n.station = []
         }
      }

      ctx.status = 200;
      ctx.body = {
         total: total,
         data: nvrRes
      }
   } catch (error) {
      ctx.fs.logger.error(`path: ${ctx.path}, error: ${error}`);
      ctx.status = 400;
      ctx.body = {}
   }
}

async function del (ctx, next) {
   const transaction = await ctx.fs.dc.orm.transaction();
   try {
      const models = ctx.fs.dc.models;
      const { userId, token } = ctx.fs.api
      const { nvrId } = ctx.params

      await models.Nvr.destroy({
         where: {
            id: nvrId
         },
         transaction
      })

      const cameraRes = await models.Camera.findAll({
         where: {
            nvrId
         }
      })

      const cameraIds = cameraRes.map(c => c.id)

      await models.Camera.destroy({
         where: {
            nvrId,
         }
      })
      if (cameraIds.length) {
         await ctx.app.fs.axyRequest.delete('vcmp/camera/project', { query: { token, cameraId: cameraIds.join(',') } })
      }

      await transaction.commit();
      ctx.status = 204;
   } catch (error) {
      await transaction.rollback();
      ctx.fs.logger.error(`path: ${ctx.path}, error: ${error}`);
      ctx.status = 400;
      ctx.body = {}
   }
}

async function detail (ctx) {
   let errMsg = '获取 NVR 详情失败'
   try {
      const models = ctx.fs.dc.models;
      const { userId, token } = ctx.fs.api
      const { nvrId } = ctx.params

      const nvrRes = await models.Nvr.findOne({
         attributes: { exclude: ['delete'] },
         where: {
            id: nvrId
         },
         include: [{
            model: models.Vender
         }]
      })

      if (!nvrRes) {
         throw errMsg
      }

      // 查询对应的后端服务相关信息

      const cameraRes = await models.Camera.findAll({
         attributes: ['id', 'name', 'channelName', 'serialNo', 'rtmp'],
         where: {
            nvrId
         }
      })
      let cameras = []
      let cameraIds = []
      for (let c of cameraRes) {
         cameraIds.push(c.id)
         cameras.push(c.dataValues)
      }

      const cameraProject = await ctx.app.fs.axyRequest.get('vcmp/camera/project', { query: { token, cameraId: cameraIds.join(',') } })
      const bindStations = []
      for (let c of cameraProject) {
         for (let s of c.stations) {
            bindStations.push(s)
         }
      }

      const corUser = await ctx.app.fs.authRequest.get(`user/${nvrRes.createUserId}/message`, { query: { token } })

      let nvrDetail = {
         ...nvrRes.dataValues,
         station: bindStations,
         camera: cameras,
         createUser: {
            namePresent: corUser[0].namePresent
         },
         accessWay: 'GB/T28181'
      }

      ctx.status = 200;
      ctx.body = nvrDetail
   } catch (error) {
      ctx.fs.logger.error(`path: ${ctx.path}, error: ${error}`);
      ctx.status = 400;
      ctx.body = { message: errMsg }
   }
}

async function nvrExport (ctx) {
   try {
      const { userId, token } = ctx.fs.api

      const nvrRes = await models.Nvr.findAll({
         where: {
            createUserId: userId,
         }
      })
      const nvrIds = nvrRes.map(r => r.id)
      const cameraRes = await models.Camera.findAll({
         where: {
            nvrId: { $in: nvrIds }
         }
      })
      const cameraIds = cameraRes.map(c => c.id)

      // 查在安心云绑定的数据
      const axbindCameraRes =
         cameraIds.length ?
            await ctx.app.fs.axyRequest.get('vcmp/camera/project', { query: { token, cameraId: cameraIds.join(',') } })
            : []

      for (let { dataValues: n } of nvrRes) {
         const corCameras = cameraRes.filter(c => c.nvrId == n.id)
         const corBind = axbindCameraRes.filter(b => corCameras.some(c => c.id == b.cameraId))
         if (corBind.length) {
            n.station = []
            for (let c of corBind) {
               n.station = n.station.concat(c.stations)
            }
         } else {
            n.station = []
         }
      }

      const filePath = ''
      const fileData = fs.readFileSync(filePath);
      let fileName = filePath.split('/').pop()
      ctx.status = 200;
      ctx.set('Content-Type', 'application/x-xls');
      ctx.set('Content-disposition', 'attachment; filename=' + encodeURI(fileName));
      ctx.body = fileData;
   } catch (error) {
      ctx.fs.logger.error(`path: ${ctx.path}, error: ${error}`);
      ctx.status = 400;
      ctx.body = {}
   }
}

module.exports = {
   edit,
   get,
   del,
   detail,
   nvrExport,
};