vaulterm/frontend/pages/auth/hooks.ts

import { useMutation, useQuery } from "@tanstack/react-query";
import {
  loginResultSchema,
  LoginSchema,
  loginSchema,
  RegisterSchema,
} from "./schema";
import authStore from "@/stores/auth";
import { router } from "expo-router";
import api from "@/lib/api";

export const useLoginMutation = () => {
  return useMutation({
    mutationFn: async (body: LoginSchema) => {
      const res = await api("/auth/login", { method: "POST", body });
      const { data } = loginResultSchema.safeParse(res);
      if (!data) {
        throw new Error("Invalid response!");
      }
      return data;
    },
    onSuccess(data) {
      authStore.setState({ token: data.sessionId });
      router.replace("/");
    },
  });
};

export const useRegisterMutation = () => {
  return useMutation({
    mutationFn: async (body: RegisterSchema) => {
      const res = await api("/auth/register", { method: "POST", body });
      const { data } = loginResultSchema.safeParse(res);
      if (!data) {
        throw new Error("Invalid response!");
      }
      return data;
    },
    onSuccess(data) {
      authStore.setState({ token: data.sessionId });
      router.replace("/");
    },
  });
};

export const useOAuthCallback = (type: string) => {
  return useMutation({
    mutationFn: async (body: { code: string; verifier?: string }) => {
      const res = await api(`/auth/${type}`, { body, method: "POST" });
      const { data } = loginResultSchema.safeParse(res);
      if (!data) {
        throw new Error("Invalid response!");
      }
      return data;
    },
    onError: (err) => {
      console.log(err);
    },
    onSuccess(data) {
      authStore.setState({ token: data.sessionId });
      router.replace("/");
    },
  });
};