index.js

const path = require('path');
const fs = require('fs');
const os = require('os');
const crypto = require('crypto');

// Try to load the native addon
let addon;
try {
  addon = require('./build/Release/baja_xlsx.node');
} catch (err) {
  try {
    addon = require('./build/Debug/baja_xlsx.node');
  } catch (err2) {
    throw new Error(
      'Native addon not found. Please run "npm install" or "npm run build" first.\n' +
      'Make sure you have installed xlnt library via vcpkg or system package manager.'
    );
  }
}

/**
 * 处理不同类型的输入，统一转换为文件路径
 * @param {string|Buffer} input - 文件路径、Buffer 或 base64 字符串
 * @returns {{filepath: string, cleanup: function}} 文件路径和清理函数
 * @private
 */
function prepareFilePath(input) {
  // 情况1: Buffer
  if (Buffer.isBuffer(input)) {
    // 创建临时文件
    const tempDir = os.tmpdir();
    const tempFile = path.join(tempDir, `excel-${crypto.randomBytes(16).toString('hex')}.xlsx`);
    
    fs.writeFileSync(tempFile, input);
    
    return {
      filepath: tempFile,
      cleanup: () => {
        try {
          fs.unlinkSync(tempFile);
        } catch (err) {
          // 忽略删除错误
        }
      }
    };
  }
  
  // 情况2: 字符串
  if (typeof input === 'string') {
    // 检查是否是 base64
    // base64 特征：很长的字符串（>500），只包含 base64 字符，且包含常见的文件头（如 UEs/PK）
    const isBase64 = input.length > 500 &&
                     /^[A-Za-z0-9+/=\s]+$/.test(input) &&
                     (input.startsWith('UEs') || input.startsWith('PK') || 
                      input.includes('AAAA') || input.includes('////'));
    
    if (isBase64) {
      // base64 字符串，转换为 Buffer 然后写入临时文件
      const buffer = Buffer.from(input, 'base64');
      const tempDir = os.tmpdir();
      const tempFile = path.join(tempDir, `excel-${crypto.randomBytes(16).toString('hex')}.xlsx`);
      
      fs.writeFileSync(tempFile, buffer);
      
      return {
        filepath: tempFile,
        cleanup: () => {
          try {
            fs.unlinkSync(tempFile);
          } catch (err) {
            // 忽略删除错误
          }
        }
      };
    } else {
      // 文件路径
      const absolutePath = path.isAbsolute(input) ? input : path.resolve(input);
      
      if (!fs.existsSync(absolutePath)) {
        throw new Error(`File not found: ${absolutePath}`);
      }
      
      return {
        filepath: absolutePath,
        cleanup: () => {} // 不需要清理
      };
    }
  }
  
  throw new Error('Input must be a file path (string), Buffer, or base64 string');
}


/**
 * 读取Excel表格并返回JSON数组
 * @param {string|Buffer} input - Excel文件路径、Buffer 或 base64 字符串
 * @param {Object} options - 配置选项
 * @param {string} [options.sheetName] - 指定Sheet名称，不传则读取第一个Sheet
 * @param {number} [options.headerRow=0] - 表头所在行索引（从0开始）
 * @param {number[]} [options.skipRows=[]] - 需要跳过的行索引数组
 * @param {Object<string, string>} [options.headerMap={}] - 表头映射，将原表头映射为新的属性名
 * @returns {Array<Object>} JSON数组，每个元素代表一行数据
 * 
 * @example
 * // 使用文件路径
 * const data1 = readTableAsJSON('./sample.xlsx', {
 *   sheetName: 'Sheet1',
 *   headerRow: 0,
 *   skipRows: [1, 2],
 *   headerMap: {
 *     '名称': 'name',
 *     '年龄': 'age'
 *   }
 * });
 * 
 * // 使用 Buffer
 * const buffer = fs.readFileSync('./sample.xlsx');
 * const data2 = readTableAsJSON(buffer, { headerRow: 0 });
 * 
 * // 使用 base64
 * const base64 = buffer.toString('base64');
 * const data3 = readTableAsJSON(base64, { headerRow: 0 });
 */
function readTableAsJSON(input, options = {}) {
  if (!input) {
    throw new Error('Input is required (filepath, Buffer, or base64 string)');
  }
  
  const { filepath, cleanup } = prepareFilePath(input);
  
  try {
    // 默认选项
    const {
      sheetName = null,
      headerRow = 0,
      skipRows = [],
      headerMap = {}
    } = options;
    
    // 读取Excel数据
    const excelData = addon.readExcel(filepath);
  
  // 选择目标Sheet
  let targetSheet;
  if (sheetName) {
    targetSheet = excelData.sheets.find(sheet => sheet.name === sheetName);
    if (!targetSheet) {
      throw new Error(`未找到名为 "${sheetName}" 的Sheet`);
    }
  } else {
    if (excelData.sheets.length === 0) {
      throw new Error('Excel文件中没有Sheet');
    }
    targetSheet = excelData.sheets[0];
  }
  
  const sheetData = targetSheet.data;
  
  // 检查数据是否足够
  if (sheetData.length <= headerRow) {
    throw new Error(`表头行索引 ${headerRow} 超出数据范围（共 ${sheetData.length} 行）`);
  }
  
  // 获取表头
  const headers = sheetData[headerRow];
  
  // 应用表头映射
  const mappedHeaders = headers.map(header => {
    return headerMap[header] || header;
  });
  
  // 创建跳过行的Set（包含表头行）
  const skipRowsSet = new Set([headerRow, ...skipRows]);
  
  // 构建JSON数组
  const result = [];
  
  for (let rowIndex = 0; rowIndex < sheetData.length; rowIndex++) {
    // 跳过指定的行
    if (skipRowsSet.has(rowIndex)) {
      continue;
    }
    
    const row = sheetData[rowIndex];
    const rowObj = {};
    
    // 填充数据
    // 注意：所有图片（嵌入式和浮动图片）都已经在 C++ 层直接转换为 { data: Buffer, name, type } 对象
    for (let colIndex = 0; colIndex < mappedHeaders.length; colIndex++) {
      const header = mappedHeaders[colIndex];
      const value = row[colIndex] || '';
      
      // 只有在表头不为空时才添加属性
      if (header) {
        rowObj[header] = value;
      }
    }
    
    result.push(rowObj);
  }
  
  return result;
  } finally {
    cleanup();
  }
}


module.exports = {
  readTableAsJSON
};