haskell - 如何使用 Haskell Snap 框架获取上传的文件？

Question

Snap 框架提供 Snap.Util.FileUploads module ，但这很不友好。我正在编写一个简单的 HTTP 服务器，我并不真正关心文件大小限制、上传速率限制、担心找到一些与操作系统无关的临时目录来放置上传的文件、编写多部分数据流处理器等。

当然，如果我需要的话，Snap 可以为我提供如此详细的信息，这真是太棒了，但我不需要。更糟糕的是，关于如何编写文件上传处理程序的示例为零。我本来期望的是 handleFileUploadsSimple::MonadSnap snap => snap [File]。类似于 Scotty's files action ，你猜对了，它给了我上传的文件，bish-bash-bosh。

如果确实存在，它在哪里？如果没有的话应该怎么写？

Answer 1

这些政策是重要的安全功能；想要它们。

政策

首先您需要制定默认策略。我将坚持使用其默认的速率限制等，但将最大文件大小设置为 2Mb，而不是默认的 128Kb。

maxMb = 2
megaByte = 2^(20::Int)

myDefaultPolicy :: UploadPolicy
myDefaultPolicy = setMaximumFormInputSize (maxMb * megaByte) defaultUploadPolicy 

接下来我们需要一个针对每个部分的策略。您似乎不想检查有关上传的任何内容，我认为这很奇怪，但我们将保留 2Mb 的限制。我们将使用 allowWithMaximumSize::Int64 -> PartUploadPolicy 函数:

myPerPartPolicy :: PartInfo -> PartUploadPolicy
myPerPartPolicy _ = allowWithMaximumSize (maxMb * megaByte)

请记住，这是 Snap 希望您查看 mime 类型等的地方，因为

data PartInfo =
    PartInfo { partFieldName   :: !ByteString
             , partFileName    :: !(Maybe ByteString)
             , partContentType :: !ByteString
             }

这样您就可以定义一个更智能的 myPerPartPolicy 来检查它进入的字段以及它是否在您的 MIME 类型白名单中，而不是忽略该信息。也许您正计划这样做，因为 Scotty 的文件操作在输出中提供了这一点，而不是以这种处理程序的方式。

上传处理程序

您需要实际的文件处理程序。这是对上传进行繁重工作的函数。要求您将其编写为处理程序，而不是仅仅为您提供数据，可以让您以更静态、模块化的方式编写代码，因此我认为它比仅仅抛出数据有一个优势。

myBusinessLogic :: MonadSnap m => PartInfo -> FilePath -> m ()
myBusinessLogic = undefined -- whatever you actually want to do with your uploads

警告:

After the user handler runs (but before the Response body Enumerator is streamed to the client), the files are deleted from disk; so if you want to retain or use the uploaded files in the generated response, you would need to move or otherwise process them.

让我们将其包装成一种忽略错误上传的东西，并在有效的上传上实现您的业务逻辑，一次一个。请注意，我们限制自己不返回任何内容，因此我们可以使用 mapM_ 逐一运行整个过程，但如果您愿意，您可以通过过滤掉您想要的内容来做一些更聪明的事情。不想要列表中的内容，并返回比 () 更有趣的内容。无论如何，我已经使用过滤器来指向该方向:

myUploadHandler :: MonadSnap m => 
                [(PartInfo, Either PolicyViolationException FilePath)] -> m ()
myUploadHandler xs = mapM_ handleOne (filter wanted xs) where
 wanted (_,Left _) = False
 wanted (_,Right _) = True
 handleOne (partInfo,Right filepath) = myBusinessLogic partInfo filepath       

您不需要担心与操作系统无关的临时文件夹，因为如果您导入System.Directory，您可以使用getTemporaryDirectory::IO FilePath来获取一个临时目录。 (如果您愿意，您可以在当前用户空间中使用特定于应用程序的目录，使用 createDirectoryIfMissing::Bool -> FilePath -> IO () 和 getAppUserDataDirectory::String -> IO FilePath。)在任何情况下，您都需要将从那里获得的文件路径传递给您的处理程序。

一起

现在

handleFileUploads :: MonadSnap m => 
     FilePath -> UploadPolicy -> (PartInfo -> PartUploadPolicy)
       -> ([(PartInfo, Either PolicyViolationException FilePath)] -> m a) 
       -> m a

这样你就可以写

myHandleFileUploads :: MonadSnap m => FilePath -> m ()
myHandleFileUploads tempDir = 
   handleFileUploads tempDir myDefaultPolicy myPerPartPolicy myUploadHandler

您仍然需要从 System.Directory 中传递 tempDir，就像我之前提到的那样，但是这个 myHandleFileUploads 已准备好滚动。