minpkg/lib/min_io.nim
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 |
import strutils, logging, critbits, terminal import ../packages/nimline/nimline, ../packages/nim-sgregex/sgregex, ../core/parser, ../core/value, ../core/env, ../core/interpreter, ../core/utils var ORIGKEYMAP {.threadvar.}: CritBitTree[KeyCallback] for key, value in KEYMAP.pairs: ORIGKEYMAP[key] = value proc io_module*(i: In) = let def = i.define() def.symbol("clear") do (i: In): stdout.eraseScreen stdout.setCursorPos(0, 0) def.symbol("unmapkey") do (i: In): let vals = i.expect("'sym") let key = vals[0].getString.toLowerAscii if not KEYNAMES.contains(key) and not KEYSEQS.contains(key): raiseInvalid("Unrecognized key: " & key) if KEYMAP.hasKey(key): if ORIGKEYMAP.hasKey(key): KEYMAP[key] = ORIGKEYMAP[key] else: KEYMAP.excl(key) def.symbol("mapkey") do (i: In): let vals = i.expect("'sym", "quot") let key = vals[0].getString.toLowerAscii var q = vals[1] if not KEYNAMES.contains(key) and not KEYSEQS.contains(key): raiseInvalid("Unrecognized key: " & key) var ic = i.copy(i.filename) KEYMAP[key] = proc (ed: var LineEditor) {.gcsafe.} = ic.apply(q) def.symbol("newline") do (i: In): echo "" def.symbol("notice") do (i: In): let a = i.peek notice $$a def.symbol("info") do (i: In): let a = i.peek info $$a def.symbol("error") do (i: In): let a = i.peek error $$a def.symbol("warn") do (i: In): let a = i.peek warn $$a def.symbol("debug") do (i: In): let a = i.peek debug $$a def.symbol("fatal") do (i: In): let a = i.peek fatal $$a quit(100) def.symbol("column-print") do (i: In): let vals = i.expect("int", "quot") let n = vals[0] let q = vals[1] var c = 0 for s in q.qVal: c.inc stdout.write $$s & spaces(max(0, 15 - ($$s).len)) if c mod n.intVal == 0: echo "" echo "" def.symbol("getchr") do (i: In): i.push getchr().newVal def.symbol("putchr") do (i: In): let ch = i.expect("string") if ch[0].getString.len != 1: raiseInvalid("Symbol putch requires a string containing a single character.") putchr(ch[0].getString[0].cint) def.symbol("password") do (i: In) {.gcsafe.}: var ed = initEditor() i.push ed.password("Enter Password: ").newVal def.symbol("ask") do (i: In) {.gcsafe.}: var ed = initEditor() let vals = i.expect("string") let s = vals[0] i.push ed.readLine(s.getString & ": ").newVal def.symbol("confirm") do (i: In) {.gcsafe.}: var ed = initEditor() let vals = i.expect("string") let s = vals[0] proc confirm(): bool = let answer = ed.readLine(s.getString & " [yes/no]: ") if answer.match("^y(es)?$", "i"): return true elif answer.match("^no?$", "i"): return false else: stdout.write "Invalid answer. Please enter 'yes' or 'no': " return confirm() i.push confirm().newVal def.symbol("choose") do (i: In) {.gcsafe.}: var ed = initEditor() let vals = i.expect("'sym", "quot") let s = vals[0] var q = vals[1] if q.qVal.len <= 0: raiseInvalid("No choices to display") stdout.writeLine(s.getString) proc choose(): int = var c = 0 for item in q.qVal: if not item.isQuotation or not item.qVal.len == 2 or not item.qVal[0].isString or not item.qVal[1].isQuotation: raiseInvalid("Each item of the quotation must be a quotation containing a string and a quotation") c.inc echo "$1 - $2" % [$c, item.qVal[0].getString] let answer = ed.readLine("Enter your choice ($1 - $2): " % ["1", $c]) var choice: int try: choice = answer.parseInt except: choice = 0 if choice <= 0 or choice > c: echo "Invalid choice." return choose() else: return choice let choice = choose() i.dequote(q.qVal[choice-1].qVal[1]) def.symbol("print") do (i: In): let a = i.peek a.print def.symbol("fread") do (i: In): let vals = i.expect("string") let file = vals[0].strVal var contents = "" if MINCOMPILED: var compiledFile = strutils.replace(strutils.replace(file, "\\", "/"), "./", "") if COMPILEDASSETS.hasKey(compiledFile): contents = COMPILEDASSETS[compiledFile] if contents == "": contents = file.readFile i.push newVal(contents) def.symbol("fwrite") do (i: In): let vals = i.expect("string", "string") let a = vals[0] let b = vals[1] a.strVal.writeFile(b.strVal) def.symbol("fappend") do (i: In): let vals = i.expect("string", "string") let a = vals[0] let b = vals[1] var f:File discard f.open(a.strVal, fmAppend) f.write(b.strVal) f.close() def.finalize("io") |