From a627addad57c7a7af685863aac15e72812849fc7 Mon Sep 17 00:00:00 2001
From: strNophix <nvdpoel01@gmail.com>
Date: Sun, 16 Oct 2022 21:02:41 +0200
Subject: [PATCH] Added login flow

---
 .docker/kratos/kratos.yml              |  4 --
 client/components/login/LoginForm.tsx  | 76 ++++++++++++++++++--------
 client/components/login/SignupForm.tsx | 16 +-----
 client/components/nav/NavBar.tsx       | 67 +++++++++++++++--------
 client/config/index.ts                 |  2 -
 client/config/validation.ts            | 18 ++++++
 client/hooks/useLogInFlow.ts           | 53 ++++++++++++++++++
 client/hooks/useLogout.ts              | 38 +++++++++++++
 client/hooks/useSession.ts             | 31 +++++++++++
 client/hooks/useSignUpFlow.ts          | 29 +++++-----
 client/package.json                    |  3 +-
 client/pages/_app.tsx                  | 11 +++-
 client/pages/index.tsx                 |  9 +++
 client/pnpm-lock.yaml                  | 26 +++++++++
 14 files changed, 303 insertions(+), 80 deletions(-)
 create mode 100644 client/config/validation.ts
 create mode 100644 client/hooks/useLogInFlow.ts
 create mode 100644 client/hooks/useLogout.ts
 create mode 100644 client/hooks/useSession.ts
 create mode 100644 client/pages/index.tsx

diff --git a/.docker/kratos/kratos.yml b/.docker/kratos/kratos.yml
index 5bb6918..4bdcaba 100644
--- a/.docker/kratos/kratos.yml
+++ b/.docker/kratos/kratos.yml
@@ -47,10 +47,6 @@ selfservice:
     registration:
       lifespan: 10m
       ui_url: http://127.0.0.1:3000/signup
-      after:
-        password:
-          hooks:
-            - hook: session
 
 log:
   level: debug
diff --git a/client/components/login/LoginForm.tsx b/client/components/login/LoginForm.tsx
index 1236cbf..ecfe0ff 100644
--- a/client/components/login/LoginForm.tsx
+++ b/client/components/login/LoginForm.tsx
@@ -5,34 +5,66 @@ import FormField from "../common/form/FormField"
 import InlineLink from "../common/InlineLink"
 import SubmitButton from "../common/form/SubmitButton"
 
-interface LoginFormValues {
-  username: string
-  password: string
-}
+import * as validation from "../../config/validation"
+import { z } from "zod"
+import { zodResolver } from "@hookform/resolvers/zod"
+import useLogInFlow from "../../hooks/useLogInFlow"
+import Input from "../common/Input"
+
+const LogInFormSchema = z.object({
+  csrfToken: z.string(),
+  password: validation.password,
+  email: z.string().email({ message: "Not a valid email address" }).trim(),
+})
+
+type LogInFormValues = z.infer<typeof LogInFormSchema>
+
+const formFields = [
+  { id: "email", label: "Email", type: "email" },
+  { id: "password", label: "Password", type: "password" },
+]
 
 const LoginForm: FC = () => {
-  const { register, handleSubmit } = useForm<LoginFormValues>()
-  const onSubmit: SubmitHandler<LoginFormValues> = (data) => console.log(data)
+  const logInFlow = useLogInFlow()
+  const { register, handleSubmit, formState } = useForm<LogInFormValues>({
+    resolver: zodResolver(LogInFormSchema),
+  })
+  const onSubmit: SubmitHandler<LogInFormValues> = (data) =>
+    logInFlow.submitData({
+      csrf_token: data.csrfToken,
+      method: "password",
+      identifier: data.email,
+      password: data.password,
+    })
 
   return (
     <form onSubmit={handleSubmit(onSubmit)} className="space-y-4">
-      <FormField
-        label="Username"
-        className="py-2 px-2 outline-2 w-full"
-        {...register("username")}
-      />
-      <FormField
-        label="Password"
-        type="password"
-        {...register("password")}
-        className="py-2 px-2 outline-2 w-full"
-        bottomElement={
-          <InlineLink to="#" className="block mt-2">
-            Trouble logging in?
-          </InlineLink>
-        }
-      />
+      {logInFlow.flow && (
+        <Input
+          hidden={true}
+          value={logInFlow.flow.ui.nodes[0].attributes.value}
+          {...register("csrfToken")}
+        />
+      )}
+      {formFields.map((field) => (
+        <FormField
+          key={field.id}
+          id={field.id}
+          type={field.type}
+          label={field.label}
+          {...register(field.id as any)}
+          className="py-2 px-2 outline-2 w-full"
+          bottomElement={
+            <p className="text-xs">
+              {formState.errors[(field.id as any) || ""]?.message}
+            </p>
+          }
+        />
+      ))}
       <SubmitButton className="w-full" value="Log In" />
+      <div className="text-center">
+        <InlineLink to="#">Trouble logging in?</InlineLink>
+      </div>
     </form>
   )
 }
