黄尔冬 / net_iou

/**
 * Created by Tommy Huang on 18/04/02.
 */
const config = require('config-lite')({
  config_basedir: __dirname,
  config_dir: 'config'
})
const moment = require('moment')
const uuid = require('uuid/v4')
const qr = require('qr-image')
const Helper = require('./helper')
const Op = require('sequelize').Op

const User = require('../models').User
const BorrowList = require('../models').BorrowList
const Friendship = require('../models').Friendship
const Redis = require('../models/redis')

exports.create = async function(req, res) {
  try {
    const idFromToken = req.user.userId
    const id = req.body.userId
    const amount = parseInt(req.body.amount) || 0
    const role = req.body.role
    const target = req.body.target
    const start = req.body.start
    const end = req.body.end
    const rate = req.body.rate
    const rateValue = parseFloat(req.body.rateValue) || 0
    const usage = req.body.usage
    const explanation = req.body.explanation || ''
    const imgs = req.files || []
    if (!id || !role || !target || !start || !end || !usage ||!amount)  throw new Error('参数错误')
    if (idFromToken !== id) throw new Error('token check fail')
    if (amount <= 0) throw new Error('非法的借款金额')
    if (parseInt(rate.slice(0, rate.length - 1)) !== rateValue * 100 || rateValue < 0.00 || rateValue > 0.24) throw new Error('非法利率参数')
    const startDate = moment(start)
    const endDate = moment(end)
    if (startDate.isAfter(endDate)) throw new Erro('还款日期最早为借款当日')
    const user = await User.findOne({where: {id: id}})
    if (!user) throw new Erro('用户不存在')
    let imgSrc = []
    if (imgs.length) {
      imgSrc = await Promise.all(imgs.map(async (item) => {
        let fileName = `${uuid()}.${item.mimetype.split('/')[1]}`
        let buffer = item.buffer
        if (buffer.size >= 2000000) {
          buffer = await Helper.imageCompression(item.buffer, 1200)
        }
        await Helper.writeFile(`${config.img.iouImg}/${fileName}`, buffer)
        return fileName
      }))
    }
    imgSrc = imgSrc.join(',')
    let newBorrowList = {
      id: uuid(),
      amount: amount,
      startDate: startDate.format(),
      endDate: endDate.format(), 
      rate: rateValue,
      usage: usage,
      explanation: explanation,
      imgs: imgSrc,
      targetName: target,
      sponsorId: user.id
    }
    if (role === 'borrower') {
      newBorrowList.borrowerId = user.id
    } else {
      newBorrowList.creditorId = user.id
    }
    let result = await BorrowList.create(newBorrowList)
    res.json({
      success: 1,
      id: result.id
    })
  } catch (e) {
    console.log(e)
    res.json({
      success: 0,
      msg: `创建失败:${e.message}`
    })
  }
}

exports.get = async function (req, res) {
  try {
    const idFromToken = req.user.userId
    const userId = req.query.userId
    const id = req.query.id
    
    if (!id || !userId) throw new Error('参数错误')
    if (idFromToken !== userId) throw new Error('token check fail')
    let info = await BorrowList.findOne({
      where: {
        id: id,
        deleted: false
      },
      include: [{
        model: User,
        required: false,
        as: 'Borrower',
        attributes: ['id', 'name', 'username', 'headimgurl']
      }, {
        model: User,
        required: false,
        as: 'Creditor',
        attributes: ['id', 'name', 'username', 'headimgurl']
      }]
    })
    if (!info) throw new Error('未找到对应的借条')
    if (info.sponsorId !== userId) {
      if (info.status === '未发起') {
        throw new Error('未找到对应的借条')
      } else if (info.status === '待确认') {
        if (!info.borrowerId || !info.creditorId) {
          const target = await User.findOne({where: {id: userId}})
          if (target.name) {
            if (target.name === info.targetName) {
              let update = {}
              !!info.borrowerId ? update.creditorId = userId : update.borrowerId = userId
              await info.update(update)
            } else {
              let update = {status: '已驳回'}
              update.reason = !!info.borrowerId ? '填写的出借人姓名与邀请对象的姓名不符，系统自动驳回' : '填写的借款人姓名与邀请对象的姓名不符，系统自动驳回'
              await info.update(update)
              throw new Error('未找到对应的借条')
            }
          }
        } else {
          if (info.creditorId !== userId && info.borrowerId !== userId) throw new Error('未找到对应的借条')
        }
      } else {
        if (info.creditorId !== userId && info.borrowerId !== userId) throw new Error('未找到对应的借条')
      } 
    }
    res.json({
      success: 1,
      info: info
    })
  } catch (e) {
    console.log(e)
    res.json({
      success: 0,
      msg: `获取失败:${e.message}`
    })
  }
}

