// recipe-audience-demographics.ts
// Audience geography for a creator + sampled age/gender for one of their videos.
// Run with: SOCIALCRAWL_KEY=sc_... npx tsx recipe-audience-demographics.ts

const KEY = process.env.SOCIALCRAWL_KEY;
if (!KEY) throw new Error("Set SOCIALCRAWL_KEY");

const BASE = "https://www.socialcrawl.dev/v1";
const SAMPLE_SIZE = 10;

async function get<T>(path: string, params: Record<string, string>): Promise<T> {
  const url = new URL(`${BASE}/${path}`);
  for (const [k, v] of Object.entries(params)) url.searchParams.set(k, v);
  const res = await fetch(url, { headers: { "x-api-key": KEY! } });
  return (await res.json()) as T;
}

// 1. Which countries is the creator's audience in? (5 credits)
type AudienceRes = {
  data?: {
    audienceLocations: Array<{
      country: string;
      countryCode: string;
      count: number;
      percentage: string;
    }>;
  };
};
const audience = await get<AudienceRes>("tiktok/user/audience", {
  handle: "charlidamelio",
});
console.table(audience.data?.audienceLocations.slice(0, 5));

// 2. Who engages with this specific video? Sample its commenters. (1 credit)
type CommentsRes = {
  data?: { items: Array<{ author: { username: string | null } }> };
};
const comments = await get<CommentsRes>("tiktok/post/comments", {
  url: "https://www.tiktok.com/@charlidamelio/video/7667257618544495886",
});
const usernames = (comments.data?.items ?? [])
  .map((c) => c.author.username)
  .filter((u): u is string => Boolean(u))
  .slice(0, SAMPLE_SIZE);

// 3. Estimate age + gender per commenter from their profile avatar. (10 credits each)
type AgeGenderRes = {
  data?: {
    ageRange: { low: number; high: number };
    gender: string;
    confidence: { gender: number };
  };
};
const sampled = await Promise.all(
  usernames.map(async (username) => {
    const r = await get<AgeGenderRes>("utility/age-gender", {
      url: `https://www.tiktok.com/@${username}`,
    });
    return r.data
      ? { username, age: `${r.data.ageRange.low}-${r.data.ageRange.high}`, gender: r.data.gender }
      : null;
  }),
);
const rows = sampled.filter((r) => r !== null);
console.table(rows);

// 4. Aggregate the sample into a demographic estimate.
const genderShare = rows.reduce<Record<string, number>>((acc, r) => {
  acc[r.gender] = (acc[r.gender] ?? 0) + 1;
  return acc;
}, {});
console.log("Gender mix of sampled commenters:", genderShare);

// Step 1 — audienceLocations (top 5 of ~60 countries):
// ┌─────────┬─────────────────┬─────────────┬───────┬────────────┐
// │ (index) │ country         │ countryCode │ count │ percentage │
// ├─────────┼─────────────────┼─────────────┼───────┼────────────┤
// │    0    │ "United States" │ "US"        │  185  │ "19.49%"   │
// │    1    │ "Mexico"        │ "MX"        │  156  │ "16.44%"   │
// │    2    │ "Pakistan"      │ "PK"        │  70   │ "7.38%"    │
// │    3    │ "Argentina"     │ "AR"        │  51   │ "5.37%"    │
// │    4    │ "Colombia"      │ "CO"        │  47   │ "4.95%"    │
// └─────────┴─────────────────┴─────────────┴───────┴────────────┘

// Step 3 — sampled commenters:
// ┌─────────┬──────────────────┬─────────┬──────────┐
// │ (index) │ username         │ age     │ gender   │
// ├─────────┼──────────────────┼─────────┼──────────┤
// │    0    │ "dandefleurette" │ "23-29" │ "Female" │
// │   ...   │                  │         │          │
// └─────────┴──────────────────┴─────────┴──────────┘