import { Builder, By, Key, until } from 'selenium-webdriver';
import fs from 'fs';
import XLSX from 'xlsx';
import path from 'path';

const loginTest = (req, res) => {
  res.send('<h2>Automated Login Test Script: Completed</h2>');
  (async function example() {
    let driver = await new Builder().forBrowser('chrome').build();
    let testStatus = 'Failed';
    let workbook;
    let worksheet;
    const startTime = new Date().toISOString();

    const steps = [
      { description: 'Navigate to login page', action: async () => await driver.navigate().to('https://app-qa.journeysmap.com/UI/User/Login') },
      { description: 'Expand browser', action: async () => await driver.manage().window().maximize() },
      { description: 'Locate username field', action: async () => await driver.wait(until.elementLocated(By.id('txtU')), 5000) },
      { description: 'Enter username', action: async () => await driver.findElement(By.id('txtU')).sendKeys('gnight147', Key.RETURN) },
      { description: 'Wait for password field to be present', action: async () => await driver.sleep(3000) },
      { description: 'Enter password', action: async () => await driver.findElement(By.id('txtP')).sendKeys('EDmin123', Key.RETURN) },
      { description: 'Wait for login process to complete', action: async () => await driver.sleep(10000) },
    ];

    const logStep = (worksheet, stepDescription, status, stepIndex, timestamp, screenshotPath) => {
      const json_data = XLSX.utils.sheet_to_json(worksheet, { header: 1 });
      json_data.push([stepIndex, stepDescription, status, timestamp, screenshotPath]);
      return XLSX.utils.aoa_to_sheet(json_data);
    };

    const takeScreenshot = async (driver, stepIndex, stepDescription) => {
      const screenshot = await driver.takeScreenshot();

      const dir = path.join('screenshots', 'loginScript');
      if (!fs.existsSync(dir)) {
        fs.mkdirSync(dir, { recursive: true });
      }

      const screenshotFileName = `step_${stepIndex}_${stepDescription.replace(/\s+/g, '_')}.png`;
      const screenshotPath = path.join(dir, screenshotFileName);
      fs.writeFileSync(screenshotPath, screenshot, 'base64');
      return screenshotPath;
    };

    try {
      // Initialize workbook and worksheet
      if (!fs.existsSync('loginResult.xlsx')) {
        workbook = XLSX.utils.book_new();
        worksheet = XLSX.utils.aoa_to_sheet([
          ['Step Number', 'Step Description', 'Status', 'Timestamp', 'Screenshot']
        ]);
        XLSX.utils.book_append_sheet(workbook, worksheet, 'Sheet1');
        XLSX.writeFile(workbook, 'loginResult.xlsx');
      } else {
        workbook = XLSX.readFile('loginResult.xlsx', { cellStyles: true });
        const sheetName = workbook.SheetNames[0];
        worksheet = workbook.Sheets[sheetName];
      }

      // Log test start time
      worksheet = logStep(worksheet, 'Test Start', 'Started', '', startTime, '');

      for (const step of steps) {
        const stepIndex = steps.indexOf(step) + 1;
        const timestamp = new Date().toISOString();

        try {
          await step.action();
          const screenshotPath = await takeScreenshot(driver, stepIndex, step.description);
          worksheet = logStep(worksheet, step.description, 'Passed', stepIndex, timestamp, screenshotPath);
        } catch (error) {
          const screenshotPath = await takeScreenshot(driver, stepIndex, `${step.description}_Failed`);
          worksheet = logStep(worksheet, step.description, 'Failed', stepIndex, timestamp, screenshotPath);
          throw error; // Re-throw the error to stop further execution if necessary
        }
      }

      // Check if login is successful
      const logoutButton = await driver.wait(until.elementLocated(By.id('headerContainer')), 5000).catch(() => false);

      if (logoutButton) {
        console.log('Login successful');
        testStatus = 'Passed';
      } else {
        console.log('Login failed');
      }

      // Log test end time
      const endTime = new Date().toISOString();
      worksheet = logStep(worksheet, 'Test End', 'Ended', '', endTime, '');

      // Write the updated worksheet back to the workbook and save it
      workbook.Sheets['Sheet1'] = worksheet;
      XLSX.writeFile(workbook, 'loginResult.xlsx');
    } finally {
      await driver.quit();
    }
  })();
}

export default loginTest;