Ben / Nujel

;;; Nujel - Copyright (C) 2020-2021 - Benjamin Vincent Schulenburg
;;; This project uses the MIT license, a copy should be included under /LICENSE
;;;
;;; This File contains various functions generating ansi escape sequences for colorful output


[def disabled #f]
[defn disable! [v]
      :export
      [set! disabled [if [nil? v] #t v]]]

[export reset [def reset "\e[0m"]]
[export fg-reset [def fg-reset "\e[0;39m"]]
[export bg-reset [def bg-reset "\e[49m"]]

[export ansi-fg [def ansi-fg #[
                               "\e[0;30m"
                               "\e[0;31m"
                               "\e[0;32m"
                               "\e[0;33m"
                               "\e[0;34m"
                               "\e[0;35m"
                               "\e[0;36m"
                               "\e[0;37m"
                               "\e[1;30m"
                               "\e[1;31m"
                               "\e[1;32m"
                               "\e[1;33m"
                               "\e[1;34m"
                               "\e[1;35m"
                               "\e[1;36m"
                               "\e[1;37m"
                               ]]]

[export ansi-bg [def ansi-bg #[
                               "\e[40m"
                               "\e[41m"
                               "\e[42m"
                               "\e[43m"
                               "\e[44m"
                               "\e[45m"
                               "\e[46m"
                               "\e[47m"
                               ]]]


[defn wrap [code string]
      "Wrap STRING in the ansi color CODE"
      :export
      [cat [or disabled [array/ref ansi-fg code]]
           string
           [or disabled reset]]]

[defn black      args :export [wrap  0 [apply cat args]]]
[defn dark-red   args :export [wrap  1 [apply cat args]]]
[defn dark-green args :export [wrap  2 [apply cat args]]]
[defn brown      args :export [wrap  3 [apply cat args]]]
[defn dark-blue  args :export [wrap  4 [apply cat args]]]
[defn purple     args :export [wrap  5 [apply cat args]]]
[defn teal       args :export [wrap  6 [apply cat args]]]
[defn dark-gray  args :export [wrap  7 [apply cat args]]]
[defn gray       args :export [wrap  8 [apply cat args]]]
[defn red        args :export [wrap  9 [apply cat args]]]
[defn green      args :export [wrap 10 [apply cat args]]]
[defn yellow     args :export [wrap 11 [apply cat args]]]
[defn blue       args :export [wrap 12 [apply cat args]]]
[defn pink       args :export [wrap 13 [apply cat args]]]
[defn cyan       args :export [wrap 14 [apply cat args]]]
[defn white      args :export [wrap 15 [apply cat args]]]

[defn rainbow args
      "Wrap ARGS in the colors of the rainbow!"
      :export
      [let* [def count 0]
            [cat [join [map [split [apply cat args] ""]
                            [fn [a]
                                [set! count [bit-and [+ 1 count] #x7]]
                              [cat [or disabled [array/ref ansi-fg [if [zero? count] 7 [+ count 8]]]] a]]] ""]
                 [or disabled fg-reset]]]]

[defn rainbow-bg args
      "Wrap ARGS in the colors of the rainbow!"
      :export
      [def count 0]
      [def colored-list [map [split [apply cat args] ""]
                             [fn [a]
                                 [set! count [bit-and [+ 1 count] #x7]]
                               [cat [or disabled [array/ref ansi-fg [bit-xor count #x7]]] [or disabled [array/ref ansi-bg count]] a]]]]
      [cat [join colored-list ""]
           [or disabled reset]]]

[defn reprint-line [text width]
      :export
      [when-not width [set! width 20]]
      [print "\r"]
      [dotimes [i width]
        [print " "]]
      [print "\r"]
      [print text]]

[deftest "\e[0;33mabc\e[0m" [ansi/wrap 3 "abc"]]
[deftest "\e[0;30m123\e[0m" [ansi/black "123"]]
[deftest "\e[0;31m123\e[0m" [ansi/dark-red "123"]]
[deftest "\e[0;32m123\e[0m" [ansi/dark-green "123"]]
[deftest "\e[0;33m123\e[0m" [ansi/brown "123"]]
[deftest "\e[0;34m123\e[0m" [ansi/dark-blue "123"]]
[deftest "\e[0;35m123\e[0m" [ansi/purple "123"]]
[deftest "\e[0;36m123\e[0m" [ansi/teal "123"]]
[deftest "\e[0;37m123\e[0m" [ansi/dark-gray "123"]]
[deftest "\e[1;30m123\e[0m" [ansi/gray "123"]]
[deftest "\e[1;31m123\e[0m" [ansi/red "123"]]
[deftest "\e[1;32m123\e[0m" [ansi/green "123"]]
[deftest "\e[1;33m123\e[0m" [ansi/yellow "123"]]
[deftest "\e[1;34m123\e[0m" [ansi/blue "123"]]
[deftest "\e[1;35m123\e[0m" [ansi/pink "123"]]
[deftest "\e[1;36m123\e[0m" [ansi/cyan "123"]]
[deftest "\e[1;37m123\e[0m" [ansi/white "123"]]
[deftest "\e[1;31mt\e[1;32me\e[1;33ms\e[1;34mt\e[1;35me\e[1;36mr\e[1;37ml\e[0;37me\e[0;39m" [ansi/rainbow "testerle"]]
[deftest "\e[0;36m\e[41mt\e[0;35m\e[42me\e[0;34m\e[43ms\e[0;33m\e[44mt\e[0;32m\e[45me\e[0;31m\e[46mr\e[0;30m\e[47ml\e[0;37m\e[40me\e[0m" [ansi/rainbow-bg "testerle"]]