boredgame/src/lib/server/api/controllers/oauth.controller.ts

import 'reflect-metadata'
import { Controller } from '$lib/server/api/common/types/controller'
import { LuciaService } from '$lib/server/api/services/lucia.service'
import { OAuthService } from '$lib/server/api/services/oauth.service'
import { github, google } from '$lib/server/auth'
import { OAuth2RequestError } from 'arctic'
import { getCookie, setCookie } from 'hono/cookie'
import { TimeSpan } from 'oslo'
import { inject, injectable } from 'tsyringe'

@injectable()
export class OAuthController extends Controller {
	constructor(
		@inject(LuciaService) private luciaService: LuciaService,
		@inject(OAuthService) private oauthService: OAuthService,
	) {
		super()
	}

	routes() {
		return this.controller
			.get('/github', async (c) => {
				try {
					const code = c.req.query('code')?.toString() ?? null
					const state = c.req.query('state')?.toString() ?? null
					const storedState = getCookie(c).github_oauth_state ?? null

					console.log('code', code, 'state', state, 'storedState', storedState)

					if (!code || !state || !storedState || state !== storedState) {
						return c.body(null, 400)
					}

					const tokens = await github.validateAuthorizationCode(code)
					const githubUserResponse = await fetch('https://api.github.com/user', {
						headers: {
							Authorization: `Bearer ${tokens.accessToken}`,
						},
					})
					const githubUser: GitHubUser = await githubUserResponse.json()

					const userId = await this.oauthService.handleOAuthUser(githubUser.id, githubUser.login, 'github')

					const session = await this.luciaService.lucia.createSession(userId, {})
					const sessionCookie = this.luciaService.lucia.createSessionCookie(session.id)

					setCookie(c, sessionCookie.name, sessionCookie.value, {
						path: sessionCookie.attributes.path,
						maxAge:
							sessionCookie?.attributes?.maxAge && sessionCookie?.attributes?.maxAge < new TimeSpan(365, 'd').seconds()
								? sessionCookie.attributes.maxAge
								: new TimeSpan(2, 'w').seconds(),
						domain: sessionCookie.attributes.domain,
						sameSite: sessionCookie.attributes.sameSite as any,
						secure: sessionCookie.attributes.secure,
						httpOnly: sessionCookie.attributes.httpOnly,
						expires: sessionCookie.attributes.expires,
					})

					return c.json({ message: 'ok' })
				} catch (error) {
					console.error(error)
					// the specific error message depends on the provider
					if (error instanceof OAuth2RequestError) {
						// invalid code
						return c.body(null, 400)
					}
					return c.body(null, 500)
				}
			})
			.get('/google', async (c) => {
				try {
					const code = c.req.query('code')?.toString() ?? null
					const state = c.req.query('state')?.toString() ?? null
					const storedState = getCookie(c).google_oauth_state ?? null
					const storedCodeVerifier = getCookie(c).google_oauth_code_verifier ?? null

					console.log('code', code, 'state', state, 'storedState', storedState)

					if (!code || !storedState || !storedCodeVerifier || state !== storedState) {
						return c.body(null, 400)
					}

					const tokens = await google.validateAuthorizationCode(code, storedCodeVerifier)
					console.log('tokens', tokens)
					const googleUserResponse = await fetch('https://www.googleapis.com/oauth2/v3/userinfo', {
						headers: {
							Authorization: `Bearer ${tokens.accessToken}`,
						},
					})
					console.log('googleUserResponse', googleUserResponse)
					const googleUser: GoogleUser = await googleUserResponse.json()

					const userId = await this.oauthService.handleOAuthUser(googleUser.id, googleUser.login, 'github')

					const session = await this.luciaService.lucia.createSession(userId, {})
					const sessionCookie = this.luciaService.lucia.createSessionCookie(session.id)

					setCookie(c, sessionCookie.name, sessionCookie.value, {
						path: sessionCookie.attributes.path,
						maxAge:
							sessionCookie?.attributes?.maxAge && sessionCookie?.attributes?.maxAge < new TimeSpan(365, 'd').seconds()
								? sessionCookie.attributes.maxAge
								: new TimeSpan(2, 'w').seconds(),
						domain: sessionCookie.attributes.domain,
						sameSite: sessionCookie.attributes.sameSite as any,
						secure: sessionCookie.attributes.secure,
						httpOnly: sessionCookie.attributes.httpOnly,
						expires: sessionCookie.attributes.expires,
					})

					return c.json({ message: 'ok' })
				} catch (error) {
					console.error(error)
					// the specific error message depends on the provider
					if (error instanceof OAuth2RequestError) {
						// invalid code
						return c.body(null, 400)
					}
					return c.body(null, 500)
				}
			})
	}
}

interface GitHubUser {
	id: number
	login: string
}

interface GoogleUser {
	id: number
	login: string
}