340 lines
9.0 KiB
TypeScript
340 lines
9.0 KiB
TypeScript
import { postgrestGet, postgrestPut, postgrestPost, type PostgrestParams } from '../postgrest-client';
|
|
import dayjs from 'dayjs';
|
|
// import { API_BASE_URL } from '../client';
|
|
|
|
/**
|
|
* 格式化日期
|
|
* @param dateString 日期字符串
|
|
* @returns 格式化后的日期字符串
|
|
*/
|
|
function formatDate(dateString: string): string {
|
|
if (!dateString) return '';
|
|
try {
|
|
return dayjs(dateString).format('YYYY-MM-DD HH:mm:ss');
|
|
} catch (error) {
|
|
console.error('日期格式化失败:', error);
|
|
return dateString;
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* 从不同格式的 API 响应中提取数据
|
|
* @param responseData API 响应数据
|
|
* @returns 提取后的数据或 null
|
|
*/
|
|
function extractApiData<T>(responseData: unknown): T | null {
|
|
if (!responseData) return null;
|
|
|
|
// 格式1: { code: number, msg: string, data: T }
|
|
if (typeof responseData === 'object' && responseData !== null &&
|
|
'code' in responseData &&
|
|
'data' in responseData &&
|
|
(responseData as { data: unknown }).data) {
|
|
return (responseData as { data: T }).data;
|
|
}
|
|
|
|
// 格式2: 直接是数据对象
|
|
return responseData as T;
|
|
}
|
|
|
|
// 文档状态枚举
|
|
export enum DocumentStatus {
|
|
WAITING = "Waiting",
|
|
CUTTING = "Cutting",
|
|
EXTRACTIONING = "Extractioning",
|
|
EVALUATIONING = "Evaluationing",
|
|
FAILED = "Failed",
|
|
PROCESSED = "Processed"
|
|
}
|
|
|
|
// 文档类型接口
|
|
export interface DocumentType {
|
|
id: number;
|
|
name: string;
|
|
}
|
|
|
|
// 提取结果接口
|
|
interface ExtractedResult {
|
|
[key: string]: unknown;
|
|
}
|
|
|
|
// 摘要接口
|
|
interface Summary {
|
|
[key: string]: unknown;
|
|
}
|
|
|
|
// 文档接口
|
|
export interface Document {
|
|
id: number;
|
|
name: string;
|
|
type_id: number;
|
|
file_size: number;
|
|
status: DocumentStatus;
|
|
created_at: string;
|
|
document_number?: string;
|
|
path?: string;
|
|
storage_type?: string;
|
|
is_test_document?: boolean;
|
|
evaluation_level?: string;
|
|
ocr_result?: Record<string, string>;
|
|
extracted_results?: ExtractedResult;
|
|
sumary?: Summary;
|
|
remark?: string;
|
|
}
|
|
|
|
// 文件上传响应接口
|
|
export interface FileUploadResponse {
|
|
success: boolean;
|
|
result?: {
|
|
id: number;
|
|
file_name: string;
|
|
file_size: number;
|
|
file_url: string;
|
|
type_id: number;
|
|
type_description: string;
|
|
document_number: string | null;
|
|
storage_type: string;
|
|
is_test_document: boolean;
|
|
remark: string | null;
|
|
background_processing: boolean;
|
|
evaluation_level: string;
|
|
};
|
|
error: string | null;
|
|
}
|
|
|
|
/**
|
|
* 将文件转换为二进制数据
|
|
*/
|
|
export async function uploadFileToBinary(file: File): Promise<ArrayBuffer> {
|
|
return new Promise((resolve, reject) => {
|
|
const reader = new FileReader();
|
|
reader.onload = () => {
|
|
if (reader.result instanceof ArrayBuffer) {
|
|
resolve(reader.result);
|
|
} else {
|
|
reject(new Error('文件读取失败'));
|
|
}
|
|
};
|
|
reader.onerror = () => reject(new Error('文件读取失败'));
|
|
reader.readAsArrayBuffer(file);
|
|
});
|
|
}
|
|
|
|
/**
|
|
* 上传文件到文档审核系统
|
|
* @param binaryData 文件的二进制数据
|
|
* @param fileName 文件名
|
|
* @param fileType 文件类型
|
|
* @param typeId 文档类型ID
|
|
* @param priority 优先级
|
|
* @param documentNumber 文档编号（可选）
|
|
* @param remark 备注信息（可选）
|
|
* @param isTestDocument 是否为测试文档
|
|
* @returns 上传结果
|
|
*/
|
|
export async function uploadDocumentToServer(
|
|
binaryData: ArrayBuffer,
|
|
fileName: string,
|
|
fileType: string,
|
|
typeId: string | number,
|
|
priority: string,
|
|
documentNumber?: string | null,
|
|
remark?: string | null,
|
|
isTestDocument: boolean = false
|
|
): Promise<{data: FileUploadResponse; error?: never} | {data?: never; error: string; status?: number}> {
|
|
try {
|
|
// 创建FormData对象
|
|
const formData = new FormData();
|
|
|
|
// 将二进制数据转换为Blob并添加到FormData
|
|
const blob = new Blob([binaryData], { type: fileType });
|
|
formData.append('file', blob, fileName);
|
|
|
|
// 将信息添加到一个JSON对象中
|
|
const uploadInfo = {
|
|
type_id: Number(typeId),
|
|
evaluation_level: priority,
|
|
document_number: documentNumber || null,
|
|
remark: remark || null,
|
|
is_test_document: isTestDocument
|
|
};
|
|
|
|
// 添加JSON字符串到FormData
|
|
formData.append('upload_info', JSON.stringify(uploadInfo));
|
|
|
|
console.log('上传信息:', {
|
|
fileName,
|
|
fileType,
|
|
typeId: Number(typeId),
|
|
priority,
|
|
documentNumber: documentNumber || null,
|
|
remark: remark || null,
|
|
isTestDocument
|
|
});
|
|
|
|
// 发送请求
|
|
// const response = await fetch(`${API_BASE_URL}/admin/documents/upload`, {
|
|
const response = await fetch('http://172.16.0.55:8000/admin/documents/upload', {
|
|
method: 'POST',
|
|
headers: {
|
|
'X-File-Name': encodeURIComponent(fileName)
|
|
},
|
|
body: formData
|
|
});
|
|
|
|
if (!response.ok) {
|
|
const errorText = await response.text();
|
|
console.error(`上传失败 (${response.status}): ${errorText}`);
|
|
return {
|
|
error: `上传失败: ${response.status} ${response.statusText}`,
|
|
status: response.status
|
|
};
|
|
}
|
|
|
|
const responseData = await response.json();
|
|
const extractedData = extractApiData<FileUploadResponse>(responseData);
|
|
|
|
if (!extractedData) {
|
|
return { error: '处理上传响应失败', status: 500 };
|
|
}
|
|
|
|
return { data: extractedData };
|
|
} catch (error) {
|
|
console.error('上传错误:', error);
|
|
return {
|
|
error: error instanceof Error ? error.message : '上传失败',
|
|
status: 500
|
|
};
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 获取当天的文档列表
|
|
* @returns 文档列表
|
|
*/
|
|
export async function getTodayDocuments(): Promise<{data: Document[]; error?: never} | {data?: never; error: string; status?: number}> {
|
|
try {
|
|
const today = dayjs().startOf('day').format('YYYY-MM-DD');
|
|
console.log('查询当天文档，日期范围:', today);
|
|
|
|
const params: PostgrestParams = {
|
|
select: `
|
|
id,
|
|
name,
|
|
type_id,
|
|
file_size,
|
|
status,
|
|
created_at,
|
|
document_number,
|
|
path,
|
|
storage_type,
|
|
is_test_document,
|
|
evaluation_level,
|
|
ocr_result,
|
|
extracted_results,
|
|
sumary,
|
|
remark
|
|
`,
|
|
order: 'created_at.desc',
|
|
filter: {
|
|
'created_at': `gte.${today}`
|
|
}
|
|
};
|
|
|
|
// console.log('发送请求参数:', params);
|
|
const response = await postgrestGet<Document[]>('documents', params);
|
|
// console.log('API 响应:', response);
|
|
|
|
if (response.error) {
|
|
console.error('API 返回错误:', response.error);
|
|
return { error: response.error, status: response.status };
|
|
}
|
|
|
|
const extractedData = extractApiData<Document[]>(response.data);
|
|
// console.log('提取后的数据:', extractedData);
|
|
|
|
if (!extractedData) {
|
|
console.error('数据提取失败');
|
|
return { error: '获取数据失败', status: 500 };
|
|
}
|
|
|
|
return { data: extractedData };
|
|
} catch (error) {
|
|
console.error('获取当天文档列表失败:', error);
|
|
return {
|
|
error: error instanceof Error ? error.message : '获取当天文档列表失败',
|
|
status: 500
|
|
};
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 获取文档类型列表
|
|
* @returns 文档类型列表
|
|
*/
|
|
export async function getDocumentTypes(): Promise<{data: DocumentType[]; error?: never} | {data?: never; error: string; status?: number}> {
|
|
try {
|
|
const params: PostgrestParams = {
|
|
select: 'id, name'
|
|
};
|
|
|
|
const response = await postgrestGet<DocumentType[]>('document_types', params);
|
|
|
|
if (response.error) {
|
|
return { error: response.error, status: response.status };
|
|
}
|
|
|
|
const extractedData = extractApiData<DocumentType[]>(response.data);
|
|
if (!extractedData) {
|
|
return { error: '获取数据失败', status: 500 };
|
|
}
|
|
|
|
return { data: extractedData };
|
|
} catch (error) {
|
|
console.error('获取文档类型列表失败:', error);
|
|
return {
|
|
error: error instanceof Error ? error.message : '获取文档类型列表失败',
|
|
status: 500
|
|
};
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 获取指定文档的状态
|
|
* @param documentIds 文档ID列表
|
|
* @returns 文档状态列表
|
|
*/
|
|
export async function getDocumentsStatus(documentIds: number[]): Promise<{data: Document[]; error?: never} | {data?: never; error: string; status?: number}> {
|
|
try {
|
|
if (!documentIds || documentIds.length === 0) {
|
|
return { data: [] };
|
|
}
|
|
|
|
const params: PostgrestParams = {
|
|
select: 'id, status',
|
|
filter: {
|
|
'id': `in.(${documentIds.join(',')})`
|
|
}
|
|
};
|
|
|
|
const response = await postgrestGet<Document[]>('documents', params);
|
|
|
|
if (response.error) {
|
|
return { error: response.error, status: response.status };
|
|
}
|
|
|
|
const extractedData = extractApiData<Document[]>(response.data);
|
|
if (!extractedData) {
|
|
return { error: '获取数据失败', status: 500 };
|
|
}
|
|
|
|
return { data: extractedData };
|
|
} catch (error) {
|
|
console.error('获取文档状态失败:', error);
|
|
return {
|
|
error: error instanceof Error ? error.message : '获取文档状态失败',
|
|
status: 500
|
|
};
|
|
}
|
|
} |