codeceptjs 4.0.0-beta.2 → 4.0.0-beta.20

package/lib/pause.js

@@ -1,210 +1,238 @@
-import colors from 'chalk';
-import readline from 'readline';
-import ora from 'ora-classic';
-import debug from 'debug';
-
-import container from './container.js';
-import history from './history.js';
-import { store } from './store.js';
-import AiAssistant from './ai.js';
-import recorder from './recorder.js';
-import * as event from './event.js';
-import * as output from './output.js';
-import { methodsOfObject } from './utils.js';
-
-debug('codeceptjs:pause');
+import colors from 'chalk'
+import readline from 'readline'
+import ora from 'ora-classic'
+import debugModule from 'debug'
+const debug = debugModule('codeceptjs:pause')
+import container from './container.js'
+import history from './history.js'
+import store from './store.js'
+import aiAssistant from './ai.js'
+import recorder from './recorder.js'
+import event from './event.js'
+import output from './output.js'
+import { methodsOfObject, searchWithFusejs } from './utils.js'
 
 // npm install colors
-let rl;
-let nextStep;
-let finish;
-let next;
-let registeredVariables = {};
-let aiAssistant;
+let rl
+let nextStep
+let finish
+let next
+let registeredVariables = {}
 /**
  * Pauses test execution and starts interactive shell
  * @param {Object<string, *>} [passedObject]
  */
 const pause = function (passedObject = {}) {
-  if (store.dryRun) return;
+  if (store.dryRun) return
 
-  next = false;
+  next = false
   // add listener to all next steps to provide next() functionality
   event.dispatcher.on(event.step.after, () => {
     recorder.add('Start next pause session', () => {
-      if (!next) return;
-      return pauseSession();
-    });
-  });
-  recorder.add('Start new session', () => pauseSession(passedObject));
-};
+      // test already finished, nothing to pause
+      if (!store.currentTest) return
+      if (!next) return
+      return pauseSession()
+    })
+  })
+
+  event.dispatcher.on(event.test.finished, () => {
+    finish()
+    recorder.session.restore('pause')
+    rl.close()
+    history.save()
+  })
+
+  recorder.add('Start new session', () => pauseSession(passedObject))
+}
 
 function pauseSession(passedObject = {}) {
-  registeredVariables = passedObject;
-  recorder.session.start('pause');
+  registeredVariables = passedObject
+  recorder.session.start('pause')
   if (!next) {
-    let vars = Object.keys(registeredVariables).join(', ');
-    if (vars) vars = `(vars: ${vars})`;
-
-    aiAssistant = AiAssistant.getInstance();
+    let vars = Object.keys(registeredVariables).join(', ')
+    if (vars) vars = `(vars: ${vars})`
 
-    output.print(colors.yellow(' Interactive shell started'));
-    output.print(colors.yellow(' Use JavaScript syntax to try steps in action'));
-    output.print(colors.yellow(` - Press ${colors.bold('ENTER')} to run the next step`));
-    output.print(colors.yellow(` - Press ${colors.bold('TAB')} twice to see all available commands`));
-    output.print(colors.yellow(` - Type ${colors.bold('exit')} + Enter to exit the interactive shell`));
-    output.print(colors.yellow(` - Prefix ${colors.bold('=>')} to run js commands ${colors.bold(vars)}`));
+    output.print(colors.yellow(' Interactive shell started'))
+    output.print(colors.yellow(' Use JavaScript syntax to try steps in action'))
+    output.print(colors.yellow(` - Press ${colors.bold('ENTER')} to run the next step`))
+    output.print(colors.yellow(` - Press ${colors.bold('TAB')} twice to see all available commands`))
+    output.print(colors.yellow(` - Type ${colors.bold('exit')} + Enter to exit the interactive shell`))
+    output.print(colors.yellow(` - Prefix ${colors.bold('=>')} to run js commands ${colors.bold(vars)}`))
 
     if (aiAssistant.isEnabled) {
-      output.print(colors.blue(` ${colors.bold('OpenAI is enabled! (experimental)')} Write what you want and make OpenAI run it`));
-      output.print(colors.blue(' Please note, only HTML fragments with interactive elements are sent to OpenAI'));
-      output.print(colors.blue(' Ideas: ask it to fill forms for you or to click'));
-    } else {
-      output.print(colors.blue(` Enable OpenAI assistant by setting ${colors.bold('OPENAI_API_KEY')} env variable`));
+      output.print(colors.blue(` ${colors.bold('AI is enabled! (experimental)')} Write what you want and make AI run it`))
+      output.print(colors.blue(' Please note, only HTML fragments with interactive elements are sent to AI provider'))
+      output.print(colors.blue(' Ideas: ask it to fill forms for you or to click'))
     }
   }
-  rl = readline.createInterface(process.stdin, process.stdout, completer);
 
-  rl.on('line', parseInput);
+  rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+    terminal: true,
+    completer,
+    history: history.load(),
+    historySize: 50, // Adjust the history size as needed
+  })
+
+  store.onPause = true
+  rl.on('line', parseInput)
   rl.on('close', () => {
-    if (!next) console.log('Exiting interactive shell....');
-  });
-  return new Promise(((resolve) => {
-    finish = resolve;
-    // eslint-disable-next-line
-    return askForStep();
-  }));
+    if (!next) console.log('Exiting interactive shell....')
+    store.onPause = false
+  })
+  return new Promise(resolve => {
+    finish = resolve
+    return askForStep()
+  })
 }
 
