umami/lib/filters.js

import firstBy from 'thenby';
import { BROWSERS, ISO_COUNTRIES } from './constants';
import { removeTrailingSlash, removeWWW, getDomainName } from './url';

export const urlFilter = (data, { raw }) => {
  const isValidUrl = url => {
    return url !== '' && !url.startsWith('#');
  };

  if (raw) {
    return data.filter(({ x }) => isValidUrl(x));
  }

  const cleanUrl = url => {
    try {
      const { pathname, search } = new URL(url);

      if (search.startsWith('?/')) {
        return `${pathname}${search}`;
      }

      return removeTrailingSlash(pathname);
    } catch {
      return null;
    }
  };

  const map = data.reduce((obj, { x, y }) => {
    if (!isValidUrl(x)) {
      return obj;
    }

    const url = cleanUrl(`http://x${x}`);

    if (url) {
      if (!obj[url]) {
        obj[url] = y;
      } else {
        obj[url] += y;
      }
    }

    return obj;
  }, {});

  return Object.keys(map)
    .map(key => ({ x: key, y: map[key] }))
    .sort(firstBy('y', -1).thenBy('x'));
};

export const refFilter = (data, { domain, domainOnly, raw }) => {
  const domainName = getDomainName(domain);
  const regex = new RegExp(`http[s]?://${domainName}`);
  const links = {};

  const isValidRef = ref => {
    return ref !== '' && !ref.startsWith('/') && !ref.startsWith('#');
  };

  if (raw) {
    return data.filter(({ x }) => isValidRef(x) && !regex.test(x));
  }

  const cleanUrl = url => {
    try {
      const { hostname, origin, pathname, searchParams, protocol } = new URL(url);

      if (hostname === domainName) {
        return null;
      }

      if (domainOnly && hostname) {
        return hostname;
      }

      if (!origin || origin === 'null') {
        return `${protocol}${removeTrailingSlash(pathname)}`;
      }

      if (protocol.startsWith('http')) {
        const path = removeTrailingSlash(pathname);
        const ref = searchParams.get('ref');
        const query = ref ? `?ref=${ref}` : '';

        return removeTrailingSlash(`${removeWWW(hostname)}${path}`) + query;
      }

      return null;
    } catch {
      return null;
    }
  };

  const map = data.reduce((obj, { x, y }) => {
    if (!isValidRef(x)) {
      return obj;
    }

    const url = cleanUrl(x);

    if (!domainOnly && !raw) {
      links[url] = x;
    }

    if (url) {
      if (!obj[url]) {
        obj[url] = y;
      } else {
        obj[url] += y;
      }
    }

    return obj;
  }, {});

  return Object.keys(map)
    .map(key => ({ x: key, y: map[key], w: links[key] }))
    .sort(firstBy('y', -1).thenBy('x'));
};

export const browserFilter = data =>
  data.map(({ x, y }) => ({ x: BROWSERS[x] || x, y })).filter(({ x }) => x);

export const osFilter = data => data.filter(({ x }) => x);

export const deviceFilter = data => data.filter(({ x }) => x);

export const countryFilter = data =>
  data.map(({ x, y }) => ({ x: ISO_COUNTRIES[x] || x, y })).filter(({ x }) => x);

export const percentFilter = data => {
  const total = data.reduce((n, { y }) => n + y, 0);
  return data.map(({ x, y, ...props }) => ({ x, y, z: total ? (y / total) * 100 : 0, ...props }));
};