rsp2k/headplane
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat: add logout and dropdown

Browse source
This commit is contained in:
Aarnav Tale 2024-03-30 02:44:06 -04:00
parent 3f65c7e610
commit 50bcfa9c04
No known key found for this signature in database
7 changed files with 144 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

49
app/routes/login.tsx
View file

@ -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>