@ng-atomic/schematics 4.1.1 → 4.3.0

package/README.md

@@ -1,7 +1,6 @@
-# schematics
+# @NgAtomic/Schematics
 
-This library was generated with [Nx](https://nx.dev).
-
-## Running unit tests
-
-Run `nx test schematics` to execute the unit tests via [Jest](https://jestjs.io).
+## How to use
+```sh
+$ ng g gpt3 
+```

package/collection.json

@@ -56,7 +56,12 @@
     "setup": {
       "description": "Setup angular-atomic-schematics",
       "factory": "./src/setup/index#setup",
-      "schema": "./src/ng-add//schema.json"
+      "schema": "./src/ng-add/schema.json"
+    },
+    "gpt3": {
+      "description": "Generate files by GPT-3",
+      "factory": "./src/gpt3/index#gpt3",
+      "schema": "./src/ng-add/schema.json"
     }
   }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ng-atomic/schematics",
-  "version": "4.1.1",
+  "version": "4.3.0",
   "license": "MIT",
   "schematics": "./collection.json",
   "ng-add": {
@@ -11,6 +11,7 @@
   "dependencies": {},
   "peerDependencies": {
     "@angular/core": "^14.0.0",
+    "openai": "^3.1.0",
     "@angular/common": "^14.0.0",
     "@angular/router": "^14.0.0",
     "string-template": "^1.0.0",

package/src/gpt3/helpers.d.ts

@@ -0,0 +1,3 @@
+export declare function parseFilePath(filePath: string, characters?: string[]): string[];
+export declare const convertByWords: (str: string, words: string[]) => string;
+export declare function getEstimateSimilarFilePaths(path: string, files: string[]): string[];

package/src/gpt3/helpers.js

@@ -0,0 +1,64 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getEstimateSimilarFilePaths = exports.convertByWords = exports.parseFilePath = void 0;
+function parseFilePath(filePath, characters = ['.', '/']) {
+    const result = [];
+    let word = '';
+    for (let i = 0; i < filePath.length; i++) {
+        const char = filePath[i];
+        if (characters.includes(char)) {
+            if (word.length)
+                result.push(word);
+            result.push(char);
+            word = '';
+        }
+        else {
+            word += char;
+        }
+    }
+    result.push(word);
+    return result;
+}
+exports.parseFilePath = parseFilePath;
+const convertByWords = (str, words) => {
+    return parseFilePath(str).map(c => ~words.indexOf(c) ? words.indexOf(c) : c).join('');
+};
+exports.convertByWords = convertByWords;
+function getEstimateSimilarFilePaths(path, files) {
+    const wordSet = new Set([path, ...files].map(path => parseFilePath(path)).flat());
+    wordSet.delete('.');
+    wordSet.delete('/');
+    const words = Array.from(wordSet);
+    const results = files.map(file => {
+        const distance = levenshteinDistance((0, exports.convertByWords)(path, words), (0, exports.convertByWords)(file, words)) + depthDistance(path, file);
+        return [distance, file];
+    });
+    const min = Math.min(...results.map(([distance]) => distance));
+    return results.filter(([distance]) => distance === min).map(([_, file]) => file);
+}
+exports.getEstimateSimilarFilePaths = getEstimateSimilarFilePaths;
+function depthDistance(str1, str2) {
+    const depth1 = str1.split('/').length;
+    const depth2 = str2.split('/').length;
+    return Math.abs(depth1 - depth2);
+}
+function levenshteinDistance(str1, str2) {
+    const cost = (a, b) => {
+        return a === b ? 0 : 1;
+    };
+    const d = [];
+    for (let i = 0; i <= str1.length; i++) {
+        d[i] = [];
+        d[i][0] = i;
+    }
+    for (let j = 0; j <= str2.length; j++) {
+        d[0][j] = j;
+    }
+    for (let i = 1; i <= str1.length; i++) {
+        for (let j = 1; j <= str2.length; j++) {
+            d[i][j] = Math.min(d[i - 1][j] + 1, d[i][j - 1] + 1, d[i - 1][j - 1] + cost(str1[i - 1], str2[j - 1]));
+        }
+    }
+    return d[str1.length][str2.length];
+}
+//# sourceMappingURL=helpers.js.map

package/src/gpt3/helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"helpers.js","sourceRoot":"","sources":["../../../../../libs/schematics/src/gpt3/helpers.ts"],"names":[],"mappings":";;;AACA,SAAgB,aAAa,CAAC,QAAgB,EAAE,UAAU,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC;IACrE,MAAM,MAAM,GAAG,EAAE,CAAC;IAClB,IAAI,IAAI,GAAG,EAAE,CAAC;IACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;QACxC,MAAM,IAAI,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QACzB,IAAI,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;YAC7B,IAAI,IAAI,CAAC,MAAM;gBAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClB,IAAI,GAAG,EAAE,CAAC;SACX;aAAM;YACL,IAAI,IAAI,IAAI,CAAC;SACd;KACF;IACD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClB,OAAO,MAAM,CAAC;AAChB,CAAC;AAfD,sCAeC;AAEM,MAAM,cAAc,GAAG,CAAC,GAAW,EAAE,KAAe,EAAU,EAAE;IACrE,OAAO,aAAa,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AACxF,CAAC,CAAA;AAFY,QAAA,cAAc,kBAE1B;AAED,SAAgB,2BAA2B,CAAC,IAAY,EAAE,KAAe;IACvE,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAClF,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IACpB,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IACpB,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAClC,MAAM,OAAO,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;QAC/B,MAAM,QAAQ,GAAG,mBAAmB,CAAC,IAAA,sBAAc,EAAC,IAAI,EAAE,KAAK,CAAC,EAAE,IAAA,sBAAc,EAAC,IAAI,EAAE,KAAK,CAAC,CAAC,GAAG,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAC3H,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAqB,CAAC;IAC9C,CAAC,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC/D,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,QAAQ,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;AACnF,CAAC;AAXD,kEAWC;AAED,SAAS,aAAa,CAAC,IAAY,EAAE,IAAY;IAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,OAAO,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,CAAC;AACnC,CAAC;AAGD,SAAS,mBAAmB,CAAC,IAAY,EAAE,IAAY;IACrD,MAAM,IAAI,GAAG,CAAC,CAAS,EAAE,CAAS,EAAU,EAAE;QAC5C,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACzB,CAAC,CAAA;IAED,MAAM,CAAC,GAAG,EAAE,CAAC;IACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;QACrC,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;QACV,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;KACb;IACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;QACrC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;KACb;IAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;QACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACrC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAChB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EACf,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EACf,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CACjD,CAAC;SACH;KACF;IAED,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AACrC,CAAC"}

package/src/gpt3/helpers.spec.ts

@@ -0,0 +1,56 @@
+import { getEstimateSimilarFilePaths, convertByWords, parseFilePath } from './helpers';
+
+describe('getEstimateSimilarFilePaths', () => {
+  const FILES = [
+    '/projects/app/src/styles.scss',
+    '/projects/app/src/app/app.component.html',
+    '/projects/app/src/app/_shared/components/example/example.module.ts',
+    '/projects/app/src/app/_shared/components/example/example.component.html',
+    '/projects/app/src/app/_shared/components/example/example.component.scss',
+    '/projects/app/src/app/_shared/components/example/example.component.spec.ts',
+    '/projects/app/src/app/_shared/components/example/example.component.ts',
+    '/projects/app/src/app/_shared/components/example/example.stories.ts',
+    '/projects/app/src/app/_shared/components/example/index.ts',
+    '/projects/app/src/app/_shared/components/test/test.module.ts',
+    '/projects/app/src/app/_shared/components/test/test.component.html',
+    '/projects/app/src/app/_shared/components/test/test.component.scss',
+    '/projects/app/src/app/_shared/components/test/test.component.spec.ts',
+    '/projects/app/src/app/_shared/components/test/test.component.ts',
+    '/projects/app/src/app/_shared/components/test/test.stories.ts',
+    '/projects/app/src/app/_shared/components/test/index.ts',
+  ];
+
+  it('', () => {
+    const file = '/projects/app/src/app/_shared/components/expected/expected.component.ts';
+    const files = getEstimateSimilarFilePaths(file, FILES);
+    expect(files).toEqual([
+      '/projects/app/src/app/_shared/components/example/example.component.ts',
+      '/projects/app/src/app/_shared/components/test/test.component.ts',
+    ]);
+  });
+
+  it('', () => {
+    const file = '/projects/app/src/app/_shared/components/expected/expected.component.html';
+    const files = getEstimateSimilarFilePaths(file, FILES);
+    expect(files).toEqual([
+      '/projects/app/src/app/_shared/components/example/example.component.html',
+      '/projects/app/src/app/_shared/components/test/test.component.html',
+    ]);
+  });
+});
+
+describe('parseFilePath', () => {
+  it('should parse file path', () => {
+    expect(parseFilePath('/projects/app/src/app/_shared/components/example/example.component.ts')).toEqual([
+      '/', 'projects', '/', 'app', '/',  'src', '/',  'app', '/',  '_shared', '/',  
+      'components', '/', 'example', '/', 'example', '.', 'component', '.', 'ts'
+    ]);
+  });
+});
+
+describe('convertByWords', () => {
+  xit('should convert from a string to a number', () => {
+    expect(convertByWords('this/is/test/path', ['this', 'is', 'test', 'path'])).toBe('0/1/2/3');
+  });
+});
+

package/src/gpt3/index.d.ts

@@ -0,0 +1,6 @@
+import { Rule, FileEntry } from '@angular-devkit/schematics';
+import { Schema } from '../atomic-component/schema';
+export declare const gpt3: (options: Schema) => Rule;
+export declare function completeToJson(text: string): string[];
+export declare function parseJsonCodeBlock(text: string): string[];
+export declare function parseCodeBlocks(text: string): FileEntry[];

package/src/gpt3/index.js

@@ -0,0 +1,101 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.parseCodeBlocks = exports.parseJsonCodeBlock = exports.completeToJson = exports.gpt3 = void 0;
+const schematics_1 = require("@angular-devkit/schematics");
+const workspace_1 = require("@schematics/angular/utility/workspace");
+const openai_1 = require("openai");
+const path_1 = require("path");
+const helpers_1 = require("./helpers");
+const config = new openai_1.Configuration({ apiKey: process.env['OPEN_AI_TOKEN'] });
+const openai = new openai_1.OpenAIApi(config);
+const gpt3 = (options) => async (tree) => {
+    options.path = (0, path_1.join)(await (0, workspace_1.createDefaultPath)(tree, options.project), options?.path ?? '');
+    const files = getFiles(tree.root);
+    const path = (0, path_1.join)(tree.root.path, options.path, options.name);
+    const estimatedFiles = (0, path_1.parse)(path).ext.length ? [path] : await getPredicatedFiles(files, path);
+    for (const file of estimatedFiles) {
+        const fileEntries = (0, helpers_1.getEstimateSimilarFilePaths)(file, files).map(file => tree.get(file));
+        const fileEntry = await getPredicatedFileEntry(file, fileEntries);
+        tree.create(file, fileEntry.content);
+    }
+    return (0, schematics_1.chain)([]);
+};
+exports.gpt3 = gpt3;
+async function getPredicatedFiles(files, dirPath) {
+    const fileStart = dirPath.split('/').slice(-1)[0];
+    const prompt = makeDirectoryPrompt(files, (0, path_1.join)(dirPath, fileStart));
+    const res = await openai.createCompletion({
+        model: 'code-davinci-002',
+        prompt,
+        temperature: 0,
+        max_tokens: 512,
+        stop: '\n\`\`\`',
+    });
+    const results = completeToJson(`${prompt}${res.data.choices?.[0].text}`);
+    return results.filter(result => result.startsWith(dirPath));
+}
+async function getPredicatedFileEntry(path, fileEntries = []) {
+    const prompt = makeFilePrompt(fileEntries, path);
+    const res = await openai.createCompletion({
+        model: 'code-davinci-002',
+        prompt,
+        temperature: 0,
+        max_tokens: 256,
+        stop: '\n\`\`\`',
+    });
+    const entries = parseCodeBlocks(`${prompt}${res.data.choices?.[0].text}`);
+    return entries.find(entry => entry.path === path);
+}
+function makeDirectoryPrompt(files, name) {
+    return `\n\`\`\`tree.json\n[\n${files.map(file => `\t\"${file}\",`).join('\n')}\n\t\"${name}`;
+}
+function makeFilePrompt(files, path) {
+    return `${makeFileCodeBlocks(files)}\n\n\`\`\`${path}\n`;
+}
+function makeFileCodeBlocks(files) {
+    return files.map(file => makeFileCodeBlock(file.path, file.content.toString())).join('\n\n');
+}
+function makeFileCodeBlock(name, content) {
+    return `\`\`\`${name}\n${content}\n\`\`\``;
+}
+function completeToJson(text) {
+    while (text.length) {
+        let suffixes = ['"]\n```', ']\n```', '\n```', '```', '``', '`', ''];
+        for (const suffix of suffixes) {
+            try {
+                return parseJsonCodeBlock(text + suffix);
+            }
+            catch { }
+        }
+        text = text.slice(0, -1);
+    }
+}
+exports.completeToJson = completeToJson;
+function parseJsonCodeBlock(text) {
+    const code = getCodeBlocks(text);
+    return JSON.parse(code);
+}
+exports.parseJsonCodeBlock = parseJsonCodeBlock;
+function getCodeBlocks(text) {
+    return text.match(/\`\`\`tree\.json\n([\s\S]*)\`\`\`/)?.[1];
+}
+function parseCodeBlocks(text) {
+    return text.split('```').filter((_, i) => i % 2).map(code => {
+        const [path, ...lines] = code.split('\n');
+        return { path, content: Buffer.from(lines.join('\n')) };
+    });
+}
+exports.parseCodeBlocks = parseCodeBlocks;
+function getFiles(dir) {
+    const files = [];
+    walkDir(dir, (path, entry) => entry.subfiles.forEach(file => files.push(`${path}/${file}`)));
+    return files;
+}
+function walkDir(dir, callback, parent = '/') {
+    dir.subdirs.forEach(path => {
+        const entry = dir.dir(path);
+        callback((0, path_1.join)(parent, path), entry);
+        walkDir(entry, callback, (0, path_1.join)(parent, path));
+    });
+}
+//# sourceMappingURL=index.js.map

package/src/gpt3/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../libs/schematics/src/gpt3/index.ts"],"names":[],"mappings":";;;AAAA,2DAAoF;AAEpF,qEAA0E;AAC1E,mCAAmD;AACnD,+BAAmC;AACnC,uCAAwD;AAExD,MAAM,MAAM,GAAG,IAAI,sBAAa,CAAC,EAAC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,EAAC,CAAC,CAAC;AACzE,MAAM,MAAM,GAAG,IAAI,kBAAS,CAAC,MAAM,CAAC,CAAC;AAE9B,MAAM,IAAI,GAAG,CAAC,OAAe,EAAQ,EAAE,CAAC,KAAK,EAAE,IAAU,EAAE,EAAE;IACnE,OAAO,CAAC,IAAI,GAAG,IAAA,WAAI,EAAC,MAAM,IAAA,6BAAiB,EAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;IAExF,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,IAAI,GAAG,IAAA,WAAI,EAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9D,MAAM,cAAc,GAAG,IAAA,YAAK,EAAC,IAAI,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,kBAAkB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAE/F,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE;QACjC,MAAM,WAAW,GAAG,IAAA,qCAA2B,EAAC,IAAI,EAAE,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;QACzF,MAAM,SAAS,GAAG,MAAM,sBAAsB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QAClE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,OAAO,CAAC,CAAC;KACtC;IAEF,OAAO,IAAA,kBAAK,EAAC,EAAE,CAAC,CAAC;AAClB,CAAC,CAAC;AAdW,QAAA,IAAI,QAcf;AAEF,KAAK,UAAU,kBAAkB,CAAC,KAAe,EAAE,OAAe;IAChE,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAClD,MAAM,MAAM,GAAG,mBAAmB,CAAC,KAAK,EAAE,IAAA,WAAI,EAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC;IACpE,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC;QACxC,KAAK,EAAE,kBAAkB;QACzB,MAAM;QACN,WAAW,EAAE,CAAC;QACd,UAAU,EAAE,GAAG;QACf,IAAI,EAAE,UAAU;KACjB,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,cAAc,CAAC,GAAG,MAAM,GAAG,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IACzE,OAAO,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC;AAC9D,CAAC;AAED,KAAK,UAAU,sBAAsB,CAAC,IAAY,EAAE,cAA2B,EAAE;IAC/E,MAAM,MAAM,GAAG,cAAc,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IACjD,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC;QACxC,KAAK,EAAE,kBAAkB;QACzB,MAAM;QACN,WAAW,EAAE,CAAC;QACd,UAAU,EAAE,GAAG;QACf,IAAI,EAAE,UAAU;KACjB,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,eAAe,CAAC,GAAG,MAAM,GAAG,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAC1E,OAAO,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;AACpD,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAe,EAAE,IAAY;IACxD,OAAO,yBAAyB,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,IAAI,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC;AAChG,CAAC;AAED,SAAS,cAAc,CAAC,KAAkB,EAAE,IAAY;IACtD,OAAO,GAAG,kBAAkB,CAAC,KAAK,CAAC,aAAa,IAAI,IAAI,CAAC;AAC3D,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAkB;IAC5C,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAC/F,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAY,EAAE,OAAe;IACtD,OAAO,SAAS,IAAI,KAAK,OAAO,UAAU,CAAC;AAC7C,CAAC;AAED,SAAgB,cAAc,CAAC,IAAY;IACzC,OAAM,IAAI,CAAC,MAAM,EAAE;QACjB,IAAI,QAAQ,GAAG,CAAC,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QAEpE,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE;YAC7B,IAAI;gBACF,OAAO,kBAAkB,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC;aAC1C;YAAC,MAAM,GAAG;SACZ;QAED,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;KAC1B;AACH,CAAC;AAZD,wCAYC;AAED,SAAgB,kBAAkB,CAAC,IAAY;IAC7C,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;IACjC,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAHD,gDAGC;AAED,SAAS,aAAa,CAAC,IAAY;IACjC,OAAO,IAAI,CAAC,KAAK,CAAC,mCAAmC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;AAC9D,CAAC;AAED,SAAgB,eAAe,CAAC,IAAY;IAC1C,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;QAC1D,MAAM,CAAC,IAAI,EAAE,GAAG,KAAK,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1C,OAAO,EAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAc,CAAC;IACrE,CAAC,CAAC,CAAC;AACL,CAAC;AALD,0CAKC;AAED,SAAS,QAAQ,CAAC,GAAa;IAC7B,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IAC7F,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,OAAO,CAAC,GAAa,EAAE,QAAiD,EAAE,MAAM,GAAG,GAAG;IAC7F,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;QACzB,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC5B,QAAQ,CAAC,IAAA,WAAI,EAAC,MAAM,EAAE,IAAI,CAAC,EAAE,KAAK,CAAC,CAAC;QACpC,OAAO,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAA,WAAI,EAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC"}

package/src/gpt3/index.spec.ts

@@ -0,0 +1,156 @@
+import { SchematicTestRunner, UnitTestTree } from '@angular-devkit/schematics/testing';
+import path from 'path';
+import { createWorkspace } from '../_testing';
+import { parseJsonCodeBlock, parseCodeBlocks } from './index';
+
+jest.setTimeout(300 * 1000);
+
+const COLLECTION_PATH = path.join(__dirname, '../../collection.json');
+
+describe('Gpt Schematics', () => {
+  const runner = new SchematicTestRunner('@ng-atomic/schematics', COLLECTION_PATH);
+  let tree: UnitTestTree;
+
+  describe('Angular Workspace', () => {
+    beforeEach(async () => {
+      tree = await createWorkspace(runner, tree);
+      tree = await runner.runSchematicAsync('atomic-component', {
+        project: 'app', path: '_shared/components', name: 'example'
+      }, tree).toPromise();
+
+      tree = await runner.runSchematicAsync('atomic-component', {
+        project: 'app', path: '_shared/components', name: 'test'
+      }, tree).toPromise();
+
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/example/example.module.ts');
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/example/example.component.html');
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/example/example.component.scss');
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/example/example.component.spec.ts');
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/example/example.component.ts');
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/example/example.stories.ts');
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/example/index.ts');
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/test/test.module.ts');
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/test/test.component.html');
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/test/test.component.scss');
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/test/test.component.spec.ts');
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/test/test.component.ts');
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/test/test.stories.ts');
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/test/index.ts');
+    });
+
+    xit('should create atomic component files', async () => {
+      tree = await runner.runSchematicAsync('gpt3', {
+        project: 'app', path: '_shared/components', name: 'expected'
+      }, tree).toPromise();
+      // expect(tree.files).toContain('/projects/app/src/app/_shared/components/expected/expected.module.ts');
+      // expect(tree.files).toContain('/projects/app/src/app/_shared/components/expected/expected.component.html');
+      // expect(tree.files).toContain('/projects/app/src/app/_shared/components/expected/expected.component.scss');
+      // expect(tree.files).toContain('/projects/app/src/app/_shared/components/expected/expected.component.spec.ts');
+      // expect(tree.files).toContain('/projects/app/src/app/_shared/components/expected/expected.component.ts');
+      // expect(tree.files).toContain('/projects/app/src/app/_shared/components/expected/expected.stories.ts');
+      // expect(tree.files).toContain('/projects/app/src/app/_shared/components/expected/index.ts');
+    });
+
+    xit('should create atomic component files', async () => {
+      tree = await runner.runSchematicAsync('gpt3', {
+        project: 'app', path: '_shared/components/', name: 'expected/expected.module.ts'
+      }, tree).toPromise();
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/expected/expected.module.ts');
+    });
+
+    it('should create atomic component files', async () => {
+      tree = await runner.runSchematicAsync('gpt3', {
+        project: 'app', path: '', name: '_shared/components/expected/expected.module.ts'
+      }, tree).toPromise();
+      expect(tree.files).toContain('/projects/app/src/app/_shared/components/expected/expected.module.ts');
+    });
+  });
+});
+
+
+const TEST = `
+\`\`\`tree.json
+[]
+\`\`\`
+`;
+
+// describe('completeToJson', () => {
+//   xit('should convert to json', () => {
+//     expect(completeToJson(TEST)).toBeTruthy();
+//   });
+// });
+
+describe('parseJsonCodeBlock', () => {
+  it('should parse json code block', () => {
+    expect(parseJsonCodeBlock(TEST)).toBeTruthy();
+  });
+});
+
+
+const CODE_BLOCKS = `
+\`\`\`/projects/app/src/app/_shared/components/test/test.component.ts
+import { ChangeDetectionStrategy, Component, OnInit } from '@angular/core';
+
+@Component({
+  selector: 'undefineds-test',
+  templateUrl: './test.component.html',
+  styleUrls: ['./test.component.scss'],
+  changeDetection: ChangeDetectionStrategy.OnPush
+})
+export class TestComponent implements OnInit {
+
+  constructor() { }
+
+  ngOnInit(): void {
+  }
+
+}
+
+\`\`\`
+
+\`\`\`/projects/app/src/app/_shared/components/example/example.component.ts
+import { ChangeDetectionStrategy, Component, OnInit } from '@angular/core';
+
+@Component({
+  selector: 'undefineds-example',
+  templateUrl: './example.component.html',
+  styleUrls: ['./example.component.scss'],
+  changeDetection: ChangeDetectionStrategy.OnPush
+})
+export class ExampleComponent implements OnInit {
+
+  constructor() { }
+
+  ngOnInit(): void {
+  }
+
+}
+
+\`\`\`
+
+\`\`\`/projects/app/src/app/_shared/components/expected/expected.component.ts
+import { ChangeDetectionStrategy, Component, OnInit } from '@angular/core';
+
+@Component({
+  selector: 'undefineds-expected',
+  templateUrl: './expected.component.html',
+  styleUrls: ['./expected.component.scss'],
+  changeDetection: ChangeDetectionStrategy.OnPush
+})
+export class ExpectedComponent implements OnInit {
+
+  constructor() { }
+
+  ngOnInit(): void {
+  }
+
+}
+
+\`\`\`
+`;
+
+describe('parseCodeBlocks', () => {
+  it('should parse code blocks', () => {
+    expect(parseCodeBlocks(CODE_BLOCKS)).toBeTruthy();
+  });
+});

package/src/gpt3/schema.json

@@ -0,0 +1,22 @@
+{
+  "$schema": "http://json-schema.org/schema",
+  "$id": "ng-atomic-schematics-gpt3",
+  "title": "Angular Atomic Schematics GPT3",
+  "type": "object",
+  "properties": {
+    "project": {
+      "type": "string",
+      "description": "Name of the project.",
+      "$default": {
+        "$source": "projectName"
+      }
+    },
+    "path": {
+      "description": "Setup components directory",
+      "type": "string",
+      "default": "_shared/components",
+      "x-prompt": "Where do you setup Angular Atomic Components?" 
+    }
+  },
+  "required": []
+}