90 lines
2.9 KiB
TypeScript
90 lines
2.9 KiB
TypeScript
|
'use server';
|
||
|
|
|
||
|
|
import { getSessionData } from '@/app/actions/session';
|
||
|
|
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';
|
||
|
|
|
||
|
|
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 (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';
|
||
|
|
|
||
|
|
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}`);
|
||
|
|
}
|
||
|
|
|
||
|
|
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;
|
||
|
|
}
|
||
|
|
}
|