const webdriver = require("selenium-webdriver");
const path = require('path');
const os = require('os');
const { By } = webdriver;
const firefox = require('selenium-webdriver/firefox');
const options = new firefox.Options()

async function runTest_DownloadFile() {
  try {
    options.set('se:downloadsEnabled', true);

    let driver = new webdriver.Builder()
    .usingServer("http://192.168.0.116:4444")
    .forBrowser("firefox")
    .setFirefoxOptions(options)
    .build()

    const fileNames = ['20MB.bin'];

    driver.get("https://myjob.page/tools/test-files").then(async () => {
      downloadLink = await driver.findElement(By.xpath('/html/body/main/div/div/div[1]/a[5]'));
      await downloadLink.click();

      await waitForDownloadableFile(driver, fileNames);
      files = await driver.getDownloadableFiles();

      if (files.sort().toString() === fileNames.sort().toString()) {
        console.log(JSON.stringify({ status: "passed", message: "Test passed successfully" }));
      } else {
        console.log(JSON.stringify({ status: 'failed', message: 'Files mismatch!' }));
      }

      const targetDirectory = path.join(os.tmpdir(), 'nodejs_downloaded');
      await driver.downloadFile(files, targetDirectory);

      try {
        await driver.quit();
      } catch (error) {
        if (error.message != "success") {
          console.error('Error during test execution:', error);
          console.log(JSON.stringify({ status: "failed", message: error.message }));
        }
      }
      console.log(JSON.stringify({ status: "passed", message: "Test passed successfully" }));
    });
  } catch (error) {
    if (driver) {
      await driver.quit();
    }
    console.error('Error during test execution:', error);
    console.log(JSON.stringify({ status: "failed", message: error.message }));
  }
}

// Function to wait for downloadable files to include the desired file
async function waitForDownloadableFile(driver, fileNames, interval = 3000, maxAttempts = 10) {
  let attempt = 0;
  while (attempt < maxAttempts) {
    const files = await driver.getDownloadableFiles();
    if (fileNames.every(fileName => files.includes(fileName))) {
      return;
    }
    await new Promise(resolve => setTimeout(resolve, interval));
    attempt++;
  }
  throw new Error(`Timeout: Files '${fileNames}' not found after ${maxAttempts} attempts`);
}

runTest_DownloadFile();