src/services/websiteAuthCookie.ts

Rama main · réplica de lectura del repositorio original.

import { getLocalConfig } from '~/services/console'

const WEBSITE_PAID_USER_COOKIE_NAME = 'capgo_paid_user'
const WEBSITE_PAID_USER_COOKIE_MAX_AGE_SECONDS = 60 * 60 * 24 * 30

interface WebsiteAuthOrganization {
  is_invite?: boolean | null
  paying?: boolean | null
  role?: string | null
}

function isCookieDomainUnsupported(hostname: string) {
  return !hostname || hostname === 'localhost' || /^[\d.]+$/.test(hostname)
}

function hostnameFromUrl(url: string | undefined) {
  if (!url)
    return null

  try {
    return new URL(url).hostname
  }
  catch {
    return null
  }
}

function getWebsiteCookieDomain() {
  const { host, hostWeb } = getLocalConfig()
  const hostname = (
    hostnameFromUrl(hostWeb)
    ?? hostnameFromUrl(host)
    ?? globalThis.location?.hostname
    ?? ''
  )
    .replace(/^www\./, '')
    .replace(/^console\./, '')

  if (isCookieDomainUnsupported(hostname))
    return null

  return `.${hostname}`
}

function getCookieAttributes(maxAgeSeconds: number, domain?: string | null) {
  const attributes = [
    'Path=/',
    `Max-Age=${maxAgeSeconds}`,
    'SameSite=Lax',
  ]

  if (domain)
    attributes.push(`Domain=${domain}`)

  if (globalThis.location.protocol === 'https:')
    attributes.push('Secure')

  return attributes.join('; ')
}

function writeWebsiteCookie(name: string, value: string, maxAgeSeconds: number, domain?: string | null) {
  document.cookie = `${name}=${value}; ${getCookieAttributes(maxAgeSeconds, domain)}`
}

function clearWebsiteCookie(name: string, domain?: string | null) {
  writeWebsiteCookie(name, '', 0, domain)
  if (domain)
    writeWebsiteCookie(name, '', 0)
}

export function clearWebsitePaidUserCookie() {
  if (typeof document === 'undefined')
    return

  const domain = getWebsiteCookieDomain()
  clearWebsiteCookie(WEBSITE_PAID_USER_COOKIE_NAME, domain)
}

export function setWebsitePaidUserCookie(isPaidUser: boolean) {
  if (typeof document === 'undefined')
    return

  if (!isPaidUser) {
    clearWebsitePaidUserCookie()
    return
  }

  const domain = getWebsiteCookieDomain()
  writeWebsiteCookie(WEBSITE_PAID_USER_COOKIE_NAME, '1', WEBSITE_PAID_USER_COOKIE_MAX_AGE_SECONDS, domain)
}

export function syncWebsitePaidUserCookieFromOrganizations(organizations: WebsiteAuthOrganization[]) {
  const hasPaidOrganization = organizations.some((organization) => {
    return organization.is_invite !== true && !!organization.paying
  })

  setWebsitePaidUserCookie(hasPaidOrganization)
}