removeFile fileName
renameFile tempName fileName)
Мы открываем файл, полное имя которого задаётся в идентификаторе fileName
, открываем временный файл, удаляем строку по индексу, записываем во временный файл, удаляем исходный файл и переименовываем временный в fileName
. Приведём полный листинг программы во всей её красе:
import System.Environment
import System.Directory
import System.IO
import Control.Exception
import Data.List
dispatch :: String -> [String] -> IO ()
dispatch "add" = add
dispatch "view" = view
dispatch "remove" = remove
main = do
(command:argList) <- getArgs
dispatch command argList
add :: [String] -> IO ()
add [fileName, todoItem] = appendFile fileName (todoItem ++ "\n")
view :: [String] -> IO ()
view [fileName] = do
contents <- readFile fileName
let todoTasks = lines contents
numberedTasks = zipWith (\n line -> show n ++ " – " ++ line)
[0..] todoTasks
putStr $ unlines numberedTasks
remove :: [String] -> IO ()
remove [fileName, numberString] = do
contents <- readFile fileName
let todoTasks = lines contents
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 fileName
renameFile tempName fileName)
Резюмируем наше решение. Мы написали функцию dispatch
, отображающую команды на функции, которые принимают аргументы командной строки в виде списка и возвращают соответствующее действие ввода-вывода. Основываясь на значении первого аргумента, функция dispatch
даёт нам необходимую функцию. В результате вызова этой функции мы получаем требуемое действие и выполняем его.
Давайте проверим, как наша программа работает:
$ ./todo view todo.txt
0 – Погладить посуду
1 – Помыть собаку
2 – Вынуть салат из печи
$ ./todo add todo.txt "Забрать детей из химчистки"
$ ./todo view todo.txt
0 – Погладить посуду
1 – Помыть собаку
2 – Вынуть салат из печи
3 – Забрать детей из химчистки
$ ./todo remove todo.txt 2
$ ./todo view todo.txt
0 – Погладить посуду
1 – Помыть собаку
2 – Забрать детей из химчистки
Большой плюс такого подхода – легко добавлять новую функциональность. Добавить вариант определения функции dispatch
, реализовать соответствующую функцию – и готово! В качестве упражнения можете реализовать функцию bump
, которая примет файл и номер задачи и вернёт действие ввода-вывода, которое поднимет указанную задачу на вершину списка задач.
Работаем с некорректным вводом
Можно было бы дописать эту программу, улучшив сообщения об ошибках, возникающих при некорректных исходных данных. Начать можно с добавления варианта функции dispatch
, который срабатывает при любой несуществующей команде:
dispatch :: String -> [String] -> IO ()
dispatch "add" = add
dispatch "view" = view
dispatch "remove" = remove
dispatch command = doesntExist command
doesntExist :: String -> [String] -> IO ()
doesntExist command _ =
putStrLn $ "Команда " ++ command ++ " не определена"
Также можно добавить варианты определения функций add
, view
и remove
для случаев, когда программе передано неправильное количество аргументов. Например:
add :: [String] -> IO ()