Convert send to TS.

lib/session.ts

@@ -0,0 +1,104 @@
+import cache from 'lib/cache';
+import clickhouse from 'lib/clickhouse';
+import { secret, uuid } from 'lib/crypto';
+import { getClientInfo, getJsonBody } from 'lib/detect';
+import { parseToken } from 'next-basics';
+import { CollectRequestBody, NextApiRequestCollect } from 'pages/api/send';
+import { createSession, getSession, getWebsite } from 'queries';
+import { validate } from 'uuid';
+
+export async function findSession(req: NextApiRequestCollect) {
+  const { payload } = getJsonBody<CollectRequestBody>(req);
+
+  if (!payload) {
+    return null;
+  }
+
+  // Check if cache token is passed
+  const cacheToken = req.headers['x-umami-cache'];
+
+  if (cacheToken) {
+    const result = await parseToken(cacheToken, secret());
+
+    if (result) {
+      return result;
+    }
+  }
+
+  // Verify payload
+  const { website: websiteId, hostname, screen, language } = payload;
+
+  if (!validate(websiteId)) {
+    return null;
+  }
+
+  // Find website
+  let website;
+
+  if (cache.enabled) {
+    website = await cache.fetchWebsite(websiteId);
+  } else {
+    website = await getWebsite({ id: websiteId });
+  }
+
+  if (!website || website.deletedAt) {
+    throw new Error(`Website not found: ${websiteId}`);
+  }
+
+  const { userAgent, browser, os, ip, country, subdivision1, subdivision2, city, device } =
+    await getClientInfo(req, payload);
+  const sessionId = uuid(websiteId, hostname, ip, userAgent);
+
+  // Clickhouse does not require session lookup
+  if (clickhouse.enabled) {
+    return {
+      id: sessionId,
+      websiteId,
+      hostname,
+      browser,
+      os,
+      device,
+      screen,
+      language,
+      country,
+      subdivision1,
+      subdivision2,
+      city,
+    };
+  }
+
+  // Find session
+  let session;
+
+  if (cache.enabled) {
+    session = await cache.fetchSession(sessionId);
+  } else {
+    session = await getSession({ id: sessionId });
+  }
+
+  // Create a session if not found
+  if (!session) {
+    try {
+      session = await createSession({
+        id: sessionId,
+        websiteId,
+        hostname,
+        browser,
+        os,
+        device,
+        screen,
+        language,
+        country,
+        subdivision1,
+        subdivision2,
+        city,
+      });
+    } catch (e: any) {
+      if (!e.message.toLowerCase().includes('unique constraint')) {
+        throw e;
+      }
+    }
+  }
+
+  return session;
+}