BiliNote/BillNote_extension/src/logic/cookies.ts

import { setDownloaderCookie } from './api'
import type { Platform } from './types'

// 后端期望的 cookie 字符串格式：name=value; name=value; ...
// 见 backend/app/downloaders/bilibili_downloader.py 的 split("; ")
const COOKIE_DOMAINS: Record<Exclude<Platform, 'local'>, string> = {
  bilibili: '.bilibili.com',
  youtube: '.youtube.com',
  douyin: '.douyin.com',
  kuaishou: '.kuaishou.com',
}

export const SUPPORTED_COOKIE_PLATFORMS: Array<Exclude<Platform, 'local'>> = [
  'bilibili',
  'douyin',
  'kuaishou',
  'youtube',
]

export async function readBrowserCookies(platform: Exclude<Platform, 'local'>): Promise<string> {
  const domain = COOKIE_DOMAINS[platform]
  const list = await browser.cookies.getAll({ domain })
  return list.map(c => `${c.name}=${c.value}`).join('; ')
}

export async function syncCookieToBackend(platform: Exclude<Platform, 'local'>): Promise<{ ok: boolean, count: number, error?: string }> {
  try {
    const cookieStr = await readBrowserCookies(platform)
    if (!cookieStr)
      return { ok: false, count: 0, error: '当前浏览器没有该域名的 cookie，先在浏览器内登录目标站点' }
    const count = cookieStr.split('; ').length
    await setDownloaderCookie(platform, cookieStr)
    return { ok: true, count }
  }
  catch (e) {
    return { ok: false, count: 0, error: (e as Error).message }
  }
}