Refactor components and styles.

lib/db.js

@@ -39,11 +39,12 @@ export async function runQuery(query) {
   });
 }
 
-export async function getWebsite(website_uuid) {
+export async function getWebsite({ website_id, website_uuid }) {
   return runQuery(
     prisma.website.findOne({
       where: {
-        website_uuid,
+        ...(website_id && { website_id }),
+        ...(website_uuid && { website_uuid }),
       },
     }),
   );
@@ -77,11 +78,12 @@ export async function createSession(website_id, data) {
   );
 }
 
-export async function getSession(session_uuid) {
+export async function getSession({ session_id, session_uuid }) {
   return runQuery(
     prisma.session.findOne({
       where: {
-        session_uuid,
+        ...(session_id && { session_id }),
+        ...(session_uuid && { session_uuid }),
       },
     }),
   );

lib/session.js

@@ -18,13 +18,13 @@ export default async req => {
     const country = await getCountry(req, ip);
 
     if (website_uuid) {
-      const website = await getWebsite(website_uuid);
+      const website = await getWebsite({ website_uuid });
 
       if (website) {
         const { website_id } = website;
         const session_uuid = uuid(website_id, hostname, ip, userAgent, os);
 
-        let session = await getSession(session_uuid);
+        let session = await getSession({ session_uuid });
 
         if (!session) {
           session = await createSession(website_id, {