aboutsummaryrefslogtreecommitdiff
path: root/stdlib/source/library/lux/control/parser/code.lux
blob: 93a2f65d9b89d00f85c6e3cf7cde088ea139b16d (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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
(.module:
  [library
   [lux (#- nat int rev local)
    [abstract
     ["." monad (#+ do)]]
    [control
     ["." try (#+ Try)]]
    [data
     ["." bit]
     ["." text ("#\." monoid)]
     ["." name]
     [collection
      ["." list ("#\." functor)]]]
    [macro
     ["." code ("#\." equivalence)]]
    [math
     [number
      ["." nat]
      ["." int]
      ["." rev]
      ["." frac]]]]]
  ["." //])

(def: (un_paired pairs)
  (All [a] (-> (List [a a]) (List a)))
  (case pairs
    #.End                   #.End
    (#.Item [[x y] pairs']) (list& x y (un_paired pairs'))))

(type: .public Parser
  {#.doc "A Lux code parser."}
  (//.Parser (List Code)))

(def: remaining_inputs
  (-> (List Code) Text)
  (|>> (list\map code.format)
       (text.interposed " ")
       ($_ text\compose text.new_line "Remaining input: ")))

(def: .public any
  {#.doc "Yields the next input without applying any logic."}
  (Parser Code)
  (function (_ tokens)
    (case tokens
      #.End
      (#try.Failure "There are no tokens to parse!")
      
      (#.Item [t tokens'])
      (#try.Success [tokens' t]))))

(template [<query> <check> <type> <tag> <eq> <desc>]
  [(with_expansions [<failure> (as_is (#try.Failure ($_ text\compose "Cannot parse " <desc> (remaining_inputs tokens))))]
     (def: .public <query>
       {#.doc (code.text ($_ text\compose "Parses the next " <desc> " input."))}
       (Parser <type>)
       (function (_ tokens)
         (case tokens
           (#.Item [[_ (<tag> x)] tokens'])
           (#try.Success [tokens' x])

           _
           <failure>)))

     (def: .public (<check> expected)
       {#.doc (code.text ($_ text\compose "Checks for a specific " <desc> " input."))}
       (-> <type> (Parser Any))
       (function (_ tokens)
         (case tokens
           (#.Item [[_ (<tag> actual)] tokens'])
           (if (\ <eq> = expected actual)
             (#try.Success [tokens' []])
             <failure>)

           _
           <failure>))))]

  [bit        bit!        Bit  #.Bit        bit.equivalence  "bit"]
  [nat        nat!        Nat  #.Nat        nat.equivalence  "nat"]
  [int        int!        Int  #.Int        int.equivalence  "int"]
  [rev        rev!        Rev  #.Rev        rev.equivalence  "rev"]
  [frac       frac!       Frac #.Frac       frac.equivalence "frac"]
  [text       text!       Text #.Text       text.equivalence "text"]
  [identifier identifier! Name #.Identifier name.equivalence "identifier"]
  [tag        tag!        Name #.Tag        name.equivalence "tag"]
  )

(def: .public (this! code)
  {#.doc "Ensures the given Code is the next input."}
  (-> Code (Parser Any))
  (function (_ tokens)
    (case tokens
      (#.Item [token tokens'])
      (if (code\= code token)
        (#try.Success [tokens' []])
        (#try.Failure ($_ text\compose "Expected a " (code.format code) " but instead got " (code.format token)
                          (remaining_inputs tokens))))

      _
      (#try.Failure "There are no tokens to parse!"))))

(template [<query> <check> <tag> <eq> <desc>]
  [(with_expansions [<failure> (as_is (#try.Failure ($_ text\compose "Cannot parse " <desc> (remaining_inputs tokens))))]
     (def: .public <query>
       {#.doc (code.text ($_ text\compose "Parse a local " <desc> " (a " <desc> " that has no module prefix)."))}
       (Parser Text)
       (function (_ tokens)
         (case tokens
           (#.Item [[_ (<tag> ["" x])] tokens'])
           (#try.Success [tokens' x])

           _
           <failure>)))

     (def: .public (<check> expected)
       {#.doc (code.text ($_ text\compose "Checks for a specific local " <desc> " (a " <desc> " that has no module prefix)."))}
       (-> Text (Parser Any))
       (function (_ tokens)
         (case tokens
           (#.Item [[_ (<tag> ["" actual])] tokens'])
           (if (\ <eq> = expected actual)
             (#try.Success [tokens' []])
             <failure>)

           _
           <failure>))))]

  [local_identifier local_identifier! #.Identifier text.equivalence "local identifier"]
  [       local_tag        local_tag! #.Tag        text.equivalence "local tag"]
  )

(template [<name> <tag> <desc>]
  [(def: .public (<name> p)
     {#.doc (code.text ($_ text\compose "Parses the contents of a " <desc> "."))}
     (All [a]
       (-> (Parser a) (Parser a)))
     (function (_ tokens)
       (case tokens
         (#.Item [[_ (<tag> members)] tokens'])
         (case (p members)
           (#try.Success [#.End x]) (#try.Success [tokens' x])
           _                          (#try.Failure ($_ text\compose "Parser was expected to fully consume " <desc> (remaining_inputs tokens))))

         _
         (#try.Failure ($_ text\compose "Cannot parse " <desc> (remaining_inputs tokens))))))]

  [ form  #.Form "form"]
  [tuple #.Tuple "tuple"]
  )

(def: .public (record p)
  {#.doc "Parses the contents of a record."}
  (All [a]
    (-> (Parser a) (Parser a)))
  (function (_ tokens)
    (case tokens
      (#.Item [[_ (#.Record pairs)] tokens'])
      (case (p (un_paired pairs))
        (#try.Success [#.End x]) (#try.Success [tokens' x])
        _                          (#try.Failure ($_ text\compose "Parser was expected to fully consume record" (remaining_inputs tokens))))

      _
      (#try.Failure ($_ text\compose "Cannot parse record" (remaining_inputs tokens))))))

(def: .public end!
  {#.doc "Verifies there are no more inputs."}
  (Parser Any)
  (function (_ tokens)
    (case tokens
      #.End (#try.Success [tokens []])
      _     (#try.Failure ($_ text\compose "Expected list of tokens to be empty!" (remaining_inputs tokens))))))

(def: .public end?
  {#.doc "Checks whether there are no more inputs."}
  (Parser Bit)
  (function (_ tokens)
    (#try.Success [tokens (case tokens
                            #.End true
                            _     false)])))

(def: .public (result parser inputs)
  {#.doc (example "Executes a parser against a stream of code, and verifies all the inputs are consumed.")}
  (All [a] (-> (Parser a) (List Code) (Try a)))
  (case (parser inputs)
    (#try.Failure error)
    (#try.Failure error)

    (#try.Success [unconsumed value])
    (case unconsumed
      #.End
      (#try.Success value)

      _
      (#try.Failure (|> unconsumed
                        (list\map code.format)
                        (text.interposed ", ")
                        (text\compose "Unconsumed inputs: "))))))

(def: .public (local inputs parser)
  {#.doc "Runs parser against the given list of inputs."}
  (All [a] (-> (List Code) (Parser a) (Parser a)))
  (function (_ real)
    (do try.monad
      [value (..result parser inputs)]
      (in [real value]))))