// websocket文件上传
import { getToken } from './auth'
export default function uploadFile (item, callback, errorCallBack) {
  const protocol = location.protocol === 'http:' ? 'ws' : 'wss'
  item.isReady = true;
  const socket = new WebSocket(`${protocol}://${location.host}/api/sms/uploadService`, getToken());
  let i = 0;
  let startSize = 0, endSize = 0;
  const paragraph = 4 * 1024 * 1024; // 以4MB为一个分片
  const count = parseInt(item.file.size / paragraph) + 1;
  socket.onopen = function () {
    item.isUploading = true;
    socket.send(JSON.stringify({
      filename: item.file.name,
      upload: 'file'
    }));
    // 取消上传
    item.cancel = function () {
      item.progress = 0;
      socket.send(JSON.stringify({
        UPLOAD_CANCEL: 'UPLOAD_CANCEL'
      }));
      item.isUploading = false;
    };
  };
  socket.onmessage = function (event) {
    const sendFile = function () {
      if (startSize < item.file.size) {
        let blob;
        endSize += paragraph;
        if (item.file.webkitSlice) {
          blob = item.file.webkitSlice(startSize, endSize);
        } else if (item.file.mozSlice) {
          blob = item.file.mozSlice(startSize, endSize);
        } else {
          blob = item.file.slice(startSize, endSize);
        }
        const reader = new FileReader();
        reader.readAsArrayBuffer(blob);

        reader.onload = function loaded (evt) {
          const result = evt.target.result;
          // i++;
          // const isok = (i / count) * 100;
          // item.progress = parseInt(isok);
          startSize = endSize;
          socket.send(result);
        };
      } else {
        item.progress = 100;
        socket.send(JSON.stringify({
          sendover: 'sendover'
        }));
      }
    };
    item.isUploading = true;
    item.isCancel = false;
    const obj = JSON.parse(event.data);
    if (obj.category == 'UPLOAD_ACK') {
      item.filePath = obj.content;
      sendFile();
    } else if (obj.category == 'UPLOAD') {
      if (obj.content == 'SAVE_FAILURE') {
        item.isUploading = false;
        errorCallBack(item);
      } else if (obj.content == 'SAVE_SUCCESS') {
        sendFile();
        i++;
        const isok = (i / count) * 100;
        item.progress = parseInt(isok);
      } else if (obj.content == 'TRUE') {
        callback(item);
        item.isReady = true;
        item.isSuccess = true;
        item.isUploading = false;
        socket.close();
      }
    } else if (obj.category == 'UPLOAD_CANCEL') {
      item.progress = 0;
      item.isCancel = true;
      socket.close();
    }
  };
}