aboutsummaryrefslogtreecommitdiff
path: root/stdlib/source/lux/world/console.lux
blob: 7bd7cfaca601095dde67f5722cda34a67bf8009b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
(.module:
  [lux #- open]
  (lux (control [monad #+ do])
       (data ["e" error]
             [text])
       (concurrency [promise]
                    [task #+ Task])
       (type (object interface))
       [io #+ IO Process io]
       [host]))

(interface: #export Console
  (read-char [] (Task Text))
  (read-line [] (Task Text))
  (write [Text] (Task Any))
  (close [] (Task Any)))

(for {"JVM"
      (as-is (host.import java/io/InputStream
               (read [] #io #try int)
               (available [] #io #try int)
               (mark [int] #io #try void)
               (reset [] #io #try void))

             (host.import java/io/Reader)

             (host.import java/io/PrintStream
               (print [String] #io #try void))

             (host.import java/lang/System
               (#static in java/io/InputStream)
               (#static out java/io/PrintStream))

             (host.import java/lang/Appendable
               (append [CharSequence] Appendable))

             (host.import java/lang/String)

             (host.import java/lang/StringBuffer
               (new [String])
               (toString [] String))

             (class: JVM-Console Console
               {#input InputStream
                #output PrintStream}

               (def: read-char
                 (|>> get@Console
                      (get@ #input)
                      (InputStream::read [])
                      (:: io.Functor<Process> map (|>> .nat text.from-code))
                      promise.future))
               
               (def: (read-line console)
                 (let [input (|> console get@Console (get@ #input))
                       buffer (StringBuffer::new [""])]
                   (promise.future
                    (loop [_ []]
                      (do io.Monad<Process>
                        [char (<| (:: @ map (|>> .nat text.from-code))
                                  (InputStream::read [] input))]
                        (case char
                          "\n"
                          (wrap (StringBuffer::toString [] buffer))
                          
                          "\r"
                          (do @
                            [available (InputStream::available [] input)]
                            (if (i/> 0 available)
                              (do @
                                [_ (InputStream::mark [10] input)
                                 next (<| (:: @ map (|>> .nat text.from-code))
                                          (InputStream::read [] input))]
                                (case next
                                  "\n"
                                  (wrap (StringBuffer::toString [] buffer))
                                  
                                  _
                                  (do @
                                    [_ (InputStream::reset [] input)]
                                    (wrap (StringBuffer::toString [] buffer)))))
                              (wrap (StringBuffer::toString [] buffer))))
                          
                          _
                          (exec (Appendable::append [(:! String char)] buffer)
                            (recur []))))))))
               
               (def: (write message)
                 (|>> get@Console
                      (get@ #output)
                      (PrintStream::print [message])
                      promise.future))
               
               (def: (close self)
                 (task.return [])))

             (def: #export open
               (Process Console)
               (io (#e.Success (new@JVM-Console {#input System::in
                                                 #output System::out})))))
      })