Blame view

lapp.nim 11.3 KB
4aa613cb   Göran Krampe   First commit
1
  import strutils
809a975c   Göran Krampe   Cleanups and simp...
2
  import os
4aa613cb   Göran Krampe   First commit
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
  import tables
  export tables.`[]`
  
  #### Simple string lexer ###
  type
    PLexer = ref TLexer
    TLexer = object
      str: string
      idx: int
    TLexType = enum
      tend
      tword
      tint
      tfloat
      trange
      telipsis
      tchar
1b2c46b8   Göran Krampe   Added groups supp...
20
21
      toption
      tdivider
4aa613cb   Göran Krampe   First commit
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
  
  proc thisChar(L: PLexer):char = L.str[L.idx]
  proc next(L: PLexer) = L.idx += 1
  
  proc skipws(L: PLexer) =
    while thisChar(L) in Whitespace: next(L)
      
  proc get(L: PLexer; t: var TLexType): string =
    skipws(L)
    let c = thisChar(L)
    t = tend
    if c == '\0': return nil
    result = ""
    result.add(c)
    next(L) 
    t = tchar
    case c
    of '-':  # '-", "--"
1b2c46b8   Göran Krampe   Added groups supp...
40
      t = toption
4aa613cb   Göran Krampe   First commit
41
42
43
      if thisChar(L) == '-':
        result.add('-')
        next(L)
1b2c46b8   Göran Krampe   Added groups supp...
44
45
46
47
48
        if thisChar(L) == '-': # "---..."
          t = tdivider
          result.add('-') 
          while thisChar(L) == '-':
            next(L)
4aa613cb   Göran Krampe   First commit
49
50
51
52
53
54
55
56
57
58
59
60
    of Letters: # word
      t = tword
      while thisChar(L) in Letters:
        result.add(thisChar(L))
        next(L)
    of Digits: # number
      t = tint
      while thisChar(L) in Digits:
        result.add(thisChar(L))
        next(L)
      if thisChar(L) == '.':
        t = tfloat
7be5e09d   Göran Krampe   Fixed float parsi...
61
        result.add(thisChar(L))
4aa613cb   Göran Krampe   First commit
62
63
        next(L)
        while thisChar(L) in Digits:
7be5e09d   Göran Krampe   Fixed float parsi...
64
          result.add(thisChar(L))
4aa613cb   Göran Krampe   First commit
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
          next(L)
    of '.': # ".", "..", "..."
      if thisChar(L) == '.':
        t = trange
        result.add('.')
        next(L)
        if thisChar(L) == '.':
          t = telipsis
          result.add('.')
          next(L)
    else: discard
      
  proc get(L: PLexer): string = 
    var t: TLexType
    get(L,t)
      
  proc reset(L: PLexer, s: string) =
    L.str  = s
    L.idx = 0
  
  proc newLexer(s: string): PLexer =
    new(result)
    result.reset(s)
      
  ### a container for values ###
      
  type
    TValueKind = enum
      vInt,
      vFloat,
      vString,
      vBool,
      vFile,
      vSeq
  
7be5e09d   Göran Krampe   Fixed float parsi...
100
101
102
103
104
105
106
    PValue* = ref TValue
    TValue* = object
      case kind*: TValueKind
        of vInt: asInt*: int
        of vFloat: asFloat*: float
        of vString: asString*: string
        of vBool: asBool*: bool
4aa613cb   Göran Krampe   First commit
107
        of vFile:
7be5e09d   Göran Krampe   Fixed float parsi...
108
109
110
          asFile*: File
          fileName*: string
        of vSeq: asSeq*: seq[PValue]
4aa613cb   Göran Krampe   First commit
111
112
113
114
115
116
117
118
119
120
121
122
  
  proc boolValue(c: bool): PValue =  PValue(kind: vBool, asBool: c)
      
  proc fileValue(f: File, name: string): PValue =  PValue(kind: vFile, asFile: f, fileName: name)
      
  proc strValue(s: string): PValue =  PValue(kind: vString, asString: s) 
      
  proc intValue(v: int): PValue =   PValue(kind: vInt, asInt: v)
      
  proc floatValue(v: float): PValue = PValue(kind: vFloat, asFloat: v)
      
  proc seqValue(v: seq[PValue]): PValue = PValue(kind: vSeq, asSeq: v)
4aa613cb   Göran Krampe   First commit
123
124
125
126
127
128
      
  type
    PSpec = ref TSpec
    TSpec = object
      defVal: string
      ptype: string
1b2c46b8   Göran Krampe   Added groups supp...
129
      group: int
