69 lines · 2.7 KB
import { createServerFn } from "@tanstack/react-start";
import { enforceRateLimit, requestIp } from "~/server/limits";
import { z } from "zod";
import { viewerMiddleware } from "./middleware";
import { createSession, destroySession, loadUserByLogin } from "~/server/auth";
import { verifyPassword } from "~/server/crypto";
import { first } from "~/server/db";
import { createUser, githubOAuthEnabled, signupsEnabled } from "~/server/accounts";
import { badRequest } from "~/server/errors";
import { sendEmailVerification } from "~/server/account-security";
import { requiresSecondFactor, startPendingLogin } from "~/server/two-factor";
import { accountName, email, password } from "~/lib/validation";

export const getSession = createServerFn({ method: "GET" })
  .middleware([viewerMiddleware])
  .handler(async ({ context }) => {
    const unread = context.viewer
      ? ((
          await first<{ count: number }>(
            "SELECT COUNT(*) AS count FROM notifications WHERE user_id = ? AND read_at IS NULL",
            context.viewer.id,
          )
        )?.count ?? 0)
      : 0;
    return { viewer: context.viewer, unreadNotifications: unread };
  });

export const getAuthOptions = createServerFn({ method: "GET" }).handler(async () => {
  return { signupsEnabled: signupsEnabled(), githubEnabled: githubOAuthEnabled() };
});

export const signUp = createServerFn({ method: "POST" })
  .validator(z.object({ username: accountName, email, password }))
  .handler(async ({ data }) => {
    await enforceRateLimit("AUTH_LIMITER", `signup:${requestIp()}`);
    const id = await createUser({
      name: data.username,
      email: data.email,
      password: data.password,
      githubId: null,