exports.getIndexList = async function (req, res) {
  try {
    const idFromToken = req.user.userId
    const userId = req.query.userId
    if (!userId) throw new Error('参数错误')
    // if (idFromToken !== userId) throw new Error('token check fail')
    let [borrowList, creditList, borrowedAmount, creditedAmout] = await Promise.all([
      BorrowList.findAll({
        where: {
          deleted: false,
          borrowerId: userId,
          status: ['未发起', '待确认', '已驳回']
        },
        attributes: ['id', 'amount', 'start_date', 'end_date', 'rate', 'status', 'targetName'],
        include: [{
          model: User,
          required: false,
          as: 'Creditor',
          attributes: ['id', 'name', 'username', 'headimgurl']
        }],
        order: [['created_at', 'DESC']],
        limit: 10
      }),
      BorrowList.findAll({
        where: {
          deleted: false,
          creditorId: userId,
          status: ['未发起', '待确认', '已驳回']
        },
        attributes: ['id', 'amount', 'start_date', 'end_date', 'rate', 'status', 'targetName'],
        include: [{
          model: User,
          required: false,
          as: 'Borrower',
          attributes: ['id', 'name', 'username', 'headimgurl']
        }],
        order: [['created_at', 'DESC']],
        limit: 10
      }),
      BorrowList.sum('amount', {where: {
        deleted: false,
        borrowerId: userId,
        status: ['已生效']
      }}),
      BorrowList.sum('amount', {where: {
        deleted: false,
        creditorId: userId,
        status: ['已生效']
      }}),
    ])
    borrowList = borrowList.map((item) => {
      return buildIndexListInfo(item.dataValues, 'borrow')
    })
    creditList = creditList.map((item) => {
      return buildIndexListInfo(item.dataValues, 'credit')
    })
    res.json({
      success: 1,
      borrowedAmount: borrowedAmount || 0,
      creditedAmout: creditedAmout || 0,
      list: {
        borrow: borrowList,
        credit: creditList
      }
    })
  } catch (e) {
    console.log(e)
    res.json({
      success: 0,
      msg: `获取失败:${e.message}`
    })
  }
}

/**
 * 生成首页需要的借条列表格式
 * @param {Object} item 借条数据对象
 * @param {String} type 正在借款：'borrow', 正在出借： ’credit‘
 */
function buildIndexListInfo(item, type) {
  console.log(item)
  const duration = moment(item.end_date).diff(moment(item.start_date), 'days')
  const timeRemain = moment(item.end_date).diff(moment(), 'days')
  const target = type === 'borrow' ? item.Creditor : item.Borrower
  const avatar = target ? target.headimgurl : null
  item = Object.assign(item, {
    duration: duration,
    timeRemain: timeRemain,
    target: target,
    avatar: avatar,
  })
  return item
}

exports.submitIOU = async function(req, res) {
  try {
    const idFromToken = req.user.userId
    const userId = req.body.userId
    const id = req.body.id
    const password = req.body.password
    if (!id || !userId || !password )  throw new Error('参数错误')
    if (idFromToken !== userId) throw new Error('token check fail')
    let [user, list] = await Promise.all([
      User.findOne({where: {id: userId}}),
      BorrowList.findOne({where: {id: id, sponsorId: userId, status: ['未发起', '已驳回']}})
    ]) 
    if (!user) throw new Error('用户不存在')
    if (!list) throw new Error('未找到对应的借条')
    if (!Helper.passwdCheck(password, user.password)) throw new Error('交易密码不正确')
    const shareUrl = Helper.buildShareUrl(list.id)
    const qrImg = qr.imageSync(shareUrl, { type: 'svg' })        
    let update = {
      qr: qrImg
    }
    if (user.id === list.creditorId) {
      update.creditorAgree = true
      update.status = '待确认'
      update.borrowerId = null
    }
    if (user.id === list.borrowerId) {
      update.borrowerAgree = true
      update.status = '待确认'
      update.creditorId = null 
    }
    let result = await list.update(update)
    res.json({
      success: 1,
    })
  } catch (e) {
    console.log(e)
    res.json({
      success: 0,
      msg: `借条提交失败：${e.message}`
    })
  }
}

