From 6bb61d7a53fe11a2ab03161e1b0513792ca3095a Mon Sep 17 00:00:00 2001
From: Brian Cao <brianbcao@gmail.com>
Date: Thu, 6 Oct 2022 13:13:44 -0700
Subject: [PATCH] Revert "remove postgres collect"

This reverts commit 86bd9a7793adb53bc2b0252a53e4339cdc7a4442.
---
 lib/session.js | 61 ++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 47 insertions(+), 14 deletions(-)

diff --git a/lib/session.js b/lib/session.js
index f4498596..d635f75a 100644
--- a/lib/session.js
+++ b/lib/session.js
@@ -1,13 +1,14 @@
+import { parseToken } from 'next-basics';
+import { validate } from 'uuid';
 import { uuid } from 'lib/crypto';
 import redis, { DELETED } from 'lib/redis';
 import { getClientInfo, getJsonBody } from 'lib/request';
-import { parseToken } from 'next-basics';
-import { getWebsiteByUuid } from 'queries';
-import { validate } from 'uuid';
+import { createSession, getSessionByUuid, getWebsiteByUuid } from 'queries';
 
 export async function getSession(req) {
   const { payload } = getJsonBody(req);
   const hasRedis = process.env.REDIS_URL;
+  const hasClickhouse = process.env.CLICKHOUSE_URL;
 
   if (!payload) {
     throw new Error('Invalid request');
@@ -52,17 +53,49 @@ export async function getSession(req) {
   let sessionId = null;
   let session = null;
 
-  session = {
-    session_id: sessionId,
-    session_uuid,
-    hostname,
-    browser,
-    os,
-    screen,
-    language,
-    country,
-    device,
-  };
+  if (!hasClickhouse) {
+    // Check if session exists
+    if (hasRedis) {
+      sessionId = Number(await redis.client.get(`session:${session_uuid}`));
+    }
+
+    // Check database if does not exists in Redis
+    if (!sessionId) {
+      session = await getSessionByUuid(session_uuid);
+      sessionId = session ? session.session_id : null;
+    }
+
+    if (!sessionId) {
+      try {
+        session = await createSession(websiteId, {
+          session_uuid,
+          hostname,
+          browser,
+          os,
+          screen,
+          language,
+          country,
+          device,
+        });
+      } catch (e) {
+        if (!e.message.toLowerCase().includes('unique constraint')) {
+          throw e;
+        }
+      }
+    }
+  } else {
+    session = {
+      session_id: sessionId,
+      session_uuid,
+      hostname,
+      browser,
+      os,
+      screen,
+      language,
+      country,
+      device,
+    };
+  }
 
   return {
     website_id: websiteId,