Finished secret snapshots

frontend/pages/dashboard/[id].tsx

@@ -6,8 +6,9 @@ import { useTranslation } from "next-i18next";
 import {
   faArrowDownAZ,
   faArrowDownZA,
+  faArrowLeft,
   faCheck,
-  faCopy,
+  faClockRotateLeft,
   faDownload,
   faEye,
   faEyeSlash,
@@ -16,6 +17,8 @@ import {
   faPlus,
 } from '@fortawesome/free-solid-svg-icons';
 import { FontAwesomeIcon } from '@fortawesome/react-fontawesome';
+import getProjectSercetSnapshotsCount from 'ee/api/secrets/GetProjectSercetSnapshotsCount';
+import PITRecoverySidebar from 'ee/components/PITRecoverySidebar';
 
 import Button from '~/components/basic/buttons/Button';
 import ListBox from '~/components/basic/Listbox';
@@ -30,12 +33,13 @@ import pushKeys from '~/components/utilities/secrets/pushKeys';
 import { getTranslatedServerSideProps } from '~/components/utilities/withTranslateProps';
 import guidGenerator from '~/utilities/randomId';
 
-import { envMapping } from '../../public/data/frequentConstants';
+import { envMapping, reverseEnvMapping } from '../../public/data/frequentConstants';
 import getUser from '../api/user/getUser';
 import checkUserAction from '../api/userActions/checkUserAction';
 import registerUserAction from '../api/userActions/registerUserAction';
 import getWorkspaces from '../api/workspace/getWorkspaces';
 
+const queryString = require("query-string");
 
 interface SecretDataProps {
   type: 'personal' | 'shared';
@@ -46,6 +50,19 @@ interface SecretDataProps {
   comment: string;
 }
 
+interface SnapshotProps {
+  id: string;
+  createdAt: string;
+  secretVersions: {
+    id: string;
+    pos: number;
+    type: "personal" | "shared";
+    environment: string;
+    key: string;
+    value: string;
+  }[];
+}
+
 /**
  * this function finds the teh duplicates in an array
  * @param arr - array of anything (e.g., with secret keys and types (personal/shared))
@@ -76,21 +93,20 @@ export default function Dashboard() {
   const [workspaceId, setWorkspaceId] = useState('');
   const [blurred, setBlurred] = useState(true);
   const [isKeyAvailable, setIsKeyAvailable] = useState(true);
-  const [env, setEnv] = useState(
-    router.asPath.split('?').length == 1
-      ? 'Development'
-      : Object.keys(envMapping).includes(router.asPath.split('?')[1])
-      ? router.asPath.split('?')[1]
-      : 'Development'
-  );
+  const [env, setEnv] = useState('Development');
+  const [snapshotEnv, setSnapshotEnv] = useState('Development');
   const [isNew, setIsNew] = useState(false);
+  const [isLoading, setIsLoading] = useState(false);
   const [searchKeys, setSearchKeys] = useState('');
   const [errorDragAndDrop, setErrorDragAndDrop] = useState(false);
   const [sortMethod, setSortMethod] = useState('alphabetical');
   const [checkDocsPopUpVisible, setCheckDocsPopUpVisible] = useState(false);
   const [hasUserEverPushed, setHasUserEverPushed] = useState(false);
   const [sidebarSecretId, toggleSidebar] = useState("None");
+  const [PITSidebarOpen, togglePITSidebar] = useState(false);
   const [sharedToHide, setSharedToHide] = useState<string[]>([]);
+  const [snapshotData, setSnapshotData] = useState<SnapshotProps>();
+  const [numSnapshots, setNumSnapshots] = useState<number>();
 
   const { t } = useTranslation();
   const { createNotification } = useNotificationContext();
@@ -141,17 +157,39 @@ export default function Dashboard() {
   useEffect(() => {
     (async () => {
       try {
+        console.log(1, 'reloaded')
+        const tempNumSnapshots = await getProjectSercetSnapshotsCount({ workspaceId: String(router.query.id) })
+        setNumSnapshots(tempNumSnapshots);
         const userWorkspaces = await getWorkspaces();
         const listWorkspaces = userWorkspaces.map((workspace) => workspace._id);
         if (
-          !listWorkspaces.includes(router.asPath.split('/')[2].split('?')[0])
+          !listWorkspaces.includes(router.asPath.split('/')[2])
         ) {
           router.push('/dashboard/' + listWorkspaces[0]);
         }
 
-        if (env != router.asPath.split('?')[1]) {
-          router.push(router.asPath.split('?')[0] + '?' + env);
-        }
+        const user = await getUser();
+        setIsNew(
+          (Date.parse(String(new Date())) - Date.parse(user.createdAt)) / 60000 < 3
+            ? true
+            : false
+        );
+
+        const userAction = await checkUserAction({
+          action: 'first_time_secrets_pushed'
+        });
+        setHasUserEverPushed(userAction ? true : false);
+      } catch (error) {
+        console.log('Error', error);
+        setData(undefined);
+      }
+    })();
+  }, []);
+
+  useEffect(() => {
+    (async () => {
+      try {
+        setIsLoading(true);
         setBlurred(true);
         setWorkspaceId(String(router.query.id));
 
@@ -173,18 +211,7 @@ export default function Dashboard() {
               dataToSort?.map((item) => item.key).indexOf(item)
           ).includes(row.key) && row.type == 'shared'))?.map((item) => item.id)
         )
-
-        const user = await getUser();
-        setIsNew(
-          (Date.parse(String(new Date())) - Date.parse(user.createdAt)) / 60000 < 3
-            ? true
-            : false
-        );
-
-        const userAction = await checkUserAction({
-          action: 'first_time_secrets_pushed'
-        });
-        setHasUserEverPushed(userAction ? true : false);
+        setIsLoading(false);
       } catch (error) {
         console.log('Error', error);
         setData(undefined);
@@ -321,12 +348,21 @@ export default function Dashboard() {
   /**
    * Save the changes of environment variables and push them to the database
    */
-  const savePush = async () => {
-    // Format the new object with environment variables
-    const obj = Object.assign(
-      {},
-      ...data!.map((row: SecretDataProps) => ({ [row.type.charAt(0) + row.key]: [row.value, row.comment] }))
-    );
+  const savePush = async (dataToPush?: any[], envToPush?: string) => {
+    let obj;
+    // dataToPush is mostly used for rollbacks, otherwise we always take the current state data
+    if ((dataToPush ?? [])?.length > 0) {
+      obj = Object.assign(
+        {},
+        ...dataToPush!.map((row: SecretDataProps) => ({ [row.type.charAt(0) + row.key]: [row.value, row.comment ?? ''] }))
+      );
+    } else {
+      // Format the new object with environment variables
+      obj = Object.assign(
+        {},
+        ...data!.map((row: SecretDataProps) => ({ [row.type.charAt(0) + row.key]: [row.value, row.comment ?? ''] }))
+      );
+    }
 
     // Checking if any of the secret keys start with a number - if so, don't do anything
     const nameErrors = !Object.keys(obj)
@@ -350,13 +386,17 @@ export default function Dashboard() {
 
     // Once "Save changed is clicked", disable that button
     setButtonReady(false);
-    pushKeys({ obj, workspaceId: String(router.query.id), env });
+    console.log(envToPush ? envToPush : env, env, envToPush)
+    pushKeys({ obj, workspaceId: String(router.query.id), env: envToPush ? envToPush : env });
 
     // If this user has never saved environment variables before, show them a prompt to read docs
     if (!hasUserEverPushed) {
       setCheckDocsPopUpVisible(true);
       await registerUserAction({ action: 'first_time_secrets_pushed' });
     }
+
+    // increasing the number of project commits
+    setNumSnapshots(numSnapshots ?? 0 + 1);
   };
 
   const addData = (newData: SecretDataProps[]) => {
@@ -427,6 +467,11 @@ export default function Dashboard() {
           setSharedToHide={setSharedToHide}
           deleteRow={deleteCertainRow}
         />}
+        {PITSidebarOpen && <PITRecoverySidebar 
+          toggleSidebar={togglePITSidebar} 
+          chosenSnapshot={String(snapshotData?.id ? snapshotData.id : "")}
+          setSnapshotData={setSnapshotData}
+        />}
         <div className="w-full max-h-96 pb-2">
           <NavHeader pageName={t("dashboard:title")} isProjectRelated={true} />
           {checkDocsPopUpVisible && (
@@ -441,9 +486,22 @@ export default function Dashboard() {
             />
           )}
           <div className="flex flex-row justify-between items-center mx-6 mt-6 mb-3 text-xl max-w-5xl">
+            {snapshotData && 
+            <div className={`flex justify-start max-w-sm mt-1 mr-2`}>
+              <Button
+                text={String(t("Go back to current"))}
+                onButtonPressed={() => setSnapshotData(undefined)}
+                color="mineshaft"
+                size="md"
+                icon={faArrowLeft}
+              />
+            </div>}
             <div className="flex flex-row justify-start items-center text-3xl">
-              <p className="font-semibold mr-4 mt-1">{t("dashboard:title")}</p>
-              {data?.length == 0 && (
+              <div className="font-semibold mr-4 mt-1 flex flex-row items-center">
+                <p>{snapshotData ? "Secret Snapshot" : t("dashboard:title")}</p>
+                {snapshotData && <span className='bg-primary-800 text-sm ml-4 mt-1 px-1.5 rounded-md'>{new Date(snapshotData.createdAt).toLocaleString()}</span>}
+              </div>
+              {!snapshotData && data?.length == 0 && (
                 <ListBox
                   selected={env}
                   data={['Development', 'Staging', 'Production', 'Testing']}
@@ -452,8 +510,17 @@ export default function Dashboard() {
               )}
             </div>
             <div className="flex flex-row">
-              {(data?.length !== 0 || buttonReady) && (
-                <div className={`flex justify-start max-w-sm mt-2`}>
+              <div className={`flex justify-start max-w-sm mt-1 mr-2`}>
+                <Button
+                  text={String(numSnapshots + " " + t("Commits"))}
+                  onButtonPressed={() => togglePITSidebar(true)}
+                  color="mineshaft"
+                  size="md"
+                  icon={faClockRotateLeft}
+                />
+              </div>
+              {(data?.length !== 0 || buttonReady) && !snapshotData && (
+                <div className={`flex justify-start max-w-sm mt-1`}>
                   <Button
                     text={String(t("common:save-changes"))}
                     onButtonPressed={savePush}
@@ -465,18 +532,64 @@ export default function Dashboard() {
                   />
                 </div>
               )}
+              {snapshotData && <div className={`flex justify-start max-w-sm mt-1`}>
+                <Button
+                  text={String(t("Rollback to this snapshot"))}
+                  onButtonPressed={async () => {
+                    const envsToRollback = snapshotData.secretVersions.map(sv => sv.environment).filter((v, i, a) => a.indexOf(v) === i);
+
+                    // Update secrets in the state only for the current environment
+                    setData(
+                      snapshotData.secretVersions
+                      .filter(row => reverseEnvMapping[row.environment] == env)
+                      .map((sv, position) => { 
+                        return {
+                          id: sv.id, pos: position, type: sv.type, key: sv.key, value: sv.value, comment: ''
+                        }
+                      })
+                    );
+
+                    // Rollback each of the environments in the snapshot
+                    // #TODO: clean up other environments
+                    envsToRollback.map(async (envToRollback) => {
+                      await savePush(
+                        snapshotData.secretVersions
+                        .filter(row => row.environment == envToRollback)
+                        .map((sv, position) => { 
+                          return {id: sv.id, pos: position, type: sv.type, key: sv.key, value: sv.value, comment: ''}
+                        }),
+                        reverseEnvMapping[envToRollback]
+                      );
+                    });
+                    setSnapshotData(undefined);
+                    createNotification({
+                      text: `Rollback has been performed successfully.`,
+                      type: 'success'
+                    });
+                  }}
+                  color="primary"
+                  size="md"
+                  active={buttonReady}
+                />
+              </div>}
             </div>
           </div>
           <div className="mx-6 w-full pr-12">
             <div className="flex flex-col max-w-5xl pb-1">
               <div className="w-full flex flex-row items-start">
-                {data?.length !== 0 && (
+                {(!snapshotData || data?.length !== 0) && (
                   <>
-                    <ListBox
+                    {!snapshotData 
+                    ? <ListBox
                       selected={env}
                       data={['Development', 'Staging', 'Production', 'Testing']}
                       onChange={setEnv}
                     />
+                    : <ListBox
+                      selected={snapshotEnv}
+                      data={['Development', 'Staging', 'Production', 'Testing']}
+                      onChange={setSnapshotEnv}
+                    />}
                     <div className="h-10 w-full bg-white/5 hover:bg-white/10 ml-2 flex items-center rounded-md flex flex-row items-center">
                       <FontAwesomeIcon
                         className="bg-white/5 rounded-l-md py-3 pl-4 pr-2 text-gray-400"
@@ -489,7 +602,7 @@ export default function Dashboard() {
                         placeholder={String(t("dashboard:search-keys"))}
                       />
                     </div>
-                    <div className="ml-2 min-w-max flex flex-row items-start justify-start">
+                    {!snapshotData && <div className="ml-2 min-w-max flex flex-row items-start justify-start">
                       <Button
                         onButtonPressed={() => reorderRows(1)}
                         color="mineshaft"
@@ -500,15 +613,15 @@ export default function Dashboard() {
                             : faArrowDownZA
                         }
                       />
-                    </div>
-                    <div className="ml-2 min-w-max flex flex-row items-start justify-start">
+                    </div>}
+                    {!snapshotData && <div className="ml-2 min-w-max flex flex-row items-start justify-start">
                       <Button
                         onButtonPressed={download}
                         color="mineshaft"
                         size="icon-md"
                         icon={faDownload}
                       />
-                    </div>
+                    </div>}
                     <div className="ml-2 min-w-max flex flex-row items-start justify-start">
                       <Button
                         onButtonPressed={changeBlurred}
@@ -517,7 +630,7 @@ export default function Dashboard() {
                         icon={blurred ? faEye : faEyeSlash}
                       />
                     </div>
-                    <div className="relative ml-2 min-w-max flex flex-row items-start justify-end">
+                    {!snapshotData && <div className="relative ml-2 min-w-max flex flex-row items-start justify-end">
                       <Button
                         text={String(t("dashboard:add-key"))}
                         onButtonPressed={addRow}
@@ -531,18 +644,29 @@ export default function Dashboard() {
                           <span className="relative inline-flex rounded-full h-3 w-3 bg-primary"></span>
                         </span>
                       )}
-                    </div>
+                    </div>}
                   </>
                 )}
               </div>
             </div>
-            {data?.length !== 0 ? (
+            {isLoading ? (
+            <div className="flex items-center justify-center h-full my-48">
+              <Image
+                src="/images/loading/loading.gif"
+                height={60}
+                width={100}
+                alt="infisical loading indicator"
+              ></Image>
+            </div> 
+            ) : (
+            data?.length !== 0 ? (
               <div className="flex flex-col w-full mt-1 mb-2">
                 <div
                   className={`max-w-5xl mt-1 max-h-[calc(100vh-280px)] overflow-hidden overflow-y-scroll no-scrollbar no-scrollbar::-webkit-scrollbar`}
                 >
                   <div className="px-1 pt-2 bg-mineshaft-800 rounded-md p-2">
-                    {data?.filter(row => !(sharedToHide.includes(row.id) && row.type == 'shared')).map((keyPair) => (
+                    {!snapshotData && data?.filter(row => row.key.toUpperCase().includes(searchKeys.toUpperCase()))
+                    .filter(row => !(sharedToHide.includes(row.id) && row.type == 'shared')).map((keyPair) => (
                       <KeyPair 
                         key={keyPair.id}
                         keyPair={keyPair}
@@ -552,10 +676,33 @@ export default function Dashboard() {
                         isDuplicate={findDuplicates(data?.map((item) => item.key + item.type))?.includes(keyPair.key + keyPair.type)}
                         toggleSidebar={toggleSidebar}
                         sidebarSecretId={sidebarSecretId}
+                        isSnapshot={false}
+                      />
+                    ))}
+                    {snapshotData && snapshotData.secretVersions?.sort((a, b) => a.key.localeCompare(b.key))
+                    .filter(row => reverseEnvMapping[row.environment] == snapshotEnv)
+                    .filter(row => row.key.toUpperCase().includes(searchKeys.toUpperCase()))
+                    .filter(row => !(snapshotData.secretVersions?.filter(row => (snapshotData.secretVersions
+                      ?.map((item) => item.key)
+                      .filter(
+                        (item, index) =>
+                          index !==
+                          snapshotData.secretVersions?.map((item) => item.key).indexOf(item)
+                      ).includes(row.key) && row.type == 'shared'))?.map((item) => item.id).includes(row.id) && row.type == 'shared')).map((keyPair) => (
+                      <KeyPair 
+                        key={keyPair.id}
+                        keyPair={keyPair}
+                        modifyValue={listenChangeValue}
+                        modifyKey={listenChangeKey}
+                        isBlurred={blurred}
+                        isDuplicate={findDuplicates(data?.map((item) => item.key + item.type))?.includes(keyPair.key + keyPair.type)}
+                        toggleSidebar={toggleSidebar}
+                        sidebarSecretId={sidebarSecretId}
+                        isSnapshot={true}
                       />
                     ))}
                   </div>
-                  <div className="w-full max-w-5xl px-2 pt-3">
+                  {!snapshotData && <div className="w-full max-w-5xl px-2 pt-3">
                     <DropZone
                       setData={addData}
                       setErrorDragAndDrop={setErrorDragAndDrop}
@@ -565,12 +712,12 @@ export default function Dashboard() {
                       keysExist={true}
                       numCurrentRows={data.length}
                     />
-                  </div>
+                  </div>}
                 </div>
               </div>
             ) : (
               <div className="flex flex-col items-center justify-center h-full text-xl text-gray-400 max-w-5xl mt-28">
-                {isKeyAvailable && (
+                {isKeyAvailable && !snapshotData && (
                   <DropZone
                     setData={setData}
                     setErrorDragAndDrop={setErrorDragAndDrop}
@@ -599,7 +746,7 @@ export default function Dashboard() {
                     </>
                   ))}
               </div>
-            )}
+            ))}
           </div>
         </div>
       </div>