From fbc106b5c1f03c0f3fa7abc435fca4cf84f28a95 Mon Sep 17 00:00:00 2001
From: Joshen Lim <joshenlimek@gmail.com>
Date: Thu, 17 Nov 2022 10:06:37 +0800
Subject: [PATCH] Scaffold enc keys ui part 2

---
 .../Vault/EncryptionKeysManagement.tsx        | 75 +++++++++++++++----
 studio/pages/project/[ref]/settings/vault.tsx |  2 +-
 studio/stores/project/VaultStore.ts           | 30 ++++++--
 3 files changed, 87 insertions(+), 20 deletions(-)

diff --git a/studio/components/interfaces/Settings/Vault/EncryptionKeysManagement.tsx b/studio/components/interfaces/Settings/Vault/EncryptionKeysManagement.tsx
index 08ca7f372b7..810dd46b676 100644
--- a/studio/components/interfaces/Settings/Vault/EncryptionKeysManagement.tsx
+++ b/studio/components/interfaces/Settings/Vault/EncryptionKeysManagement.tsx
@@ -1,13 +1,27 @@
 import { observer } from 'mobx-react-lite'
-import { FC, Fragment, useEffect, useState } from 'react'
-import { Input, IconSearch, Listbox, Button, Divider, Modal, Form, IconHelpCircle } from 'ui'
+import { FC, Fragment, useState } from 'react'
+import {
+  Input,
+  IconSearch,
+  Listbox,
+  Button,
+  Divider,
+  Modal,
+  Form,
+  IconHelpCircle,
+  IconTrash,
+  Badge,
+} from 'ui'
 import EncryptionKeySelector from './EncryptionKeySelector'
 import InformationBox from 'components/ui/InformationBox'
 import { useStore } from 'hooks'
+import dayjs from 'dayjs'
 
-interface Props {}
+interface Props {
+  onSelectRemove: (key: any) => void
+}
 
