Files
cunkebao_v3/Cunkebao/app/contact-import/page.tsx

331 lines
11 KiB
TypeScript
Raw Normal View History

2025-04-08 16:01:59 +08:00
"use client";
import React, { useState } from "react";
import * as XLSX from "xlsx";
import { Button } from "@/components/ui/button";
import { Card } from "@/components/ui/card";
import { Input } from "@/components/ui/input";
import { Alert, AlertDescription } from "@/components/ui/alert";
import { AlertCircle } from "lucide-react";
interface ContactData {
mobile: number;
from: string;
alias: string;
}
export default function ContactImportPage() {
const [parsedData, setParsedData] = useState<ContactData[]>([]);
const [error, setError] = useState<string | null>(null);
const [fileName, setFileName] = useState<string>("");
const [isImportSuccessful, setIsImportSuccessful] = useState(false);
const [isProcessing, setIsProcessing] = useState(false);
const [detectedColumns, setDetectedColumns] = useState<{
mobile?: string;
from?: string;
alias?: string;
}>({});
const handleFileChange = (e: React.ChangeEvent<HTMLInputElement>) => {
const files = e.target.files;
if (!files || files.length === 0) {
return;
}
// 重置状态
setError(null);
setIsImportSuccessful(false);
setParsedData([]);
setDetectedColumns({});
setIsProcessing(true);
const file = files[0];
setFileName(file.name);
const reader = new FileReader();
reader.onload = (event) => {
try {
const data = event.target?.result;
if (!data) {
setError("文件内容为空,请检查文件是否有效");
setIsProcessing(false);
return;
}
let workbook;
try {
workbook = XLSX.read(data, { type: "binary" });
} catch (parseErr) {
console.error("解析Excel内容失败:", parseErr);
setError("无法解析文件内容请确保上传的是有效的Excel文件(.xlsx或.xls格式)");
setIsProcessing(false);
return;
}
if (!workbook.SheetNames || workbook.SheetNames.length === 0) {
setError("Excel文件中没有找到工作表");
setIsProcessing(false);
return;
}
const sheetName = workbook.SheetNames[0];
const worksheet = workbook.Sheets[sheetName];
if (!worksheet) {
setError(`无法读取工作表 "${sheetName}",请检查文件是否损坏`);
setIsProcessing(false);
return;
}
// 转换为JSON
const jsonData = XLSX.utils.sheet_to_json(worksheet, { header: "A" });
// 检查是否有数据
if (!jsonData || jsonData.length === 0) {
setError("Excel 文件中没有数据");
setIsProcessing(false);
return;
}
// 查找栏位对应的列
let mobileColumn: string | null = null;
let fromColumn: string | null = null;
let aliasColumn: string | null = null;
// 遍历第一行查找栏位
const firstRow = jsonData[0] as Record<string, any>;
if (!firstRow) {
setError("Excel 文件的第一行为空,无法识别栏位");
setIsProcessing(false);
return;
}
for (const key in firstRow) {
if (!firstRow[key]) continue; // 跳过空值
const value = String(firstRow[key]).toLowerCase();
// 扩展匹配列表,提高识别成功率
if (value.includes("手机") || value.includes("电话") || value.includes("mobile") ||
value.includes("phone") || value.includes("tel") || value.includes("cell")) {
mobileColumn = key;
} else if (value.includes("来源") || value.includes("source") || value.includes("from") ||
value.includes("channel") || value.includes("渠道")) {
fromColumn = key;
} else if (value.includes("微信") || value.includes("alias") || value.includes("wechat") ||
value.includes("wx") || value.includes("id") || value.includes("账号")) {
aliasColumn = key;
}
}
// 保存检测到的列名
if (mobileColumn && firstRow[mobileColumn]) {
setDetectedColumns(prev => ({ ...prev, mobile: String(firstRow[mobileColumn]) }));
}
if (fromColumn && firstRow[fromColumn]) {
setDetectedColumns(prev => ({ ...prev, from: String(firstRow[fromColumn]) }));
}
if (aliasColumn && firstRow[aliasColumn]) {
setDetectedColumns(prev => ({ ...prev, alias: String(firstRow[aliasColumn]) }));
}
if (!mobileColumn) {
setError("未找到手机号码栏位请确保Excel中包含手机、电话、mobile或phone等栏位名称");
setIsProcessing(false);
return;
}
// 解析数据,跳过首行
const importedData: ContactData[] = [];
for (let i = 1; i < jsonData.length; i++) {
const row = jsonData[i] as Record<string, any>;
// 检查行是否存在且有手机号列
if (!row || row[mobileColumn] === undefined || row[mobileColumn] === null) {
continue;
}
// 安全地转换并清理手机号
const mobileStr = row[mobileColumn] !== undefined && row[mobileColumn] !== null
? String(row[mobileColumn]).trim()
: "";
// 过滤非数字字符,确保手机号是纯数字
const mobile = mobileStr.replace(/\D/g, "");
// 手机号为空的跳过
if (!mobile) continue;
// 安全地获取来源和别名
const from = fromColumn && row[fromColumn] !== undefined && row[fromColumn] !== null
? String(row[fromColumn]).trim()
: "";
const alias = aliasColumn && row[aliasColumn] !== undefined && row[aliasColumn] !== null
? String(row[aliasColumn]).trim()
: "";
importedData.push({
mobile: Number(mobile),
from,
alias
});
}
if (importedData.length === 0) {
setError("未找到有效数据请确保Excel中至少有一行有效的手机号码");
setIsProcessing(false);
return;
}
setParsedData(importedData);
setIsProcessing(false);
} catch (err) {
console.error("解析Excel文件出错:", err);
setError("解析Excel文件时出错请确保文件格式正确");
setIsProcessing(false);
}
};
reader.onerror = () => {
setError("读取文件时出错,请重试");
setIsProcessing(false);
};
reader.readAsBinaryString(file);
};
const handleImport = () => {
if (parsedData.length > 0) {
// 打印解析数据
console.log("导入的联系人数据:");
console.log(JSON.stringify(parsedData, null, 2));
// 在界面上显示成功消息
setIsImportSuccessful(true);
}
};
const handleReset = () => {
setParsedData([]);
setFileName("");
setError(null);
setIsImportSuccessful(false);
const fileInput = document.getElementById("file-input") as HTMLInputElement;
if (fileInput) {
fileInput.value = "";
}
};
return (
<div className="container mx-auto py-10 max-w-2xl">
<Card className="p-6 shadow-md">
<h1 className="text-2xl font-bold mb-6 text-center"></h1>
<div className="space-y-6">
<div className="space-y-2">
<label className="text-sm font-medium">Excel文件</label>
<Input
id="file-input"
type="file"
accept=".xlsx,.xls"
onChange={handleFileChange}
className="w-full"
disabled={isProcessing}
/>
{fileName && (
<p className="text-sm text-gray-500">
: {fileName}
</p>
)}
<div className="text-xs text-gray-500 mt-1">
请确保Excel文件包含以下列: 手机号码()()()
</div>
</div>
{isProcessing && (
<div className="text-center py-4">
<div className="inline-block h-6 w-6 animate-spin rounded-full border-2 border-solid border-blue-500 border-r-transparent"></div>
<p className="mt-2 text-sm text-gray-600">Excel文件...</p>
</div>
)}
{error && (
<Alert variant="destructive">
<AlertCircle className="h-4 w-4" />
<AlertDescription>{error}</AlertDescription>
</Alert>
)}
{isImportSuccessful && (
<Alert className="bg-green-50 text-green-800 border-green-200">
<AlertDescription>
{parsedData.length}
</AlertDescription>
</Alert>
)}
{parsedData.length > 0 && !isImportSuccessful && (
<div className="space-y-4">
<p className="text-sm">
{parsedData.length}
</p>
{Object.keys(detectedColumns).length > 0 && (
<div className="text-xs p-2 bg-blue-50 rounded border border-blue-100">
<p className="font-medium text-blue-700 mb-1">:</p>
<ul className="list-disc pl-5 space-y-1">
{detectedColumns.mobile && <li>: {detectedColumns.mobile}</li>}
{detectedColumns.from && <li>: {detectedColumns.from}</li>}
{detectedColumns.alias && <li>: {detectedColumns.alias}</li>}
</ul>
</div>
)}
<div className="grid grid-cols-2 gap-4">
<div>
<p className="font-medium text-sm mb-1"></p>
<div className="text-xs bg-gray-50 p-2 rounded overflow-hidden">
<pre>{JSON.stringify(parsedData.slice(0, 3), null, 2)}</pre>
{parsedData.length > 3 && <p className="mt-1 text-gray-500">... {parsedData.length} </p>}
</div>
</div>
<div>
<p className="font-medium text-sm mb-1"></p>
<div className="text-xs bg-gray-50 p-2 rounded">
<pre>{`[
{
"mobile": 13800000000,
"from": "小红书",
"alias": "xxxxxx"
},
...
]`}</pre>
</div>
</div>
</div>
</div>
)}
<div className="flex space-x-4">
<Button
onClick={handleImport}
className="flex-1"
disabled={parsedData.length === 0 || isImportSuccessful || isProcessing}
>
</Button>
<Button
variant="outline"
onClick={handleReset}
className="flex-1"
disabled={isProcessing}
>
</Button>
</div>
</div>
</Card>
</div>
);
}