Zero/apps/mail/lib/server-utils.ts

import type { HonoContext } from '@/trpc/hono';
import { connection } from '@zero/db/schema';
import { createDriver } from './driver';
import { and, eq } from 'drizzle-orm';

export const getActiveConnection = async (c: HonoContext) => {
  const { session, db } = c.var;
  if (!session?.user) throw new Error('Session Not Found');
  if (!session.connectionId) throw new Error('No active connection found for the user');

  const activeConnection = await db.query.connection.findFirst({
    where: and(eq(connection.userId, session.user.id), eq(connection.id, session.connectionId)),
  });

  if (!activeConnection) throw new Error('Active connection not found');

  if (!activeConnection.refreshToken || !activeConnection.accessToken)
    throw new Error(
      'Active Connection is not properly authorized, please reconnect the connection',
    );
  return activeConnection;
};

export const connectionToDriver = (
  activeConnection: typeof connection.$inferSelect,
  c: HonoContext,
) => {
  const driver = createDriver(activeConnection.providerId, {
    auth: {
      accessToken: activeConnection.accessToken,
      refreshToken: activeConnection.refreshToken!,
      email: activeConnection.email,
    },
    c,
  });
  return driver;
};