From 4e36b2d902c76f914bbf3f13a4e14fe18f2548e9 Mon Sep 17 00:00:00 2001
From: Ren Amamiya <123083837+reyamir@users.noreply.github.com>
Date: Mon, 10 Jul 2023 13:39:53 +0700
Subject: [PATCH] add migrate page

---
 src/app.tsx                    |   2 +
 src/app/auth/create/step-2.tsx |   2 +-
 src/app/auth/import/step-2.tsx |   2 +-
 src/app/auth/migrate.tsx       | 208 +++++++++++++++++++++++++++++++++
 src/libs/storage.tsx           |   9 ++
 src/shared/protected.tsx       |   4 +
 6 files changed, 225 insertions(+), 2 deletions(-)
 create mode 100644 src/app/auth/migrate.tsx

diff --git a/src/app.tsx b/src/app.tsx
index e644b4b8..509838f4 100644
--- a/src/app.tsx
+++ b/src/app.tsx
@@ -10,6 +10,7 @@ import { AuthImportScreen } from '@app/auth/import';
 import { ImportStep1Screen } from '@app/auth/import/step-1';
 import { ImportStep2Screen } from '@app/auth/import/step-2';
 import { ImportStep3Screen } from '@app/auth/import/step-3';
+import { MigrateScreen } from '@app/auth/migrate';
 import { OnboardingScreen } from '@app/auth/onboarding';
 import { UnlockScreen } from '@app/auth/unlock';
 import { WelcomeScreen } from '@app/auth/welcome';
@@ -69,6 +70,7 @@ const router = createBrowserRouter([
         ],
       },
       { path: 'unlock', element: <UnlockScreen /> },
