Initial commit

Created from https://vercel.com/new

pages/api/collect.js

@@ -0,0 +1,60 @@
+import isbot from 'isbot';
+import ipaddr from 'ipaddr.js';
+import { savePageView, saveEvent } from 'lib/queries';
+import { useCors, useSession } from 'lib/middleware';
+import { getIpAddress } from 'lib/request';
+import { ok, badRequest } from 'lib/response';
+import { createToken } from 'lib/crypto';
+
+export default async (req, res) => {
+  await useCors(req, res);
+
+  if (isbot(req.headers['user-agent'])) {
+    return ok(res);
+  }
+
+  if (process.env.IGNORE_IP) {
+    const ips = process.env.IGNORE_IP.split(',').map(n => n.trim());
+    const ip = getIpAddress(req);
+    const blocked = ips.find(i => {
+      if (i === ip) return true;
+
+      // CIDR notation
+      if (i.indexOf('/') > 0) {
+        const addr = ipaddr.parse(ip);
+        const range = ipaddr.parseCIDR(i);
+
+        if (addr.kind() === range[0].kind() && addr.match(range)) return true;
+      }
+
+      return false;
+    });
+
+    if (blocked) {
+      return ok(res);
+    }
+  }
+
+  await useSession(req, res);
+
+  const { type, payload } = req.body;
+  const {
+    session: { website_id, session_id },
+  } = req;
+
+  if (type === 'pageview') {
+    const { url, referrer } = payload;
+
+    await savePageView(website_id, session_id, url, referrer);
+  } else if (type === 'event') {
+    const { url, event_type, event_value } = payload;
+
+    await saveEvent(website_id, session_id, url, event_type, event_value);
+  } else {
+    return badRequest(res);
+  }
+
+  const token = await createToken({ website_id, session_id });
+
+  return ok(res, token);
+};