testdriverai 4.0.25 → 4.0.27

package/index.js

@@ -16,6 +16,7 @@ const http = require('http');
 const path = require('path');
 const chalk = require('chalk')
 const yaml = require('js-yaml');
+const sanitizeFilename = require("sanitize-filename");
 const macScreenPerms = require('mac-screen-capture-permissions');
 
 // local modules
@@ -398,6 +399,36 @@ const humanInput = async (currentTask, validateAndLoop = false) => {
   
 }
 
+const generate = async (type) => {
+
+  log.log('debug', 'generate called', type)
+
+  speak('thinking...');
+  notify('thinking...');
+  log.log('info', chalk.dim('thinking...'), true);
+
+  log.log('info', '');
+
+  let image = await system.captureScreenBase64();
+  let message = await sdk.req('generate', {
+    type,
+    image});
+
+  log.prettyMarkdown(message)
+
+  let testPrompts = await parser.findGenerativePrompts(message);
+
+  // for each testPrompt
+  for (const testPrompt of testPrompts) {
+    // write a file called testprompt.headings[0].replace(' ', '-').toLowerCase().md
+    // with the contents of the testPrompt
+    let fileName = testPrompt.headings[0].trim().replace(/ /g, '-').toLowerCase() + '.md';
+    let path1 = path.join(process.cwd(), 'testdriver', '.generate', fileName);
+    let contents = testPrompt.listsOrdered[0].map((item, index) => `${index + 1}. /explore ${item}`).join('\n');
+    fs.writeFileSync(path1, contents);
+  }
+}
+
 const popFromHistory = async (fullStep) => {
 
   log.log('info', chalk.dim('undoing...'), true)
@@ -501,7 +532,9 @@ const firstPrompt = async (text) => {
     analytics.track('input', {input});
 
     console.log('') // adds a nice break between submissions
-    
+
+    let commands = input.split(' ');
+
     // if last character is a question mark, we assume the user is asking a question
     if (input.indexOf('/summarize') == 0) {
       await summarize();
@@ -510,19 +543,17 @@ const firstPrompt = async (text) => {
     } else if (input.indexOf('/save') == 0) {
       await save();
     } else if (input.indexOf('/explore') == 0) {
-      let commands = input.split(' ');
       await humanInput(commands.slice(1).join(' '), true)
     } else if (input.indexOf('/undo') == 0) {
       await undo();
     } else if (input.indexOf('/assert') == 0) {
-      let commands = input.split(' ');
       await assert(commands.slice(1).join(' '))
     } else if (input.indexOf('/manual') == 0) {
-      let commands = input.split(' ');
       await manualInput(commands.slice(1).join(' '))
     } else if (input.indexOf('/run') == 0) {
-      let commands = input.split(' ');
       await run(commands[1], commands[2]);
+    } else if (input.indexOf('/generate') == 0) {
+      await generate(commands[1]);
     } else {
       await humanInput(input, false)
     }

package/lib/commands.js

@@ -41,7 +41,7 @@ const findImageOnScreen = async (relativePath, haystack, restrictToWindow) => {
   }
 
   // check if the file exists
-  if (await !fs.exists(path.join(rootpath, relativePath))) {
+  if (await !fs.existsSync(path.join(rootpath, relativePath))) {
     throw new AiError(`Image not found: ${relativePath}`)
   }
 

package/lib/init.js

@@ -117,7 +117,7 @@ module.exports = async () => {
       await fs.appendFileSync(append, env);
 
     
-  let resolvedPath = await getLatestRelease('dashcamio', 'testdriver-web');
+  let resolvedPath = await getLatestRelease('testdriverai', 'testdriver-web');
 
   if (resolvedPath) {
     const directory = await decompress(resolvedPath, process.cwd(), {
@@ -136,6 +136,11 @@ module.exports = async () => {
     if (!fs.existsSync(testdriverFolder)) {
       fs.mkdirSync(testdriverFolder);
     }
+
+    const testdriverGenerateFolder = path.join(process.cwd(), 'testdriver', '.generate');
+    if (!fs.existsSync(testdriverGenerateFolder)) {
+      fs.mkdirSync(testdriverGenerateFolder);
+    }
   }
 
   console.log('');

package/lib/parser.js

@@ -5,7 +5,7 @@ const yaml = require('js-yaml');
 let parser = new Parser();
 
 // use markdown parser to find code blocks within AI response
-const findCodeBlocks = (markdownContent) => {
+const findCodeBlocks = async function(markdownContent) {
   
   return new Promise((resolve, reject) => {
     parser.parse(markdownContent, (err, result) => {
@@ -26,6 +26,40 @@ const findCodeBlocks = (markdownContent) => {
 
 }
 
+// use markdown parser to find code blocks within AI response
+const findGenerativePrompts = async function(markdownContent) {
+  
+  return new Promise((resolve, reject) => {
+    parser.parse(markdownContent, async (err, result) => {
+
+      if (err) {
+        return reject(err);
+      }
+
+      // parse the markdown content of each code block
+      let codes = result.codes.map((code) => {
+        return new Promise((resolve2, reject2) => {
+          parser.parse(code.code, (err, result) => {
+            if (err) {
+              reject2(err);
+            } else {
+              resolve2(result);
+            }
+          });
+        });
+      });
+
+      // use Promise.all to wait for all the promises to resolve
+      let parsedCodes = await Promise.all(codes);
+
+      return resolve(parsedCodes);
+
+    });
+    
+  });
+
+}
+
 // parse the yml from the included codeblock and clean it up
 const getYAMLFromCodeBlock = function(codeblock) {
 
@@ -58,6 +92,7 @@ const parseYAML = async function(inputYaml) {
 
 module.exports = { 
   findCodeBlocks,
+  findGenerativePrompts,
   getYAMLFromCodeBlock,
   getCommands: async function(codeBlock) {
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "testdriverai",
-  "version": "4.0.25",
+  "version": "4.0.27",
   "description": "Next generation autonomous AI agent for end-to-end testing of web & desktop",
   "main": "index.js",
   "bin": {
@@ -35,6 +35,7 @@
     "remark-parse": "^11.0.0",
     "rimraf": "^5.0.5",
     "robotjs": "^0.6.0",
+    "sanitize-filename": "^1.6.3",
     "say": "^0.16.0",
     "screenshot-desktop": "^1.15.0",
     "semver": "^7.6.2",