|
|
import ExcelJS from "exceljs";
|
|
|
|
|
|
interface TrainReportData {
|
|
|
train_id: string;
|
|
|
arrive_at: string;
|
|
|
leave_at: string | null;
|
|
|
train_data: { carriage_number: string }[];
|
|
|
appearance_data: {
|
|
|
carriage_number: string;
|
|
|
carriage_type: string;
|
|
|
occurrence_time: string;
|
|
|
warning_type: string;
|
|
|
fault_type: string;
|
|
|
defect_image: string;
|
|
|
}[];
|
|
|
}
|
|
|
|
|
|
const formatDateTime = (dateTimeStr: string) => {
|
|
|
const date = new Date(dateTimeStr);
|
|
|
return date.toLocaleString("zh-CN", {
|
|
|
year: "numeric",
|
|
|
month: "2-digit",
|
|
|
day: "2-digit",
|
|
|
hour: "2-digit",
|
|
|
minute: "2-digit",
|
|
|
second: "2-digit",
|
|
|
hour12: false,
|
|
|
});
|
|
|
};
|
|
|
|
|
|
// 使用您指定的图片URL
|
|
|
const DEFAULT_IMAGE_URL =
|
|
|
"https://img1.baidu.com/it/u=967792105,51180745&fm=253&fmt=auto?w=1200&h=800";
|
|
|
const convertImageToBase64 = async (url: string) => {
|
|
|
try {
|
|
|
const response = await fetch(url);
|
|
|
const blob = await response.blob();
|
|
|
return new Promise<string>((resolve, reject) => {
|
|
|
const reader = new FileReader();
|
|
|
reader.onloadend = () => resolve(reader.result as string);
|
|
|
reader.onerror = reject;
|
|
|
reader.readAsDataURL(blob);
|
|
|
});
|
|
|
} catch (error) {
|
|
|
console.error("Error loading image:", error);
|
|
|
return null;
|
|
|
}
|
|
|
};
|
|
|
const getImageDimensions = (
|
|
|
url: string
|
|
|
): Promise<{ width: number; height: number }> => {
|
|
|
return new Promise((resolve) => {
|
|
|
const img = new Image();
|
|
|
const timer = setTimeout(() => {
|
|
|
resolve({ width: 80, height: 60 }); // 默认尺寸
|
|
|
}, 3000);
|
|
|
|
|
|
img.crossOrigin = "Anonymous";
|
|
|
img.onload = () => {
|
|
|
clearTimeout(timer);
|
|
|
resolve({ width: img.width, height: img.height });
|
|
|
};
|
|
|
|
|
|
img.onerror = () => {
|
|
|
clearTimeout(timer);
|
|
|
resolve({ width: 80, height: 60 });
|
|
|
};
|
|
|
|
|
|
img.src = url;
|
|
|
});
|
|
|
};
|
|
|
|
|
|
// 下载图片并转换为ArrayBuffer
|
|
|
const downloadImage = async (url: string): Promise<ExcelJS.Image | null> => {
|
|
|
try {
|
|
|
const finalUrl = url || DEFAULT_IMAGE_URL;
|
|
|
let response: Response;
|
|
|
|
|
|
try {
|
|
|
response = await fetch(finalUrl);
|
|
|
if (!response.ok) throw new Error("直接下载失败");
|
|
|
} catch (directError) {
|
|
|
// 尝试使用CORS代理
|
|
|
try {
|
|
|
const proxyUrl = `https://corsproxy.io/?${encodeURIComponent(
|
|
|
finalUrl
|
|
|
)}`;
|
|
|
response = await fetch(proxyUrl);
|
|
|
if (!response.ok) throw new Error("代理下载失败");
|
|
|
} catch (proxyError) {
|
|
|
console.error("图片下载失败:", proxyError);
|
|
|
return null;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
const arrayBuffer = await response.arrayBuffer();
|
|
|
const extension = finalUrl.split(".").pop()?.toLowerCase() || "jpeg";
|
|
|
const { width, height } = await getImageDimensions(finalUrl);
|
|
|
|
|
|
// 计算缩放比例（保持宽高比）
|
|
|
const maxWidth = 80;
|
|
|
const maxHeight = 60;
|
|
|
const scale = Math.min(maxWidth / width, maxHeight / height, 1);
|
|
|
|
|
|
return {
|
|
|
buffer: arrayBuffer, // 直接使用ArrayBuffer
|
|
|
extension: extension === "png" ? "png" : "jpeg",
|
|
|
width: width * scale,
|
|
|
height: height * scale,
|
|
|
};
|
|
|
} catch (error) {
|
|
|
console.error("图片处理失败:", error);
|
|
|
return null;
|
|
|
}
|
|
|
};
|
|
|
|
|
|
// 像素转换为Excel的EMU单位 (Excel使用EMU作为度量单位)
|
|
|
const pixelsToEMU = (pixels: number) => Math.round(pixels * 9525);
|
|
|
|
|
|
export const exportTrainReport = async (data: TrainReportData) => {
|
|
|
const workbook = new ExcelJS.Workbook();
|
|
|
workbook.creator = "Railway Inspection System";
|
|
|
workbook.created = new Date();
|
|
|
|
|
|
// 1. 创建"整车信息"工作表 (无表头行)
|
|
|
const trainSheet = workbook.addWorksheet("整车信息");
|
|
|
trainSheet.getColumn(1).width = 20;
|
|
|
trainSheet.getColumn(2).width = 30;
|
|
|
|
|
|
// 直接添加数据行
|
|
|
const trainInfoData = [
|
|
|
{ key: "车头编号", value: data.train_id },
|
|
|
{ key: "进场时间", value: formatDateTime(data.arrive_at) },
|
|
|
{
|
|
|
key: "出场时间",
|
|
|
value: data.leave_at ? formatDateTime(data.leave_at) : "未出场",
|
|
|
},
|
|
|
{ key: "车厢列表", value: "" },
|
|
|
];
|
|
|
|
|
|
trainInfoData.forEach((info, index) => {
|
|
|
const row = index + 1;
|
|
|
trainSheet.getCell(`A${row}`).value = info.key;
|
|
|
trainSheet.getCell(`B${row}`).value = info.value;
|
|
|
|
|
|
// 设置第一列加粗
|
|
|
trainSheet.getCell(`A${row}`).font = { bold: true };
|
|
|
|
|
|
// 设置第二列加粗（除车厢列表行）
|
|
|
if (row < 4) {
|
|
|
trainSheet.getCell(`B${row}`).font = { bold: true };
|
|
|
}
|
|
|
});
|
|
|
|
|
|
// 添加车厢列表
|
|
|
const startRow = trainInfoData.length + 1;
|
|
|
data.train_data.forEach((carriage, index) => {
|
|
|
const row = startRow + index;
|
|
|
trainSheet.getCell(`A${row}`).value = `车厢 ${index + 1}`;
|
|
|
trainSheet.getCell(`B${row}`).value = carriage.carriage_number;
|
|
|
});
|
|
|
|
|
|
// 2. 创建"车身缺陷记录"工作表
|
|
|
const defectSheet = workbook.addWorksheet("车身缺陷记录");
|
|
|
|
|
|
// 修改列定义
|
|
|
defectSheet.columns = [
|
|
|
{ header: "车厢号", key: "carriage_number", width: 15 },
|
|
|
// { header: "车型", key: "carriage_type", width: 20 },
|
|
|
{ header: "发生时间", key: "occurrence_time", width: 20 },
|
|
|
{ header: "告警类型", key: "warning_type", width: 15 },
|
|
|
{ header: "缺陷类型", key: "fault_type", width: 15 },
|
|
|
{ header: "缺陷图片", key: "thumbnail", width: 15 }, // 缩略图列
|
|
|
// { header: "查看原图", key: "defect_image", width: 15 }, // 原图链接列
|
|
|
];
|
|
|
|
|
|
// 设置表头样式
|
|
|
defectSheet.getRow(1).eachCell((cell) => {
|
|
|
cell.font = { bold: true };
|
|
|
cell.fill = {
|
|
|
type: "pattern",
|
|
|
pattern: "solid",
|
|
|
fgColor: { argb: "FFDDEBF7" }, // 浅蓝色背景
|
|
|
};
|
|
|
cell.border = {
|
|
|
top: { style: "thin" },
|
|
|
left: { style: "thin" },
|
|
|
bottom: { style: "thin" },
|
|
|
right: { style: "thin" },
|
|
|
};
|
|
|
});
|
|
|
|
|
|
// 设置所有行高（60像素≈45磅）
|
|
|
defectSheet.properties.defaultRowHeight = 45;
|
|
|
|
|
|
// 添加所有缺陷记录
|
|
|
for (let i = 0; i < data.appearance_data.length; i++) {
|
|
|
const item = data.appearance_data[i];
|
|
|
const rowIndex = i + 2; // 从第二行开始（第一行是表头）
|
|
|
|
|
|
// 添加文本数据
|
|
|
defectSheet.addRow({
|
|
|
carriage_number: item.carriage_number,
|
|
|
carriage_type: item.carriage_type,
|
|
|
occurrence_time: formatDateTime(item.occurrence_time),
|
|
|
warning_type: item.warning_type,
|
|
|
fault_type: item.fault_type,
|
|
|
});
|
|
|
|
|
|
// 使用指定图片URL
|
|
|
const imageUrl = item.defect_image || DEFAULT_IMAGE_URL;
|
|
|
|
|
|
try {
|
|
|
// 下载图片
|
|
|
// const image = await downloadImage(imageUrl);
|
|
|
// console.log("图片下载成功", image);
|
|
|
const base64Image = await convertImageToBase64(imageUrl);
|
|
|
if (base64Image) {
|
|
|
const imageId = workbook.addImage({
|
|
|
base64: base64Image,
|
|
|
extension: "png", // Adjust extension if needed (jpeg, etc.)
|
|
|
});
|
|
|
|
|
|
// TODO : 调整图片大小和位置 (根据需要调整)
|
|
|
// // 计算缩放比例和显示尺寸
|
|
|
// const maxSize = 30; // 最大边长30像素
|
|
|
// const scale = Math.min(maxSize / image.width, maxSize / image.height);
|
|
|
// const displayWidth = image.width * scale;
|
|
|
// const displayHeight = image.height * scale;
|
|
|
|
|
|
// // 列宽（像素）和行高（像素）
|
|
|
// const colWidthPixels = 105; // 15字符 * 7
|
|
|
// const rowHeightPixels = 60; // 45磅 * (96/72) = 60像素
|
|
|
|
|
|
// // 计算偏移量（居中）
|
|
|
// const offsetXPixels = (colWidthPixels - displayWidth) / 2;
|
|
|
// const offsetYPixels = (rowHeightPixels - displayHeight) / 2;
|
|
|
|
|
|
// // 转换为EMU
|
|
|
// const offsetXEMU = pixelsToEMU(offsetXPixels);
|
|
|
// const offsetYEMU = pixelsToEMU(offsetYPixels);
|
|
|
// const widthEMU = pixelsToEMU(displayWidth);
|
|
|
// const heightEMU = pixelsToEMU(displayHeight);
|
|
|
defectSheet.addImage(imageId, {
|
|
|
tl: { col: defectSheet.columns.length - 1, row: rowIndex - 1 }, // F列是第5列（0-based）
|
|
|
// br: { col: 6, row: rowIndex },
|
|
|
ext: {
|
|
|
width: 80,
|
|
|
height: 45,
|
|
|
},
|
|
|
editAs: "oneCell",
|
|
|
});
|
|
|
|
|
|
|
|
|
}
|
|
|
} catch (error) {
|
|
|
console.error(`图片添加失败 (行 ${rowIndex}):`, error);
|
|
|
const errorCell = defectSheet.getCell(`E${rowIndex}`);
|
|
|
errorCell.value = "图片加载失败";
|
|
|
errorCell.font = {
|
|
|
color: { argb: "FFFF0000" }, // 红色
|
|
|
italic: true,
|
|
|
};
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// 3. 生成Excel文件
|
|
|
const buffer = await workbook.xlsx.writeBuffer();
|
|
|
try {
|
|
|
const blob = new Blob([buffer], {
|
|
|
type: "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
|
|
|
});
|
|
|
const url = URL.createObjectURL(blob);
|
|
|
const link = document.createElement("a");
|
|
|
link.href = url;
|
|
|
link.download = "车辆信息.xlsx";
|
|
|
link.click();
|
|
|
URL.revokeObjectURL(url);
|
|
|
} catch (error) {
|
|
|
console.error("导出 Excel 文件时出错:", error);
|
|
|
}
|
|
|
};
|
|
|
// ... formatDateTime 函数保持不变 ...
|
|
|
|
|
|
export const useTrainSaveToExcel = (data) => {
|
|
|
// 模拟数据 (实际应用中应通过API获取)
|
|
|
const reportData = {
|
|
|
train_id: "DF4-1234",
|
|
|
arrive_at: "2023-06-15 08:30:00",
|
|
|
leave_at: "2023-06-15 16:45:22",
|
|
|
train_data: [
|
|
|
{ carriage_number: "CK-1001" },
|
|
|
{ carriage_number: "CK-1002" },
|
|
|
{ carriage_number: "CK-1003" },
|
|
|
],
|
|
|
appearance_data: [
|
|
|
{
|
|
|
carriage_number: "CK-1001",
|
|
|
carriage_type: "C80型运煤敞车",
|
|
|
occurrence_time: "2023-06-15 09:15:34",
|
|
|
warning_type: "AI智能识别告警",
|
|
|
fault_type: "车体变形",
|
|
|
defect_image:
|
|
|
"http://192.168.10.14:8123/sftp/2025-07-01/20250701_131010.jpg",
|
|
|
},
|
|
|
{
|
|
|
carriage_number: "CK-1002",
|
|
|
carriage_type: "P70型棚车",
|
|
|
occurrence_time: "2023-06-15 10:22:18",
|
|
|
warning_type: "人工巡检报告",
|
|
|
fault_type: "油漆剥落",
|
|
|
defect_image:
|
|
|
"http://192.168.10.14:8123/sftp/2025-07-01/20250701_131010.jpg",
|
|
|
},
|
|
|
{
|
|
|
carriage_number: "CK-1001",
|
|
|
carriage_type: "C80型运煤敞车",
|
|
|
occurrence_time: "2023-06-15 09:15:34",
|
|
|
warning_type: "AI智能识别告警",
|
|
|
fault_type: "车体变形",
|
|
|
defect_image:
|
|
|
"http://192.168.10.14:8123/sftp/2025-07-01/20250701_131010.jpg",
|
|
|
},
|
|
|
{
|
|
|
carriage_number: "CK-1002",
|
|
|
carriage_type: "P70型棚车",
|
|
|
occurrence_time: "2023-06-15 10:22:18",
|
|
|
warning_type: "人工巡检报告",
|
|
|
fault_type: "油漆剥落",
|
|
|
defect_image:
|
|
|
"http://192.168.10.14:8123/sftp/2025-07-01/20250701_131010.jpg",
|
|
|
},
|
|
|
{
|
|
|
carriage_number: "CK-1001",
|
|
|
carriage_type: "C80型运煤敞车",
|
|
|
occurrence_time: "2023-06-15 09:15:34",
|
|
|
warning_type: "AI智能识别告警",
|
|
|
fault_type: "车体变形",
|
|
|
defect_image:
|
|
|
"http://192.168.10.14:8123/sftp/2025-07-01/20250701_131010.jpg",
|
|
|
},
|
|
|
{
|
|
|
carriage_number: "CK-1002",
|
|
|
carriage_type: "P70型棚车",
|
|
|
occurrence_time: "2023-06-15 10:22:18",
|
|
|
warning_type: "人工巡检报告",
|
|
|
fault_type: "油漆剥落",
|
|
|
defect_image:
|
|
|
"http://192.168.10.14:8123/sftp/2025-07-01/20250701_131010.jpg",
|
|
|
},
|
|
|
{
|
|
|
carriage_number: "CK-1001",
|
|
|
carriage_type: "C80型运煤敞车",
|
|
|
occurrence_time: "2023-06-15 09:15:34",
|
|
|
warning_type: "AI智能识别告警",
|
|
|
fault_type: "车体变形",
|
|
|
defect_image:
|
|
|
"http://192.168.10.14:8123/sftp/2025-07-01/20250701_131010.jpg",
|
|
|
},
|
|
|
{
|
|
|
carriage_number: "CK-1002",
|
|
|
carriage_type: "P70型棚车",
|
|
|
occurrence_time: "2023-06-15 10:22:18",
|
|
|
warning_type: "人工巡检报告",
|
|
|
fault_type: "油漆剥落",
|
|
|
defect_image:
|
|
|
"http://192.168.10.14:8123/sftp/2025-07-01/20250701_131010.jpg",
|
|
|
},
|
|
|
{
|
|
|
carriage_number: "CK-1001",
|
|
|
carriage_type: "C80型运煤敞车",
|
|
|
occurrence_time: "2023-06-15 09:15:34",
|
|
|
warning_type: "AI智能识别告警",
|
|
|
fault_type: "车体变形",
|
|
|
defect_image:
|
|
|
"http://192.168.10.14:8123/sftp/2025-07-01/20250701_131010.jpg",
|
|
|
},
|
|
|
{
|
|
|
carriage_number: "CK-1002",
|
|
|
carriage_type: "P70型棚车",
|
|
|
occurrence_time: "2023-06-15 10:22:18",
|
|
|
warning_type: "人工巡检报告",
|
|
|
fault_type: "油漆剥落",
|
|
|
defect_image:
|
|
|
"http://192.168.10.14:8123/sftp/2025-07-01/20250701_131010.jpg",
|
|
|
},
|
|
|
],
|
|
|
};
|
|
|
return {
|
|
|
saveToExcel: () => exportTrainReport(reportData),
|
|
|
};
|
|
|
};
|