helper.js 3.88 KB
/**
 * Created by Tommy Huang on 18/03/21.
 */

const config = require('config-lite')({
  config_basedir: __dirname,
  config_dir: 'config'
})
const axios = require('axios')
const {URL} = require('url')
const crypto = require('crypto')
const FormData = require('form-data')

/**
 * 获取用户微信 Token
 * @param {string} code 微信返回的用户 code
 */
exports.getUserWxAccessTokenInfo = async function(code) {
  const {data} = await axios.get(`https://api.weixin.qq.com/sns/oauth2/access_token?appid=${config.wx.appId}&secret=${config.wx.appSecret}&code=${code}&grant_type=authorization_code`)
  if (data.access_token && data.refresh_token && data.openid) {
    return data
  } else {
    return false
  }
}

/**
 * 刷新用户的 Token
 * @param {string} refreshToken 微信返回的用户 refresh_token
 */
exports.refreshUserWxToken = async function(refreshToken) {
  const {data} = await axios.get(`https://api.weixin.qq.com/sns/oauth2/refresh_token?appid=${config.wx.appId}&grant_type=refresh_token&refresh_token=${refreshToken}`)
  if (data.access_token && data.refresh_token && data.openid) {
    return data
  } else {
    return false
  }
}
/**
 * 拉取用户信息
 * @param {string} token 微信返回的用户 access_token
 * @param {string} openId 微信返回的用户 openid
 */
exports.getUserWxInfo = async function(token, openId) {
  const {data} = await axios.get(`https://api.weixin.qq.com/sns/userinfo?access_token=${token}&openid=${openId}&lang=zh_CN`)
  if (data.openid && data.nickname && data.headimgurl) {
    return data
  } else {
    return false
  }
}

/**
 * 发送验证码
 * @param {string} phone 要发送验证码的手机号
 */
exports.sendVerificationCode = async function (phone) {
  if (!phone) return {success: false, msg: '缺少参数'}
  const code = randomWord(false, 4, 4)
  var url = `${config.sms.url}?account=${config.sms.account}&password=${config.sms.password}&mobile=${phone}&content=【七彩贷】验证码:${code},如非本人操作,请忽略本短信`
  url = new URL(url).href
  let { data } = await axios.get(url)
  console.log(data)
  if (data.status === 0) {
    return {success: true, code: code}
  } else {
    return {success: false, msg: data.desc}
  }
}

/**
 * 产生任意长度随机数字组合
 * @param {bool} randomFlag 是否任意长度
 * @param {num} min 任意长度最小位[固定位数]
 * @param {num} max 任意长度最大位
 */
function randomWord (randomFlag, min, max) {
  let str = ""
  let range = min
  let arr = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']

  if (randomFlag) {
    range = Math.round(Math.random() * (max - min)) + min
  }
  for (let i = 0 ; i < range; i++) {
    let pos = Math.round(Math.random() * (arr.length-1))
    str += arr[pos]
  }
  return str
}

/**
 * 生成腾讯云接口鉴权签名
 */
exports.tencentCloudAuthorization = function () {
  const current = Math.floor(Date.now() / 1000)
  const rand = Math.floor(Math.random() * Math.pow(10, 10))
  const expire = current + 30
  const origin = `a=${config.idOCR.appid}&b=${config.idOCR.bucket}&k=${config.idOCR.SecretId}&e=${expire}&t=${current}&r=${rand}&u=0&f=`
  var auth = crypto.createHmac('sha1', config.idOCR.SecretKey).update(origin).digest()
  auth = Buffer.concat([auth, Buffer.from(origin, 'utf-8')])
  return auth.toString('base64')
}

/**
 * 获取腾讯云身份证识别接口结果
 * @param {object File} file 身份证图片文件
 * @param {integer} type 图片类型:正面:0, 反面:1
 * @param {string} auth 接口鉴权签名
 */
exports.getIdORCResult = async function(file, type, auth) {
  let form = new FormData()
  form.append('appid', config.idOCR.appid)
  form.append('bucket', config.idOCR.bucket)
  form.append('card_type', type)
  form.append('image', file)
  let formHead = form.getHeaders()
  formHead.authorization = auth
  let {data} = await axios.post(config.idOCR.url, form, {headers: formHead})
  return data.result_list[0]
}