diff --git a/client/components/login/SignupForm.tsx b/client/components/login/SignupForm.tsx
index d4ccc51..a0e7ea2 100644
--- a/client/components/login/SignupForm.tsx
+++ b/client/components/login/SignupForm.tsx
@@ -7,24 +7,14 @@ import FormField from "../common/form/FormField"
 import InlineLink from "../common/InlineLink"
 import Input from "../common/Input"
 import SubmitButton from "../common/form/SubmitButton"
-import { PASSWORD_REGEX } from "../../config"
 import useSignUpFlow from "../../hooks/useSignUpFlow"
+import * as validation from "../../config/validation"
 
 const SignupFormSchema = z
   .object({
     csrfToken: z.string(),
-    username: z
-      .string()
-      .trim()
-      .min(1, { message: "Username must be at least 1 character long." })
-      .max(16, { message: "Username can't be longer than 16 characters.." }),
-    password: z
-      .string()
-      .trim()
-      .regex(
-        PASSWORD_REGEX,
-        "Password must be 8-64 long and must contain a number, uppercase, lowercase and special character.",
-      ),
+    username: validation.username,
+    password: validation.password,
     passwordRepeat: z.string().trim(),
     email: z.string().email({ message: "Not a valid email address" }).trim(),
   })
diff --git a/client/components/nav/NavBar.tsx b/client/components/nav/NavBar.tsx
index dc3744e..8ea5ead 100644
--- a/client/components/nav/NavBar.tsx
+++ b/client/components/nav/NavBar.tsx
@@ -1,11 +1,19 @@
-import { UserIcon } from "@heroicons/react/24/outline"
+import {
+  ArrowRightIcon,
+  ArrowRightOnRectangleIcon,
+  UserIcon,
+} from "@heroicons/react/24/outline"
 import { FC, useState } from "react"