exports.deleteIOU = async function(req, res) {
  try {
    const idFromToken = req.user.userId
    const userId = req.body.userId
    const id = req.body.id
    const password = req.body.password
    if (!id || !userId || !password )  throw new Error('参数错误')
    if (idFromToken !== userId) throw new Error('token check fail')
    let [user, list] = await Promise.all([
      User.findOne({where: {id: userId}}),
      BorrowList.findOne({where: {id: id, sponsorId: userId, status: '未发起'}})
    ]) 
    if (!user) throw new Error('用户不存在')
    if (!list) throw new Error('未找到对应的借条')
    if (!Helper.passwdCheck(password, user.password)) throw new Error('交易密码不正确')
    let result = await list.update({
      deleted: true
    })
    res.json({
      success: 1,
    })
  } catch (e) {
    console.log(e)
    res.json({
      success: 0,
      msg: `借条删除失败：${e.message}`
    })
  }
}

exports.comfirmIOU = async function(req, res) {
  try {
    const idFromToken = req.user.userId
    const userId = req.body.userId
    const id = req.body.id
    const password = req.body.password
    if (!id || !userId || !password )  throw new Error('参数错误')
    if (idFromToken !== userId) throw new Error('token check fail')
    let [user, list] = await Promise.all([
      User.findOne({where: {id: userId}}),
      BorrowList.findOne({where: {id: id, status: '待确认'}})
    ]) 
    if (!user) throw new Error('用户不存在')
    if (!list) throw new Error('未找到对应的借条')
    if (list.creditorId !== userId && list.borrowerId !== userId) throw new Error('未找到对应的借条')
    if (!Helper.passwdCheck(password, user.password)) throw new Error('交易密码不正确')
    if (user.name !== list.targetName) throw new Error('借条信息与您的身份信息不符，无法确认该借条')
    let update = {}
    if (list.borrowerId === user.id && list.sponsorId !== user.id) {
      update.borrowerAgree = true      
      update.status = '已生效'
    } else if (list.creditorId === user.id && list.sponsorId !== user.id) {
      update.creditorAgree = true   
      update.status = '已生效'
    }
    let [result, friends] = await Promise.all([
      list.update(update),
      Friendship.findOne({
        where: {userId: user.id, friendId: list.sponsorId},
      })
    ]) 
    if (!friends) {
      friends = await Friendship.bulkCreate([
        {userId: user.id, friendId: list.sponsorId},
        {friendId: user.id, userId: list.sponsorId},        
      ])
    }
    res.json({
      success: 1,
    })
  } catch (e) {
    console.log(e)
    res.json({
      success: 0,
      msg: `借条确认失败：${e.message}`
    })
  }
}

exports.deleteIOU = async function(req, res) {
  try {
    const idFromToken = req.user.userId
    const userId = req.body.userId
    const id = req.body.id
    const password = req.body.password
    if (!id || !userId || !password )  throw new Error('参数错误')
    if (idFromToken !== userId) throw new Error('token check fail')
    let [user, list] = await Promise.all([
      User.findOne({where: {id: userId}}),
      BorrowList.findOne({where: {id: id, sponsorId: userId, status: ['未发起', '已驳回']}})
    ]) 
    if (!user) throw new Error('用户不存在')
    if (!list) throw new Error('未找到对应的借条')
    if (!Helper.passwdCheck(password, user.password)) throw new Error('交易密码不正确')
    let result = await list.update({
      deleted: true
    })
    res.json({
      success: 1,
    })
  } catch (e) {
    console.log(e)
    res.json({
      success: 0,
      msg: `借条删除失败：${e.message}`
    })
  }
}

