diff --git a/src/lib/util.ts b/src/lib/util.ts index 633cfde..402f77b 100644 --- a/src/lib/util.ts +++ b/src/lib/util.ts @@ -14,7 +14,7 @@ export function jsonResponse (data: any, status = 200) { }) } -export function getDatabaseConnection (type: number = 0) { +export function getDatabaseConnection (type: number) { if (type !== 0 && type !== 1) return null const env = diff --git a/src/routes/launcher/latest.ts b/src/routes/launcher/latest.ts index dd3c24d..ec5f805 100644 --- a/src/routes/launcher/latest.ts +++ b/src/routes/launcher/latest.ts @@ -1,9 +1,12 @@ import { launcherUpdates } from '../../lib/tables' import { desc, eq } from 'drizzle-orm' -import { getDatabaseConnection } from '../../lib/util' +import { getDatabaseConnection, jsonResponse } from '../../lib/util' export async function handler () { - const { connection, db } = getDatabaseConnection() + const dbResult = getDatabaseConnection(0) + if (!dbResult) + return jsonResponse({ error: 'Failed to connect to database' }, 500) + const { connection, db } = dbResult const version = await db .select({ diff --git a/src/routes/launcher/loader/update-data.ts b/src/routes/launcher/loader/update-data.ts index e45de5f..e82adfc 100644 --- a/src/routes/launcher/loader/update-data.ts +++ b/src/routes/launcher/loader/update-data.ts @@ -4,7 +4,10 @@ import { getDatabaseConnection, jsonResponse } from '../../../lib/util' import { Context } from 'elysia' export async function handler (context: Context) { - const { connection, db } = getDatabaseConnection() + const dbResult = getDatabaseConnection(0) + if (!dbResult) + return jsonResponse({ error: 'Failed to connect to database' }, 500) + const { connection, db } = dbResult const platform = context.query.platform as string | undefined const arch = context.query.arch as string | undefined diff --git a/src/routes/launcher/versions.ts b/src/routes/launcher/versions.ts index 6c1578e..755ac28 100644 --- a/src/routes/launcher/versions.ts +++ b/src/routes/launcher/versions.ts @@ -4,7 +4,10 @@ import { getDatabaseConnection, jsonResponse } from '../../lib/util' import { Context } from 'elysia' export async function handler (context: Context) { - const { connection, db } = getDatabaseConnection() + const dbResult = getDatabaseConnection(0) + if (!dbResult) + return jsonResponse({ error: 'Failed to connect to database' }, 500) + const { connection, db } = dbResult const platform = context.query.platform as string | undefined const arch = context.query.arch as string | undefined