import { GoogleGenAI } from '@google/genai';
import fs from 'fs';
import path from 'path';
 
const ai = new GoogleGenAI({});

const fileStoreName = 'my-example-store';
 
const createStoreOp = await ai.fileSearchStores.create({
  config: { displayName: fileStoreName }
});
 
console.log(`Đã tạo Kho với tên: `);

let fileStore = null;
// Liệt kê các kho với giới hạn 10 kết quả mỗi trang
const pager = await ai.fileSearchStores.list({ config: { pageSize: 10 } });
let page = pager.page;
 
// Lặp qua các trang cho đến khi tìm thấy kết quả khớp
searchLoop: while (true) {
  for (const store of page) {
    if (store.displayName === fileStoreName) {
      fileStore = store;
      break searchLoop;
    }
  }
  if (!pager.hasNextPage()) break;
  page = await pager.nextPage();
}
 
if (!fileStore) {
  throw new Error(`Không tìm thấy kho có tên hiển thị ''.`);
}
console.log(`Đã tìm thấy kho: `);

const docsDir = "docs"; // Đảm bảo bạn có thư mục 'docs' chứa các tệp văn bản
const files = fs.readdirSync(docsDir).map(file => path.join(docsDir, file));
 
await Promise.all(files.map(async (filePath) => {
  // 1. Khởi tạo tải lên và lập chỉ mục
  let operation = await ai.fileSearchStores.uploadToFileSearchStore({
    file: filePath,
    fileSearchStoreName: fileStore.name,
    config: {
      displayName: path.basename(filePath),
    }
  });
 
  // 2. Kiểm tra trạng thái cho đến khi tài liệu được xử lý xong
  while (!operation.done) {
    await new Promise(resolve => setTimeout(resolve, 1000)); // đợi 1 giây
    operation = await ai.operations.get({ operation });
  }
  console.log(`Xử lý hoàn tất cho: `);
  return operation;
}));

const specialDocPath = 'special-docs/technical-manual.txt';
 
let advancedUploadOp = await ai.fileSearchStores.uploadToFileSearchStore({
  file: specialDocPath,
  fileSearchStoreName: fileStore.name,
  config: {
    displayName: 'technical-manual.txt',
    customMetadata: [
      { key: "doc_type", stringValue: "manual" },
    ],
    chunkingConfig: {
      whiteSpaceConfig: {
        maxTokensPerChunk: 500, // Các đoạn nhỏ hơn để tra cứu chính xác hơn
        maxOverlapTokens: 50    // Đảm bảo ngữ cảnh không bị mất giữa các đoạn
      }
    }
  }
});
 
// Đợi tệp xử lý xong
while (!advancedUploadOp.done) {
  await new Promise(resolve => setTimeout(resolve, 1000));
  advancedUploadOp = await ai.operations.get({ operation: advancedUploadOp });
}
console.log("Đã xử lý xong tệp nâng cao.");

const response = await ai.models.generateContent({
  model: "gemini-2.5-flash",
  contents: "Gemini là gì và File API là gì?",
  config: {
    tools: [{
      fileSearch: {
        fileSearchStoreNames: [fileStore.name]
      }
    }]
  }
});
 
console.log("Phản hồi từ mô hình:", response.text);
// Bạn có thể kiểm tra response.candidates[0].groundingMetadata để xem các nguồn trích dẫn!

const responseFiltered = await ai.models.generateContent({
  model: "gemini-2.5-flash",
  contents: "Làm thế nào để khởi động lại thiết bị theo hướng dẫn sử dụng?",
  config: {
    tools: [{
      fileSearch: {
        fileSearchStoreNames: [fileStore.name],
        metadataFilter: 'doc_type="manual"'
      }
    }]
  }
});
 
console.log("Phản hồi đã lọc:", responseFiltered.text);

const docToFind = 'doc1.txt';
let targetDoc = null;
 
let documentPager = await ai.fileSearchStores.documents.list({
  parent: fileStore.name,
});
 
// Lặp qua danh sách tài liệu trong kho
searchDocsLoop: while (true) {
  for (const document of documentPager.page) {
    if (document.displayName === docToFind) {
      targetDoc = document;
      break searchDocsLoop;
    }
  }
  if (!documentPager.hasNextPage()) break;
  documentPager = await documentPager.nextPage();
}
 
if (!targetDoc) throw new Error(`Không tìm thấy tài liệu ''.`);

await ai.fileSearchStores.documents.delete({
    name: targetDoc.name,
    config: { force: true } // Bắt buộc để xóa vĩnh viễn khỏi kho
});

const docToUpdate = 'doc1.txt'; // giả sử tệp này đã có nội dung mới
const localDocPath = path.join(docsDir, docToUpdate);
 
// 1. Tìm ID của tài liệu hiện có trong kho dựa trên tên hiển thị
let documentPager = await ai.fileSearchStores.documents.list({ parent: fileStore.name });
let foundDoc = null;
findLoop: while (true) {
  for (const doc of documentPager.page) {
    if (doc.displayName === docToUpdate) {
      foundDoc = doc;
      break findLoop;
    }
  }
  if (!documentPager.hasNextPage()) break;
  documentPager = await documentPager.nextPage();
}
 
// 2. Nếu tìm thấy, tiến hành xóa nó
if (foundDoc) {
   await ai.fileSearchStores.documents.delete({
     name: foundDoc.name,
     config: { force: true } // 'force' là bắt buộc để xóa các tài liệu đã lập chỉ mục
   });
}
 
// 3. Tải lên phiên bản mới
let updateOp = await ai.fileSearchStores.uploadToFileSearchStore({
  file: localDocPath,
  fileSearchStoreName: fileStore.name,
  config: { displayName: docToUpdate }
});

while (!updateOp.done) {
   await new Promise(resolve => setTimeout(resolve, 1000));
   updateOp = await ai.operations.get({ operation: updateOp });
}
 
console.log("Bản hiệu chỉnh đã được tải lên và lập chỉ mục thành công.");

await ai.fileSearchStores.delete({
    name: fileStore.name,
    config: { force: true } // Xóa toàn bộ kho và các tài liệu bên trong
});