TidGi-Desktop/features/stepDefinitions/cleanup.ts

import { After, Before } from '@cucumber/cucumber';
import fs from 'fs-extra';
import path from 'path';
import { logsDirectory, screenshotsDirectory } from '../supports/paths';
import { clearAISettings } from './agent';
import { ApplicationWorld } from './application';
import { clearTidgiMiniWindowSettings } from './tidgiMiniWindow';
import { clearGitTestData, clearHibernationTestData, clearSubWikiRoutingTestData } from './wiki';

Before(async function(this: ApplicationWorld, { pickle }) {
  // Create necessary directories under userData-test/logs to match appPaths in dev/test
  if (!(await fs.pathExists(logsDirectory))) {
    await fs.ensureDir(logsDirectory);
  }

  // Create screenshots subdirectory in logs
  if (!(await fs.pathExists(screenshotsDirectory))) {
    await fs.ensureDir(screenshotsDirectory);
  }

  if (pickle.tags.some((tag) => tag.name === '@ai-setting')) {
    await clearAISettings();
  }
  if (pickle.tags.some((tag) => tag.name === '@tidgi-mini-window')) {
    await clearTidgiMiniWindowSettings();
  }
});

After(async function(this: ApplicationWorld, { pickle }) {
  // IMPORTANT: Close app FIRST before cleaning up files
  // This releases file locks so wiki folders can be deleted
  if (this.app) {
    try {
      // Close all windows including tidgi mini window before closing the app, otherwise it might hang, and refused to exit until ctrl+C
      const allWindows = this.app.windows();
      await Promise.all(
        allWindows.map(async (window) => {
          try {
            if (!window.isClosed()) {
              // Add timeout protection for window.close() to prevent hanging
              await Promise.race([
                window.close(),
                new Promise((_, reject) => setTimeout(() => reject(new Error('Window close timeout')), 5000)),
              ]);
            }
          } catch (error) {
            console.error('Error closing window:', error);
          }
        }),
      );
      
      // Add timeout protection for app.close() to prevent hanging
      await Promise.race([
        this.app.close(),
        new Promise((_, reject) => setTimeout(() => reject(new Error('App close timeout')), 10000)),
      ]);
    } catch (error) {
      console.error('Error during cleanup:', error);
      // Force kill the app if it hangs
      try {
        if (this.app) {
          await this.app.context().close();
        }
      } catch (forceCloseError) {
        console.error('Error force closing app:', forceCloseError);
      }
    }
    this.app = undefined;
    this.mainWindow = undefined;
    this.currentWindow = undefined;
  }

  // Clean up settings and test data AFTER app is closed
  if (pickle.tags.some((tag) => tag.name === '@tidgi-mini-window')) {
    await clearTidgiMiniWindowSettings();
  }
  if (pickle.tags.some((tag) => tag.name === '@ai-setting')) {
    await clearAISettings();
  }
  if (pickle.tags.some((tag) => tag.name === '@subwiki')) {
    await clearSubWikiRoutingTestData();
  }
  // Clean up git test data to prevent state pollution between git tests
  // Removes entire wiki folder - it will be recreated on next test start
  if (pickle.tags.some((tag) => tag.name === '@git')) {
    await clearGitTestData();
  }
  // Clean up hibernation test data - remove wiki2 folder created during tests
  if (pickle.tags.some((tag) => tag.name === '@hibernation')) {
    await clearHibernationTestData();
  }
  // Clean up move workspace test data - remove wiki-test-moved folder
  if (pickle.tags.some((tag) => tag.name === '@move-workspace')) {
    const wikiTestMovedPath = path.resolve(process.cwd(), 'wiki-test-moved');
    if (await fs.pathExists(wikiTestMovedPath)) {
      await fs.remove(wikiTestMovedPath);
    }
  }

  // Separate logs by test scenario for easier debugging
  try {
    const today = new Date().toISOString().split('T')[0];
    const wikiLogFile = `${logsDirectory}/wiki-${today}.log`;
    const tidgiLogFile = `${logsDirectory}/TidGi-${today}.log`;

    // Create a sanitized scenario name for the log files
    const scenarioName = pickle.name.replace(/[^a-z0-9]/gi, '_').substring(0, 50);

    if (await fs.pathExists(wikiLogFile)) {
      const targetWikiLog = `${logsDirectory}/${scenarioName}_wiki.log`;
      await fs.move(wikiLogFile, targetWikiLog, { overwrite: true });
    }

    if (await fs.pathExists(tidgiLogFile)) {
      const targetTidgiLog = `${logsDirectory}/${scenarioName}_TidGi.log`;
      await fs.move(tidgiLogFile, targetTidgiLog, { overwrite: true });
    }
  } catch (error) {
    console.error('Error moving log files:', error);
  }
});