+      { path: 'migrate', element: <MigrateScreen /> },
     ],
   },
   {
diff --git a/src/app/auth/create/step-2.tsx b/src/app/auth/create/step-2.tsx
index 57abe0f5..7ecf2fb4 100644
--- a/src/app/auth/create/step-2.tsx
+++ b/src/app/auth/create/step-2.tsx
@@ -124,7 +124,7 @@ export function CreateStep2Screen() {
             <button
               type="submit"
               disabled={!isDirty || !isValid}
-              className="inline-flex h-11 w-full items-center justify-center rounded-md bg-fuchsia-500 font-medium text-zinc-100 hover:bg-fuchsia-600"
+              className="inline-flex h-11 w-full items-center justify-center rounded-md bg-fuchsia-500 font-medium text-zinc-100 hover:bg-fuchsia-600 disabled:pointer-events-none disabled:opacity-50"
             >
               {loading ? (
                 <LoaderIcon className="h-4 w-4 animate-spin text-black dark:text-zinc-100" />
diff --git a/src/app/auth/import/step-2.tsx b/src/app/auth/import/step-2.tsx
index 9adc20e4..f3ae5127 100644
--- a/src/app/auth/import/step-2.tsx
+++ b/src/app/auth/import/step-2.tsx
@@ -124,7 +124,7 @@ export function ImportStep2Screen() {
             <button
               type="submit"
               disabled={!isDirty || !isValid}
-              className="inline-flex h-11 w-full items-center justify-center rounded-md bg-fuchsia-500 font-medium text-zinc-100 hover:bg-fuchsia-600"
+              className="inline-flex h-11 w-full items-center justify-center rounded-md bg-fuchsia-500 font-medium text-zinc-100 hover:bg-fuchsia-600 disabled:pointer-events-none disabled:opacity-50"
             >
               {loading ? (
                 <LoaderIcon className="h-4 w-4 animate-spin text-black dark:text-zinc-100" />
diff --git a/src/app/auth/migrate.tsx b/src/app/auth/migrate.tsx
new file mode 100644
index 00000000..84a86549
--- /dev/null
+++ b/src/app/auth/migrate.tsx
@@ -0,0 +1,208 @@
+import { useState } from 'react';
+import { Resolver, useForm } from 'react-hook-form';
+import { useNavigate } from 'react-router-dom';
+
+import { removePrivkey } from '@libs/storage';
+
+import { CheckCircleIcon, EyeOffIcon, EyeOnIcon, LoaderIcon } from '@shared/icons';
+
+import { useStronghold } from '@stores/stronghold';
+
+import { useAccount } from '@utils/hooks/useAccount';
+import { useSecureStorage } from '@utils/hooks/useSecureStorage';
+
+type FormValues = {
+  password: string;
+};
+
+const resolver: Resolver<FormValues> = async (values) => {
+  return {
+    values: values.password ? values : {},
+    errors: !values.password
+      ? {
+          password: {
+            type: 'required',
+            message: 'This is required.',
+          },
+        }
+      : {},
+  };
+};
+
+export function MigrateScreen() {
+  const navigate = useNavigate();
+
+  const [passwordInput, setPasswordInput] = useState('password');
+  const [passwordStep, setPasswordStep] = useState({ loading: false, done: false });
+  const [privkeyStep, setPrivkeyStep] = useState({ loading: false, done: false });
+
+  const { account } = useAccount();
+  const { save } = useSecureStorage();
+
+  const setPassword = useStronghold((state) => state.setPassword);
+
+  // toggle private key
+  const showPassword = () => {
+    if (passwordInput === 'password') {
+      setPasswordInput('text');
+    } else {
+      setPasswordInput('password');
+    }
+  };
+
+  const clearPrivkey = async () => {
+    setPrivkeyStep((prev) => ({ ...prev, loading: true }));
+    const res = await removePrivkey();
+    if (res) {
+      setPrivkeyStep({ done: true, loading: false });
+    } else {
+      setPrivkeyStep((prev) => ({ ...prev, loading: false }));
+    }
+  };
+
+  const {
+    register,
+    setError,
+    handleSubmit,
+    formState: { errors, isDirty, isValid },
+  } = useForm<FormValues>({ resolver });
+
+  const onSubmit = async (data: { [x: string]: string }) => {
+    setPasswordStep((prev) => ({ ...prev, loading: true }));
+    if (data.password.length > 3) {
+      // add password to local state
+      setPassword(data.password);
+
+      // load private in secure storage
+      try {
+        await save(account.pubkey, account.privkey, data.password);
+        // redirect to home
+        navigate('/', { replace: true });
+      } catch {
+        setPasswordStep((prev) => ({ ...prev, loading: false }));
+        setError('password', {
+          type: 'custom',
+          message: 'Wrong password',
+        });
+      }
+    } else {
+      setPasswordStep((prev) => ({ ...prev, loading: false }));
+      setError('password', {
+        type: 'custom',
+        message: 'Password is required and must be greater than 3',
+      });
+    }
+  };
+
+  return (
+    <div className="flex h-full w-full items-center justify-center">
+      <div className="mx-auto w-full max-w-md">
+        <div className="mb-8 text-center">
+          <h1 className="text-xl font-semibold text-zinc-100">
+            Upgrade security for your account
+          </h1>
+        </div>
+        <div className="w-full rounded-xl border-t border-zinc-800/50 bg-zinc-900 px-3 py-3">
+          <div className="flex flex-col gap-4">
+            <div>
+              <h3 className="font-medium text-zinc-200">
+                Remove plaintext privkey store in local database
+              </h3>
+              <div className="mt-1">
+                <p className="text-sm text-zinc-400">
+                  You&apos;re using old Lume version which store your private key as
+                  plaintext in database, this is huge security risk.
+                </p>
+                <p className="mt-2 text-sm text-zinc-400">To secure your private key</p>
+                <ul className="mt-2 list-outside list-disc pl-5 text-sm text-zinc-400">
+                  <li>Firstly, click button below to remove it from local database</li>
+                  <li>
+                    Then set password and Lume will put your private key in secure storage
+                  </li>
+                </ul>
+              </div>
+              <div className="mt-3">
+                <div className="relative">
+                  <input
+                    readOnly
+                    value={privkeyStep.done ? 'Nothing to see here' : account.privkey}
+                    className="relative w-full rounded-lg bg-zinc-800 px-3 py-3 text-zinc-100 !outline-none placeholder:text-zinc-500"
+                  />
+                  <div className="absolute left-0 top-0 flex h-full w-full items-center justify-center rounded-lg bg-black/10 backdrop-blur-sm">
+                    {privkeyStep.done ? (
+                      privkeyStep.loading ? (
+                        <LoaderIcon className="h-4 w-4 animate-spin text-zinc-100" />
+                      ) : (
+                        <CheckCircleIcon className="h-5 w-5 text-green-500" />
+                      )
+                    ) : (
+                      <button
+                        type="button"
+                        onClick={() => clearPrivkey()}
+                        className="inline-flex w-max items-center justify-center rounded bg-fuchsia-500 px-2.5 py-1.5 text-sm hover:bg-fuchsia-600"
+                      >
+                        Click to remove
+                      </button>
+                    )}
+                  </div>
+                </div>
+              </div>
+            </div>
+            <form onSubmit={handleSubmit(onSubmit)} className="mb-0">
+              <div className="flex flex-col gap-1">
+                <span className="font-medium text-zinc-200">
+                  Set password to protect your key
+                </span>
+                <div className="relative">
+                  <input
+                    {...register('password', { required: true })}
+                    type={passwordInput}
+                    placeholder="min. 4 characters"
+                    className="relative w-full rounded-lg bg-zinc-800 py-3 pl-3.5 pr-11 text-zinc-100 !outline-none placeholder:text-zinc-400"
+                  />
+                  <button
+                    type="button"
+                    onClick={() => showPassword()}
+                    className="group absolute right-2 top-1/2 -translate-y-1/2 transform rounded p-1 hover:bg-zinc-700"
+                  >
+                    {passwordInput === 'password' ? (
+                      <EyeOffIcon
+                        width={20}
+                        height={20}
+                        className="text-zinc-500 group-hover:text-zinc-100"
+                      />
+                    ) : (
+                      <EyeOnIcon
+                        width={20}
+                        height={20}
+                        className="text-zinc-500 group-hover:text-zinc-100"
+                      />
+                    )}
+                  </button>
+                </div>
+                <span className="text-sm text-red-400">
+                  {errors.password && <p>{errors.password.message}</p>}
+                </span>
+              </div>
+              <div className="flex items-center justify-center">
+                {privkeyStep.done && (
+                  <button
+                    type="submit"
+                    disabled={!isDirty || !isValid}
+                    className="mt-3 inline-flex h-11 w-full items-center justify-center rounded-md bg-fuchsia-500 font-medium text-zinc-100 hover:bg-fuchsia-600 disabled:pointer-events-none disabled:opacity-50"
+                  >
+                    {passwordStep.loading ? (
+                      <LoaderIcon className="h-4 w-4 animate-spin text-black dark:text-zinc-100" />
+                    ) : (
+                      'Continue →'
+                    )}
+                  </button>
+                )}
+              </div>
+            </form>
+          </div>
+        </div>
+      </div>
+    </div>
+  );
+}
diff --git a/src/libs/storage.tsx b/src/libs/storage.tsx
index cb4284a7..00e69c1c 100644
--- a/src/libs/storage.tsx
+++ b/src/libs/storage.tsx
@@ -447,3 +447,12 @@ export async function getUserMetadata(pubkey: string) {
     return null;
   }
 }
+
+// delete privkey
+export async function removePrivkey() {
+  const db = await connect();
+  const activeAccount = await getActiveAccount();
+  return await db.execute(
+    `UPDATE accounts SET privkey = "privkey is stored in secure storage" WHERE id = "${activeAccount.id}";`
+  );
+}
diff --git a/src/shared/protected.tsx b/src/shared/protected.tsx
index c538a9ad..9f9f7576 100644
--- a/src/shared/protected.tsx
+++ b/src/shared/protected.tsx
@@ -13,6 +13,10 @@ export function Protected({ children }: { children: ReactNode }) {
     return <Navigate to="/auth/welcome" replace />;
   }
 
+  if (status === 'success' && account && account.privkey.length > 35) {
+    return <Navigate to="/auth/migrate" replace />;
+  }
+
   if (status === 'success' && account && !password) {
     return <Navigate to="/auth/unlock" replace />;
   }