Миран Липовача - Изучай Haskell во имя добра!

Сохраните программу в файле с именем deletetodo.hs , скомпилируйте её и проверьте:

$ ./deletetodo

Ваши задания:

0 – Погладить посуду

1 – Помыть собаку

2 – Вынуть салат из печи

Что вы хотите удалить?

1

Смотрим, что осталось:

$ cat todo.txt

Погладить посуду

Вынуть салат из печи

Круто! Удалим ещё что-нибудь:

$ ./deletetodo

Ваши задания:

0 – Погладить посуду

1 – Вынуть салат из печи

Что вы хотите удалить?

0

Проверяя файл с заданиями, убеждаемся, что осталось только одно:

$ cat todo.txt

Вынуть салат из печи

Итак, всё работает. Осталась только одна вещь, которую мы в этой программе не учли. Если после открытия временного файла что-то произойдёт и программа неожиданно завершится, то временный файл не будет удалён. Давайте это исправим.

Чтобы гарантировать удаление временного файла, воспользуемся функцией bracketOnErrorиз модуля Control.Exception. Она очень похожа на bracket, но если последняя получает ресурс и гарантирует, что освобождение ресурса будет выполнено всегда, то функция bracketOnErrorвыполнит завершающие действия только в случае возникновения исключения. Вот исправленный код:

import System.IO

import System.Directory

import Data.List

import Control.Exception

main = do

contents <���– readFile "todo.txt"

let todoTasks = lines contents

numberedTasks = zipWith (\n line –> show n ++ " – " ++ line)

[0..] todoTasks

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 входят удобные функции для работы с ними.

В предыдущей главе мы написали программы для добавления и удаления элемента в список заданий. Но у нашего подхода есть две проблемы. Во-первых, мы жёстко задали имя файла со списком заданий в тексте программы. Мы решили, что файл будет называться todo.txt , и что пользователь никогда не захочет вести несколько списков.

Эту проблему можно решить, спрашивая пользователя каждый раз, какой файл он хочет использовать как файл со списком заданий. Мы использовали такой подход, когда спрашивали пользователя, какой элемент он хочет удалить. Это, конечно, работает, но не идеально, поскольку пользователь должен запустить программу, подождать, пока она спросит что-нибудь, и затем дать ответ. Такая программа называется интерактивной , и сложность здесь заключается вот в чём: вдруг вам понадобится автоматизировать выполнение этой программы, например, с помощью скрипта? Гораздо сложнее написать скрипт, который будет взаимодействовать с программой, чем обычный скрипт, который просто вызовет её один или несколько раз!

Вот почему иногда лучше сделать так, чтобы пользователь сообщал, чего он хочет, при запуске программы, вместо того чтобы она сама спрашивала его после запуска. И что может послужить этой цели лучше командной строки!..

В модуле 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.txt .

Назовём программу просто todo, она сможет делать три разные вещи:

• просматривать задания;

• добавлять задания;

• удалять задания.

Для добавления нового задания в список дел в файле todo.txt мы будем писать:

$ ./todo add todo.txt "Найти магический меч силы"

Просмотреть текущие задания можно будет командой view:

$ ./todo view todo.txt

Для удаления задания потребуется дополнительно указать его индекс:

$ ./todo remove todo.txt 2

Начнём с реализации функции, которая принимает команду в виде строки (например, "add"или "view") и возвращает функцию, которая в свою очередь принимает список аргументов и возвращает действие ввода-вывода, выполняющее в точности то, что необходимо:

import System.Environment

import System.Directory

import System.IO

import Data.List

import Control.Exception

dispatch :: String -> [String] –> IO ()

dispatch "add" = add

dispatch "view" = view

dispatch "remove" = remove

Функция mainбудет выглядеть так:

main = do