Fix deleting API key

UdaraJay · Nov 11, 2023 · a128b0b · a128b0b
1 parent b49ef45
commit a128b0b
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/src/main/ipc.ts b/src/main/ipc.ts
@@ -20,6 +20,10 @@ ipcMain.handle('set-ai-key', async (event, secretKey) => {
   return await keytar.setPassword('pile', 'aikey', secretKey);
 });
 
+ipcMain.handle('delete-ai-key', async (event) => {
+  return await keytar.deletePassword('pile', 'aikey');
+});
+
 // Link preview
 ipcMain.handle('get-link-preview', async (event, url) => {
   const preview = await getLinkPreview(url)

diff --git a/src/renderer/context/AIContext.js b/src/renderer/context/AIContext.js
@@ -52,6 +52,10 @@ export const AIContextProvider = ({ children }) => {
     return window.electron.ipc.invoke('set-ai-key', secretKey);
   };
 
+  const deleteKey = () => {
+    return window.electron.ipc.invoke('delete-ai-key');
+  };
+
   const getCompletion = async (model = 'gpt-3', context) => {
     const response = await ai.chat.completions.create({
       model: 'gpt-4',
@@ -67,6 +71,7 @@ export const AIContextProvider = ({ children }) => {
     prompt,
     setKey,
     getKey,
+    deleteKey,
     getCompletion,
   };
 

diff --git a/src/renderer/pages/Pile/Settings/index.jsx b/src/renderer/pages/Pile/Settings/index.jsx
@@ -9,7 +9,7 @@ import {
 } from 'renderer/context/PilesContext';
 
 export default function Settings() {
-  const { ai, prompt, getKey, setKey } = useAIContext();
+  const { ai, prompt, getKey, setKey, deleteKey } = useAIContext();
   const [key, setCurrentKey] = useState('');
   const { currentTheme, setTheme } = usePilesContext();
 
@@ -27,7 +27,9 @@ export default function Settings() {
   };
 
   const handleSaveChanges = () => {
-    if (key != '') {
+    if (key == '') {
+      deleteKey();
+    } else {
       setKey(key);
     }
   };