68 lines
2.7 KiB
TypeScript
68 lines
2.7 KiB
TypeScript
import { GoogleGenAI, Type } from "@google/genai";
|
|
import { FileItem, FileType } from "../types";
|
|
|
|
const ai = new GoogleGenAI({ apiKey: process.env.API_KEY || '' });
|
|
|
|
export const generateRemoteFileList = async (host: string, path: string): Promise<FileItem[]> => {
|
|
try {
|
|
const model = 'gemini-2.5-flash-latest';
|
|
const prompt = `
|
|
You are simulating an FTP server file listing for the host: "${host}" at path: "${path}".
|
|
Generate a realistic list of 8-15 files and folders that might exist on this specific type of server.
|
|
If the host sounds corporate, use corporate files. If it sounds like a game server, use game files.
|
|
If it sounds like NASA, use space data files.
|
|
|
|
Return a JSON array of objects.
|
|
`;
|
|
|
|
const response = await ai.models.generateContent({
|
|
model: model,
|
|
contents: prompt,
|
|
config: {
|
|
responseMimeType: "application/json",
|
|
responseSchema: {
|
|
type: Type.ARRAY,
|
|
items: {
|
|
type: Type.OBJECT,
|
|
properties: {
|
|
name: { type: Type.STRING },
|
|
type: { type: Type.STRING, enum: ["FILE", "FOLDER"] },
|
|
size: { type: Type.INTEGER, description: "Size in bytes. Folders should be 0 or 4096." },
|
|
},
|
|
required: ["name", "type", "size"]
|
|
}
|
|
}
|
|
}
|
|
});
|
|
|
|
const data = JSON.parse(response.text || '[]');
|
|
|
|
return data.map((item: any, index: number) => ({
|
|
id: `gen-${Date.now()}-${index}`,
|
|
name: item.name,
|
|
type: item.type === 'FOLDER' ? FileType.FOLDER : FileType.FILE,
|
|
size: item.size,
|
|
date: new Date(Date.now() - Math.floor(Math.random() * 10000000000)).toISOString(),
|
|
permissions: item.type === 'FOLDER' ? 'drwxr-xr-x' : '-rw-r--r--'
|
|
}));
|
|
} catch (error) {
|
|
console.error("Gemini generation failed", error);
|
|
// Fallback if API fails
|
|
return [
|
|
{ id: 'err1', name: 'connection_retry.log', type: FileType.FILE, size: 1024, date: new Date().toISOString(), permissions: '-rw-r--r--' },
|
|
{ id: 'err2', name: 'backup', type: FileType.FOLDER, size: 0, date: new Date().toISOString(), permissions: 'drwxr-xr-x' }
|
|
];
|
|
}
|
|
};
|
|
|
|
export const generateServerMessage = async (host: string): Promise<string> => {
|
|
try {
|
|
const response = await ai.models.generateContent({
|
|
model: 'gemini-3-flash-preview',
|
|
contents: `Write a short, single-line FTP welcome message (220 Service ready) for a server named "${host}". Make it sound authentic to the domain info.`,
|
|
});
|
|
return response.text?.trim() || `220 ${host} FTP Server ready.`;
|
|
} catch (e) {
|
|
return `220 ${host} FTP Server ready.`;
|
|
}
|
|
} |