advanced-sitemap-generator/src/SitemapRotator.js

Easily create XML sitemaps for your website.

const SitemapStream = require('./SitemapStream');

module.exports = function SitemapRotator(options) {
  const maxEntries = options.maxEntriesPerFile;

  const sitemaps = [];
  let count = 0;
  let current = null;

  // return temp sitemap paths
  const getPaths = () =>
    sitemaps.reduce((arr, map) => {
      arr.push(map.getPath());
      return arr;
    }, []);

  // adds url to stream
  const addURL = (url) => {
    // create stream if none exists
    if (current === null) {
      current = SitemapStream(options);
      sitemaps.push(current);
    }

    // rotate stream
    if (count === maxEntries) {
      current = SitemapStream(options);
      sitemaps.push(current);
      count = 0;
    }
    current.addURL(url);
    count += 1;
  };
  const flush = () => {
    for (const sitemap of sitemaps) {
      sitemap.flush();
    }
  }
  // close stream
  const finish = () => {
    for (const sitemap of sitemaps) {
      sitemap.end();
    }
  };

  return {
    getPaths,
    addURL,
    flush,
    finish,
  };
};