"use server";

import { cookies } from "next/headers";

type RequestCookies = Awaited<ReturnType<typeof cookies>>;

const decodeValue = (value?: string) => {
  if (!value) return value;
  return value.startsWith("base64-") ? Buffer.from(value.slice(7), "base64").toString("utf8") : value;
};

const encodeValue = (value: string) => {
  if (!value) return value;
  // Next.js 会自动 base64 编码，我们只需处理已有 base64 前缀的情况
  if (value.startsWith("base64-")) {
    return value;
  }
  return value;
};

const wrapCookies = (store: RequestCookies) => {
  return {
    get: (name: string) => {
      const cookie = store.get(name);
      if (!cookie) return cookie;
      return { ...cookie, value: decodeValue(cookie.value) };
    },
    getAll: (...args: Parameters<RequestCookies["getAll"]>) =>
      store.getAll(...args).map((cookie) => ({ ...cookie, value: decodeValue(cookie.value) })),
    set: (...args: Parameters<RequestCookies["set"]>) => {
      const [name, value, options] = args;
      if (typeof value === "string") {
        store.set(name, encodeValue(value), options);
      } else {
        store.set(name, value);
      }
    },
    delete: (...args: Parameters<RequestCookies["delete"]>) => store.delete(...args),
    has: (...args: Parameters<RequestCookies["has"]>) => store.has(...args),
  };
};

export const getDecodedCookies = async () => {
  const store = await cookies();
  return wrapCookies(store) as RequestCookies;
};