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 182 183 184 185 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 217 218 |
import strutils, logging, nre, critbits, terminal, noise import ../core/parser, ../core/value, ../core/env, ../core/interpreter, ../core/utils when defined(windows): proc putchr*(c: cint): cint {.discardable, header: "<conio.h>", importc: "_putch".} ## Prints an ASCII character to stdout. proc getchr*(): cint {.header: "<conio.h>", importc: "_getch".} ## Retrieves an ASCII character from stdin. else: proc putchr*(c: cint) {.header: "stdio.h", importc: "putchar"} = ## Prints an ASCII character to stdout. stdout.write(c.chr) stdout.flushFile() proc getchr*(): cint = ## Retrieves an ASCII character from stdin. stdout.flushFile() return getch().ord.cint #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) when false: 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("str") 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.}: let newline = '\n'.ord.cint stdout.write("Enter Password: ") var c = getchr() var s = "" while c != newline: putchr('*'.ord.cint) s &= c.chr i.push s.newVal def.symbol("ask") do (i: In) {.gcsafe.}: let vals = i.expect("str") let s = vals[0] var ed = Noise.init() ed.setPrompt(s.getString & ": ") discard ed.readLine() i.push ed.getLine.newVal def.symbol("confirm") do (i: In) {.gcsafe.}: var ed = Noise.init() let vals = i.expect("str") let s = vals[0] proc confirm(): bool = ed.setPrompt(s.getString & " [yes/no]: ") discard ed.readLine() let answer = ed.getLine() if answer.contains(re"(?i)^y(es)?$"): return true elif answer.contains(re"(?i)^no?$"): return false else: stdout.write "Invalid answer. Please enter 'yes' or 'no': " stdout.flushFile() return confirm() i.push confirm().newVal def.symbol("choose") do (i: In) {.gcsafe.}: var ed = Noise.init() 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) stdout.flushFile() 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] ed.setPrompt("Enter your choice ($1 - $2): " % ["1", $c]) discard ed.readLine() let answer = ed.getLine() var choice: int try: choice = answer.parseInt except CatchableError: 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("str") 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("str", "str") let a = vals[0] let b = vals[1] a.strVal.writeFile(b.strVal) def.symbol("fappend") do (i: In): let vals = i.expect("str", "str") let a = vals[0] let b = vals[1] var f:File discard f.open(a.strVal, fmAppend) f.write(b.strVal) f.close() def.symbol("write") do (i: In): i.pushSym("fwrite") def.symbol("read") do (i: In): i.pushSym("fread") def.finalize("io") |