fluxent-web/lib/actions/account.ts

"use server";

import { z } from "zod";
import { and, desc, eq, isNull } from "drizzle-orm";
import { revalidatePath } from "next/cache";
import { auth } from "@/lib/auth";
import { db } from "@/lib/db";
import { accounts, channels, type Account } from "@/lib/db/schema";

async function requireUser() {
  const session = await auth();
  if (!session?.user?.id) {
    throw new Error("请先登录");
  }
  return session.user.id;
}

const accountSchema = z.object({
  name: z.string().min(1, "请输入账户名称").max(100),
  accountType: z.enum(["BANK", "E_WALLET", "CASH"] as const),
  balanceType: z.enum(["ASSET", "LIABILITY", "EQUITY"] as const),
  primaryCurrency: z.string().trim().max(10).optional().nullable(),
  supportedCurrencies: z.array(z.string().trim()).optional().nullable(),
  remark: z.string().max(500).optional().nullable(),
  ext: z.string().optional().nullable(),

  // 银行账户字段
  issuerName: z.string().max(100).optional().nullable(),
  accountNumber: z.string().max(100).optional().nullable(),

  // 电子钱包字段
  platform: z.string().max(64).optional().nullable(),
  accountId: z.string().max(100).optional().nullable(),

  // 现金账户字段
  location: z.string().max(150).optional().nullable(),
});

export type AccountInput = z.infer<typeof accountSchema>;

export type AccountWithChannels = Account & {
  channelCount: number;
};

export async function getAccountsAction(): Promise<{
  success: boolean;
  data?: AccountWithChannels[];
  error?: string;
}> {
  try {
    const userId = await requireUser();

    const userAccounts = await db
      .select()
      .from(accounts)
      .where(and(eq(accounts.userId, userId), isNull(accounts.deletedAt)))
      .orderBy(desc(accounts.createdAt));

    const userChannels = await db
      .select({ id: channels.id, refAccounts: channels.refAccounts })
      .from(channels)
      .where(and(eq(channels.userId, userId), isNull(channels.deletedAt)));

    const result: AccountWithChannels[] = userAccounts.map((acc) => {
      const count = userChannels.filter((ch) =>
        Array.isArray(ch.refAccounts) && ch.refAccounts.includes(acc.id)
      ).length;
      return {
        ...acc,
        channelCount: count,
      };
    });

    return { success: true, data: result };
  } catch (err) {
    console.error("getAccountsAction error:", err);
    return { success: false, error: err instanceof Error ? err.message : "获取账户失败" };
  }
}

export async function createAccountAction(
  data: AccountInput
): Promise<{ success: boolean; data?: Account; error?: string }> {
  try {
    const userId = await requireUser();
    const parsed = accountSchema.safeParse(data);
    if (!parsed.success) {
      return { success: false, error: parsed.error.issues[0]?.message || "参数错误" };
    }

    const val = parsed.data;
    const [newAccount] = await db
      .insert(accounts)
      .values({
        userId,
        name: val.name.trim(),
        accountType: val.accountType,
        balanceType: val.balanceType,
        primaryCurrency: val.primaryCurrency ? val.primaryCurrency.toUpperCase() : null,
        supportedCurrencies: val.supportedCurrencies && val.supportedCurrencies.length > 0 ? val.supportedCurrencies : null,
        remark: val.remark?.trim() || null,
        ext: val.ext?.trim() || null,
        issuerName: val.issuerName?.trim() || null,
        accountNumber: val.accountNumber?.trim() || null,
        platform: val.platform?.trim() || null,
        accountId: val.accountId?.trim() || null,
        location: val.location?.trim() || null,
        isActive: true,
      })
      .returning();

    revalidatePath("/accounts");
    revalidatePath("/channels");
    revalidatePath("/");
    return { success: true, data: newAccount };
  } catch (err) {
    console.error("createAccountAction error:", err);
    return { success: false, error: err instanceof Error ? err.message : "创建账户失败" };
  }
}

export async function updateAccountAction(
  id: string,
  data: AccountInput
): Promise<{ success: boolean; data?: Account; error?: string }> {
  try {
    const userId = await requireUser();
    const parsed = accountSchema.safeParse(data);
    if (!parsed.success) {
      return { success: false, error: parsed.error.issues[0]?.message || "参数错误" };
    }

    const val = parsed.data;
    const [updated] = await db
      .update(accounts)
      .set({
        name: val.name.trim(),
        accountType: val.accountType,
        balanceType: val.balanceType,
        primaryCurrency: val.primaryCurrency ? val.primaryCurrency.toUpperCase() : null,
        supportedCurrencies: val.supportedCurrencies && val.supportedCurrencies.length > 0 ? val.supportedCurrencies : null,
        remark: val.remark?.trim() || null,
        ext: val.ext?.trim() || null,
        issuerName: val.issuerName?.trim() || null,
        accountNumber: val.accountNumber?.trim() || null,
        platform: val.platform?.trim() || null,
        accountId: val.accountId?.trim() || null,
        location: val.location?.trim() || null,
        updatedAt: new Date(),
      })
      .where(and(eq(accounts.id, id), eq(accounts.userId, userId), isNull(accounts.deletedAt)))
      .returning();

    if (!updated) {
      return { success: false, error: "未找到该账户或无权修改" };
    }

    revalidatePath("/accounts");
    revalidatePath("/channels");
    revalidatePath("/");
    return { success: true, data: updated };
  } catch (err) {
    console.error("updateAccountAction error:", err);
    return { success: false, error: err instanceof Error ? err.message : "更新账户失败" };
  }
}

export async function toggleAccountActiveAction(
  id: string,
  isActive: boolean
): Promise<{ success: boolean; error?: string }> {
  try {
    const userId = await requireUser();
    await db
      .update(accounts)
      .set({
        isActive,
        updatedAt: new Date(),
      })
      .where(and(eq(accounts.id, id), eq(accounts.userId, userId), isNull(accounts.deletedAt)));

    revalidatePath("/accounts");
    return { success: true };
  } catch (err) {
    console.error("toggleAccountActiveAction error:", err);
    return { success: false, error: err instanceof Error ? err.message : "状态切换失败" };
  }
}

export async function deleteAccountAction(
  id: string
): Promise<{ success: boolean; error?: string }> {
  try {
    const userId = await requireUser();
    // 软删除
    const [deleted] = await db
      .update(accounts)
      .set({
        deletedAt: new Date(),
        updatedAt: new Date(),
      })
      .where(and(eq(accounts.id, id), eq(accounts.userId, userId), isNull(accounts.deletedAt)))
      .returning();

    if (!deleted) {
      return { success: false, error: "账户不存在或已删除" };
    }

    revalidatePath("/accounts");
    revalidatePath("/channels");
    revalidatePath("/");
    return { success: true };
  } catch (err) {
    console.error("deleteAccountAction error:", err);
    return { success: false, error: err instanceof Error ? err.message : "删除账户失败" };
  }
}