From 50bcfa9c04ae4dba187f369a674baa13ea960ae3 Mon Sep 17 00:00:00 2001
From: Aarnav Tale <aarnavtale@icloud.com>
Date: Sat, 30 Mar 2024 02:44:06 -0400
Subject: [PATCH] feat: add logout and dropdown

---
 app/routes/_data.tsx  | 81 +++++++++++++++++++++++++++++++++++++++----
 app/routes/login.tsx  | 49 +++++++++++++++++++-------
 app/routes/logout.tsx | 15 ++++++++
 app/types/Key.ts      |  7 ++++
 app/types/index.ts    |  1 +
 app/utils/oidc.ts     |  5 +++
 app/utils/sessions.ts |  5 +++
 7 files changed, 144 insertions(+), 19 deletions(-)
 create mode 100644 app/routes/logout.tsx
 create mode 100644 app/types/Key.ts

diff --git a/app/routes/_data.tsx b/app/routes/_data.tsx
index d035ee9..e7b56f5 100644
--- a/app/routes/_data.tsx
+++ b/app/routes/_data.tsx
@@ -1,6 +1,9 @@
-import { Cog8ToothIcon, CpuChipIcon, GlobeAltIcon, LockClosedIcon, ServerStackIcon, UsersIcon } from '@heroicons/react/24/outline'
+import { Menu, Transition } from '@headlessui/react'
+import { Cog8ToothIcon, CpuChipIcon, GlobeAltIcon, LockClosedIcon, ServerStackIcon, UserCircleIcon, UsersIcon } from '@heroicons/react/24/outline'
 import { type LoaderFunctionArgs, redirect } from '@remix-run/node'
-import { Outlet, useLoaderData, useRouteError } from '@remix-run/react'
+import { Form, Outlet, useLoaderData, useRouteError } from '@remix-run/react'
+import clsx from 'clsx'
+import { Fragment } from 'react/jsx-runtime'
 
 import { ErrorPopup } from '~/components/Error'
 import TabLink from '~/components/TabLink'
@@ -34,7 +37,10 @@ export async function loader({ request }: LoaderFunctionArgs) {
 	}
 
 	const context = await getContext()
