<template>
  <view>
    <button @click="getList">获取设备列表</button>
    <view v-if="showList">
      <button class="device" v-for="item in list"
              :key="item.deviceId" @click="connect(item)">{{item.name}}
        - {{item.pair ? '已配对' : '未配对'}}-{{item.sleep ? '休眠中' :
            '已唤醒'}}</button>
    </view>
    <view v-else>
      <button @click="getPublicKey">获取锁公钥</button>
      <button @click="getCommKey">获取锁私钥</button>
    </view>
  </view>
</template>

<script>
import crc from 'crc'
import { md5 } from 'js-md5'
import { sm4 } from 'sm-crypto'

export default {
  data() {
    return {
      list: [],
      getting: false,
      deviceId: '',
      serviceId: '',
      characteristicId1: '',
      characteristicId2: '',
      lockId: '',
      publicKey: new Uint8Array(16),
      commKey: new Uint8Array(16),
      signKey: new Uint8Array(16),
      showList: true,
      number: 1
    }
  },
  onLoad() {
    uni.openBluetoothAdapter({
      success(res) {
        console.log('初始化成功')
        console.log(res)
      },
      fail(res) {
        console.log('初始化失败')
        console.log(res)
      }
    })
  },
  onUnload() {
    uni.closeBluetoothAdapter({
      success(res) {
        console.log('关闭蓝牙成功')
        console.log(res)
      },
      fail(res) {
        console.log('关闭蓝牙失败')
        console.log(res)
      }
    })
  },
  methods: {
    // 十六进制字符串转Uint8Array
    hexToUint8Array(hex) {
      if (hex.length % 2 !== 0) {
        throw new Error('Invalid hex string')
      }
      const array = new Uint8Array(hex.length / 2)
      for (let i = 0; i < hex.length; i += 2) {
        array[i / 2] = parseInt(hex.substr(i, 2), 16)
      }
      return array
    },
    // 获取私钥
    getCommKey() {
      const cmdId = 0x3091
      const keyId = '0'
      const authUserId = '294'
      const nowTime = parseInt(new Date().getTime() / 1000)

      const originAuthCodeLen = keyId.length + authUserId.length + 4 + 16

      let authCodeBuffer = new ArrayBuffer(originAuthCodeLen)
      let authCodeBinaryData = new Uint8Array(authCodeBuffer)

      for(let i=0; i < authUserId.length; i++){
        authCodeBinaryData[i] = authUserId.charCodeAt(i)
      }

      for(let i=0; i < keyId.length; i++){
        authCodeBinaryData[authUserId.length + i] = keyId.charCodeAt(i)
      }

      const number = keyId.length + authUserId.length
      authCodeBinaryData[number + 0] = (nowTime & 0xff000000) >> 24
      authCodeBinaryData[number + 1] = (nowTime & 0xff0000) >> 16
      authCodeBinaryData[number + 2] = (nowTime & 0xff00) >> 8
      authCodeBinaryData[number + 3] = (nowTime & 0xff)

      for(let i=0; i < 16; i++){
        authCodeBinaryData[number + 4 + i] = this.publicKey[i]
      }

      console.log('钥ID', keyId, '用户ID', authUserId, '时间戳', nowTime,'公钥', Array.from(this.publicKey))
      console.log('md5前的数据', Array.from(authCodeBinaryData, byte =>
          byte.toString(10)).join(','))
      const md5Hex = md5(authCodeBinaryData)
      const codeMd5 = new Uint8Array(md5Hex.match(/.{1,2}/g).map(byte => parseInt(byte, 16)))
      console.log('md5后的数据', codeMd5, codeMd5.length)

      const bufferLength = codeMd5.length + 12 + 107

      let buffer = new ArrayBuffer(bufferLength)
      let binaryData = new Uint8Array(buffer)

      // 固定包头
      binaryData[0] = 0xEF
      binaryData[1] = 0x01
      binaryData[2] = 0xEE
      binaryData[3] = 0x02

      // 包类型 发送
      binaryData[4] = 0x01

      // 包序号
      binaryData[5] = this.number / 256
      binaryData[6] = this.number % 256
      this.number++

      // 包标识
      binaryData[7] = 0x22

      // 数据长度
      binaryData[10] = (codeMd5.length + 107) / 256
      binaryData[11] = (codeMd5.length + 107) % 256

      // 命令
      binaryData[12] = 0x3091 / 256
      binaryData[13] = 0x3091 % 256

      for(let i=0; i < this.lockId.length; i++){
        binaryData[14 + i] = this.lockId.charCodeAt(i)
      }
      let flagNumber = 14 + 40
      for (let i = 0; i < keyId.length; i++) {
        binaryData[flagNumber + i] = keyId.charCodeAt(i)
      }
      flagNumber += 40
      for (let i = 0; i < authUserId.length; i++) {
        binaryData[flagNumber + i] = authUserId.charCodeAt(i)
      }

      flagNumber += 20
      binaryData[flagNumber + 0] = (nowTime & 0xff000000) >> 24
      binaryData[flagNumber + 1] = (nowTime & 0xff0000) >> 16
      binaryData[flagNumber + 2] = (nowTime & 0xff00) >> 8
      binaryData[flagNumber + 3] = (nowTime & 0xff)

      flagNumber += 4
      binaryData[flagNumber] = codeMd5.length
      flagNumber += 1
      for (let i = 0; i < codeMd5.length; i++) {
        binaryData[flagNumber + i] = codeMd5[i]
      }

      binaryData[8] = Math.ceil((bufferLength - 12)/16) * 16 / 256
      binaryData[9] = Math.ceil((bufferLength - 12)/16) * 16 % 256

      let cebBuffer = new ArrayBuffer(bufferLength - 12)
      let cebBinaryData = new Uint8Array(cebBuffer)

      for(let i = 0; i < bufferLength - 12; i++){
        cebBinaryData[i] = binaryData[i + 12]
      }

      const key = new Uint8Array(16)
      for (let i = 0; i < this.lockId.length; i++) {
        key[i] = this.lockId.charCodeAt(i)
      }

      console.log('未加密的数据', Array.from(cebBinaryData))
      const encrypted = sm4.encrypt(cebBinaryData, key, { mode: 'ecb' })
      const newBinaryData = this.hexToUint8Array(encrypted)
      console.log('ecb加密后的数据', Array.from(newBinaryData))

      const encrypteBuffer = new ArrayBuffer(12 + newBinaryData.length)
      const encrypteBinaryData = new Uint8Array(encrypteBuffer)
      for(let i = 0; i< 12; i++){
        encrypteBinaryData[i] = binaryData[i]
      }
      for(let i = 12; i < encrypteBinaryData.length; i++){
        encrypteBinaryData[i] = newBinaryData[i - 12]
      }
      console.log('crc前的数据', Array.from(encrypteBinaryData))

      const resultBuffer = new ArrayBuffer(encrypteBinaryData.length + 2)
      const resultBinaryData = new Uint8Array(resultBuffer)
      for(let i = 0; i< encrypteBinaryData.length; i++){
        resultBinaryData[i] = encrypteBinaryData[i]
      }
      resultBinaryData[encrypteBinaryData.length] = crc.crc16kermit(encrypteBuffer) / 256
      resultBinaryData[1 + encrypteBinaryData.length] = crc.crc16kermit(encrypteBuffer) % 256

      this.writeBLECharacteristicValue(resultBuffer)
    },
    // 获取公钥
    getPublicKey() {
      let buffer = new ArrayBuffer(54)
      let binaryData = new Uint8Array(buffer)

      // 固定包头
      binaryData[0] = 0xEF
      binaryData[1] = 0x01
      binaryData[2] = 0xEE
      binaryData[3] = 0x02

      // 包类型 发送
      binaryData[4] = 0x01

      // 包序号
      binaryData[5] = this.number / 256
      binaryData[6] = this.number % 256
      this.number++

      // 包标识
      binaryData[7] = 0x20

      // 数据长度
      binaryData[8] = 42 / 256
      binaryData[9] = 42 % 256
      binaryData[10] = 42 / 256
      binaryData[11] = 42 % 256

      // 命令
      binaryData[12] = 0x3090 / 256
      binaryData[13] = 0x3090 % 256

      // // 设备ID
      for(let i=0; i < this.lockId.length; i++){
        binaryData[14 + i] = this.lockId.charCodeAt(i);
      }

      // crc

      let newBuffer = new ArrayBuffer(56)
      let newBinaryData = new Uint8Array(newBuffer)
      for(let i = 0; i < 54; i++){
        newBinaryData[i] = binaryData[i]
      }
      newBinaryData[54] = crc.crc16kermit(buffer) / 256
      newBinaryData[55] = crc.crc16kermit(buffer) % 256
      this.writeBLECharacteristicValue(newBuffer)
    },
    // 写入输入
    writeBLECharacteristicValue(data) {
      const that = this
      console.log('写入设备的设备 ID',that.deviceId)
      console.log('写入设备的serviceId',that.serviceId)
      console.log('写入设备的characteristicId',that.characteristicId2)
      console.log('写入设备的lockID',that.lockId)

      let Uint8ArrayData = new Uint8Array(data)

      console.log('未加工的数据', Array.from(Uint8ArrayData))

      for(let i = 0; i < Math.ceil(data.byteLength / 20); i++){
        let buffer = new ArrayBuffer(20)
        let binaryData = new Uint8Array(buffer)
        for(let j = 0; j < 20; j++){
          if(i * 20 + j < data.byteLength){
            binaryData[j] = Uint8ArrayData[i * 20 + j]
          }
        }
        uni.writeBLECharacteristicValue({
          deviceId: that.deviceId,
          serviceId: that.serviceId,
          characteristicId: that.characteristicId2,
          value: buffer,
          success(write) {
            const hexString = Array.from(binaryData, byte =>
                byte.toString(10)).join(',')
            console.log('写入的数据', hexString)
            console.log('写入成功')
          },
          fail(res) {
            console.log('写入失败')
            console.log(res)
          }
        })
      }
    },
    // 监听特征值变化
    notifyBLECharacteristicValueChange() {
      const that = this
      uni.notifyBLECharacteristicValueChange({
        state: true,
        deviceId: that.deviceId,
        serviceId: that.serviceId,
        characteristicId: that.characteristicId1,
        type: 'notification',
        success(res) {
          console.log(res)
          uni.onBLECharacteristicValueChange(function (res) {
            console.log('设备特征值改变')
            console.log(res)
            let binaryData = new Uint8Array(res.value)
            if(binaryData[0] === 239 && binaryData[1] === 1 && binaryData[2] === 238 && binaryData[3] === 2) {
              console.log('设备返回的数据', Array.from(binaryData), Array.from(binaryData, byte =>
                  byte.toString(16)).join(','))
              // 明文 0x20 SM4事先约定密钥 0x22
              if(binaryData[7] === 0x20) {
                if(binaryData[12] === 48 && binaryData[13] === 144){
                  const publicKey = binaryData.slice(15, 31)
                  console.log('公钥', Array.from(publicKey))
                  that.publicKey = publicKey
                }
              } else if(binaryData[7] === 0x22) {
                const cebBinaryData = binaryData.slice(12, binaryData.length - 2)
                console.log('sm4返回参数', Array.from(cebBinaryData))
                const key = new Uint8Array(16)
                for (let i = 0; i < that.lockId.length; i++) {
                  key[i] = that.lockId.charCodeAt(i)
                }
                const decrypted = sm4.decrypt(cebBinaryData, key, { mode: 'ecb', output: 'array' })
                console.log('ecb解密后的数据', decrypted)
                if(decrypted[0] === 48 && decrypted[1] === 145) {
                  that.commKey = decrypted.slice(3, 19)
                  that.signKey = decrypted.slice(19, 35)
                  console.log('commKey', Array.from(that.commKey))
                  console.log('signKey', Array.from(that.signKey))
                }
              }
            }
          })
        }
      })
    },
    // 连接蓝牙设备
    connect(item) {
      const that = this
      if(item.pair) {
        return
      }
      uni.createBLEConnection({
        deviceId: item.deviceId,
        success(res) {
          console.log('连接成功')
          console.log(res)
          uni.getBLEDeviceServices({
            deviceId: item.deviceId,
            success(res) {
              console.log('获取服务成功')
              console.log(res.services)
              const uuid = res.services[0].uuid
              that.lockId = item.name
              console.log('锁ID', that.lockId)
              uni.getBLEDeviceCharacteristics({
                deviceId: item.deviceId,
                serviceId: uuid,
                success(res) {
                  console.log('获取特征值成功')
                  console.log(res)
                  that.deviceId = item.deviceId
                  that.serviceId = uuid
                  for (let i = 0; i < res.characteristics.length; i++) {
                    if (res.characteristics[i].uuid.indexOf("FFF1") !== -1) {
                      that.characteristicId1 = res.characteristics[i].uuid
                    }
                    if (res.characteristics[i].uuid.indexOf("FFF2") !== -1) {
                      that.characteristicId2 = res.characteristics[i].uuid
                    }
                  }
                  that.showList = false
                  that.notifyBLECharacteristicValueChange()
                  if(uni.$u.os() === 'android') {
                    uni.setBLEMTU({
                      deviceId: that.deviceId,
                      mtu: 256,
                      success() {
                        console.log('mtu设置成功')
                      },
                      fail(res) {
                        console.log('mtu设置失败', res)
                      }
                    })
                  }
                  uni.showToast({
                    title: '数据获取成功',
                    icon: 'none',
                    duration: 2000
                  })
                },
                fail(res) {
                  console.log('获取特征值失败')
                  console.log(res)
                }
              })
            },
            fail(res) {
              console.log('获取服务失败')
              console.log(res)
            }
          })
        },
        fail(res) {
          console.log('连接失败')
          console.log(res)
        }
      })
    },
    // 监听蓝牙设备列表变化
    startGetList() {
      const that = this
      this.getting = true
      console.log('开始监听蓝牙列表更新')
      uni.startBluetoothDevicesDiscovery({
        allowDuplicatesKey: false,
        powerLevel: 'high',
        success: function (res) {
          // 更新列表
          console.log('列表发生变化', res)
          that.getList()
        }
      })
    },
    // 获取蓝牙设备列表
    getList() {
      const that = this
      uni.getBluetoothDevices({
        success(res) {
          console.log('获取设备列表成功')
          console.log(res)
          that.list = []
          for(let i =0; i < res.devices.length; i++){
            if(res.devices[i]?.advertisServiceUUIDs) {
              const advertisServiceUUIDs =
                  res.devices[i]?.advertisServiceUUIDs[0]
              if(advertisServiceUUIDs &&
                  advertisServiceUUIDs.slice(2,8)==='758824'){
                res.devices[i].pair =
                    advertisServiceUUIDs.slice(30,32) === '01'
                res.devices[i].sleep =
                    advertisServiceUUIDs.slice(32,34) === '00'
                that.list.push(res.devices[i])
              }
            }
          }
          console.log(that.list)
          if(!that.getting) {
            that.startGetList()
          }
        },
        fail(res) {
          console.log('获取设备列表失败')
          console.log(res)
        }
      })
    }
  }
}
</script>

<style>
.device {
  text-align: center;
}
</style>