+import { useLogout } from "../../hooks/useLogout"
+import useSession from "../../hooks/useSession"
 
 import Button from "../common/Button"
 import Logo from "../common/Logo"
 import LoginModal, { LoginModelProps } from "../login/LoginModal"
 
 const NavBar: FC = () => {
+  const logout = useLogout()
+  const session = useSession((state) => state.session)
   const [modalProps, setModalProps] = useState<LoginModelProps>({
     isOpen: false,
     defaultPage: 0,
@@ -23,6 +31,7 @@ const NavBar: FC = () => {
       isOpen: true,
     })
 
+  console.log({ session })
   return (
     <nav className="bg-zinc-800 w-screen font-semibold border-b border-b-black">
       <div className="flex flex-row justify-between items-center h-12 mx-2">
@@ -35,27 +44,41 @@ const NavBar: FC = () => {
         </div>
         <div>
           <ul className="justify-end flex flex-row space-x-3 items-center">
-            <li>
-              <Button
-                className="text-sm px-3 py-2 bg-neutral-700"
-                onClick={showLoginTab}
-              >
-                Log In
-              </Button>
-            </li>
-            <li>
-              <Button
-                className="text-sm px-3 py-2 bg-violet-500"
-                onClick={showSignupTab}
-              >
-                Sign Up
-              </Button>
-            </li>
-            <li>
-              <Button variant="subtle" className="p-[0.4rem]">
-                <UserIcon className="h-5 w-5 inline-block" />
-              </Button>
-            </li>
+            {session ? (
+              <>
+                <li>
+                  <UserIcon className="h-5 w-5 inline-block" />
+                </li>
+                <li>
+                  <Button
+                    variant="subtle"
+                    className="p-[0.4rem]"
+                    onClick={logout}
+                  >
+                    <ArrowRightOnRectangleIcon className="h-5 w-5 inline-block" />
+                  </Button>
+                </li>
+              </>
+            ) : (
+              <>
+                <li>
+                  <Button
+                    className="text-sm px-3 py-2 bg-neutral-700"
+                    onClick={showLoginTab}
+                  >
+                    Log In
+                  </Button>
+                </li>
+                <li>
+                  <Button
+                    className="text-sm px-3 py-2 bg-violet-500"
+                    onClick={showSignupTab}
+                  >
+                    Sign Up
+                  </Button>
+                </li>
+              </>
+            )}
           </ul>
         </div>
       </div>
diff --git a/client/config/index.ts b/client/config/index.ts
index 831ff05..3a2a2b0 100644
--- a/client/config/index.ts
+++ b/client/config/index.ts
@@ -1,3 +1 @@
 export const KRATOS_URL = "http://127.0.0.1:4433"
-export const PASSWORD_REGEX =
-  /^(?=.*?[A-Z])(?=.*?[a-z])(?=.*?[0-9])(?=.*?[#?!@$ %^&*-]).{8,64}$/
diff --git a/client/config/validation.ts b/client/config/validation.ts
new file mode 100644
index 0000000..ccd5e2d
--- /dev/null
+++ b/client/config/validation.ts
@@ -0,0 +1,18 @@
+import { z } from "zod"
+
+export const PASSWORD_REGEX =
+  /^(?=.*?[A-Z])(?=.*?[a-z])(?=.*?[0-9])(?=.*?[#?!@$ %^&*-]).{8,64}$/
+
+export const username = z
+  .string()
+  .trim()
+  .min(3, { message: "Username must be at least 3 character long." })
+  .max(16, { message: "Username can't be longer than 16 characters.." })
+
+export const password = z
+  .string()
+  .trim()
+  .regex(
+    PASSWORD_REGEX,
+    "Password must be 8-64 long and must contain a number, uppercase, lowercase and special character.",
+  )
diff --git a/client/hooks/useLogInFlow.ts b/client/hooks/useLogInFlow.ts
new file mode 100644
index 0000000..7788a0b
--- /dev/null
+++ b/client/hooks/useLogInFlow.ts
@@ -0,0 +1,53 @@
+import {
+  SelfServiceRegistrationFlow,
+  SubmitSelfServiceLoginFlowBody,
+} from "@ory/client"
+import { useRouter } from "next/router"
+import { useEffect, useState } from "react"
+import ory from "../services/ory"
+import useSession from "./useSession"
+
+const useLogInFlow = () => {
+  const router = useRouter()
+  const [flow, setFlow] = useState<SelfServiceRegistrationFlow>()
+  const { flow: flowId } = router.query
+  const updateSession = useSession((state) => state.update)
+
+  useEffect(() => {
+    const func = async () => {
+      if (!router.isReady || flow) {
+        return
+      }
+
+      let serviceFlow
+      if (flowId) {
+        serviceFlow = await ory.getSelfServiceLoginFlow(String(flowId))
+      } else {
+        serviceFlow = await ory.initializeSelfServiceLoginFlowForBrowsers()
+      }
+
+      setFlow(serviceFlow.data)
+    }
+
+    func()
+  }, [flowId, router, router.isReady, flow])
+
+  const submitData = async (data: SubmitSelfServiceLoginFlowBody) => {
+    await router.push(`/login?flow=${flow?.id}`, undefined, {
+      shallow: true,
+    })
+    ory
+      .submitSelfServiceLoginFlow(String(flow?.id), undefined, data)
+      .then(async ({ data }) => {
+        updateSession(data.session)
+        await router.push(flow?.return_to || "/")
+      })
+      .catch((err) => {
+        console.log({ err })
+      })
+  }
+
+  return { flow, submitData }
+}
+
+export default useLogInFlow
diff --git a/client/hooks/useLogout.ts b/client/hooks/useLogout.ts
new file mode 100644
index 0000000..168f8e8
--- /dev/null
+++ b/client/hooks/useLogout.ts
@@ -0,0 +1,38 @@
+import { AxiosError } from "axios"
+import { useRouter } from "next/router"
+import { useState, useEffect, DependencyList } from "react"
+
+import ory from "../services/ory"
+import useSession from "./useSession"
+
+export function useLogout(deps?: DependencyList) {
+  const session = useSession()
+  const [logoutToken, setLogoutToken] = useState<string>("")
+  const router = useRouter()
+
+  useEffect(() => {
+    ory
+      .createSelfServiceLogoutFlowUrlForBrowsers()
+      .then(({ data }) => {
+        setLogoutToken(data.logout_token)
+      })
+      .catch((err: AxiosError) => {
+        switch (err.response?.status) {
+          case 401:
+            return
+        }
+
+        return Promise.reject(err)
+      })
+  }, deps)
+
+  return () => {
+    if (logoutToken) {
+      session.drop()
+      ory
+        .submitSelfServiceLogoutFlow(logoutToken)
+        .then(() => router.push("/"))
+        .then(() => router.reload())
+    }
+  }
+}
diff --git a/client/hooks/useSession.ts b/client/hooks/useSession.ts
new file mode 100644
index 0000000..c812831
--- /dev/null
+++ b/client/hooks/useSession.ts
@@ -0,0 +1,31 @@
+import create from "zustand"
+
+import { Session } from "@ory/client"
+import ory from "../services/ory"
+import { AxiosError } from "axios"
+
+export interface SessionState {
+  session?: Session
+
+  load: () => void
+  update: (data: Session) => void
+  drop: () => void
+}
+const useSession = create<SessionState>((set) => ({
+  session: undefined,
+
+  load: () => {
+    ory
+      .toSession()
+      .then(({ data }) => {
+        set({ session: data })
+      })
+      .catch((err: AxiosError) => {})
+  },
+  update: (session) => set({ session }),
+  drop: () => {
+    set({ session: undefined })
+  },
+}))
+
+export default useSession
diff --git a/client/hooks/useSignUpFlow.ts b/client/hooks/useSignUpFlow.ts
index 2a86672..0d73960 100644
--- a/client/hooks/useSignUpFlow.ts
+++ b/client/hooks/useSignUpFlow.ts
@@ -1,15 +1,18 @@
 import {
   SelfServiceRegistrationFlow,
+  Session,
   SubmitSelfServiceRegistrationFlowBody,
 } from "@ory/client"
 import { useRouter } from "next/router"
 import { useEffect, useState } from "react"
 import ory from "../services/ory"
+import useSession from "./useSession"
 
 export const useSignUpFlow = () => {
   const router = useRouter()
   const [flow, setFlow] = useState<SelfServiceRegistrationFlow>()
-  const { flow: flowId, return_to: returnTo } = router.query
+  const { flow: flowId } = router.query
+  const updateSession = useSession((state) => state.update)
 
   useEffect(() => {
     const func = async () => {
@@ -22,30 +25,28 @@ export const useSignUpFlow = () => {
         serviceFlow = await ory.getSelfServiceRegistrationFlow(String(flowId))
       } else {
         serviceFlow =
-          await ory.initializeSelfServiceRegistrationFlowForBrowsers(
-            returnTo ? String(returnTo) : undefined,
-          )
+          await ory.initializeSelfServiceRegistrationFlowForBrowsers()
       }
 
       setFlow(serviceFlow.data)
     }
 
     func()
-  }, [flowId, router, router.isReady, returnTo, flow])
+  }, [flowId, router, router.isReady, flow])
 
   const submitData = async (data: SubmitSelfServiceRegistrationFlowBody) => {
     await router.push(`/signup?flow=${flow?.id}`, undefined, {
       shallow: true,
     })
-    try {
-      const resp = await ory.submitSelfServiceRegistrationFlow(
-        String(flow?.id),
-        data,
-      )
-      // TODO: handle registration
-    } catch (e) {
-      // TODO: handle errors
-    }
+    ory
+      .submitSelfServiceRegistrationFlow(String(flow?.id), data)
+      .then(async ({ data }) => {
+        updateSession(data.session as Session)
+        await router.push(flow?.return_to || "/")
+      })
+      .catch((err) => {
+        console.log({ err })
+      })
   }
 
   return { flow, submitData }
diff --git a/client/package.json b/client/package.json
index f1367e0..7e09a09 100644
--- a/client/package.json
+++ b/client/package.json
@@ -28,7 +28,8 @@
     "react-dom": "17.0.2",
     "react-hook-form": "^7.37.0",
     "tailwind-scrollbar": "2.1.0-preview.0",
-    "zod": "^3.19.1"
+    "zod": "^3.19.1",
+    "zustand": "^4.1.2"
   },
   "devDependencies": {
     "@trivago/prettier-plugin-sort-imports": "^2.0.2",
diff --git a/client/pages/_app.tsx b/client/pages/_app.tsx
index 3f5c9d5..5a94b0c 100644
--- a/client/pages/_app.tsx
+++ b/client/pages/_app.tsx
@@ -1,7 +1,14 @@
-import '../styles/globals.css'
-import type { AppProps } from 'next/app'
+import "../styles/globals.css"
+import type { AppProps } from "next/app"
+import useSession from "../hooks/useSession"
+import { useEffect } from "react"
 
 function MyApp({ Component, pageProps }: AppProps) {
+  const loadSession = useSession((state) => state.load)
+  useEffect(() => {
+    loadSession()
+  }, [loadSession])
+
   return <Component {...pageProps} />
 }
 
diff --git a/client/pages/index.tsx b/client/pages/index.tsx
new file mode 100644
index 0000000..87f7ca3
--- /dev/null
+++ b/client/pages/index.tsx
@@ -0,0 +1,9 @@
+import { NextPage } from "next"
+import useSession from "../hooks/useSession"
+
+const IndexPage: NextPage = () => {
+  const session = useSession()
+  return <div>{JSON.stringify(session.session || {})}</div>
+}
+
+export default IndexPage
diff --git a/client/pnpm-lock.yaml b/client/pnpm-lock.yaml
index 86e4ee7..c5d5fb8 100644
--- a/client/pnpm-lock.yaml
+++ b/client/pnpm-lock.yaml
@@ -26,6 +26,7 @@ specifiers:
   tailwindcss: ^3.1.8
   typescript: 4.4.4
   zod: ^3.19.1
+  zustand: ^4.1.2
 
 dependencies:
   '@headlessui/react': 1.7.3_sfoxds7t5ydpegc3knd667wn6m
@@ -41,6 +42,7 @@ dependencies:
   react-hook-form: 7.37.0_react@17.0.2
   tailwind-scrollbar: 2.1.0-preview.0_tailwindcss@3.1.8
   zod: 3.19.1
+  zustand: 4.1.2_react@17.0.2
 
 devDependencies:
   '@trivago/prettier-plugin-sort-imports': 2.0.4_prettier@2.7.1
@@ -3479,6 +3481,14 @@ packages:
     dependencies:
       punycode: 2.1.1
 
+  /use-sync-external-store/1.2.0_react@17.0.2:
+    resolution: {integrity: sha512-eEgnFxGQ1Ife9bzYs6VLi8/4X6CObHMw9Qr9tPY43iKwsPw8xE8+EFsf/2cFZ5S3esXgpWgtSCtLNS41F+sKPA==}
+    peerDependencies:
+      react: ^16.8.0 || ^17.0.0 || ^18.0.0
+    dependencies:
+      react: 17.0.2
+    dev: false
+
   /util-deprecate/1.0.2:
     resolution: {integrity: sha512-EPD5q1uXyFxJpCrLnCc1nHnq3gOa6DZBocAIiI2TaSCA7VCJ1UJDMagCzIkXNsUYfD1daK//LTEQ8xiIbrHtcw==}
 
@@ -3572,3 +3582,19 @@ packages:
   /zod/3.19.1:
     resolution: {integrity: sha512-LYjZsEDhCdYET9ikFu6dVPGp2YH9DegXjdJToSzD9rO6fy4qiRYFoyEYwps88OseJlPyl2NOe2iJuhEhL7IpEA==}
     dev: false
+
+  /zustand/4.1.2_react@17.0.2:
+    resolution: {integrity: sha512-gcRaKchcxFPbImrBb/BKgujOhHhik9YhVpIeP87ETT7uokEe2Szu7KkuZ9ghjtD+/KKkcrRNktR2AiLXPIbKIQ==}
+    engines: {node: '>=12.7.0'}
+    peerDependencies:
+      immer: '>=9.0'
+      react: '>=16.8'
+    peerDependenciesMeta:
+      immer:
+        optional: true
+      react:
+        optional: true
+    dependencies:
+      react: 17.0.2
+      use-sync-external-store: 1.2.0_react@17.0.2
+    dev: false