import { getYoutubeSubscriberCount } from "./youtube";
import { getRobloxFollowerCount } from "./roblox";
import { getDiscordMemberCount } from "./discord";
import { getTiktokProfile, type TiktokProfile } from "./tiktok";

export type { TiktokProfile };

export interface FollowerCounts {
  youtube: number | null;
  roblox: number | null;
  discord: number | null;
  tiktok: number | null;
}

export interface FollowerInfo {
  count: number;
  label: string;
}

const CACHE_KEY = "erika:follower_counts";
const CACHE_KEY_TIKTOK = "erika:tiktok_profile";
const CACHE_TTL = 3600; // 1 hour

export interface FetchResult {
  counts: FollowerCounts;
  tiktokProfile: TiktokProfile;
}

export async function getFollowerCounts(): Promise<FetchResult> {
  try {
    const { getRedisClient } = await import("@/lib/redis");
    const redis = await getRedisClient();
    const [cachedCounts, cachedProfile] = await Promise.all([
      redis.get(CACHE_KEY),
      redis.get(CACHE_KEY_TIKTOK),
    ]);
    if (cachedCounts && cachedProfile) {
      return {
        counts: JSON.parse(cachedCounts),
        tiktokProfile: JSON.parse(cachedProfile),
      };
    }
  } catch {
    // Redis unavailable, fall through to fetch
  }

  const [youtube, roblox, discord, tiktokProfileResult] = await Promise.all([
    getYoutubeSubscriberCount(),
    getRobloxFollowerCount(),
    getDiscordMemberCount(),
    getTiktokProfile(),
  ]);

  const counts: FollowerCounts = {
    youtube,
    roblox,
    discord,
    tiktok: tiktokProfileResult.followerCount,
  };

  try {
    const { getRedisClient } = await import("@/lib/redis");
    const redis = await getRedisClient();
    await Promise.all([
      redis.set(CACHE_KEY, JSON.stringify(counts), { ex: CACHE_TTL }),
      redis.set(CACHE_KEY_TIKTOK, JSON.stringify(tiktokProfileResult), { ex: CACHE_TTL }),
    ]);
  } catch {
    // Redis unavailable, skip caching
  }

  return { counts, tiktokProfile: tiktokProfileResult };
}

const followerLabels: Record<keyof FollowerCounts, string> = {
  youtube: "ผู้ติดตาม",
  roblox: "ผู้ติดตาม",
  discord: "สมาชิก",
  tiktok: "ผู้ติดตาม",
};

export function getCountForLink(name: string, counts: FollowerCounts): FollowerInfo | null {
  const key = name.toLowerCase();
  for (const [platform, label] of Object.entries(followerLabels)) {
    const count = counts[platform as keyof FollowerCounts];
    if (key.includes(platform) && count !== null) {
      return { count, label };
    }
  }
  return null;
}

export function formatCount(n: number): string {
  if (n >= 1_000_000) return `${(n / 1_000_000).toFixed(1)}M`;
  if (n >= 1_000) return `${(n / 1_000).toFixed(1)}K`;
  return n.toString();
}