add file info fro document view

src/app/actions/trackingActions.ts

@@ -5,86 +5,123 @@ import { API_BASE_URL } from '@/app/actions/definitions';
 import { headers } from 'next/headers';
 
 export async function requestFileWithTracking(fileId: string) {
-    const token = await getSessionData('token');
-    const headersList = await headers();
-    const forwardedFor = headersList.get('x-forwarded-for');
-    const realIp = headersList.get('x-real-ip');
-    const ip = forwardedFor?.split(',')[0] || realIp || 'unknown';
-    const userAgent = headersList.get('user-agent') || 'unknown';
+	const token = await getSessionData('token');
+	const headersList = await headers();
+	const forwardedFor = headersList.get('x-forwarded-for');
+	const realIp = headersList.get('x-real-ip');
+	const ip = forwardedFor?.split(',')[0] || realIp || 'unknown';
+	const userAgent = headersList.get('user-agent') || 'unknown';
 
-    try {
-        const response = await fetch(`${API_BASE_URL}/api/file/link`, {
-            method: 'POST',
-            body: JSON.stringify({
-                file_id: fileId,
-                token: token,
-                ip: ip,
-                user_agent: userAgent
-            }),
-            headers: {
-                'Content-Type': 'application/json'
-            }
-        });
+	try {
+		const response = await fetch(`${API_BASE_URL}/api/file/link`, {
+			method: 'POST',
+			body: JSON.stringify({
+				file_id: fileId,
+				token: token,
+				ip: ip,
+				user_agent: userAgent
+			}),
+			headers: {
+				'Content-Type': 'application/json'
+			}
+		});
 
-        if (response.ok) {
-            const contentType = response.headers.get('content-type');
+		if (response.ok) {
+			const contentType = response.headers.get('content-type');
 
-            if (contentType?.includes('application/json')) {
-                const data = await response.json();
-                return data.message_body;
-            } else {
-                const blob = await response.blob();
-                return URL.createObjectURL(blob);
-            }
-        } else {
-            throw new Error(`${response.status}`);
-        }
-    } catch (error) {
-        return null;
-    }
+			if (contentType?.includes('application/json')) {
+				const data = await response.json();
+				return data.message_body;
+			} else {
+				const blob = await response.blob();
+				return URL.createObjectURL(blob);
+			}
+		} else {
+			throw new Error(`${response.status}`);
+		}
+	} catch (error) {
+		return null;
+	}
 }
 
 export async function requestFileAsBuffer(fileId: string): Promise<ArrayBuffer | null> {
-    const token = await getSessionData('token');
-    const headersList = await headers();
-    const forwardedFor = headersList.get('x-forwarded-for');
-    const realIp = headersList.get('x-real-ip');
-    const ip = forwardedFor?.split(',')[0] || realIp || 'unknown';
-    const userAgent = headersList.get('user-agent') || 'unknown';
+	const token = await getSessionData('token');
+	const headersList = await headers();
+	const forwardedFor = headersList.get('x-forwarded-for');
+	const realIp = headersList.get('x-real-ip');
+	const ip = forwardedFor?.split(',')[0] || realIp || 'unknown';
+	const userAgent = headersList.get('user-agent') || 'unknown';
 
-    try {
-        const response = await fetch(`${API_BASE_URL}/api/file/link`, {
-            method: 'POST',
-            body: JSON.stringify({
-                file_id: fileId,
-                token: token,
-                ip: ip,
-                user_agent: userAgent
-            }),
-            headers: {
-                'Content-Type': 'application/json'
-            }
-        });
+	try {
+		const response = await fetch(`${API_BASE_URL}/api/file/link`, {
+			method: 'POST',
+			body: JSON.stringify({
+				file_id: fileId,
+				token: token,
+				ip: ip,
+				user_agent: userAgent
+			}),
+			headers: {
+				'Content-Type': 'application/json'
+			}
+		});
 
-        if (!response.ok) {
-            throw new Error(`${response.status}`);
-        }
+		if (!response.ok) {
+			throw new Error(`${response.status}`);
+		}
 
-        const contentType = response.headers.get('content-type');
-        if (contentType?.includes('application/json')) {
-            const data = await response.json();
-            const fileUrl = data.message_body;
-            const fileResponse = await fetch(fileUrl);
-            if (!fileResponse.ok) {
-                throw new Error(`Failed to fetch file: ${fileResponse.status}`);
-            }
-            
-            return await fileResponse.arrayBuffer();
-        } 
-        else {
-            return await response.arrayBuffer();
-        }
-    } catch (error) {
-        return null;
-    }
+		const contentType = response.headers.get('content-type');
+		if (contentType?.includes('application/json')) {
+			const data = await response.json();
+			const fileUrl = data.message_body;
+			const fileResponse = await fetch(fileUrl);
+			if (!fileResponse.ok) {
+				throw new Error(`Failed to fetch file: ${fileResponse.status}`);
+			}
+
+			return await fileResponse.arrayBuffer();
+		}
+		else {
+			return await response.arrayBuffer();
+		}
+	} catch (error) {
+		return null;
+	}
+}
+
+export async function fetchDocumentInfo(fileId: string) {
+	const token = await getSessionData('token');
+
+	try {
+		const response = await fetch(`${API_BASE_URL}/api/v1/data`, {
+			method: 'POST',
+			body: JSON.stringify({
+				version: 1,
+				msg_id: 20005,
+				message_body: {
+					file_id: fileId,
+					token: token,
+					action: 'file_info'
+				}
+			}),
+			headers: {
+				'Content-Type': 'application/json'
+			}
+		});
+
+		if (response.ok) {
+			const parsed = await response.json();
+
+			if (parsed.message_code === 0) {
+				return parsed.message_body;
+			} else {
+				return null;
+			}
+
+		} else {
+			throw new Error(`${response.status}`);
+		}
+	} catch (error) {
+		return null;
+	}
 }