@pega/custom-dx-components 0.23.3 → 0.23.4

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

@@ -1,390 +0,0 @@
-import fs from 'fs';
-import {sep, join} from 'path';
-import { URL, fileURLToPath } from 'url';
-
-import chalk from 'chalk';
-import inquirer from 'inquirer';
-import _ from 'lodash';
-
-import {
-  AI_PROMPT_FILENAME,
-  AI_DAVINCI_PROMPT_FILENAME
-} from '../../constants.js';
-import {
-  getComponents,
-  getComponentsOfType,
-  getDirectoryFiles,
-  getInfinityAIComponent,
-  getComponentDirectoryPath,
-  compileMustacheTemplate,
-  getComponentConfigJson,
-  getComponentConfigType
-} from '../../util.js';
-
-export const renameFile = (oldFileName, newFileName) => {
-  try {
-    fs.renameSync(oldFileName, newFileName);
-
-  } catch(err) {
-    console.log(err)
-  }
-}
-
-export const generateAIComponent = async (componentKey, designSystemComponents, additionalPromptInfo) => {
-
-  const fileName = "." + sep + AI_PROMPT_FILENAME;
-
-  // get config file
-  //
-  const targetDirectory = await getComponentDirectoryPath(componentKey);
-  const configFileData = await getComponentConfigJson(targetDirectory);
-
- // let data = JSON.parse(configFileData);
- // data.properties = [];
- // configFileData  = JSON.stringify(data);
-
-
-  let strPrompt = compileMustacheTemplate(fileURLToPath(new URL(fileName, import.meta.url)), {
-    COMPONENT_CONFIG: configFileData,
-    COMPONENT_DSC: designSystemComponents,
-    COMPONENT_ADDITIONAL: additionalPromptInfo
-  });
-
-  strPrompt = strPrompt.replaceAll("$", "//$");
-
-  strPrompt = _.unescape(strPrompt);
-
-
-  let aiComponent = await getInfinityAIComponent(strPrompt);
-
-  if (aiComponent) {
-    const indexFile = join(targetDirectory, "index.jsx");
-
-    // remove enclosed ``` and "Answer:"
-    aiComponent = aiComponent.substring(aiComponent.indexOf("import"));
-    aiComponent = aiComponent.substring(0, aiComponent.lastIndexOf("```"));
-
-
-
-    // get directory list
-    // check for index-template.jsx.old
-    //    if not there, copy index.jsx to index-template.jsx.old and then copy aiComponent to index.jsx
-    //    if there, check if have index-genai-1.jsx.old, if not, then copy index.jsx, to index-genai-1.jsx.old
-    //    otherwise, find all index-genai-X and find biggest number and add one, and save index.jsx to that and then create index.jsx
-
-    const dirList = await getDirectoryFiles(targetDirectory);
-
-    if (dirList.includes("index-template.jsx.old")) {
-
-      const genList = dirList.map( (el) => {
-        if (el.indexOf("index-genai") >= 0) {
-          return el;
-        }
-        else {
-          return "";
-        }
-      });
-
-      if (genList.length > 0) {
-        let lastIndex = 0;
-
-        genList.forEach((el) => {
-          const theName = el.split(".")[0];
-          const theIndex = theName.split("-")[2];
-
-          if (parseInt(theIndex) > lastIndex) {
-            lastIndex = parseInt(theIndex);
-          }
-        });
-
-        lastIndex++;
-
-        const oldGenFileName = join(targetDirectory, "index.jsx");
-        const newGenFileName = join(targetDirectory, "index-genai-" + lastIndex + ".jsx.old");
-
-        renameFile(oldGenFileName, newGenFileName);
-      }
-      else {
-        const oldGenFileName = join(targetDirectory, "index.jsx");
-        const newGenFileName = join(targetDirectory, "index-genai-1.jsx.old");
-
-        renameFile(oldGenFileName, newGenFileName);
-      }
-    }
-    else {
-      // so only index.jsx, so rename index.jsx to index-template.jsx.old
-      // and create a new index.jsx
-      const oldFileName = join(targetDirectory, "index.jsx");
-      const newFileName = join(targetDirectory, "index-template.jsx.old");
-
-      renameFile(oldFileName, newFileName);
-    }
-
-    // creae new index.jsx
-    fs.writeFileSync(indexFile, aiComponent, err => {
-      if (err) {
-        Promise.reject(`${chalk.bold.red(err)}`);
-      }
-    });
-
-    console.log(`Updated file ${chalk.green('index.jsx')} in directory ${chalk.green(componentKey)} with AI generated code.`);
-  }
-
-
-
-
-}
-
-
-
-export const generateInfinityAIComponent = async (componentKey, additionalPromptInfo) => {
-
-  const fileName = "." + sep + AI_DAVINCI_PROMPT_FILENAME;
-
-  // get config file
-  //
-  const targetDirectory = await getComponentDirectoryPath(componentKey);
-  let configFileData = await getComponentConfigJson(targetDirectory);
-
-  // need to escape $
-  configFileData = configFileData.replaceAll("$", "\\$");
-
-  let strPrompt = compileMustacheTemplate(fileURLToPath(new URL(fileName, import.meta.url)), {
-    COMPONENT_CONFIG: configFileData,
-    COMPONENT_ADDITIONAL: additionalPromptInfo
-  });
-
-  strPrompt = _.unescape(strPrompt);
-
-  //let strPrompt = await getAIPromptData(fileURLToPath(new URL(fileName, import.meta.url)));
-  let aiComponent = await getInfinityAIComponent(strPrompt);
-
-  //console.log(aiData);
-
-  if (aiComponent) {
-    const indexFile = join(targetDirectory, "index-genai.jsx.tmp");
-
-    // remove enclosed ``` and "Answer:"
-    if (aiComponent.indexOf("```") >= 0) {
-      aiComponent = aiComponent.substring(aiComponent.indexOf("```"));
-      aiComponent = aiComponent.substring(0, aiComponent.lastIndexOf("```"));
-    }
-    else if (aiComponent.indexOf("'") == 0) {
-      aiComponent = aiComponent.substring(aiComponent.indexOf("'"));
-      aiComponent = aiComponent.substring(0, aiComponent.lastIndexOf("'"));
-    }
-
-
-
-
-    // get directory list
-    // check for index-template.jsx.old
-    //    if not there, copy index.jsx to index-template.jsx.old and then copy aiComponent to index.jsx
-    //    if there, check if have index-genai-1.jsx.old, if not, then copy index.jsx, to index-genai-1.jsx.old
-    //    otherwise, find all index-genai-X and find biggest number and add one, and save index.jsx to that and then create index.jsx
-
-    const dirList = await getDirectoryFiles(targetDirectory);
-
-
-
-      const genList = dirList.map( (el) => {
-        if (el.indexOf("index-genai") >= 0) {
-          return el;
-        }
-        else {
-          return "";
-        }
-      });
-
-    if (genList.length > 0) {
-      let lastIndex = -1;
-
-      genList.forEach((el) => {
-        const theName = el.split(".")[0];
-
-        if (theName.indexOf("index-genai") >=0) {
-          if (lastIndex === -1) {
-            lastIndex = 0;
-          }
-          const theIndex = theName.split("-")[2];
-
-          if (parseInt(theIndex) > lastIndex) {
-            lastIndex = parseInt(theIndex);
-          }
-        }
-
-      });
-
-      lastIndex++;
-
-      if (lastIndex > 0) {
-        const oldGenFileName = join(targetDirectory, "index-genai.jsx.tmp");
-        const newGenFileName = join(targetDirectory, "index-genai-" + lastIndex + ".jsx.tmp");
-
-        renameFile(oldGenFileName, newGenFileName);
-      }
-
-    }
-
-
-    // creae new index.jsx
-    fs.writeFileSync(indexFile, aiComponent, err => {
-      if (err) {
-        Promise.reject(`${chalk.bold.red(err)}`);
-      }
-    });
-
-    console.log(`Updated file ${chalk.green('index-genai.jsx.tmp')} in directory ${chalk.green(componentKey)} with AI generated code.`);
-  }
-
-
-
-}
-
-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/generate-ai/index.js

@@ -1,34 +0,0 @@
-import inquirer from 'inquirer';
-import ora from 'ora';
-import chalk from 'chalk';
-
-import {
-  determineAndGenerateAI
-} from './helper.js';
-
-import { showVersion } from '../../util.js';
-
-
-
-
-export default async options => {
-  showVersion();
-
-  if (options.params.length > 3 && options.params[3] != "") {
-    const sourceOfComponents = options.params[3];
-
-    //await generateAIComponent(sourceOfComponents);
-    //await getComponentAndGenerateAI(sourceOfComponents);
-    await determineAndGenerateAI(sourceOfComponents);
-  }
-  else {
-
-    //await getComponentAndGenerateAI();
-    await determineAndGenerateAI();
-  }
-
-
-
-
-
-};