Fixed missing docker middleware.

This commit is contained in:
Mike Cao 2023-08-30 23:53:05 -07:00
parent 620011a837
commit 68ffa823f8
2 changed files with 63 additions and 63 deletions

View File

@ -12,8 +12,8 @@ RUN yarn install --frozen-lockfile
FROM node:18-alpine AS builder FROM node:18-alpine AS builder
WORKDIR /app WORKDIR /app
COPY --from=deps /app/node_modules ./node_modules COPY --from=deps /app/node_modules ./node_modules
COPY docker/middleware.js .
COPY . . COPY . .
COPY docker/middleware.js ./src
ARG DATABASE_TYPE ARG DATABASE_TYPE
ARG BASE_PATH ARG BASE_PATH

View File

@ -7,7 +7,7 @@ import { getIpAddress } from 'lib/detect';
import { useCors, useSession, useValidate } from 'lib/middleware'; import { useCors, useSession, useValidate } from 'lib/middleware';
import { CollectionType, YupRequest } from 'lib/types'; import { CollectionType, YupRequest } from 'lib/types';
import { NextApiRequest, NextApiResponse } from 'next'; import { NextApiRequest, NextApiResponse } from 'next';
import { badRequest, createToken, forbidden, ok, send } from 'next-basics'; import { badRequest, createToken, forbidden, methodNotAllowed, ok, send } from 'next-basics';
import { saveEvent, saveSessionData } from 'queries'; import { saveEvent, saveSessionData } from 'queries';
import * as yup from 'yup'; import * as yup from 'yup';
@ -73,75 +73,75 @@ const schema = {
export default async (req: NextApiRequestCollect, res: NextApiResponse) => { export default async (req: NextApiRequestCollect, res: NextApiResponse) => {
await useCors(req, res); await useCors(req, res);
if (isbot(req.headers['user-agent']) && !process.env.DISABLE_BOT_CHECK) { if (req.method === 'POST') {
return ok(res); if (isbot(req.headers['user-agent']) && !process.env.DISABLE_BOT_CHECK) {
} return ok(res);
const { type, payload } = req.body;
if (!type || !payload) {
return badRequest(res);
}
req.yup = schema;
await useValidate(req, res);
if (await hasBlockedIp(req)) {
return forbidden(res);
}
const { url, referrer, name: eventName, data: eventData, title: pageTitle } = payload;
await useSession(req, res);
const session = req.session;
if (type === COLLECTION_TYPE.event) {
// eslint-disable-next-line prefer-const
let [urlPath, urlQuery] = url?.split('?') || [];
let [referrerPath, referrerQuery] = referrer?.split('?') || [];
let referrerDomain;
if (!urlPath) {
urlPath = '/';
} }
if (referrerPath?.startsWith('http')) { const { type, payload } = req.body;
const refUrl = new URL(referrer);
referrerPath = refUrl.pathname; req.yup = schema;
referrerQuery = refUrl.search.substring(1); await useValidate(req, res);
referrerDomain = refUrl.hostname.replace(/www\./, '');
if (await hasBlockedIp(req)) {
return forbidden(res);
} }
if (process.env.REMOVE_TRAILING_SLASH) { const { url, referrer, name: eventName, data: eventData, title: pageTitle } = payload;
urlPath = urlPath.replace(/.+\/$/, '');
await useSession(req, res);
const session = req.session;
if (type === COLLECTION_TYPE.event) {
// eslint-disable-next-line prefer-const
let [urlPath, urlQuery] = url?.split('?') || [];
let [referrerPath, referrerQuery] = referrer?.split('?') || [];
let referrerDomain;
if (!urlPath) {
urlPath = '/';
}
if (referrerPath?.startsWith('http')) {
const refUrl = new URL(referrer);
referrerPath = refUrl.pathname;
referrerQuery = refUrl.search.substring(1);
referrerDomain = refUrl.hostname.replace(/www\./, '');
}
if (process.env.REMOVE_TRAILING_SLASH) {
urlPath = urlPath.replace(/.+\/$/, '');
}
await saveEvent({
urlPath,
urlQuery,
referrerPath,
referrerQuery,
referrerDomain,
pageTitle,
eventName,
eventData,
...session,
sessionId: session.id,
});
} }
await saveEvent({ if (type === COLLECTION_TYPE.identify) {
urlPath, if (!eventData) {
urlQuery, return badRequest(res, 'Data required.');
referrerPath, }
referrerQuery,
referrerDomain, await saveSessionData({ ...session, sessionData: eventData, sessionId: session.id });
pageTitle, }
eventName,
eventData, const token = createToken(session, secret());
...session,
sessionId: session.id, return send(res, token);
});
} }
if (type === COLLECTION_TYPE.identify) { return methodNotAllowed(res);
if (!eventData) {
return badRequest(res, 'Data required.');
}
await saveSessionData({ ...session, sessionData: eventData, sessionId: session.id });
}
const token = createToken(session, secret());
return send(res, token);
}; };
async function hasBlockedIp(req: NextApiRequestCollect) { async function hasBlockedIp(req: NextApiRequestCollect) {