diff --git a/src/controller/order.controller.ts b/src/controller/order.controller.ts index 1f51d86..a11cf6d 100644 --- a/src/controller/order.controller.ts +++ b/src/controller/order.controller.ts @@ -2,6 +2,7 @@ import { Body, Controller, Del, + Files, Get, Inject, Param, @@ -27,7 +28,6 @@ import { } from '../dto/order.dto'; import { User } from '../decorator/user.decorator'; import { ErpOrderStatus } from '../enums/base.enum'; - @Controller('/order') export class OrderController { @Inject() @@ -264,4 +264,21 @@ export class OrderController { return errorResponse(error?.message || '导出失败'); } } + + + // 导入产品(CSV 文件) + @ApiOkResponse() + @Post('/import') + async importWintopay(@Files() files: any) { + try { + // 条件判断:确保存在文件 + const file = files?.[0]; + if (!file) return errorResponse('未接收到上传文件'); + + const result = await this.orderService.importWintopayTable(file); + return successResponse(result); + } catch (error) { + return errorResponse(error?.message || error); + } + } } diff --git a/src/service/order.service.ts b/src/service/order.service.ts index 9c55bb3..1e872f3 100644 --- a/src/service/order.service.ts +++ b/src/service/order.service.ts @@ -1,5 +1,6 @@ import { Inject, Logger, Provide } from '@midwayjs/core'; import { WPService } from './wp.service'; +import * as xlsx from 'xlsx'; import { Order } from '../entity/order.entity'; import { In, Like, Repository } from 'typeorm'; import { InjectEntityModel, TypeORMDataSourceManager } from '@midwayjs/typeorm'; @@ -32,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 { SyncOperationResult } from '../dto/api.dto'; +import { BatchErrorItem, SyncOperationResult } from '../dto/api.dto'; import * as fs from 'fs'; import * as path from 'path'; @@ -2580,8 +2581,8 @@ export class OrderService { * 导出数据为CSV格式 * @param {any[]} data 数据数组 * @param {Object} options 配置选项 - * @param {string} [options.type='string'] 输出类型:'string' | 'buffer' - * @param {string} [options.fileName] 文件名(仅当需要写入文件时使用) + * @param {string} [options.type='string'] 输出类型:'string' | 'buffer' + * @param {string} [options.fileName] 文件名(仅当需要写入文件时使用) * @param {boolean} [options.writeFile=false] 是否写入文件 * @returns {string|Buffer} 根据type返回字符串或Buffer */ @@ -2654,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 输入字符串 @@ -2729,4 +2806,108 @@ export class OrderService { return result; } + + // 从 CSV 导入产品;存在则更新,不存在则创建 + /** + * 导入 Wintopay 表格并回填物流信息 + * @param file 上传的文件 + * @returns 处理后的数据(包含更新的物流信息) + */ + async importWintopayTable(file: any): Promise { + let updated = 0; + const errors: BatchErrorItem[] = []; + + // 解析文件获取工作表 + 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; + + } + }