4aa613cb   Göran Krampe   First commit
130
131
132
133
134
      needsValue, multiple, used: bool
  var
    progname, usage: string
    aliases: array[char,string]
    parm_spec =  initTable[string,PSpec]()
809a975c   Göran Krampe   Cleanups and simp...
135
  
4aa613cb   Göran Krampe   First commit
136
137
138
139
  proc fail(msg: string)  =
    stderr.writeln(progname & ": " & msg)
    quit(usage)
  
6a614998   Göran Krampe   Fix for long opts...
140
  proc parseSpec(u: string) =    
4aa613cb   Göran Krampe   First commit
141
142
143
    var
      L: PLexer
      tok: string
1b2c46b8   Göran Krampe   Added groups supp...
144
      groupCounter: int
4aa613cb   Göran Krampe   First commit
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
      k = 1
        
    let lines = u.splitLines
    L = newLexer(lines[0])
    progname = L.get
    usage = u
    for line in lines[1..(-1)]:
      var
        isarg = false
        multiple = false
        getnext = true
        name: string
        alias: char
      L.reset(line)
      tok = L.get
      if tok == "-" or tok == "--":  # flag
        if tok == "-": #short flag
          let flag = L.get
6a614998   Göran Krampe   Fix for long opts...
163
          if len(flag) != 1: fail("short option has one character!")
4aa613cb   Göran Krampe   First commit
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
          tok = L.get
          if tok == ",": # which is alias for long flag
            tok = L.get
            if tok != "--": fail("expecting long --flag")
            name = L.get
            alias = flag[0]
          else: # only short flag
            name = flag
            alias = flag[0]
            getnext = false
        else: # only long flag
          name = L.get
          alias = '\0'
      elif tok == "<":  # argument
        isarg = true
        name = L.get
        alias = chr(k)
        k += 1
        tok = L.get
        if tok != ">": fail("argument must be enclosed in <...>")
1b2c46b8   Göran Krampe   Added groups supp...
184
185
      elif tok == "---": # divider
        inc(groupCounter)
4aa613cb   Göran Krampe   First commit
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
      if getnext:  tok = L.get
      if tok == ":": # allowed to have colon after flags
        tok = L.get
      if tok == nil: continue
      # default types for flags and arguments
      var 
        ftype = if isarg: "string" else: "bool"
        defValue = ""
      if tok == "(": # typed flag/argument
        var t = tchar
        tok = L.get(t)
        if tok == "default":  # type from default value
          defValue = L.get(t)
          if t == tint: ftype = "int"
          elif t == tfloat: ftype = "float"
          elif t == tword:
            if defValue == "stdin": ftype = "infile"
            elif defValue == "stdout": ftype = "outfile"
            else: ftype = "string"
          else: fail("unknown default value " & tok)
        else: # explicit type
          if t == tword:
            ftype = tok
            if tok == "bool": defValue = "false"
          else: fail("unknown type " & tok)
        discard L.get(t)
        multiple = t == telipsis
      elif ftype == "bool": # no type or default
        defValue = "false"
              
      if name != nil:
1b2c46b8   Göran Krampe   Added groups supp...
217
218
        #echo("Param: " & name & " type: " & $ftype & " group: " & $groupCounter & " needsvalue: " & $(ftype != "bool") & " default: " & $defValue & " multiple: " & $multiple)
        let spec = PSpec(defVal:defValue, ptype: ftype, group: groupCounter, needsValue: ftype != "bool",multiple:multiple)
4aa613cb   Göran Krampe   First commit
219
220
221
222
223
224
        aliases[alias] = name        
        parm_spec[name] = spec
  
  proc tail(s: string): string = s[1..(-1)]
  
  var 
809a975c   Göran Krampe   Cleanups and simp...
225
    files = newSeq[File]()
4aa613cb   Göran Krampe   First commit
226
227
  
  proc closeFiles() {.noconv.} =
809a975c   Göran Krampe   Cleanups and simp...
228
229
    for f in files:
      f.close()
4aa613cb   Göran Krampe   First commit
230
  
7af04aca   Göran Krampe   Exported parseArg...
231
  proc parseArguments*(usage: string, args: seq[string]): Table[string,PValue] =
4aa613cb   Göran Krampe   First commit
232
233
234
235
236
237
238
239
240
241
242
    var
      vars = initTable[string,PValue]()
      n = len(args) - 1
      i = 1
      k = 1
      flag,value, arg: string        
      info: PSpec
      short: bool
      flagvalues: seq[seq[string]]
        
    proc next(): string =
6a614998   Göran Krampe   Fix for long opts...
243
      if i > n: fail("an option required a value!")