-/* eslint-disable */
 async function parseInput(cmd) {
-  rl.pause();
-  next = false;
-  recorder.session.start('pause');
-  if (cmd === '') next = true;
+  rl.pause()
+  next = false
+  recorder.session.start('pause')
+  if (cmd === '') next = true
   if (!cmd || cmd === 'resume' || cmd === 'exit') {
-    finish();
-    recorder.session.restore();
-    rl.close();
-    history.save();
-    return nextStep();
+    finish()
+    recorder.session.restore('pause')
+    rl.close()
+    history.save()
+    return nextStep()
   }
   for (const k of Object.keys(registeredVariables)) {
-    eval(`var ${k} = registeredVariables['${k}'];`); // eslint-disable-line no-eval
+    eval(`var ${k} = registeredVariables['${k}'];`)
   }
 
-  let executeCommand = Promise.resolve();
+  let executeCommand = Promise.resolve()
 
   const getCmd = () => {
     debug('Command:', cmd)
-    return cmd;
-  };
+    return cmd
+  }
 
-  let isCustomCommand = false;
-  let lastError = null;
-  let isAiCommand = false;
-  let $res;
+  let isCustomCommand = false
+  let lastError = null
+  let isAiCommand = false
+  let $res
   try {
-    // eslint-disable-next-line
-    const locate = global.locate; // enable locate in this context
-    // eslint-disable-next-line
-    const I = container.support('I');
+    const locate = global.locate // enable locate in this context
+
+    const I = container.support('I')
     if (cmd.trim().startsWith('=>')) {
-      isCustomCommand = true;
-      cmd = cmd.trim().substring(2, cmd.length);
-    } else if (aiAssistant.isEnabled && !cmd.match(/^\w+\(/) && cmd.includes(' ')) {
-      const currentOutputLevel = output.level();
-      output.level(0);
-      const res = I.grabSource();
-      isAiCommand = true;
+      isCustomCommand = true
+      cmd = cmd.trim().substring(2, cmd.length)
+    } else if (aiAssistant.isEnabled && cmd.trim() && !cmd.match(/^\w+\(/) && cmd.includes(' ')) {
+      const currentOutputLevel = output.level()
+      output.level(0)
+      const res = I.grabSource()
+      isAiCommand = true
       executeCommand = executeCommand.then(async () => {
         try {
-          const html = await res;
-          await aiAssistant.setHtmlContext(html);
+          const html = await res
+          await aiAssistant.setHtmlContext(html)
         } catch (err) {
-          output.print(output.output.styles.error(' ERROR '), 'Can\'t get HTML context', err.stack);
-          return;
+          output.print(output.styles.error(' ERROR '), "Can't get HTML context", err.stack)
+          return
         } finally {
-          output.level(currentOutputLevel);
+          output.level(currentOutputLevel)
         }
-        // aiAssistant.mockResponse("```js\nI.click('Sign in');\n```");
-        const spinner = ora("Processing OpenAI request...").start();
-        cmd = await aiAssistant.writeSteps(cmd);
-        spinner.stop();
-        output.print('');
-        output.print(colors.blue(aiAssistant.getResponse()));
-        output.print('');
-        return cmd;
+
+        const spinner = ora('Processing AI request...').start()
+        cmd = await aiAssistant.writeSteps(cmd)
+        spinner.stop()
+        output.print('')
+        output.print(colors.blue(aiAssistant.getResponse()))
+        output.print('')
+        return cmd
       })
     } else {
-      cmd = `I.${cmd}`;
+      cmd = `I.${cmd}`
     }
-    executeCommand = executeCommand.then(async () => {
-      const cmd = getCmd();
-      if (!cmd) return;
-      return eval(cmd); // eslint-disable-line no-eval
-    }).catch((err) => {
-      debug(err);
-      if (isAiCommand) return;
-      if (!lastError) output.print(output.output.styles.error(' ERROR '), err.message);
-      debug(err.stack)
-
-      lastError = err.message;
-    })
+    executeCommand = executeCommand
+      .then(async () => {
+        const cmd = getCmd()
+        if (!cmd) return
+        return eval(cmd)
+      })
+      .catch(err => {
+        debug(err)
+        if (isAiCommand) return
+        if (!lastError) output.print(output.styles.error(' ERROR '), err.message)
+        debug(err.stack)
+
+        lastError = err.message
+      })
 
-    const val = await executeCommand;
+    const val = await executeCommand
 
     if (isCustomCommand) {
-      if (val !== undefined) console.log('Result', '$res=', val); // eslint-disable-line
-      $res = val;
+      if (val !== undefined) console.log('Result', '$res=', val)
+      $res = val
     }
 
     if (cmd?.startsWith('I.see') || cmd?.startsWith('I.dontSee')) {
-      output.print(output.output.styles.success('  OK  '), cmd);
+      output.print(output.styles.success('  OK  '), cmd)
     }
     if (cmd?.startsWith('I.grab')) {
-      output.print(output.output.styles.debug(val));
+      try {
+        output.print(output.styles.debug(JSON.stringify(val, null, 2)))
+      } catch (err) {
+        output.print(output.styles.error(' ERROR '), 'Failed to stringify result:', err.message)
+        output.print(output.styles.error(' RAW VALUE '), String(val))
+      }
     }
 
-    history.push(cmd); // add command to history when successful
+    history.push(cmd) // add command to history when successful
   } catch (err) {
-    if (!lastError) output.print(output.output.styles.error(' ERROR '), err.message);
-    lastError = err.message;
+    if (!lastError) output.print(output.styles.error(' ERROR '), err.message)
+    lastError = err.message
   }
-  recorder.session.catch((err) => {
-    const msg = err.cliMessage ? err.cliMessage() : err.message;
+  recorder.session.catch(err => {
+    const msg = err.cliMessage ? err.cliMessage() : err.message
 
     // pop latest command from history because it failed
-    history.pop();
-
-    if (isAiCommand) return;
-    if (!lastError) output.print(output.output.styles.error(' FAIL '), msg);
-    lastError = err.message;
-  });
-  recorder.add('ask for next step', askForStep);
-  nextStep();
+    history.pop()
+
+    if (isAiCommand) return
+    if (!lastError) output.print(output.styles.error(' FAIL '), msg)
+    lastError = err.message
+  })
+  recorder.add('ask for next step', askForStep)
+  nextStep()
 }
-/* eslint-enable */
 
 function askForStep() {
-  return new Promise(((resolve) => {
-    nextStep = resolve;
-    rl.setPrompt(' I.', 3);
-    rl.resume();
-    rl.prompt([false]);
-  }));
+  return new Promise(resolve => {
+    nextStep = resolve
+    rl.setPrompt(' I.', 3)
+    rl.resume()
+    rl.prompt([false])
+  })
 }
 
 function completer(line) {
-  const I = container.support('I');
-  const completions = methodsOfObject(I);
-  const hits = completions.filter((c) => {
-    if (c.indexOf(line) === 0) {
-      return c;
-    }
-    return null;
-  });
-  return [hits && hits.length ? hits : completions, line];
+  const I = container.support('I')
+  const completions = methodsOfObject(I)
+  // If no input, return all completions
+  if (!line) {
+    return [completions, line]
+  }
+
+  // Search using Fuse.js
+  const searchResults = searchWithFusejs(completions, line, {
+    threshold: 0.3,
+    distance: 100,
+    minMatchCharLength: 1,
+  })
+  const hits = searchResults.map(result => result.item)
+
+  return [hits, line]
+}
+
+function registerVariable(name, value) {
+  registeredVariables[name] = value
 }
 
-export default pause;
+export default pause
+export { registerVariable }