import type { Middleware } from "waku/config";

const setCookieMiddleware: Middleware = () => {
  return async (ctx, next) => {
    await next();
    
    // Ensure headers object exists
    ctx.res.headers ||= {};
    
    // Only set the cookie header if we have a cookie to set
    if (ctx.data.cookie) {
      ctx.res.headers["set-cookie"] = ctx.data.cookie as string;
      
      // Debugging
      console.log("Setting cookie header:", ctx.data.cookie);
    }
  };
};

export default setCookieMiddleware;