import { utils, write } from 'xlsx'
import Papa from 'papaparse'

// 获取合并头部的数据 level: 头部分有几行
function mergeHeader(worksheet, cellList, level = 3) {
  // 并获取 列名
  const colName = []
  for (const key of Object.keys(worksheet)) {
    // 根据第一行的数据
    if (key.slice(-1) === '1' && isNaN(Number(key.slice(-2, -1)))) {
      colName.push(key.slice(0, -1))
    } else {
      break
    }
  }

  // 合并行 axle true 行 false 列
  function getSpaceNumber(dataCopy, rowIndex, colIndex, axle) {
    const data = JSON.parse(JSON.stringify(dataCopy))

    let count = 0
    axle ? (rowIndex += 1) : (colIndex += 1)
    count = isSpaceString(data[rowIndex][colIndex], axle, count)

    function isSpaceString(value, axle, count) {
      if (value === '' && axle && rowIndex < level) {
        // 行合并
        count += 1
        rowIndex += 1

        if (rowIndex >= data.length - 1) {
          return count
        }

        return isSpaceString(data[rowIndex][colIndex], axle, count)
      } else if (value === '' && !axle && colIndex < colName.length) {
        // 列合并
        count += 1
        colIndex += 1

        if (colIndex >= colName.length) {
          return count
        }

        return isSpaceString(data[rowIndex][colIndex], axle, count)
      } else {
        return count
      }
    }

    return count
  }

  const merges = []
  for (const rowIndex in cellList) {
    if (rowIndex > level - 1) {
      break
    }

    const rowCurrnetIndex = Number(rowIndex) + 1
    for (const cellIndex in cellList[rowIndex]) {
      if (cellIndex > colName.length - 1) {
        break
      }

      if (cellList[rowIndex][cellIndex] !== '') {
        const colCurrnetIndex = Number(cellIndex)

        let colCount = 0
        let rowCount = 0

        // 单独判断一列（特殊情况）
        if (cellList[rowIndex][cellIndex] === '其中：三次分配（奖项/专项）') {
          // 列合并
          rowCount = getSpaceNumber(cellList, Number(rowIndex), Number(cellIndex), true)
        } else {
          // 列合并
          colCount = getSpaceNumber(cellList, Number(rowIndex), Number(cellIndex), false)
          // 行合并
          rowCount = getSpaceNumber(cellList, Number(rowIndex), Number(cellIndex), true)
        }

        if (colCount || rowCount) {
          merges.push(
            `${colName[colCurrnetIndex]}${rowCurrnetIndex}:${colName[colCurrnetIndex + colCount]}${
              rowCurrnetIndex + rowCount
            }`
          )
        }
      }
    }
  }

  return merges
}

export default function pasteToFile(event) {
  // 判断是否有粘贴的数据
  if (event.clipboardData || event.originalEvent) {
    const clipboardData = event.clipboardData || window.clipboardData

    // 获取粘贴的数据并解析
    const xlsxText = Papa.parse(clipboardData.getData('text'), { delimiter: '\t', linebreak: '\r\n' }).data

    // 两行及以下的数据不处理
    if (xlsxText.length < 1) {
      return
    }

    // windows 最后一行全是空格, 手动去掉
    xlsxText[xlsxText.length - 1].every(item => item === '') && xlsxText.pop()

    // 去掉全是空格的行
    const allIsSpace = []
    for (const index in xlsxText) {
      xlsxText[Number(index)].every(item => item === '') && allIsSpace.push(index)
    }
    for (const index of allIsSpace) {
      xlsxText.splice(Number(index), 1)
    }

    // 创建一个新的工作簿
    const workbook = utils.book_new()

    // 创建一个工作表
    const worksheet = utils.aoa_to_sheet(xlsxText)

    // 合并单元格
    // const merges = mergeHeader(worksheet, xlsxText)

    // // 判断是否有需要合并的单元格
    // if (merges.length > 0) {
    //   if (!worksheet['!merges']) {
    //     worksheet['!merges'] = []
    //   }

    //   // 将合并的单元格添加到工作表
    //   merges.forEach(item => {
    //     worksheet['!merges'].push(utils.decode_range(item))
    //   })
    // }

    // 将工作表添加到工作簿
    utils.book_append_sheet(workbook, worksheet, 'Sheet1')

    // 将工作簿保存为Excel文件
    // writeFile(workbook, "data.xlsx");

    // 将工作簿保存为ArrayBuffer
    const data = write(workbook, { type: 'array', bookType: 'xlsx' })

    // 将ArrayBuffer转为Blob
    return new Blob([data])
  }
}