4aa613cb   Göran Krampe   First commit
244
245
246
247
248
249
250
251
252
253
      result = args[i]
      i += 1
        
    proc get_alias(c: char): string =
      result = aliases[c]
      if result == nil:
        n = ord(c)
        if n < 20:
          fail("no such argument: " & $n)
        else:
6a614998   Göran Krampe   Fix for long opts...
254
          fail("no such option: " & c)
4aa613cb   Göran Krampe   First commit
255
256
257
258
        
    proc get_spec(name: string): PSpec =
      result = parm_spec[name]
      if result == nil:
6a614998   Göran Krampe   Fix for long opts...
259
        fail("no such option: " & name)
4aa613cb   Göran Krampe   First commit
260
261
262
263
264
  
    newSeq(flagvalues, 0)
    parseSpec(usage)
    addQuitProc(closeFiles)
  
c577e6e6   Göran Krampe   Added failures co...
265
266
267
    # Collect failures
    var failures = newSeq[string]()
    
4aa613cb   Göran Krampe   First commit
268
269
270
271
272
273
274
275
276
    # parse the flags and arguments
    while i <= n:
      arg = next()
      if arg[0] == '-':  #flag
        short = arg[1] != '-'
        arg = arg.tail
        if short: # all short args are aliases, even if only to themselves
          flag = get_alias(arg[0])
        else:
6a614998   Göran Krampe   Fix for long opts...
277
          flag = arg[1..high(arg)]
4aa613cb   Göran Krampe   First commit
278
279
280
281
282
283
284
285
286
287
288
289
        info = get_spec(flag)
        if info.needsValue:
          if short and len(arg) > 1: # value can follow short flag
            value = arg.tail
          else:  # grab next argument
            value = next()
        else:
          value = "true"
          if short and len(arg) > 0: # short flags can be combined
            for c in arg.tail:
              let f = get_alias(c)
              let i = get_spec(f)
c577e6e6   Göran Krampe   Added failures co...
290
291
              if i.needsValue:
                failures.add("option " & f & " needs a value")
4aa613cb   Göran Krampe   First commit
292
293
294
295
296
297
298
299
300
301
              flagvalues.add(@[f,"true"])
              i.used = true
      else: # argument (stored as \001, \002, etc
        flag = get_alias(chr(k))
        value = arg
        info = get_spec(flag)
        # don't move on if this is a varags last param
        if not info.multiple:  k += 1
      flagvalues.add(@[flag,value])
      info.used = true
c577e6e6   Göran Krampe   Added failures co...
302
303
304
305
306
307
308
    
    # Some options disables checking
    var enableChecks = true
    for flag,info in parm_spec:
      if info.used:
        if flag == "help" or flag == "version":
          enableChecks = false
1b2c46b8   Göran Krampe   Added groups supp...
309
310
311
312
313
314
315
316
          
    # Check maximum group used
    var maxGroup = 0
    for item in flagvalues:
      info = get_spec(item[0])
      if maxGroup < info.group:
        maxGroup = info.group
      
4aa613cb   Göran Krampe   First commit
317
318
319
    # any flags not mentioned?
    for flag,info in parm_spec:
      if not info.used:
1b2c46b8   Göran Krampe   Added groups supp...
320
321
        # Is there no default and we have used options in this group?
        if info.defVal == "" and info.group <= maxGroup: 
c577e6e6   Göran Krampe   Added failures co...
322
323
324
325
326
327
328
          failures.add("required option or argument missing: " & flag)
        else:
          flagvalues.add(@[flag,info.defVal])
  
    if enableChecks:
      # any failures up until now - then we fail
      if failures.len > 0:
1b2c46b8   Göran Krampe   Added groups supp...
329
        fail(failures.join("\n") & "\n")
c577e6e6   Göran Krampe   Added failures co...
330
  
4aa613cb   Göran Krampe   First commit
331
332
333
334
335
336
337
338
339
340
341
342
343
    # cool, we have the info, can convert known flags
    for item in flagvalues:
      var pval: PValue;
      let
        flag = item[0]
        value = item[1]
        info = get_spec(flag)
      case info.ptype
      of "int":
        var v: int
        try:
          v = value.parseInt
        except:
809a975c   Göran Krampe   Cleanups and simp...
344
          fail("bad integer for " & flag)
4aa613cb   Göran Krampe   First commit
345
346
347
348
349
350
        pval = intValue(v)
      of "float":
        var v: float
        try:
          v = value.parseFloat
        except:
809a975c   Göran Krampe   Cleanups and simp...
351
          fail("bad float for " & flag)
