From 9e5811fc1cf8d317d65de434e0d88acae671fd5e Mon Sep 17 00:00:00 2001
From: dev1abhi <abhilashsarangi222@gmail.com>
Date: Wed, 8 Jan 2025 13:06:42 +0530
Subject: [PATCH] feat: update environment variable handling and add config.js
 for API keys

---
 .gitignore   |  3 ++-
 extension.js | 14 ++++++++++----
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/.gitignore b/.gitignore
index 3b78273..2d1ea72 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,4 +1,5 @@
 node_modules
 .vscode-test/
 *.vsix
-.env
\ No newline at end of file
+.env
+config.js
\ No newline at end of file
diff --git a/extension.js b/extension.js
index fb1c4e0..ff98d57 100644
--- a/extension.js
+++ b/extension.js
@@ -6,6 +6,12 @@ const axios = require('axios');
 
 require('dotenv').config(); 
 
+const config = require('./config');
+
+const unsplashKey = Buffer.from(config.UNSPLASH_API_KEY, 'base64').toString('utf-8');
+const rapidApiKey = Buffer.from(config.RAPIDAPI_KEY, 'base64').toString('utf-8');
+const rapidApiHost = Buffer.from(config.RAPIDAPI_HOST, 'base64').toString('utf-8');
+
 //this calls searchImages and generateImage functions
 function activate(context) {
     // Command to search images
@@ -124,14 +130,14 @@ function activate(context) {
 }
 
 async function searchImages(query) {
-    const apiKey = process.env.UNSPLASH_API_KEY; 
+    //const apiKey = process.env.UNSPLASH_API_KEY; 
     const response = await axios.get(`https://api.unsplash.com/search/photos`, {
         params: {
             query: query,
             per_page: 5 // Number of images to retrieve
         },
         headers: {
-            Authorization: `Client-ID ${apiKey}`
+            Authorization: `Client-ID ${unsplashKey}`
         }
     });
 
@@ -144,8 +150,8 @@ async function generateImage(prompt) {
         method: 'POST',
         url: 'https://ai-image-generator10.p.rapidapi.com/image_gen_v2',
         headers: {
-          'x-rapidapi-key': process.env.RAPIDAPI_KEY,
-          'x-rapidapi-host': process.env.RAPIDAPI_HOST,
+          'x-rapidapi-key': rapidApiKey,
+          'x-rapidapi-host': rapidApiHost,
           'Content-Type': 'application/json'
         },
         data: {