interp.lua

require 'socket'

interp = {}

local macros = {}

function interp.macro (def)
    macros[def.name] = def
end

local function expand_macro (M,a)
    if M.arg then
        if not a then
            a = M.last_arg
            if not a then
                return print 'you need to pass a parameter to this macro'
            end
        else
            M.last_arg = a
        end
    else
        a = '' -- keep format happy
    end
    for i,line in ipairs(M) do
        local cmd = line:format(a)
        print('! '..cmd)
        interp.process_line(cmd)
    end
end

local f,err = io.open 'defs.lua'
if f then
    f:close()
    dofile 'defs.lua'
end

local winapi = require 'winapi'
local t,c2,err
local addr = arg[2] or 'localhost'
local c = socket.connect(addr,3333)
c:send ('yes\n')
c:receive()
local m = winapi.mutex()
t = winapi.thread(function()
    c2,err = socket.connect(addr,3334)
    if err then
      c:close()
      error('cannot connect to secondary socket '..err)
    end
    while true do
        local res = c2:receive()
        res = res:gsub('\001','\n')
        m:pcall(io.write,res)
   end
end,'ok')
winapi.sleep(50)

local log = io.open('log.txt','a')

local function prep_for_send (s)
    return (s:gsub('\n','\001'))
end

local function exists (f)
  local f,err = io.open(f)
  if f then
    f:close()
    return true
   end
end

local function mod2file (mod)
    local root = mod:gsub('%.','/')
    local file = root..'.lua'
    if not exists(file) then
        file = root..'/init.lua'
        mod = mod..'.init'
        if not exists(file) then
            return nil,"module not found"
        end
    end
    print('mod',mod,file)
    return mod,file
end

function readfile(file)
  local f,err = io.open(file)
  if not f then return nil,err end
  local contents = f:read '*a'
  f:close()
  return contents
end

function eval(line)
  local res
  m:pcall(function()
    c:send(line..'\n')
    res = c:receive()
  end)
  return (res or '?'):gsub('\001','\n')
end

print 'Lua 5.1.4  Copyright (C) 1994-2008 Lua.org, PUC-Rio'
local init,err = readfile 'init.lua'
if init then
  print 'loading init.lua'
  eval(prep_for_send(init))
end

function interp.process_line (line)
    log:write(line,'\n')
    local cmd,file = line:match '^%.(%S+)(.*)$'
    if cmd then
        file = file:gsub('^%s*','')
        file = #file > 0 and file or nil
        if macros[cmd] then
            expand_macro(macros[cmd],file)
            return
        elseif file then -- either .l (load) or .m (upload module)
            local mod,kind = file,'run'
            if cmd == 'm' then -- given in Lua module form
              mod,file = mod2file(mod)
              if not mod then return print(file) end
              kind = 'mod'
            end
            line,err = readfile(file)
            if err then
                return print(err)
            end
            if kind == 'mod' then
                local ok,err = loadstring(line)
                if not ok then return print(err) end
            end
            line = prep_for_send(line)
            line = '--'..kind..':'..mod..'\001'..line
        else
            return print 'unknown command'
        end
    else
        --  = EXPR becomes print(EXPR)
        local expr = line:match '^%s*=%s*(.+)$'
        if expr then
          line = 'print('..expr..')'
        end
    end
    if line then
        local res = eval(line)
        log:write(res,'\n')
        io.write(res)
    else
        print(err)
    end
end

io.write '> '
local line = io.read()

while line do
  interp.process_line(line)
  io.write '> '
  line = io.read()
end

log:close()
c:close()

if c2 then
    if t then t:kill() end
    c2:close()
end