diff --git a/src/service/order.service.ts b/src/service/order.service.ts index bb8d46e..1e872f3 100644 --- a/src/service/order.service.ts +++ b/src/service/order.service.ts @@ -33,7 +33,7 @@ import { UpdateStockDTO } from '../dto/stock.dto'; import { StockService } from './stock.service'; import { OrderItemOriginal } from '../entity/order_item_original.entity'; import { SiteApiService } from './site-api.service'; -import { BatchErrorItem, SyncOperationResult,BatchOperationResult } from '../dto/api.dto'; +import { BatchErrorItem, SyncOperationResult } from '../dto/api.dto'; import * as fs from 'fs'; import * as path from 'path'; @@ -2655,6 +2655,82 @@ export class OrderService { } } + /** + * 导出数据为XLSX格式 + * @param {any[]} data 数据数组 + * @param {Object} options 配置选项 + * @param {string} [options.type='buffer'] 输出类型:'buffer' | 'string' (XLSX默认返回buffer) + * @param {string} [options.fileName] 文件名(仅当需要写入文件时使用) + * @param {boolean} [options.writeFile=false] 是否写入文件 + * @returns {string|Buffer} 根据type返回字符串或Buffer + */ + async exportToXlsx(data: any[], options: { type?: 'buffer' | 'string'; fileName?: string; writeFile?: boolean } = {}): Promise { + try { + // 检查数据是否为空 + if (!data || data.length === 0) { + throw new Error('导出数据不能为空'); + } + + const { type = 'buffer', fileName, writeFile = false } = options; + + // 获取表头 + const headers = Object.keys(data[0]); + + // 构建二维数组数据(包含表头) + const aoaData = [headers]; + data.forEach(item => { + const row = headers.map(key => { + const value = item[key as keyof any]; + // 处理undefined和null + if (value === undefined || value === null) { + return ''; + } + // 处理日期类型 + if (value instanceof Date) { + return value.toISOString(); + } + return value; + }); + aoaData.push(row); + }); + + // 创建工作簿和工作表 + const workbook = xlsx.utils.book_new(); + const worksheet = xlsx.utils.aoa_to_sheet(aoaData); + xlsx.utils.book_append_sheet(workbook, worksheet, 'Sheet1'); + + // 生成XLSX buffer + const buffer = xlsx.write(workbook, { bookType: 'xlsx', type: 'buffer' }); + + // 如果需要写入文件 + if (writeFile && fileName) { + // 获取当前用户目录 + const userHomeDir = os.homedir(); + + // 构建目标路径(下载目录) + const downloadsDir = path.join(userHomeDir, 'Downloads'); + + // 确保下载目录存在 + if (!fs.existsSync(downloadsDir)) { + fs.mkdirSync(downloadsDir, { recursive: true }); + } + const filePath = path.join(downloadsDir, fileName); + // 写入文件 + fs.writeFileSync(filePath, buffer); + return filePath; + } + + // 根据类型返回不同结果 + if (type === 'string') { + return buffer.toString('base64'); + } + + return buffer; + } catch (error) { + throw new Error(`导出XLSX文件失败: ${error.message}`); + } + } + /** * 删除每个分号前面一个左右括号和最后一个左右括号包含的内容(包括括号本身) * @param str 输入字符串 @@ -2731,57 +2807,107 @@ export class OrderService { return result; } - - // 从 CSV 导入产品;存在则更新,不存在则创建 - async importWintopayTable(file: any): Promise { - let created = 0; + /** + * 导入 Wintopay 表格并回填物流信息 + * @param file 上传的文件 + * @returns 处理后的数据(包含更新的物流信息) + */ + async importWintopayTable(file: any): Promise { let updated = 0; const errors: BatchErrorItem[] = []; - const records = await this.getRecordsFromTable(file); -created++ -updated++ + + // 解析文件获取工作表 + let buffer: Buffer; + if (Buffer.isBuffer(file)) { + buffer = file; + } else if (file?.data) { + if (typeof file.data === 'string') { + buffer = fs.readFileSync(file.data); + } else { + buffer = file.data; + } + } else { + throw new Error('无效的文件输入'); + } + + const workbook = xlsx.read(buffer, { type: 'buffer', codepage: 65001 }); + const worksheet = workbook.Sheets[workbook.SheetNames[0]]; + + // 获取表头和数据 + const jsonData = xlsx.utils.sheet_to_json(worksheet, { header: 1 }); + const headers = jsonData[0] as string[]; + const dataRows = jsonData.slice(1) as any[][]; + + // 查找各列的索引 + const columnIndices = { + orderNumber: headers.indexOf('订单号'), + logisticsCompany: headers.indexOf('物流公司'), + trackingNumber: headers.indexOf('单号-单元格文本格式'), + orderCreateTime: headers.indexOf('订单创建时间'), + orderEmail: headers.indexOf('订单邮箱'), + orderSite: headers.indexOf('订单网站'), + name: headers.indexOf('姓名'), + refund: headers.indexOf('退款'), + chargeback: headers.indexOf('拒付') + }; + + // 遍历数据行 + for (let i = 0; i < dataRows.length; i++) { + const row = dataRows[i]; + const orderNumber = row[columnIndices.orderNumber]; + + if (!orderNumber) { + errors.push({ identifier: `行 ${i + 2}`, error: '订单号为空' }); + continue; + } + + try { + let orderNumbers=""; + if (orderNumber.includes('_')&&orderNumber.includes('-')) { + orderNumbers = orderNumber.split('_')[0].toString(); + orderNumbers = orderNumbers.split('-')[1]; + } + // 通过订单号查询订单 + const order = await this.orderModel.findOne({ where: { externalOrderId: orderNumbers } }); + if (order) { + // 通过orderId查询fulfillments + const fulfillments = await this.orderFulfillmentModel.find({ where: { order_id: order.id } }); + + if (fulfillments && fulfillments.length > 0) { + const fulfillment = fulfillments[0]; // 假设每个订单只有一个物流信息 + // 回填物流信息 + if (columnIndices.logisticsCompany !== -1) { + row[columnIndices.logisticsCompany] = fulfillment.shipping_provider || ''; + } + if (columnIndices.trackingNumber !== -1) { + row[columnIndices.trackingNumber] = fulfillment.tracking_number || ''; + } + + updated++; + } + } + } catch (error) { + errors.push({ identifier: `行 ${i + 2}`, error: `处理失败: ${error.message}` }); + } + } + + // 将数据转换为对象数组,与 exportOrder 方法返回格式一致 + const resultData = dataRows.map((row, index) => { + const rowData: any = {}; + headers.forEach((header, colIndex) => { + rowData[header] = row[colIndex] || ''; + }); + // 添加行号信息 + rowData['行号'] = index + 2; + return rowData; + }); + + + // 返回XLSX buffer内容给前端 + // const xlsxBuffer = await this.exportToXlsx(resultData, { type: 'buffer' }); + return resultData; - return { total: records.length, processed: records.length - errors.length, created: 0, updated: 0, errors }; } - async getRecordsFromTable(file: any) { - // 解析文件(使用 xlsx 包自动识别文件类型并解析) - try { - let buffer: Buffer; - - // 处理文件输入,获取 buffer - if (Buffer.isBuffer(file)) { - buffer = file; - } - else if (file?.data) { - if (typeof file.data === 'string') { - buffer = fs.readFileSync(file.data); - } else { - buffer = file.data; - } - } else { - throw new Error('无效的文件输入'); - } - - let records: any[] = [] - // xlsx 包会自动根据文件内容识别文件类型(CSV 或 XLSX) - // 添加codepage: 65001以确保正确处理UTF-8编码的中文 - const workbook = xlsx.read(buffer, { type: 'buffer', codepage: 65001 }); - // 获取第一个工作表 - const worksheet = workbook.Sheets[workbook.SheetNames[0]]; - // 将工作表转换为 JSON 数组 - records = xlsx.utils.sheet_to_json(worksheet); - - console.log('Parsed records count:', records.length); - if (records.length > 0) { - console.log('First record keys:', Object.keys(records[0])); - } - return records; - } catch (e: any) { - throw new Error(`文件解析失败:${e?.message || e}`); - } - - - } }