FabledCurator/extension/lib/platforms.js

/**
 * Platform definitions — mirrored from backend/app/services/platforms.py.
 * Two copies of these regexes (here and in backend extension_service.py)
 * is cheaper than building a shared definition file for two short
 * tables in different runtimes. Keep in sync by hand; reviewers catch
 * drift.
 */

const PLATFORMS = {
  patreon: {
    name: 'Patreon',
    domains: ['.patreon.com', 'www.patreon.com', 'patreon.com'],
    authType: 'cookies',
    color: '#FF424D',
    urlPattern: /^https?:\/\/(www\.)?patreon\.com/,
    // Patreon's `/api/current_user` returns 200 + the logged-in user
    // when authenticated, 401 otherwise. Cheapest definitive check.
    verify: {
      url: 'https://www.patreon.com/api/current_user',
      method: 'GET',
      okStatuses: [200],
    },
  },
  subscribestar: {
    name: 'SubscribeStar',
    domains: [
      '.subscribestar.com',
      '.subscribestar.adult',
      'www.subscribestar.com',
      'subscribestar.com',
      'subscribestar.adult',
    ],
    authType: 'cookies',
    color: '#FFD700',
    urlPattern: /^https?:\/\/(www\.)?subscribestar\.(com|adult)/,
    // No known stable auth-required endpoint that returns a definitive
    // status code; skipping verify so we don't false-positive-fail
    // good cookies. Operator can add later if a clean endpoint surfaces.
  },
  hentaifoundry: {
    name: 'Hentai Foundry',
    domains: ['.hentai-foundry.com', 'www.hentai-foundry.com', 'hentai-foundry.com'],
    authType: 'cookies',
    color: '#9C27B0',
    urlPattern: /^https?:\/\/(www\.)?hentai-foundry\.com/,
    // Mirror gallery-dl's _init_site_filters: HEAD on `?enterAgree=1`.
    // Logged in → 200, logged out → 401. Catches the exact failure mode
    // the backend extractor would hit later.
    verify: {
      url: 'https://www.hentai-foundry.com/?enterAgree=1',
      method: 'HEAD',
      okStatuses: [200],
    },
  },
  discord: {
    name: 'Discord',
    domains: ['.discord.com', 'discord.com'],
    authType: 'token',
    color: '#5865F2',
    // ptb/canary are Discord's beta clients — same channels, same token.
    urlPattern: /^https?:\/\/((www|ptb|canary)\.)?discord\.com/,
    note: 'Open Discord in browser to capture token',
  },
};

/**
 * Per-platform artist-page URL patterns — these are STRICTER than
 * urlPattern (which matches any URL on the domain). Used by the content
 * script to decide whether to show the floating "Add as source" button.
 */
const PLATFORM_ARTIST_PATTERNS = {
  // Patreon serves the same creator under three URL shapes (see backend
  // patreon_resolver._VANITY_RE): bare `patreon.com/Atole`, `c/` prefix, and
  // `cw/` "creator workspace" — the last is the URL you land on once you're
  // SUBSCRIBED, which is exactly when the button matters. Match all three, and
  // drop the single-segment end-anchor so a creator's inner page
  // (…/cw/Atole/posts, …/Atole/membership) also injects the button. Nav pages
  // (home/search/…/posts permalink) stay excluded. Mirrors extension_service
  // ._PLATFORM_PATTERNS — keep in sync (operator-flagged 2026-07-13: button
  // vanished once subscribed because the old pattern only matched the bare root).
  patreon:       /^https?:\/\/(www\.)?patreon\.com\/(?:cw\/|c\/)?(?!(?:home|search|messages|notifications|library|settings|posts)(?:[\/?#]|$))[^/?#]+/i,
  subscribestar: /^https?:\/\/(www\.)?subscribestar\.(com|adult)\/(?!feed$|messages$|library$)[^/?#]+\/?$/i,
  hentaifoundry: /^https?:\/\/(www\.)?hentai-foundry\.com\/user\/[^/?#]+/i,
  // A Discord URL names a server or channel, not a creator: the backend's slug
  // is `<server>[/<channel>]` and the Add panel asks which artist it belongs
  // to. A message jump link still names its channel; DMs (@me) and thread
  // links don't match. Mirrors extension_service._PLATFORM_PATTERNS.
  discord:       /^https?:\/\/(?:www\.|ptb\.|canary\.)?discord\.com\/channels\/\d+(?:\/\d+)?(?:\/\d+)?\/?(?:[?#].*)?$/i,
};

/**
 * `{serverId, channelId}` from a Discord channel/server URL the artist
 * pattern accepts, else null. channelId is null for a whole-server URL.
 */
function parseDiscordUrl(url) {
  if (!PLATFORM_ARTIST_PATTERNS.discord.test(url || '')) return null;
  const m = /\/channels\/(\d+)(?:\/(\d+))?/.exec(url);
  return m ? { serverId: m[1], channelId: m[2] || null } : null;
}

function getPlatformFromUrl(url) {
  for (const [key, platform] of Object.entries(PLATFORMS)) {
    if (platform.urlPattern.test(url)) return key;
  }
  return null;
}

function isArtistPage(url, platformKey) {
  const pattern = PLATFORM_ARTIST_PATTERNS[platformKey];
  return pattern ? pattern.test(url) : false;
}