-const EncryptionKeysManagement: FC<Props> = ({}) => {
+const EncryptionKeysManagement: FC<Props> = ({ onSelectRemove }) => {
   const { vault } = useStore()
 
   const [searchValue, setSearchValue] = useState<string>('')
@@ -15,7 +29,13 @@ const EncryptionKeysManagement: FC<Props> = ({}) => {
   const [showAddKeyModal, setShowAddKeyModal] = useState(false)
   const [selectedKeyToRemove, setSelectedKeyToRemove] = useState<any>()
 
-  console.log('keys', vault.listKeys())
+  const keys = searchValue
+    ? vault.listKeys(
+        (key: any) =>
+          (key?.comment ?? '').toLowerCase().includes(searchValue.toLowerCase()) ||
+          key.id.toLowerCase().includes(searchValue.toLowerCase())
+      )
+    : vault.listKeys()
 
   return (
     <>
@@ -48,22 +68,49 @@ const EncryptionKeysManagement: FC<Props> = ({}) => {
         </div>
 
         {/* Table of secrets */}
-        {/* <div className="border border-scale-500 rounded">
-          {secrets.map((secret, idx) => {
+        <div className="border border-scale-500 rounded">
+          {keys.map((key, idx) => {
             return (
-              <Fragment key={secret.key_id}>
-                <SecretRow secret={secret} onSelectRemove={setSelectedSecretToRemove} />
-                {idx !== mockSecrets.length - 1 && <Divider light />}
+              <Fragment key={key.key_id}>
+                <div className="px-6 py-4 flex items-center space-x-4">
+                  <div className="space-y-1 min-w-[37%] max-w-[37%]">
+                    <p className="text-sm truncate" title={key.comment || 'Unnamed'}>
+                      {key.comment || 'Unnamed'}
+                    </p>
+                    <p
+                      title={key.id}
+                      className="text-sm text-scale-1000 font-mono font-bold truncate"
+                    >
+                      {key.id}
+                    </p>
+                  </div>
+                  <div className="flex items-center space-x-2 w-[38%]">
+                    {key.status === 'default' && <Badge color="green">Default</Badge>}
+                  </div>
+                  <div className="flex items-center justify-end w-[25%] space-x-4">
+                    <p className="text-sm text-scale-1100">
+                      Added on {dayjs(key.created).format('MMM D, YYYY')}
+                    </p>
+                    <Button
+                      type="default"
+                      className="py-2"
+                      icon={<IconTrash />}
+                      onClick={() => onSelectRemove(key)}
+                    />
+                  </div>
+                </div>
+                {idx !== keys.length - 1 && <Divider light />}
               </Fragment>
             )
           })}
-          {secrets.length === 0 && (
+          {keys.length === 0 && (
             <>
               {searchValue.length === 0 ? (
                 <div className="px-6 py-6 space-y-1 flex flex-col items-center justify-center">
-                  <p className="text-sm text-scale-1200">No secrets added yet</p>
+                  <p className="text-sm text-scale-1200">No encryption keys added yet</p>
                   <p className="text-sm text-scale-1100">
-                    The Vault allows you to store sensitive information like API keys
+                    Encryption keys are created by the pgsodium extension and can be used to encrypt
+                    your columns and secrets
                   </p>
                 </div>
               ) : (
@@ -76,7 +123,7 @@ const EncryptionKeysManagement: FC<Props> = ({}) => {
               )}
             </>
           )}
-        </div> */}
+        </div>
       </div>
 
       <Modal
diff --git a/studio/pages/project/[ref]/settings/vault.tsx b/studio/pages/project/[ref]/settings/vault.tsx
index e19651248cf..fbff4f1f8d0 100644
--- a/studio/pages/project/[ref]/settings/vault.tsx
+++ b/studio/pages/project/[ref]/settings/vault.tsx
@@ -21,7 +21,7 @@ const VaultSettings: NextPageWithLayout = () => {
       <VaultToggle />
 
       {/* Show encryption keys table if vault extension is enabled */}
-      <EncryptionKeysManagement />
+      <EncryptionKeysManagement onSelectRemove={() => {}} />
 
       {/* Show secrets table if vault extension is enabled */}
       <SecretsManagement />
diff --git a/studio/stores/project/VaultStore.ts b/studio/stores/project/VaultStore.ts
index 4da6e993cba..125caf45bef 100644
--- a/studio/stores/project/VaultStore.ts
+++ b/studio/stores/project/VaultStore.ts
@@ -9,10 +9,20 @@ export interface IVaultStore {
   error: any
 
   load: () => void
-  listKeys: () => any[]
-  listSecrets: () => any[]
+  listKeys: (filter?: any) => EncryptionKey[]
+  listSecrets: () => Secret[]
 }
 
+interface EncryptionKey {
+  id: string
+  key_id: number
+  comment: string
+  created: string
+  status: string
+}
+
+interface Secret {}
+
 export default class VaultStore implements IVaultStore {
   rootStore: IRootStore
 
@@ -23,7 +33,7 @@ export default class VaultStore implements IVaultStore {
     LOADED: 'loaded',
   }
 
-  data: any = { keys: [], secrets: [] }
+  data: { keys: EncryptionKey[]; secrets: Secret[] } = { keys: [], secrets: [] }
   state = this.STATES.INITIAL
   error = null
 
@@ -76,12 +86,22 @@ export default class VaultStore implements IVaultStore {
     }
   }
 
+  // Always return the default key as the first result
   listKeys(filter?: any) {
     const arr = this.data.keys.slice()
+
     if (!!filter) {
-      return arr.filter(filter).sort((a: any, b: any) => b.key_id - a.key_id)
+      const filteredResults = arr.filter(filter).sort((a: any, b: any) => a.key_id - b.key_id)
+      const defaultKey = filteredResults.find((key) => key.status === 'default') as EncryptionKey
+
+      if (defaultKey === undefined) return filteredResults
+      else return [defaultKey].concat(filteredResults.filter((key) => key.status !== 'default'))
     } else {
-      return arr.sort((a: any, b: any) => b.key_id - a.key_id)
+      const results = arr.sort((a: any, b: any) => a.key_id - b.key_id)
+      const defaultKey = results.find((key) => key.status === 'default') as EncryptionKey
+
+      if (defaultKey === undefined) return results
+      return [defaultKey].concat(results.filter((key) => key.status !== 'default'))
     }
   }