rodo/util.rkt

156 lines
5.3 KiB
Racket

#lang racket/base
(require (prefix-in list: racket/list)
(prefix-in file: racket/file)
(prefix-in string: racket/string)
(prefix-in system: racket/system)
(prefix-in config: "config.rkt")
(prefix-in messages: "messages.rkt"))
(provide (all-defined-out))
(define (set-permissions permissions file-or-directory)
(let* ([converted-permissions (number->string permissions)]
[converted-file-or-directory
(cond [(path? file-or-directory) (path->string file-or-directory)]
[else file-or-directory])])
(system:system
(string-append "chmod" " " converted-permissions " " converted-file-or-directory))))
(define (check-for-file)
(file-exists? config:path))
(define (create-file)
(let ([opened-file
(open-output-file config:path
#:mode 'text
#:exists 'can-update)])
(close-output-port opened-file))
(set-permissions 600 config:path))
(define (check-for-directory)
(directory-exists? (expand-user-path
(string-append
config:program-directory))))
(define (create-directory)
(make-directory (expand-user-path
(string-append
config:program-directory)))
(set-permissions 700 config:program-directory))
(define (display-hash-ref hash-list key)
(display (hash-ref hash-list key)))
;; Just so I don't have to keep typing "#:mode...#:line-mode..." every time
(define (file->string-list config:path-to-file)
(let ([todo-list (file:file->lines config:path-to-file
#:mode 'text
#:line-mode 'any)])
todo-list))
(define (list-empty? lst)
(list:empty? (file->string-list lst)))
;; Find out which item is being removed by scooping up
;; the number the user entered in the command line
;; argument
(define (get-removed-item lst args)
;; Subtract one from what the user chose, because they are
;; are actually viewing the list numbers as human numbers
;; so (actual-number +1)
(list-ref (file->string-list lst) (sub1 (string->number args))))
(define (surround-in-quotes args)
(display (string-append "\"" args "\"")))
(define (prefix-with-period lst)
(string-append lst ". "))
(define (prefix-with-number lst)
(map string-append
;; Note: compose starts from the last element in it's
;; list, as if it were nested, so that would be add1 here
(map (compose prefix-with-period number->string add1)
(list:range (length lst)))
lst))
(define (display-prettified-list)
(display
(string:string-join
(prefix-with-number (file->string-list config:path))
"\n"
#:after-last "\n")))
;; This is a bit of ugly scheme sorcery
(define (append-to-end args lst)
(reverse (cons args (reverse (file->string-list lst)))))
(define (display-item-added args)
(display-hash-ref messages:messages 'item-added-prefix)
(surround-in-quotes args)
(display-hash-ref messages:messages 'item-added-suffix))
(define (display-item-removed args)
(display-hash-ref messages:messages 'item-removed-prefix)
(surround-in-quotes args)
(display-hash-ref messages:messages 'item-removed-suffix))
(define (show-list)
(cond [(and
(check-for-directory)
(check-for-file))
(if
;; If file exists, see if it's empty, if so
;; tell the user
(list-empty? config:path)
(display-hash-ref messages:messages 'empty-todo-list)
;; If file isn't empty, display a pretty list
(display-prettified-list))]
;; If file doesn't exist, tell the user
[else
(display-hash-ref messages:messages 'file-not-found)
(display-hash-ref messages:messages 'try-init)]))
(define (add-item-to-file args)
;; Add item to end of list and write to file
(let ([new-list (append-to-end args config:path)])
(file:display-to-file
(string:string-join new-list "\n")
config:path
#:mode 'text
#:exists 'replace)
(display-item-added args)))
(define (add-item args)
(if (and
(check-for-directory)
(check-for-file))
(add-item-to-file (vector-ref args 1))
(begin
(display-hash-ref messages:messages 'file-not-found)
(display-hash-ref messages:messages 'try-init))))
(define (remove-item-from-file args)
(let* ([removed-item (get-removed-item config:path args)]
[new-list (remove removed-item (file->string-list config:path))])
(file:display-to-file
(string:string-join new-list "\n")
config:path
#:mode 'text
#:exists 'replace)
(display-item-removed removed-item)))
(define (remove-item args)
(cond [(list-empty? config:path)
(display-hash-ref messages:messages 'empty-todo-list)]
[(and
(check-for-directory)
(check-for-file))
(remove-item-from-file (vector-ref args 1))]
[(and (not (check-for-directory)) (not (check-for-file)))
(begin
(display-hash-ref messages:messages 'file-not-found)
(display-hash-ref messages:messages 'try-init))]))