-	return context
+	return {
+		...context,
+		user: session.get('user')
+	}
 }
 
 export default function Layout() {
@@ -43,9 +49,72 @@ export default function Layout() {
 		<>
 			<header className='mb-16 bg-gray-800 text-white dark:bg-gray-700'>
 				<nav className='container mx-auto'>
-					<div className='flex items-center gap-x-2 mb-8 pt-4'>
-						<CpuChipIcon className='w-8 h-8'/>
-						<h1 className='text-2xl'>Headplane</h1>
+					<div className='flex items-center justify-between mb-8 pt-4'>
+						<div className='flex items-center gap-x-2'>
+							<CpuChipIcon className='w-8 h-8'/>
+							<h1 className='text-2xl'>Headplane</h1>
+						</div>
+						<div className='flex items-center gap-x-4'>
+							<a href='https://tailscale.com/download' target='_blank' rel='noreferrer' className='text-gray-300 hover:text-white'>
+								Download
+							</a>
+							<a href='https://github.com/tale/headplane' target='_blank' rel='noreferrer' className='text-gray-300 hover:text-white'>
+								GitHub
+							</a>
+							<a href='https://github.com/juanfont/headscale' target='_blank' rel='noreferrer' className='text-gray-300 hover:text-white'>
+								Headscale
+							</a>
+							<div className='relative'>
+								<Menu>
+									<Menu.Button>
+										<UserCircleIcon className='w-8 h-8'/>
+									</Menu.Button>
+									<Transition
+										as={Fragment}
+										enter='transition ease-out duration-100'
+										enterFrom='transform opacity-0 scale-95'
+										enterTo='transform opacity-100 scale-100'
+										leave='transition ease-in duration-75'
+										leaveFrom='transform opacity-100 scale-100'
+										leaveTo='transform opacity-0 scale-95'
+									>
+										<Menu.Items className={clsx(
+											'absolute right-0 w-36 mt-2 rounded-md',
+											'text-gray-700 dark:text-gray-300',
+											'bg-white dark:bg-zinc-800 text-right',
+											'overflow-hidden',
+											'border border-gray-200 dark:border-zinc-500',
+											'divide-y divide-gray-200 dark:divide-zinc-500'
+										)}
+										>
+											<Menu.Item>
+												{() => (
+													<div className='px-4 py-2'>
+														<p>{data.user?.name}</p>
+														<p>{data.user?.email}</p>
+													</div>
+												)}
+											</Menu.Item>
+											<Menu.Item>
+												{({ active }) => (
+													<Form method='POST' action='/logout'>
+														<button
+															type='submit'
+															className={clsx(
+																'px-4 py-2 w-full text-right',
+																active ? 'bg-gray-200 dark:bg-zinc-500' : ''
+															)}
+														>
+															Logout
+														</button>
+													</Form>
+												)}
+											</Menu.Item>
+										</Menu.Items>
+									</Transition>
+								</Menu>
+							</div>
+						</div>
 					</div>
 					<div className='flex items-center gap-x-4'>
 						<TabLink to='/machines' name='Machines' icon={<ServerStackIcon className='w-5 h-5'/>}/>
diff --git a/app/routes/login.tsx b/app/routes/login.tsx
index 5ac5919..6a31cbe 100644
--- a/app/routes/login.tsx
+++ b/app/routes/login.tsx
@@ -1,8 +1,9 @@
 import { type ActionFunctionArgs, json, type LoaderFunctionArgs, redirect } from '@remix-run/node'
-import { Form, Link, useActionData, useLoaderData } from '@remix-run/react'
+import { Form, useActionData, useLoaderData } from '@remix-run/react'
 import { useMemo } from 'react'
 
 import Code from '~/components/Code'
+import { type Key } from '~/types'
 import { pull } from '~/utils/headscale'
 import { startOidc } from '~/utils/oidc'
 import { commitSession, getSession } from '~/utils/sessions'
@@ -46,26 +47,47 @@ export async function loader({ request }: LoaderFunctionArgs) {
 
 export async function action({ request }: ActionFunctionArgs) {
 	const formData = await request.formData()
+	const oidcStart = String(formData.get('oidc-start'))
+	if (oidcStart) {
+		const issuer = process.env.OIDC_ISSUER
+		const id = process.env.OIDC_CLIENT_ID
+		// eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+		return startOidc(issuer!, id!, request)
+	}
+
 	const apiKey = String(formData.get('api-key'))
 	const session = await getSession(request.headers.get('Cookie'))
 
 	// Test the API key
 	try {
-		await pull('v1/apikey', apiKey)
+		const apiKeys = await pull<{ apiKeys: Key[] }>('v1/apikey', apiKey)
+		const key = apiKeys.apiKeys.find(k => apiKey.startsWith(k.prefix))
+		if (!key) {
+			throw new Error('Invalid API key')
+		}
+
+		const expiry = new Date(key.expiration)
+		const expiresIn = expiry.getTime() - Date.now()
+		const expiresDays = Math.round(expiresIn / 1000 / 60 / 60 / 24)
+
+		session.set('hsApiKey', apiKey)
+		session.set('user', {
+			name: key.prefix,
+			email: `${expiresDays} days`
+		})
+
+		return redirect('/machines', {
+			headers: {
+				// eslint-disable-next-line @typescript-eslint/naming-convention
+				'Set-Cookie': await commitSession(session)
+			}
+		})
 	} catch (error) {
 		console.error(error)
 		return json({
 			error: 'Invalid API key'
 		})
 	}
-
-	session.set('hsApiKey', apiKey)
-	return redirect('/machines', {
-		headers: {
-			// eslint-disable-next-line @typescript-eslint/naming-convention
-			'Set-Cookie': await commitSession(session)
-		}
-	})
 }
 
 export default function Page() {
@@ -118,11 +140,12 @@ export default function Page() {
 					</div>
 				) : undefined}
 				{data.oidc ? (
-					<Link to='/oidc/start'>
-						<button className='bg-gray-800 text-white rounded-md p-2 w-full' type='button'>
+					<Form method='POST'>
+						<input type='hidden' name='oidc-start' value='true'/>
+						<button className='bg-gray-800 text-white rounded-md p-2 w-full' type='submit'>
 							Login with SSO
 						</button>
-					</Link>
+					</Form>
 				) : undefined}
 			</div>
 		</div>
diff --git a/app/routes/logout.tsx b/app/routes/logout.tsx
new file mode 100644
index 0000000..8539078
--- /dev/null
+++ b/app/routes/logout.tsx
@@ -0,0 +1,15 @@
+import { type ActionFunctionArgs, redirect } from '@remix-run/node'
+
+import { destroySession, getSession } from '~/utils/sessions'
+
+export async function action({ request }: ActionFunctionArgs) {
+	const session = await getSession(request.headers.get('Cookie'))
+	const returnTo = new URL(request.url).pathname
+
+	return redirect(`/login?returnTo=${returnTo}`, {
+		headers: {
+			// eslint-disable-next-line @typescript-eslint/naming-convention
+			'Set-Cookie': await destroySession(session)
+		}
+	})
+}
diff --git a/app/types/Key.ts b/app/types/Key.ts
new file mode 100644
index 0000000..d496d01
--- /dev/null
+++ b/app/types/Key.ts
@@ -0,0 +1,7 @@
+export type Key = {
+	id: string;
+	prefix: string;
+	expiration: string;
+	createdAt: Date;
+	lastSeen: Date;
+}
diff --git a/app/types/index.ts b/app/types/index.ts
index 64063d4..08d36c3 100644
--- a/app/types/index.ts
+++ b/app/types/index.ts
@@ -1,3 +1,4 @@
+export * from './Key'
 export * from './Machine'
 export * from './Route'
 export * from './User'
diff --git a/app/utils/oidc.ts b/app/utils/oidc.ts
index e5f7aa2..77ab240 100644
--- a/app/utils/oidc.ts
+++ b/app/utils/oidc.ts
@@ -126,6 +126,11 @@ export async function finishOidc(issuer: string, client: string, secret: string,
 	})
 
 	session.set('hsApiKey', keyResponse.apiKey)
+	session.set('user', {
+		name: claims.name ? String(claims.name) : 'Anonymous',
+		email: claims.email ? String(claims.email) : undefined
+	})
+
 	return redirect('/machines', {
 		headers: {
 			// eslint-disable-next-line @typescript-eslint/naming-convention
diff --git a/app/utils/sessions.ts b/app/utils/sessions.ts
index a42b3b8..1e9f087 100644
--- a/app/utils/sessions.ts
+++ b/app/utils/sessions.ts
@@ -5,6 +5,10 @@ type SessionData = {
 	authState: string;
 	authNonce: string;
 	authVerifier: string;
+	user: {
+		name: string;
+		email?: string;
+	};
 }
 
 type SessionFlashData = {
@@ -23,6 +27,7 @@ export const {
 			maxAge: 60 * 60 * 24, // 24 hours
 			path: '/',
 			sameSite: 'lax',
+			// eslint-disable-next-line @typescript-eslint/no-non-null-assertion
 			secrets: [process.env.COOKIE_SECRET!],
 			secure: true
 		}