summaryrefslogtreecommitdiff
path: root/vendor/bundle/ruby/3.4.0/gems/rouge-4.7.0/lib/rouge/lexers/nial.rb
blob: 42b22602d2c812fd4abe6bba71a5150d681af71b (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
# -*- coding: utf-8 -*- #
# frozen_string_literal: true

module Rouge
  module Lexers
    class Nial < RegexLexer
      title 'Nial'
      desc 'The Nial programming language (nial-array-language.org)'
      tag 'nial'
      filenames '*.ndf', '*.nlg'
      
      def self.keywords
        @keywords ||= Set.new ["is", "gets", "op", "tr", ";",
                     "if", "then", "elseif", "else",
                     "endif", "case", "from", "endcase",
                     "begin", "end", "for", "with",
                     "endfor", "while", "do", "endwhile",
                     "repeat", "until", "endrepeat"]
      end

      def self.operators
        @operators||= Set.new [".",  "!", "#", "+", "*", "-", "<<",
                 "/", "<", ">>", "<=", ">", "=", ">=", "@", "|", "~="]
      end

      def self.punctuations
        @punctuations ||= Set.new [ "{", "}", "[", "]", ",", "(", ")", ":=", ":", ";"]
      end

      def self.transformers
        @transformers ||= Set.new ["accumulate", "across",
        "bycols", "bykey",  "byrows",
        "converse",  "down",
        "eachboth", "eachall", "each",
        "eachleft", "eachright",
        "filter", "fold", "fork",
        "grade",  "inner", "iterate",
        "leaf",  "no_tr", "outer",
        "partition", "rank", "recur",
        "reduce", "reducecols", "reducerows",
        "sort", "team", "timeit", "twig"]
      end
      
      def self.funcs
        @funcs ||= Set.new ["operation", "expression", "and", "abs",
                  "allbools", "allints", "allchars", "allin",
                  "allreals", "allnumeric", "append",
                  "arcsin", "arccos", "appendfile", "apply",
                  "arctan", "atomic", "assign", "atversion",
                  "axes", "cart", "break", "blend", "breaklist",
                  "breakin", "bye", "callstack", "choose", "char",
                  "ceiling", "catenate", "charrep", "check_socket",
                  "cos", "content", "close", "clearws",
                  "clearprofile", "cols", "continue", "copyright",
                  "cosh", "cull", "count", "diverse", "deepplace",
                  "cutall", "cut", "display", "deparse",
                  "deepupdate", "descan", "depth", "diagram",
                  "div", "divide", "drop", "dropright", "edit",
                  "empty", "expression", "exit", "except", "erase",
                  "equal", "eval", "eraserecord", "execute", "exp",
                  "external", "exprs", "findall", "find", 
                  "fault", "falsehood", "filestatus", "filelength",
                  "filepath", "filetally", "floor", "first",
                  "flip", "fuse", "fromraw", "front",
                  "gage", "getfile", "getdef", "getcommandline",
                  "getenv", "getname", "hitch", "grid", "getsyms",
                  "gradeup", "gt", "gte", "host", "in", "inverse",
                  "innerproduct", "inv", "ip", "ln", "link", "isboolean",
                  "isinteger", "ischar", "isfault", "isreal", "isphrase",
                  "isstring", "istruthvalue", "last", "laminate",
                  "like", "libpath", "library", "list", "load",
                  "loaddefs", "nonlocal", "max", "match", "log",
                  "lt", "lower", "lte", "mate", "min", "maxlength",
                  "mod", "mix", "minus", "nialroot", "mold", "not",
                  "numeric", "no_op", "no_expr", "notin",
                  "operation", "open", "or", "opposite", "opp",
                  "operators", "plus", "pick", "pack", "pass", "pair", "parse",
                  "paste", "phrase", "place", "picture", "placeall",
                  "power", "positions", "post", "quotient", "putfile",
                  "profile", "prod", "product", "profiletree",
                  "profiletable", "quiet_fault", "raise", "reach",
                  "random", "reciprocal", "read", "readfile",
                  "readchar", "readarray", "readfield",
                  "readscreen", "readrecord", "recip", "reshape",
                  "seek", "second", "rest", "reverse", "restart",
                  "return_status", "scan", "save", "rows", "rotate",
                  "seed", "see", "sublist", "sin", "simple", "shape",
                  "setformat", "setdeftrace", "set", "seeusercalls",
                  "seeprimcalls", "separator", "setwidth", "settrigger",
                  "setmessages", "setlogname", "setinterrupts",
                  "setprompt", "setprofile", "sinh", "single",
                  "sqrt", "solitary", "sketch", "sleep",
                  "socket_listen", "socket_accept", "socket_close",
                  "socket_bind", "socket_connect", "socket_getline",
                  "socket_receive", "socket_peek", "socket_read",
                  "socket_send", "socket_write", "solve", "split",
                  "sortup", "string", "status", "take", "symbols",
                  "sum", "system", "tan", "tally", "takeright",
                  "tanh", "tell", "tr", "times", "third", "time",
                  "toupper", "tolower", "timestamp", "tonumber",
                  "toraw", "toplevel", "transformer", "type",
                  "transpose", "trs", "truth", "unequal",
                  "variable", "valence", "up", "updateall",
                  "update", "vacate", "value", "version", "vars",
                  "void", "watch", "watchlist", "write", "writechars",
                  "writearray", "writefile", "writefield",
                  "writescreen", "writerecord"]
      end

      def self.consts
        @consts ||= Set.new %w(false null pi true)
      end

      state :root do
        rule %r/'/, Str::Single, :str
        rule %r/\b[lo]+\b/, Num::Bin
        rule %r/-?\d+((\.\d*)?[eE][+-]?\d|\.)\d*/, Num::Float
        rule %r/\-?\d+/, Num::Integer
        rule %r/`./, Str::Char
        rule %r/"[^\s()\[\]{}#,;]*/, Str::Symbol
        rule %r/\?[^\s()\[\]{}#,;]*/, Generic::Error
        rule %r/%[^;]+;/, Comment::Multiline
        rule %r/^#(.+\n)+\n/, Comment::Multiline
        rule %r/:=|[\{\}\[\]\(\),:;]/ do |m|
          if self.class.punctuations.include?(m[0])
            token Punctuation
          else
            token Text
          end
        end
        # [".",  "!", "#", "+", "*", "-", "<<",
        #   "/", "<", ">>", "<=", ">", "=", ">=", "@", "|", "~="]
        rule %r'>>|>=|<=|~=|[\.!#+*\-=></|@]' do |m|
          if self.class.operators.include?(m[0])
            token Operator
          else
            token Text
          end
        end
        
        rule %r/\b[_A-Za-z]\w*\b/ do |m|
          lower = m[0].downcase
          if self.class.keywords.include?(lower)
            token Keyword
          elsif self.class.funcs.include?(lower)
            token Keyword::Pseudo
          elsif self.class.transformers.include?(lower)
            token Name::Builtin
          elsif self.class.consts.include?(lower)
            token Keyword::Constant
          else
            token Name::Variable
          end
        end

        rule %r/\s+/, Text
      end
      
      state :str do
        rule %r/''/, Str::Escape
        rule %r/[^']+/, Str::Single
        rule %r/'|$/, Str::Single, :pop!
      end
    end
  end
end