使用 getArgs 或用户输入

Question

Haskell 新手，正在努力解决问题。我知道为什么我的代码不起作用，但我无法找出解决方案。

目的是让用户通过参数提供文件名，如果没有提供，则提示用户输入数据。带有文件名的消息将打印到屏幕上并处理文件。

import System.Environment
main = do
    args <- getArgs
    let file = if null args
        then do putStr "File: " ; getLine
        else head args
    putStr "processing"
    putStrLn file
    writeFile file "some very nice text"

当然，代码是错误的，但它展示了我一直在努力解决的逻辑。 Happy Learn Haskell 和 Learn you Haskell 都无法让我克服困难。我能找到的最接近的线程是this one。非常感谢您的帮助。

Answer 1

我们来分析file的类型：

let file = if null args
    then do putStr "File: " ; getLine
    else head args

这里if-then-else 语句的条件为null args，确实是Bool，所以我们是安全的。但是then 部分的类型为IO String。所以 Haskell 的理由是 args 应该是 [IO String]，但它不是：它是 Strings 的列表，所以 [String]。

后来又出现了一个问题：你使用file作为String，但它不是：它仍然是IO String。

有几种方法可以修复它。可能最小的更改是使用pure 将head args 包装回IO，并使用<- 语句替换let 子句：

import System.Environment

main = do
    args <- getArgs
    file <- if null args
        then do putStr "File: " ; getLine
        else pure (head args)
    putStr "processing "
    putStrLn file
    writeFile file "some very nice text"