@pega/custom-dx-components 0.23.2 → 0.23.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pega/custom-dx-components",
-  "version": "0.23.2",
+  "version": "0.23.3",
   "description": "Utility for building custom UI components",
   "bin": "./bin/index.js",
   "scripts": {
@@ -33,6 +33,7 @@
     "gitignore": "^0.7.0",
     "inquirer": "^9.1.1",
     "listr": "^0.14.3",
+    "lodash": "^4.17.21",
     "mini-css-extract-plugin": "2.6.1",
     "mustache": "^4.2.0",
     "ncp": "^2.0.0",

package/src/tasks/generate-ai/helper.js

@@ -1,139 +1,25 @@
+import fs from 'fs';
+import {sep, join} from 'path';
+import { URL, fileURLToPath } from 'url';
+
 import chalk from 'chalk';
 import inquirer from 'inquirer';
-import ora from 'ora';
-import path from 'path';
-import { URL, fileURLToPath } from 'url';
-import { join } from 'path';
-import fs from 'fs';
+import _ from 'lodash';
 
 import {
   AI_PROMPT_FILENAME,
   AI_DAVINCI_PROMPT_FILENAME
 } from '../../constants.js';
-
-import _ from 'lodash';
-
-
 import {
   getComponents,
   getComponentsOfType,
-  getPegaConfig,
-  getAIPromptData,
   getDirectoryFiles,
-  getAIComponent,
   getInfinityAIComponent,
   getComponentDirectoryPath,
   compileMustacheTemplate,
   getComponentConfigJson,
   getComponentConfigType
 } from '../../util.js';
-import { str } from 'ajv';
-
-
-export const getSelectAComponent = async() => {
-  //let localComponents = await getComponents();
-  const localComponents = await getComponentsOfType("Widget,Template");
-
-  let answers;
-
-  if (localComponents.length > 0) {
-    answers = await inquirer.prompt([
-      {
-        name: 'componentKey',
-        type: 'rawlist',
-        message: 'Select component to generateAI',
-        choices: localComponents
-      }]);
-  }
-
-  return answers;
-}
-
-export const determineAndGenerateAI = async (enteredKey) => {
-
-  if (!enteredKey) {
-    let answer = await getSelectAComponent();
-    if (answer) {
-      enteredKey = answer.componentKey;
-    }
-    else {
-      console.log(chalk.bold.redBright(`No custom components to generate AI. Please create a Widget or Template component first.`));
-      process.exit(1);
-    }
-
-  }
-
-  const targetDirectory = await getComponentDirectoryPath(enteredKey);
-  const configType = await getComponentConfigType(targetDirectory);
-
-  switch (configType.toLowerCase()) {
-    case "widget":
-    case "template":
-      await getWidgetTemplateAndGenerateAI(enteredKey);
-      break;
-    case "field":
-      await getFieldAndGenerateAI(enteredKey);
-      break;
-  }
-
-
-}
-
-export const getWidgetTemplateAndGenerateAI = async (enteredKey) => {
-  const localComponents = await getComponents();
-
-  let answers;
-
-  if (localComponents.length > 0) {
-
-    answers = await inquirer.prompt([
-
-      {
-        name: 'additionalPromptInfo',
-        message: 'Enter what you would like this component to do? ',
-      }
-
-
-    ]);
-
-    console.log("Generating component...");
-
-    await generateInfinityAIComponent(enteredKey, answers.additionalPromptInfo);
-
-  } else {
-    console.log(chalk.bold.redBright(`No custom components in ${sourceOfComponents}`));
-  }
-};
-
-
-export const getFieldAndGenerateAI = async (enteredKey) => {
-  const localComponents = await getComponents();
-
-  let answers;
-
-  if (localComponents.length > 0) {
-
-    answers = await inquirer.prompt([
-      {
-        name: 'designSystemComponents',
-        message: 'Enter comma delimited string of Design system components (ie. Input, TextArea, etc.)'
-      },
-      {
-        name: 'additionalPromptInfo',
-        message: 'Enter any additional generation information',
-      }
-
-    ]);
-
-
-    console.log("Generating component...");
-
-    await generateAIComponent(enteredKey || answers.componentKey, answers.designSystemComponents, answers.additionalPropmtInfo);
-
-  } else {
-    console.log(chalk.bold.redBright(`No custom components in ${sourceOfComponents}`));
-  }
-};
 
 export const renameFile = (oldFileName, newFileName) => {
   try {
@@ -146,12 +32,12 @@ export const renameFile = (oldFileName, newFileName) => {
 
 export const generateAIComponent = async (componentKey, designSystemComponents, additionalPromptInfo) => {
 
-  const fileName = "." + path.sep + AI_PROMPT_FILENAME;
+  const fileName = "." + sep + AI_PROMPT_FILENAME;
 
   // get config file
   //
   const targetDirectory = await getComponentDirectoryPath(componentKey);
-  let configFileData = await getComponentConfigJson(targetDirectory);
+  const configFileData = await getComponentConfigJson(targetDirectory);
 
  // let data = JSON.parse(configFileData);
  // data.properties = [];
@@ -169,12 +55,9 @@ export const generateAIComponent = async (componentKey, designSystemComponents,
   strPrompt = _.unescape(strPrompt);
 
 
-  //let strPrompt = await getAIPromptData(fileURLToPath(new URL(fileName, import.meta.url)));
-  //let aiComponent = await getAIComponent(strPrompt);
   let aiComponent = await getInfinityAIComponent(strPrompt);
 
   if (aiComponent) {
-    //console.log(aiData);
     const indexFile = join(targetDirectory, "index.jsx");
 
     // remove enclosed ``` and "Answer:"
@@ -252,9 +135,11 @@ export const generateAIComponent = async (componentKey, designSystemComponents,
 
 }
 
+
+
 export const generateInfinityAIComponent = async (componentKey, additionalPromptInfo) => {
 
-  const fileName = "." + path.sep + AI_DAVINCI_PROMPT_FILENAME;
+  const fileName = "." + sep + AI_DAVINCI_PROMPT_FILENAME;
 
   // get config file
   //
@@ -354,4 +239,152 @@ export const generateInfinityAIComponent = async (componentKey, additionalPrompt
 
 
 
+}
+
+export const getWidgetTemplateAndGenerateAI = async (enteredKey) => {
+  const localComponents = await getComponents();
+
+  let answers;
+
+  if (localComponents.length > 0) {
+
+    answers = await inquirer.prompt([
+      {
+        name: "fileOrText",
+        type: "rawlist",
+        choices: [ {name: "Prompt via a file", value:"File"}, {name: "Prompt entered text", value:"Text"}],
+        message: 'Enter how you want to propmt for component instructions '
+      },
+      {
+        name: 'fileName',
+        message: 'Enter file name? ',
+        default: 'prompt.txt',
+        when(answers) {
+          return answers.fileOrText === 'File';
+        }
+      },
+      {
+        name: 'additionalPromptInfo',
+        message: 'Enter what you would like this component to do? ',
+        when(answers) {
+          return answers.fileOrText === 'Text';
+        }
+      }
+
+    ]);
+
+    console.log("Generating component...");
+
+    if (answers.fileOrText === "Text") {
+      await generateInfinityAIComponent(enteredKey, answers.additionalPromptInfo);
+    }
+    else if (answers.fileOrText == "File") {
+      const strPrompt = await getFilePromptData(enteredKey, answers.fileName);
+
+      await generateInfinityAIComponent(enteredKey, strPrompt);
+    }
+
+
+
+  } else {
+    console.log(chalk.bold.redBright(`No custom components in ${sourceOfComponents}`));
+  }
+};
+
+export const getFilePromptData = async (componentKey, fileName) => {
+  const targetDirectory = await getComponentDirectoryPath(componentKey);
+
+  const fullFileName = join(targetDirectory, fileName);
+
+  if (fs.existsSync(fullFileName)) {
+    const promptData = fs.readFileSync(fullFileName, { encoding: 'utf8' });
+    return promptData;
+  }
+  else {
+    console.log(chalk.bold.redBright(`File ${fileName} does not exist.`));
+    process.exit(1);
+  }
+
+
+}
+
+export const getFieldAndGenerateAI = async (enteredKey) => {
+  const localComponents = await getComponents();
+
+  let answers;
+
+  if (localComponents.length > 0) {
+
+    answers = await inquirer.prompt([
+      {
+        name: 'designSystemComponents',
+        message: 'Enter comma delimited string of Design system components (ie. Input, TextArea, etc.)'
+      },
+      {
+        name: 'additionalPromptInfo',
+        message: 'Enter any additional generation information',
+      }
+
+    ]);
+
+
+    console.log("Generating component...");
+
+    await generateAIComponent(enteredKey || answers.componentKey, answers.designSystemComponents, answers.additionalPropmtInfo);
+
+  } else {
+    console.log(chalk.bold.redBright(`No custom components in ${sourceOfComponents}`));
+  }
+};
+
+
+
+export const getSelectAComponent = async() => {
+  const localComponents = await getComponentsOfType("Widget,Template");
+
+  let answers;
+
+  if (localComponents.length > 0) {
+    answers = await inquirer.prompt([
+      {
+        name: 'componentKey',
+        type: 'rawlist',
+        message: 'Select component to generateAI',
+        choices: localComponents
+      }]);
+  }
+
+  return answers;
+}
+
+export const determineAndGenerateAI = async (enteredKey) => {
+
+  if (!enteredKey) {
+    const answer = await getSelectAComponent();
+    if (answer) {
+      enteredKey = answer.componentKey;
+    }
+    else {
+      console.log(chalk.bold.redBright(`No custom components to generate AI. Please create a Widget or Template component first.`));
+      process.exit(1);
+    }
+
+  }
+
+  const targetDirectory = await getComponentDirectoryPath(enteredKey);
+  const configType = await getComponentConfigType(targetDirectory);
+
+  switch (configType.toLowerCase()) {
+    case "widget":
+    case "template":
+      await getWidgetTemplateAndGenerateAI(enteredKey);
+      break;
+    case "field":
+      await getFieldAndGenerateAI(enteredKey);
+      break;
+    default:
+      break;
+  }
+
+
 }

package/src/tasks/init/template/package.json.mustache

@@ -57,7 +57,7 @@
   },
   "devDependencies": {
     "@pega/configs": "^0.5.3",
-    "@pega/custom-dx-components": "^0.23.2",
+    "@pega/custom-dx-components": "^0.23.3",
     "@pega/eslint-config": "^0.3.2",
     "@storybook/addon-actions": "^6.5.12",
     "@storybook/addon-essentials": "^6.5.12",