exports.rejectIOU = async function(req, res) {
  try {
    const idFromToken = req.user.userId
    const userId = req.body.userId
    const id = req.body.id
    const password = req.body.password
    if (!id || !userId || !password )  throw new Error('参数错误')
    if (idFromToken !== userId) throw new Error('token check fail')
    let [user, list] = await Promise.all([
      User.findOne({where: {id: userId}}),
      BorrowList.findOne({where: {id: id, status: '待确认'}})
    ]) 
    if (!user) throw new Error('用户不存在')
    if (!list) throw new Error('未找到对应的借条')
    if (list.creditorId !== userId && list.borrowerId !== userId) throw new Error('未找到对应的借条')
    if (!Helper.passwdCheck(password, user.password)) throw new Error('交易密码不正确')
    if (user.name !== list.targetName) throw new Error('借条信息与您的身份信息不符，无法驳回该借条')    
    let result = await list.update({
      status: '已驳回',
      reason: `${list.borrowerId === user.id ? '借款' : '出借'}方主动驳回`
    })
    res.json({
      success: 1,
    })
  } catch (e) {
    console.log(e)
    res.json({
      success: 0,
      msg: `借条驳回失败：${e.message}`
    })
  }
}

exports.shareInfo = async function(req, res) {
  try {
    const idFromToken = req.user.userId
    const userId = req.query.userId
    const id = req.query.id
    if (!id || !userId)  throw new Error('未找到对应的借条')
    if (idFromToken !== userId) throw new Error('token check fail')
    let list = await BorrowList.findOne({
      where: {
        id: id, 
        sponsorId: userId, 
        status: '待确认'
      }
    })
    if (!list) throw new Error('未找到对应的借条')
    const shareUrl = Helper.buildShareUrl(list.id)
    let qrImg = list.qr
    if (!qrImg) {
      qrImg = qr.imageSync(shareUrl, { type: 'svg' })        
      await list.update({qr: qrImg})
    }
    res.json({
      success: 1,
      img: qrImg,
      url: shareUrl
    })
  } catch (e) {
    console.log(e)
    res.json({
      success: 0,
      msg: `${e.message}`
    })
  }
}

exports.getHistory = async function (req, res) {
  try {
    const idFromToken = req.user.userId
    const userId = req.query.userId
    const type = req.query.type
    const search = req.query.search || ''
    const status = req.query.status || ''
    let sort = req.query.sort || ''
    if (!userId || !type || (type !== 'borrow' && type !== 'credit')) throw new Error('参数错误')
    if (idFromToken !== userId) throw new Error('token check fail')

    let where = {
      deleted: false,
      status: ['已生效', '已逾期', '已还清']
    }
    let order = [['endDate', 'DESC']]
    if (type === 'borrow') {
      where.borrowerId = userId
      target = 'Creditor'
    } else {
      where.creditorId = userId
      target = 'Borrower'
    }
    if (status !== '') where.status = status
    sort = sort.split(',')
    if (sort.length === 2 && (sort[1] === 'DESC' || sort[1] === 'ASC')) order = [sort]
    let list = await BorrowList.findAll({
      where: where,
      attributes: ['id', 'amount', 'start_date', 'end_date', 'rate', 'status', 'targetName'],
      include: [{
        model: User,
        where: {
          name: {
            [Op.like]: `%${search}%`
          }
        },
        as: target,
        attributes: ['id', 'name', 'username', 'headimgurl']
      }],
      order: order,
    })
    const times = list.length
    let total = 0
    let interest = 0
    let result = []
    list = list.map((item) => {
      let result =  buildIndexListInfo(item.dataValues, type)
      total += result.amount
      interest += (result.amount * result.rate / 365 * result.duration)
      return result
    })
    interest = Math.floor(interest * 100) / 100
    res.json({
      success: 1,
      list: list,
      overview: {
        total: total,
        interest: interest,
        times: times
      }
    })
  } catch (e) {
    console.log(e)
    res.json({
      success: 0,
      msg: `获取失败:${e.message}`
    })
  }
}