import { Referral } from "@opencode-ai/console-core/referral.js"

const REFERRAL_COOKIE = "oc_referral"
const REFERRAL_MAX_AGE = 60 * 60 * 24 * 30

export function normalizeReferralCode(code?: string | null) {
  return Referral.normalizeCode(code)
}

export function referralCookie(code: string) {
  return `${REFERRAL_COOKIE}=${encodeURIComponent(code)}; Path=/; Max-Age=${REFERRAL_MAX_AGE}; SameSite=Lax; HttpOnly`
}

export function clearReferralCookie() {
  return `${REFERRAL_COOKIE}=; Path=/; Max-Age=0; SameSite=Lax; HttpOnly`
}

export function referralCodeFromCookieHeader(header: string | null) {
  if (!header) return undefined

  return normalizeReferralCode(
    header
      .split(";")
      .map((x) => x.trim())
      .find((x) => x.startsWith(`${REFERRAL_COOKIE}=`))
      ?.slice(`${REFERRAL_COOKIE}=`.length),
  )
}