P3) VSCode Extension: Add a Refresh Button to Tree View #4107 (#5)

* # Feature (4107): P3) VSCode Extension: Add a Refresh Button to Tree View

- load config now can return value
- Add or Update config function added

* # Feature (4107): P3) VSCode Extension: Add a Refresh Button to Tree View

- load config now can return value
- Add or Update config function added
- Version updated

Author: sudiptatj

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "polyapi",
-  "version": "0.23.15",
+  "version": "0.23.16",
   "description": "Poly is a CLI tool to help create and manage your Poly definitions.",
   "license": "MIT",
   "repository": {

package.json

@@ -14,7 +14,7 @@ import {
   WebhookHandleSpecification,
 } from '../../types';
 import { getSpecs } from '../../api';
-import { loadConfig } from '../../config';
+import { loadConfig, addOrUpdateConfig } from '../../config';
 import {
   generateContextDataFile,
   getContextDataFileContent,
@@ -393,6 +393,9 @@ const generate = async ({
 
   try {
     specs = await getSpecs(contexts, names, functionIds, noTypes);
+    if (contexts) {
+      addOrUpdateConfig(polyPath, 'LAST_GENERATE_CONTEXTS_USED', contexts?.join(','));
+    }
   } catch (error) {
     showErrGettingSpecs(error);
     return;

src/commands/generate/index.ts

@@ -8,11 +8,17 @@ const getPolyConfigDirPath = (polyPath: string) =>
 const getPolyConfigFilePath = (polyPath: string) =>
   `${getPolyConfigDirPath(polyPath)}/.config.env`;
 
-export const loadConfig = (polyPath: string) => {
+export const loadConfig = (polyPath: string): Record<string, string> | undefined => {
   const configFilePath = getPolyConfigFilePath(polyPath);
   if (fs.existsSync(configFilePath)) {
-    dotenv.config({ path: configFilePath, override: process.env.CONFIG_ENV_PATH_PRIORITY === 'true' });
+    const result = dotenv.config({
+      path: configFilePath,
+      override: process.env.CONFIG_ENV_PATH_PRIORITY === 'true',
+    });
+
+    return result.parsed;
   }
+  return undefined;
 };
 
 export const saveConfig = (polyPath: string, config: Record<string, string>) => {
@@ -24,3 +30,11 @@ export const saveConfig = (polyPath: string, config: Record<string, string>) =>
       .join('\n'),
   );
 };
+
+export const addOrUpdateConfig = (polyPath: string, key: string, value: string) => {
+  const existingConfig = loadConfig(polyPath) ?? {};
+
+  existingConfig[key] = value;
+
+  saveConfig(polyPath, existingConfig);
+};