SukkaSurge/Build/lib/domain-deduper.ts

import { createTrie } from './trie';

export function domainDeduper(inputDomains: string[], toArray?: true): string[];
export function domainDeduper(inputDomains: string[], toArray: false): Set<string>;
export function domainDeduper(inputDomains: string[], toArray = true): string[] | Set<string> {
  const trie = createTrie(inputDomains, true);
  const sets = new Set(inputDomains);

  for (let i = 0, len1 = inputDomains.length; i < len1; i++) {
    const d = inputDomains[i];
    if (d[0] !== '.') {
      continue;
    }

    trie.substractSetInPlaceFromFound(d, sets);
    sets.delete(d.slice(1));
  }

  if (toArray) {
    return Array.from(sets);
  }

  return sets;
}