diff options
Diffstat (limited to 'tex/context/third/enigma/enigma.lua')
| -rw-r--r-- | tex/context/third/enigma/enigma.lua | 54 |
1 files changed, 40 insertions, 14 deletions
diff --git a/tex/context/third/enigma/enigma.lua b/tex/context/third/enigma/enigma.lua index 4514ca7..0e53e0a 100644 --- a/tex/context/third/enigma/enigma.lua +++ b/tex/context/third/enigma/enigma.lua @@ -29,6 +29,7 @@ libraries. \stopparagraph --ichd]]-- local iowrite = io.write +local mathfloor = math.floor local mathrandom = math.random local next = next local nodecopy = node.copy @@ -228,17 +229,18 @@ do return stringformat("\027[4;37m%s\027[0m", s) end - local s_steps = [[Total characters encoded: ]] + local s_steps = [[Total characters encoded with machine “]] local f_warnsteps = [[ (%d over permitted maximum)]] - pprint_machine_step = function (n) + pprint_machine_step = function (n, name) local sn + name = colorize(name, 36) if n > max_msg_length then sn = colorize(n, 31) .. stringformat(f_warnsteps, n - max_msg_length) else sn = colorize(n, 37) end - emit(1, s_steps .. sn .. ".") + emit(1, s_steps .. name .. "”: " .. sn .. ".") end local rotorstate = "[s \027[1;37m%s\027[0m n\027[1;37m%2d\027[0m]> " pprint_rotor = function (rotor) @@ -957,7 +959,7 @@ consists of three elements: end local processed_chars = function (machine) - pprint_machine_step(machine.step) + pprint_machine_step(machine.step, machine.name) end new = function (setup_string, pattern) local raw_settings = lpegmatch(p_init, setup_string) @@ -1013,7 +1015,7 @@ Exported functionality will be collected in the table \stopparagraph --ichd]]-- -local enigma = { } +local enigma = { machines = { }, callbacks = { } } --[[ichd \startparagraph @@ -1122,16 +1124,33 @@ do if type(str) ~= "string" then return "" end return lpegmatch(p_ans, str) end + local ensure_int = function (n) + n = tonumber(n) + if not n then return 0 end + return mathfloor(n + 0.5) + end + p_alpha = Cs((alpha + (1 - alpha / ""))^1) + local ensure_alpha = function (s) + s = tostring(s) + return lpegmatch(p_alpha, s) + end local sanitizers = { - other_chars = toboolean, - day_key = alphanum_or_space, + other_chars = toboolean, + day_key = alphanum_or_space, + rotor_setting = ensure_alpha, + verbose = ensure_int, } enigma.parse_args = function (raw) local args = lpegmatch(p_args, raw) for k, v in next, args do local f = sanitizers[k] - args[k] = f(v) + if f then + args[k] = f(v) + else + -- OPTIONAL be fascist and permit only predefined args + args[k] = v + end end return args end @@ -1148,9 +1167,9 @@ This is the interface to \TEX. \stopparagraph --ichd]]-- -enigma.new_callback = function (machine) - enigma.current_machine = machine - return function (head) +enigma.new_callback = function (machine, name) + enigma.machines [name] = machine + enigma.callbacks[name] = function (head) for n in nodetraverse(head) do --print(node, node.id) if n.id == glyph_node then @@ -1162,7 +1181,12 @@ enigma.new_callback = function (machine) local insertion = nodecopy(n) insertion.char = utf8byte(replacement) nodeinsert_before(head, n, insertion) - print(n.char, insertion.char) + elseif type(replacement) == "table" then + for i=1, #replacement do + local insertion = nodecopy(n) + insertion.char = utf8byte(replacement[i]) + nodeinsert_before(head, n, insertion) + end end noderemove(head, n) elseif n.id == glue_node then @@ -1175,8 +1199,10 @@ enigma.new_callback = function (machine) end --local teststring = [[B I II III 01 01 01]] -enigma.new_machine = function (args, pattern) - local machine = new(args.day_key, pattern) +enigma.new_machine = function (args, name) + local machine = new(args.day_key, args.rotor_setting) + machine.name = name + verbose_level = args.verbose return machine end --stub |