在Haskell WAI中，如何添加標題作爲中間件？

Question

我想通過編寫一個非常簡單的會話管理器來了解中間件。

我需要在響應中添加SetCookie標頭。我看了一下wai-extra包，發現wai-session。

我正在使用wai-3.0.2，它似乎沒有給我直接訪問響應的類型構造函數，並且我發現在Response(..)上找到模式匹配以添加標題的所有示例。

您能否指點我正確的方向？

Answer 1

編輯：Wai版本3.0.3.0引入了一個輔助函數mapResponseHeaders，與以下示例中的mapHeader相同。這意味着該示例不再需要在Response上進行模式匹配。

import Network.HTTP.Types (ResponseHeaders, Header) 
import Network.Wai (Middleware, Response, mapResponseHeaders) 

withHeader :: Header -> Middleware 
withHeader h app req respond = app req $ respond . addHeader h 

addHeader :: Header -> Response -> Response 
addHeader h = mapResponseHeaders (\hs -> h:hs) 

---

我有什麼工作，並認爲我的理解，但會真的喜歡的反饋和建議。我是Haskell的新手，這是我第一次使用Wai。我最大的絆腳石是沒有意識到應用程序類型在Wai 3.0.0中更改爲延續傳遞樣式。 （該文件指出這非常清楚，我只是錯過了它的第15次我讀了它。）

import Network.HTTP.Types (ResponseHeaders, Header) 
import Network.Wai (Middleware) 
import Network.Wai.Internal (Response(..)) 

withHeader :: Header -> Middleware 
withHeader h app req respond = app req $ respond . addHeader h 

mapHeader :: (ResponseHeaders -> ResponseHeaders) -> Response -> Response 
mapHeader f (ResponseFile s h b1 b2) = ResponseFile s (f h) b1 b2 
mapHeader f (ResponseBuilder s h b) = ResponseBuilder s (f h) b 
mapHeader f (ResponseStream s h b) = ResponseStream s (f h) b 
mapHeader _ r@(ResponseRaw _ _) = r 

addHeader :: Header -> Response -> Response 
addHeader h = mapHeader (\hs -> h:hs) 

我沒有試圖修改標題爲ResponseRaw，因爲我無法弄清楚如何。

我不確定這是否足夠清楚，addHeader部分應用，並且是傳遞給內部應用程序的延續功能。這種形式可能是更清晰的（或醜陋）：

withHeader h app req respond = app req $ \resp -> respond $ addHeader h resp 

我從wai-session複製mapHeader，但添加的情況下ResponseRaw。