Kutt/server/queues/visit.js

const useragent = require("useragent");
const geoip = require("geoip-lite");
const URL = require("url");

const { getStatsLimit, removeWww } = require("../utils");
const query = require("../queries");

const browsersList = ["IE", "Firefox", "Chrome", "Opera", "Safari", "Edge"];
const osList = ["Windows", "Mac OS", "Linux", "Android", "iOS"];

function filterInBrowser(agent) {
  return function(item) {
    return agent.family.toLowerCase().includes(item.toLocaleLowerCase());
  }
}

function filterInOs(agent) {
  return function(item) {
    return agent.os.family.toLowerCase().includes(item.toLocaleLowerCase());
  }
}

module.exports = function({ data }) {
  const tasks = [];
  
  tasks.push(query.link.incrementVisit({ id:  data.link.id }));
  
  if (data.link.visit_count < getStatsLimit()) {
    // the following line is for backward compatibility
    // used to send the whole header to get the user agent
    const userAgent = data.userAgent || data.headers?.["user-agent"];
    const agent = useragent.parse(userAgent);
    const [browser = "Other"] = browsersList.filter(filterInBrowser(agent));
    const [os = "Other"] = osList.filter(filterInOs(agent));
    const referrer =
    data.referrer && removeWww(URL.parse(data.referrer).hostname);
    const location = geoip.lookup(data.ip);
    const country = location && location.country;

    
    tasks.push(
      query.visit.add({
        browser: browser.toLowerCase(),
        country: country || "Unknown",
        id: data.link.id,
        os: os.toLowerCase().replace(/\s/gi, ""),
        referrer: (referrer && referrer.replace(/\./gi, "[dot]")) || "Direct"
      })
    );
  }

  return Promise.all(tasks);
}