summaryrefslogtreecommitdiffstats
path: root/repl.lua
blob: a3b799ea69101485ca5e8a2b75b87ef44dd71338 (plain) (blame)
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
--
-- This file is part of LEM, a Lua Event Machine.
-- Copyright 2011 Emil Renner Berthing
--
-- LEM is free software: you can redistribute it and/or
-- modify it under the terms of the GNU General Public License as
-- published by the Free Software Foundation, either version 3 of
-- the License, or (at your option) any later version.
--
-- LEM is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with LEM.  If not, see <http://www.gnu.org/licenses/>.
--

local utils   = require 'lem.utils'

local loadstring = loadstring
local format = string.format
local concat = table.concat
local tostring = tostring
local select = select

local function repl(done, name, ins, outs)
	if not outs then outs = ins end

	local getcode, onreturn, onerror

	name = '=' .. name

	function getcode()
		local res, err = outs:write('> ')
		if not res then return done(nil, err) end

		local line
		line, err = ins:read('*l')
		if not line then return done(nil, err) end

		line = line:gsub('^=', 'return ')

		while true do
			res, err = loadstring(line, name)
			if res then break end

			if not err:match("'<eof>'") then
				return onerror(err)
			end

			res, err = outs:write('>> ')
			if not res then return done(nil, err) end

			res, err = ins:read('*l')
			if not res then return done(nil, err) end

			line = line .. res
		end

		utils.sethandler(onerror)
		return onreturn(res())
	end

	function onreturn(...)
		utils.sethandler()
		local args = select('#', ...)
		if args == 0 then return getcode() end

		local rstr
		do
			local t, ti = { ... }, nil
			for i = 1, args - 1 do
				t[i] = tostring(t[i])
			end
			t[args] = tostring(t[args])..'\n'

			rstr = concat(t, '\t')
		end

		local ok, err = outs:write(rstr)
		if not ok then return done(nil, err) end

		return getcode()
	end

	function onerror(err)
		local ok, err = outs:write(format("ERROR: %s\n", err))
		if not ok then return done(nil, err) end

		return getcode()
	end

	return getcode()
end

return {
	wait = function(name, ins, outs)
		local sleeper = utils.sleeper()
		local function done(...)
			return sleeper:wakeup(...)
		end

		utils.spawn(repl, done, name, ins, outs)

		return sleeper:sleep()
	end,

	go = function(name, ins, outs)
		local function done()
			return outs:write('\n')
		end
		return repl(done, name, ins, outs)
	end,
}

-- vim: syntax=lua ts=2 sw=2 noet: