import { db, Post, PostType, UserProfile, VisitType } from '@nice/common';

export async function setPostRelation(params: {
  data: Post;
  staff?: UserProfile;
}) {
  const { data, staff } = params;
  // 在函数开始时计算一次时间
  const thirtyMinutesAgo = new Date(Date.now() - 30 * 60 * 1000);
  const clientIp = (data?.meta as any)?.ip;
  const commentsCount = await db.post.count({
    where: {
      parentId: data.id,
      type: PostType.POST_COMMENT,
    },
  });
  const readed =
    (await db.visit.count({
      where: {
        postId: data.id,
        type: VisitType.READED,
        visitorId: staff?.id,
      },
    })) > 0;
  const liked = await db.visit.count({
    where: {
      postId: data.id,
      type: VisitType?.LIKE,
      ...(staff?.id
        ? // 如果有 staff，查找对应的 visitorId
          { visitorId: staff.id }
        : // 如果没有 staff，查找相同 IP 且 visitorId 为 null 且 30 分钟内的记录
          {
            visitorId: null,
            meta: { path: ['ip'], equals: clientIp },
            updatedAt: {
              gte: thirtyMinutesAgo,
            },
          }),
    },
  });
  const readedCount = await db.visit.count({
    where: {
      postId: data.id,
      type: VisitType.READED,
    },
  });

  Object.assign(data, {
    readed,
    readedCount,
    liked,
    // limitedComments,
    commentsCount,
    // trouble
  });
}

export function getClientIp(req: any): string {
  let ip =
    req.ip ||
    (Array.isArray(req.headers['x-forwarded-for'])
      ? req.headers['x-forwarded-for'][0]
      : req.headers['x-forwarded-for']) ||
    req.socket.remoteAddress;

  // 如果是 IPv4-mapped IPv6 地址，转换为 IPv4
  if (typeof ip === 'string' && ip.startsWith('::ffff:')) {
    ip = ip.substring(7);
  }

  return ip || '';
}