import { stripTrailingSlash } from '../src/string';

import referrers from './referrers';

function getHostname(url: string | undefined) {
  if (!url) {
    return '';
  }

  try {
    return new URL(url).hostname;
  } catch (e) {
    return '';
  }
}

export function parseReferrer(url: string | undefined) {
  const hostname = getHostname(url);
  const match = referrers[hostname] ?? referrers[hostname.replace('www.', '')];

  return {
    name: match?.name ?? '',
    type: match?.type ?? '',
    url: stripTrailingSlash(url ?? ''),
  };
}

export function getReferrerWithQuery(
  query: Record<string, string> | undefined,
) {
  if (!query) {
    return null;
  }

  const source = (
    query.utm_source ??
    query.ref ??
    query.utm_referrer ??
    ''
  ).toLowerCase();

  if (source === '') {
    return null;
  }

  const match =
    referrers[source] ||
    referrers[`${source}.com`] ||
    Object.values(referrers).find(
      (referrer) => referrer.name.toLowerCase() === source,
    );

  if (match) {
    return {
      name: match.name,
      type: match.type,
      url: '',
    };
  }

  return {
    name: source,
    type: '',
    url: '',
  };
}