overleaf/services/web/app/src/Features/Analytics/AnalyticsRegistrationSourceMiddleware.mjs

import logger from '@overleaf/logger'
import OError from '@overleaf/o-error'
import AnalyticsRegistrationSourceHelper from './AnalyticsRegistrationSourceHelper.mjs'
import SessionManager from '../../Features/Authentication/SessionManager.mjs'

function setSource(medium, source) {
  return function (req, res, next) {
    if (req.session) {
      req.session.required_login_from_product_medium = medium
      if (source) {
        req.session.required_login_from_product_source = source
      }
    }
    next()
  }
}

function clearSource() {
  return function (req, res, next) {
    AnalyticsRegistrationSourceHelper.clearSource(req.session)
    next()
  }
}

function setInbound() {
  return function setInbound(req, res, next) {
    if (req.session.inbound) {
      return next() // don't overwrite referrer
    }

    if (SessionManager.isUserLoggedIn(req.session)) {
      return next() // don't store referrer if user is already logged in
    }

    const referrer = req.header('referrer')
    try {
      AnalyticsRegistrationSourceHelper.setInbound(
        req.session,
        req.url,
        req.query,
        referrer
      )
    } catch (error) {
      // log errors and fail silently
      OError.tag(error, 'failed to parse inbound referrer', {
        referrer,
      })
      logger.warn({ error }, error.message)
    }
    next()
  }
}

export default {
  setSource,
  clearSource,
  setInbound,
}