4aa613cb   Göran Krampe   First commit
352
353
354
355
356
357
358
359
360
        pval = floatValue(v)
      of "bool":
        pval = boolValue(value.parseBool)
      of "string":
        pval = strValue(value)
      of "infile","outfile": # we open files for the app...
        var f: File
        try:
          if info.ptype == "infile":
809a975c   Göran Krampe   Cleanups and simp...
361
362
363
364
            if value == "stdin":
              f = stdin
            else:
              f = open(value, fmRead)
4aa613cb   Göran Krampe   First commit
365
          else:
809a975c   Göran Krampe   Cleanups and simp...
366
367
368
369
            if value == "stdout":
              f = stdout
            else:
              f = open(value, fmWrite)
4aa613cb   Göran Krampe   First commit
370
          # they will be closed automatically on program exit
809a975c   Göran Krampe   Cleanups and simp...
371
          files.add(f)
4aa613cb   Göran Krampe   First commit
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
        except:
          fail("cannot open " & value)
        pval = fileValue(f,value)
      else: discard
      
      var oval = vars[flag]
      if info.multiple: # multiple flags are sequence values
        if oval == nil: # first value!
          pval = seqValue(@[pval])
        else: # just add to existing sequence
          oval.asSeq.add(pval)
          pval = oval
      elif oval != nil: # cannot repeat a single flag!
        fail("cannot use '" & flag & "' more than once")
      vars[flag] = pval
  
    return vars
  
6a614998   Göran Krampe   Fix for long opts...
390
  proc parse*(usage: string): Table[string,PValue] =
4aa613cb   Göran Krampe   First commit
391
392
393
394
395
396
397
398
    var 
      args: seq[string]
      n = paramCount()
    newSeq(args,n+1)
    for i in 0..n:
      args[i] = paramStr(i)
    return parseArguments(usage,args)    
  
c577e6e6   Göran Krampe   Added failures co...
399
  # Helper proc for verbosity level.
1b2c46b8   Göran Krampe   Added groups supp...
400
  proc verbosityLevel*(args: Table[string,PValue]): int =
c577e6e6   Göran Krampe   Added failures co...
401
402
403
404
405
406
407
408
409
    if args.hasKey("verbose"):
      let verbosity = args["verbose"].asSeq
      result = verbosity.len
      if not verbosity[0].asBool:
        result = 0
    else:
      result = 0
  
  # Helper to check if we should show version
1b2c46b8   Göran Krampe   Added groups supp...
410
  proc showVersion*(args: Table[string,PValue]): bool =
c577e6e6   Göran Krampe   Added failures co...
411
412
413
    args["version"].asBool
  
  # Helper to check if we should show help
1b2c46b8   Göran Krampe   Added groups supp...
414
  proc showHelp*(args: Table[string,PValue]): bool =
c577e6e6   Göran Krampe   Added failures co...
415
416
417
418
    args["help"].asBool
  
  
  # Typical usage
4aa613cb   Göran Krampe   First commit
419
  when isMainModule:
c577e6e6   Göran Krampe   Added failures co...
420
421
422
423
424
425
426
427
    let help = """
    head [flags] files... [out]
      -h,--help                     Show this help
      --version                     Show version
      -n: (default 10)              Number of lines to show
      -v,--verbose: (bool...)       Verbosity level, ignored
      -o,--out: (default stdout)    Optional outfile, defaults to stdout
      <files>: (default stdin...)   Files to take head of
4aa613cb   Göran Krampe   First commit
428
    """
c577e6e6   Göran Krampe   Added failures co...
429
430
    # On parsing failure this will show usage automatically
    var args = parse(help)
4aa613cb   Göran Krampe   First commit
431
  
c577e6e6   Göran Krampe   Added failures co...
432
433
434
    # These two are special, they short out
    if args.showHelp: quit(help)
    if args.showVersion: quit("Version: 1.99")
4aa613cb   Göran Krampe   First commit
435
    
c577e6e6   Göran Krampe   Added failures co...
436
437
438
439
440
441
442
443
    # Ok, so what did we get...
    let n = args["n"].asInt
    
    # This one is a helper
    let v = verbosityLevel(args)
    
    echo "Lines to show: " & $n
    echo "Verbosity level: " & $verbosityLevel(args)
4aa613cb   Göran Krampe   First commit
444
445
    
    let myfiles = args["files"].asSeq
c577e6e6   Göran Krampe   Added failures co...
446
447
    var outFile = args["out"].asFile
    
4aa613cb   Göran Krampe   First commit
448
    for f in myfiles:
c577e6e6   Göran Krampe   Added failures co...
449
450
      for i in 1..n:
        writeln(outFile, string(f.asFile.readLine()))