import { NextMiddleware } from 'next/server';

const withBfcacheHeaders = (middleware: NextMiddleware): NextMiddleware => {
  return async (req, event) => {
    const response = await middleware(req, event);

    if (process.env.BF_CACHE === 'true' && response) {
      response.headers.set(
        'Cache-Control',
        'private, no-cache, max-age=0, must-revalidate'
      );
    }

    return response;
  };
};

export default withBfcacheHeaders;