111 lines
2.6 KiB
TypeScript
111 lines
2.6 KiB
TypeScript
"use server";
|
|
|
|
import { z } from "zod";
|
|
|
|
import { createUser, getUser, getAuthSettings } from "@/lib/db/queries";
|
|
|
|
import { signIn } from "./auth";
|
|
|
|
const authFormSchema = z.object({
|
|
email: z.string().email(),
|
|
password: z.string().min(6),
|
|
});
|
|
|
|
export type LoginActionState = {
|
|
status: "idle" | "in_progress" | "success" | "failed" | "invalid_data";
|
|
};
|
|
|
|
export const login = async (
|
|
_: LoginActionState,
|
|
formData: FormData
|
|
): Promise<LoginActionState> => {
|
|
try {
|
|
const provider = formData.get("provider") as string;
|
|
|
|
if (provider === "dingtalk") {
|
|
const token = formData.get("token");
|
|
console.log("[DingTalk Login] Received token:", token);
|
|
if (!token) return { status: "invalid_data" };
|
|
|
|
try {
|
|
const result = await signIn("dingtalk", {
|
|
token,
|
|
redirect: false,
|
|
});
|
|
console.log("[DingTalk Login] signIn result:", result);
|
|
return { status: "success" };
|
|
} catch (signInError) {
|
|
console.error("[DingTalk Login] signIn error:", signInError);
|
|
throw signInError;
|
|
}
|
|
}
|
|
|
|
const validatedData = authFormSchema.parse({
|
|
email: formData.get("email"),
|
|
password: formData.get("password"),
|
|
});
|
|
|
|
await signIn("credentials", {
|
|
email: validatedData.email,
|
|
password: validatedData.password,
|
|
redirect: false,
|
|
});
|
|
|
|
return { status: "success" };
|
|
} catch (error) {
|
|
if (error instanceof z.ZodError) {
|
|
return { status: "invalid_data" };
|
|
}
|
|
|
|
return { status: "failed" };
|
|
}
|
|
};
|
|
|
|
export type RegisterActionState = {
|
|
status:
|
|
| "idle"
|
|
| "in_progress"
|
|
| "success"
|
|
| "failed"
|
|
| "user_exists"
|
|
| "invalid_data";
|
|
};
|
|
|
|
export const register = async (
|
|
_: RegisterActionState,
|
|
formData: FormData
|
|
): Promise<RegisterActionState> => {
|
|
try {
|
|
// 检查是否开放注册
|
|
const settings = await getAuthSettings();
|
|
if (settings.isRegistrationEnabled === false) {
|
|
return { status: "failed" };
|
|
}
|
|
|
|
const validatedData = authFormSchema.parse({
|
|
email: formData.get("email"),
|
|
password: formData.get("password"),
|
|
});
|
|
|
|
const [user] = await getUser(validatedData.email);
|
|
|
|
if (user) {
|
|
return { status: "user_exists" } as RegisterActionState;
|
|
}
|
|
await createUser(validatedData.email, validatedData.password);
|
|
await signIn("credentials", {
|
|
email: validatedData.email,
|
|
password: validatedData.password,
|
|
redirect: false,
|
|
});
|
|
|
|
return { status: "success" };
|
|
} catch (error) {
|
|
if (error instanceof z.ZodError) {
|
|
return { status: "invalid_data" };
|
|
}
|
|
|
|
return { status: "failed" };
|
|
}
|
|
};
|