Haskellでプロンプトを表示させつつ標準入力を行いたい

Question

Haskellでプロンプトを表示しつつ標準入力を受け付ける方法を探しています。

質問の背景

「すごいHaskellたのしく学ぼう!」という書籍を参考に、Haskellを勉強しています。

関数の定義や四則演算といった部分がなんとか理解できましたので、標準入出力――IOモナドの分野に着手し始めています。

色々調べてputStrやprintを使って標準出力すること。また、getLineやreadLnを用いて標準入力を受け付けることはできるようになりました。

ただ、そこから一歩進んで、プロンプトを表示させつつ標準入力を受け付けるプログラムを組もうとしたところで躓いています。

本題

作りたいプログラムは以下のようなものです。

#include <stdio.h>
#include <stdbool.h>

int main(void) {
    printf("処理開始\n");
    while (true) {
        printf("[1] グー    [2] チョキ    [3] パー : ");
        int input = 0;
        scanf("%d", &input);
        switch (input) {
            case 1:
                printf("グー");
                return 0;
            case 2:
                printf("チョキ");
                return 0;
            case 3:
                printf("パー");
                return 0;
            default:
                printf("※再入力してください\n");
        }
    }
}

上記コードはCですが、これと同じ挙動をするようにHaskellで書きたいのです。

現状は以下の通りです。

loop :: IO()
loop = do
    putStr "[1] グー    [2] チョキ    [3] パー : " -- ここが問題の部分
    x <- getLine
    case x of
        "1" -> putStrLn "グー"
        "2" -> putStrLn "チョキ"
        "3" -> putStrLn "パー"
        x -> do
            putStrLn "※再入力してください"
            loop

main :: IO()
main = do
    putStrLn "処理開始"
    loop

基本的にはほぼ同じ挙動を再現するところまで来たのですが、ソースコード3行目のところで問題を起こしています。

私の予想では以下のように動くはずでした。

-- こう動いて欲しかった
処理開始
[1] グー    [2] チョキ    [3] パー : ※ここに入力できるようになる

それが実際に動かしてみると以下のようにプロンプトが表示さないでいます。

-- 何故かプロンプトが表示されない
処理開始
※プロンプトが表示されず、ここに入力できるようになっている

そして何らかの値を入力するとプロンプト部分の末尾に入力した値が表示されてしまいます。

-- 何故かプログラムの末尾に出力される
処理開始
1
[1] グー    [2] チョキ    [3] パー : グー

質問

どうしてこのような挙動になるのでしょうか？

アクションかIOモナドあたりの理解が足りていないのでしょうか？

長くなりましたが、ご回答よろしくお願いします。

Answer 1

バッファリングされているので手動でフラッシュする必要があります。

System.IO.hFlush を使います。

import System.IO (hFlush, stdout)

loop :: IO ()
loop = do
    putStr "[1] グー    [2] チョキ    [3] パー : " -- ここが問題の部分
    hFlush stdout -- フラッシュする
    x <- getLine
    case x of
        "1" -> putStrLn "グー"
        "2" -> putStrLn "チョキ"
        "3" -> putStrLn "パー"
        x -> do
            putStrLn "※再入力してください"
            loop

main :: IO ()
main = do
    putStrLn "処理開始"
    loop

Comments

[AGadget]

ご回答ありがとうございます！

早速試してみたところ、期待した通りの結果を得ることができました！

> バッファリングされているので手動でフラッシュする必要があります。

恥ずかしながら「バッファリング」も「フラッシュ」も知らなかったので調べてきました。

今回の問題は「putStr関数の引数で指定された文字列がバッファ上に残ったままであり、getLine関数の処理が終わったタイミングでフラッシュされていた」ことが原因ということですね。

大変勉強になりました。

ありがとうございました！