feat: make all env optional (#94)

* feat: make all env optional

* fix: fixing typecheck

studio/drizzle.config.ts

@@ -6,7 +6,7 @@ export default {
   out: "./drizzle",
   driver: "turso",
   dbCredentials: {
-    url: env.DATABASE_URL,
+    url: env.DATABASE_URL ?? "",
     authToken: env.DATABASE_AUTH_TOKEN,
   },
 } satisfies Config;

studio/src/app/api/database/[database_id]/ops/handle-assign-user.ts

@@ -1,5 +1,5 @@
 import { HttpStatus } from "@studio/constants/http-status";
-import { db } from "@studio/db";
+import { get_database } from "@studio/db";
 import { database_role, database_user_role } from "@studio/db/schema-database";
 import { user as userTable } from "@studio/db/schema-user";
 import { ApiError } from "@studio/lib/api-error";
@@ -12,6 +12,7 @@ const handleAssignUser: DatabaseOperationHandler<
   RequestDatabaseAssignUser
 > = async ({ database: databaseInfo, body, user, permission }) => {
   const { roleId, userId } = body;
+  const db = get_database();
 
   // Validate if user input all fields
   if (!userId || !roleId)

studio/src/app/api/database/[database_id]/ops/handle-delete-user.ts

@@ -1,5 +1,5 @@
 import { HttpStatus } from "@studio/constants/http-status";
-import { db } from "@studio/db";
+import { get_database } from "@studio/db";
 import { database_role, database_user_role } from "@studio/db/schema-database";
 import { ApiError } from "@studio/lib/api-error";
 import { RequestDatabaseDeleteUser } from "@studio/lib/api/api-database-request";
@@ -11,6 +11,7 @@ const handleDeleteUser: DatabaseOperationHandler<
   RequestDatabaseDeleteUser
 > = async ({ database: databaseInfo, body, user, permission }) => {
   const { userId } = body;
+  const db = get_database();
 
   // Validate if user input all fields
   if (!userId)

studio/src/app/api/database/[database_id]/ops/handle-role-list.ts

@@ -1,10 +1,12 @@
-import { db } from "@studio/db";
+import { get_database } from "@studio/db";
 import { database_role } from "@studio/db/schema-database";
 import { DatabaseOperationHandler } from "@studio/lib/with-database-ops";
 import { eq } from "drizzle-orm";
 import { NextResponse } from "next/server";
 
 const handleRoleList: DatabaseOperationHandler = async ({ database }) => {
+  const db = get_database();
+
   const roles = await db.query.database_role.findMany({
     where: eq(database_role.databaseId, database.id),
   });

studio/src/app/api/database/[database_id]/ops/handle-users-list.ts

@@ -1,4 +1,4 @@
-import { db } from "@studio/db";
+import { get_database } from "@studio/db";
 import { user } from "@studio/db/schema";
 import { database_role, database_user_role } from "@studio/db/schema-database";
 import { DatabaseOperationHandler } from "@studio/lib/with-database-ops";
@@ -11,6 +11,7 @@ const handleUserList: DatabaseOperationHandler = async ({
 }) => {
   const assigned_user = alias(user, "assigned_user");
 
+  const db = get_database();
   const users = await db
     .select({
       id: user.id,

studio/src/app/api/database/[database_id]/route.ts

@@ -3,10 +3,10 @@ import withDatabaseOperation from "@studio/lib/with-database-ops";
 import { NextResponse } from "next/server";
 import { database } from "@studio/db/schema";
 import { eq } from "drizzle-orm";
-import { db } from "@studio/db";
 import { encrypt } from "@studio/lib/encryption-edge";
 import { env } from "@studio/env";
 import { SavedConnectionItemConfig } from "@studio/app/connect/saved-connection-storage";
+import { get_database } from "@studio/db";
 
 export const runtime = "edge";
 
@@ -40,6 +40,7 @@ export const GET = withDatabaseOperation(async ({ database: databaseInfo }) => {
 export const DELETE = withDatabaseOperation(
   async ({ database: databaseInfo, user }) => {
     const isOriginalOwner = databaseInfo.userId === user.id;
+    const db = get_database();
 
     if (!isOriginalOwner) {
       return NextResponse.json({ error: "Unauthorization" }, { status: 500 });
@@ -59,6 +60,7 @@ export const DELETE = withDatabaseOperation(
 export const PUT = withDatabaseOperation(
   async ({ database: databaseInfo, body, user }) => {
     const parsed = databaseSchema.safeParse(body);
+    const db = get_database();
 
     if (!parsed.success) {
       return NextResponse.json(

studio/src/app/api/database/route.ts

@@ -1,12 +1,12 @@
 import zod from "zod";
 import withUser from "@studio/lib/with-user";
-import { db } from "@studio/db";
 import { generateId } from "lucia";
 import { database, database_role, database_user_role } from "@studio/db/schema";
 import { NextResponse } from "next/server";
 import { env } from "@studio/env";
 import { encrypt } from "@studio/lib/encryption-edge";
 import { SavedConnectionItem } from "@studio/app/connect/saved-connection-storage";
+import { get_database } from "@studio/db";
 
 export const runtime = "edge";
 
@@ -43,6 +43,7 @@ export const POST = withUser(async ({ user, req }) => {
   const now = Date.now();
 
   try {
+    const db = get_database();
     await db.batch([
       db.insert(database).values({
         id: databaseId,

studio/src/app/api/databases/route.ts

@@ -1,5 +1,5 @@
 import { SavedConnectionItem } from "@studio/app/connect/saved-connection-storage";
-import { db } from "@studio/db";
+import { get_database } from "@studio/db";
 import {
   database,
   database_user_role,
@@ -12,6 +12,7 @@ import { NextResponse } from "next/server";
 export const runtime = "edge";
 
 export const GET = withUser(async ({ user }) => {
+  const db = get_database();
   const dbs = await db
     .select()
     .from(database_user_role)

studio/src/app/api/upload/route.ts

@@ -4,14 +4,14 @@ import withUser from "@studio/lib/with-user";
 import { NextResponse } from "next/server";
 import { ok, err } from "@justmiracle/result";
 import { R2 } from "@studio/lib/r2";
-import { db } from "@studio/db";
 import { user as userTable, user_file } from "@studio/db/schema";
 import { eq, sql } from "drizzle-orm";
 import { generateId } from "lucia";
 import { ApiError } from "@studio/lib/api-error";
 import { HttpStatus } from "@studio/constants/http-status";
 import { env } from "@studio/env";
 import { filetypeinfo } from "magic-bytes.js";
+import { get_database } from "@studio/db";
 
 export const runtime = "edge";
 
@@ -34,6 +34,7 @@ export const POST = withUser(async ({ req, user }) => {
     });
   }
 
+  const db = get_database();
   const formData = await req.formData().then(ok).catch(err);
 
   if (formData.error) {

studio/src/app/client/r/page.tsx

@@ -1,7 +1,7 @@
 import "@libsqlstudio/gui/css";
 
 import type { SavedConnectionLabel } from "@studio/app/connect/saved-connection-storage";
-import { db } from "@studio/db";
+import { get_database } from "@studio/db";
 import { database } from "@studio/db/schema";
 import { getSessionFromCookie } from "@studio/lib/auth";
 import { and, eq, isNotNull } from "drizzle-orm";
@@ -22,6 +22,7 @@ export default async function SessionPage({
     return <div>Something wrong</div>;
   }
 
+  const db = get_database();
   const databaseId = searchParams?.p;
   const databaseInfo = await db.query.database.findFirst({
     where: and(eq(database.id, databaseId), isNotNull(database.id)),

studio/src/controllers/auth.ts

@@ -1,4 +1,4 @@
-import { db } from "@studio/db";
+import { get_database } from "@studio/db";
 import { user, user_oauth } from "@studio/db/schema";
 import { Provider, lucia } from "@studio/lib/auth";
 import { generateId } from "lucia";
@@ -13,6 +13,7 @@ interface UserAuth {
 export const save = async (data: UserAuth, provider: Provider) => {
   const { id, name, email } = data;
   const headerStore = headers();
+  const db = get_database();
 
   const existingUser = await db.query.user_oauth.findFirst({
     where: (field, op) =>

studio/src/db/index.ts

@@ -3,9 +3,17 @@ import * as schema from "./schema";
 import { createClient } from "@libsql/client/web";
 import { env } from "@studio/env";
 
-export const connection = createClient({
-  url: env.DATABASE_URL,
-  authToken: env.DATABASE_AUTH_TOKEN,
-});
+export function get_connection() {
+  if (env.DATABASE_URL) {
+    return createClient({
+      url: env.DATABASE_URL,
+      authToken: env.DATABASE_AUTH_TOKEN,
+    });
+  }
 
-export const db = drizzle(connection, { schema });
+  throw new Error("Database is not setup");
+}
+
+export function get_database() {
+  return drizzle(get_connection(), { schema });
+}

studio/src/db/migrate.ts

@@ -1,14 +1,13 @@
 import "dotenv/config";
 
 import { migrate } from "drizzle-orm/libsql/migrator";
-import { db, connection } from ".";
+import { get_database } from ".";
 
 (async function () {
+  const db = get_database();
+
   // This will run migrations on the database, skipping the ones already applied
   await migrate(db, { migrationsFolder: "./drizzle" });
-
-  // Don't forget to close the connection, otherwise the script will hang
-  connection.close();
 })()
   .catch(console.error)
   .finally(() => {

studio/src/env.ts

@@ -7,9 +7,9 @@ export const appVersion = pkg.version;
 
 export const env = createEnv({
   server: {
-    BASE_URL: z.string().min(1),
-    DATABASE_URL: z.string().min(1),
-    DATABASE_AUTH_TOKEN: z.string().min(1),
+    BASE_URL: z.string().min(1).optional(),
+    DATABASE_URL: z.string().min(1).optional(),
+    DATABASE_AUTH_TOKEN: z.string().min(1).optional(),
 
     GITHUB_CLIENT_ID: z.string().optional(),
     GITHUB_CLIENT_SECRET: z.string().optional(),

studio/src/lib/auth.ts

@@ -1,7 +1,7 @@
 import { Lucia } from "lucia";
 import { GitHub, Google } from "arctic";
 import { LibSQLAdapter } from "@lucia-auth/adapter-sqlite";
-import { connection } from "@studio/db";
+import { get_connection } from "@studio/db";
 import { env } from "@studio/env";
 import { cache } from "react";
 import { cookies, headers } from "next/headers";
@@ -29,7 +29,7 @@ export const google = new Google(
   `${env.BASE_URL}/login/google/callback`
 );
 
-const adapter = new LibSQLAdapter(connection, {
+const adapter = new LibSQLAdapter(get_connection(), {
   user: "user",
   session: "user_session",
 });

studio/src/lib/with-database-ops.ts

@@ -8,9 +8,9 @@ import {
 import { User } from "lucia";
 import withUser from "./with-user";
 import { NextResponse } from "next/server";
-import { db } from "@studio/db";
 import { and, eq } from "drizzle-orm";
 import { headers } from "next/headers";
+import { get_database } from "@studio/db";
 
 export interface DatabasePermission {
   isOwner: boolean;
@@ -36,6 +36,7 @@ export default function withDatabaseOperation<T = unknown>(
   return withUser<{ params: { database_id: string } }>(
     async ({ user, req, params }) => {
       const databaseId = params.params.database_id;
+      const db = get_database();
 
       if (!databaseId) {
         return NextResponse.json(