putStrLn "Ваши задания:"
mapM_ putStrLn numberedTasks
putStrLn "Что вы хотите удалить?"
numberString <– getLine
let number = read numberString
newTodoItems = unlines $ delete (todoTasks !! number) todoTasks
bracketOnError (openTempFile "." "temp")
(\(tempName, tempHandle) –> do
hClose tempHandle
removeFile tempName)
(\(tempName, tempHandle) –> do
hPutStr tempHandle newTodoItems
hClose tempHandle
removeFile "todo.txt"
renameFile tempName "todo.txt")
Вместо обычного использования функции openTempFile
мы заключаем её в bracketOnError
. Затем пишем, что должно произойти при возникновении исключения: мы хотим закрыть и удалить временный файл. Если же всё нормально, пишем новый список заданий во временный файл; все эти строки остались без изменения. Мы выводим новые задания, удаляем исходный файл и переименовываем временный.
Аргументы командной строки
Если вы пишете консольный скрипт или приложение, то вам наверняка понадобится работать с аргументами командной строки. К счастью, в стандартную библиотеку языка Haskell входят удобные функции для работы с ними.
В предыдущей главе мы написали программы для добавления и удаления элемента в список заданий. Но у нашего подхода есть две проблемы. Во-первых, мы жёстко задали имя файла со списком заданий в тексте программы. Мы решили, что файл будет называться
Эту проблему можно решить, спрашивая пользователя каждый раз, какой файл он хочет использовать как файл со списком заданий. Мы использовали такой подход, когда спрашивали пользователя, какой элемент он хочет удалить. Это, конечно, работает, но не идеально, поскольку пользователь должен запустить программу, подождать, пока она спросит что-нибудь, и затем дать ответ. Такая программа называется
Вот почему иногда лучше сделать так, чтобы пользователь сообщал, чего он хочет, при запуске программы, вместо того чтобы она сама спрашивала его после запуска. И что может послужить этой цели лучше командной строки!..
В модуле System.Environment
есть два полезных действия ввода-вывода. Первое – это функция getArgs
; её тип – getArgs :: IO [String]
. Она получает аргументы, с которыми была вызвана программа, и возвращает их в виде списка. Второе – функция getProgName
, тип которой – getProgName :: IO String
. Это действие ввода-вывода, возвращающее имя программы.
Вот простенькая программа, которая показывает, как работают эти два действия:
import System.Environment
import Data.List
main = do
args <– getArgs
progName <– getProgName
putStrLn "Аргументы командной строки:"
mapM putStrLn args
putStrLn "Имя программы:"
putStrLn progName
Мы связываем значения, возвращаемые функциями getArgs
и progName
, с именами args
и progName
. Выводим строку "Аргументы командной строки:"
и затем для каждого аргумента из списка args
выполняем функцию putStrLn
. После этого печатаем имя программы. Скомпилируем программу с именем arg-test
и проверим, как она работает:
$ ./arg-test first second w00t "multi word arg"
Аргументы командной строки:
first
second
w00t
multi word arg
Имя программы:
arg-test
Ещё больше шалостей со списком дел
В предыдущих примерах мы писали отдельные программы для добавления и удаления заданий в списке дел. Теперь мы собираемся объединить их в новое приложение, а что ему делать, будем указывать в командной строке. Кроме того, позаботимся о том, чтобы программа смогла работать с разными файлами – не только
Назовём программу просто todo
, она сможет делать три разные вещи:
• просматривать задания;
• добавлять задания;
• удалять задания.
Для добавления нового задания в список дел в файле
$ ./todo add todo.txt "Найти магический меч силы"