umami/lib/session.js

import { parseToken } from 'next-basics';
import { validate } from 'uuid';
import { secret, uuid } from 'lib/crypto';
import redis, { DELETED } from 'lib/redis';
import clickhouse from 'lib/clickhouse';
import { getClientInfo, getJsonBody } from 'lib/request';
import { createSession, getSessionByUuid, getWebsiteByUuid } from 'queries';

export async function getSession(req) {
  const { payload } = getJsonBody(req);

  if (!payload) {
    throw new Error('Invalid request');
  }

  const cache = req.headers['x-umami-cache'];

  if (cache) {
    const result = await parseToken(cache, secret());

    if (result) {
      return result;
    }
  }

  const { website: websiteUuid, hostname, screen, language } = payload;

  if (!validate(websiteUuid)) {
    return null;
  }

  let websiteId = null;

  // Check if website exists
  if (redis.enabled) {
    websiteId = Number(await redis.get(`website:${websiteUuid}`));
  }

  // Check database if does not exists in Redis
  if (!websiteId) {
    const website = await getWebsiteByUuid(websiteUuid);
    websiteId = website ? website.websiteId : null;
  }

  if (!websiteId || websiteId === DELETED) {
    throw new Error(`Website not found: ${websiteUuid}`);
  }

  const { userAgent, browser, os, ip, country, device } = await getClientInfo(req, payload);
  const sessionUuid = uuid(websiteId, hostname, ip, userAgent);

  let sessionId = null;
  let session = null;

  if (!clickhouse.enabled) {
    // Check if session exists
    if (redis.enabled) {
      sessionId = Number(await redis.get(`session:${sessionUuid}`));
    }

    // Check database if does not exists in Redis
    if (!sessionId) {
      session = await getSessionByUuid(sessionUuid);
      sessionId = session ? session.sessionId : null;
    }

    if (!sessionId) {
      try {
        session = await createSession(websiteId, {
          sessionUuid,
          hostname,
          browser,
          os,
          screen,
          language,
          country,
          device,
        });
      } catch (e) {
        if (!e.message.toLowerCase().includes('unique constraint')) {
          throw e;
        }
      }
    }
  } else {
    session = {
      sessionId,
      sessionUuid,
      hostname,
      browser,
      os,
      screen,
      language,
      country,
      device,
    };
  }

  return {
    websiteId: websiteId,
    session,
  };
}