aboutsummaryrefslogtreecommitdiff
path: root/samples/tinyirc.lua
blob: f38ab73c6b166164bdff147ecb7eab6075093c38 (plain)
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
function set_add(set, sock)
    tinsert(set, sock)
end

function set_remove(set, sock)
    for i = 1, getn(set) do
        if set[i] == sock then
           tremove(set, i)
           break
        end
    end 
end

host = host or "*"
port1 = port1 or 8080
port2 = port2 or 8081
if arg then
    host = arg[1] or host
    port1 = arg[2] or port1
    port2 = arg[3] or port2
end

server1 = bind(host, port1)
server1:timeout(1)
server1.is_server = 1
server2 = bind(host, port2)
server2:timeout(1)
server2.is_server = 1

set = {server1, server2}
number = 1

while 1 do
    local r, s, e, l, n
    r, _, e = select(set, nil)
    for i, v in r do
        if v.is_server then
            s = v:accept()
            if s then 
                s:timeout(1)
				s.number = number
				number = number + 1
                set_add(set, s) 
                write("Added client number ", s.number, ". ", 
					getn(set)-2, " total.\n")
            end
        else
            l, e = v:receive()
			n = v.number
            if e then 
                v:close()
                set_remove(set, v) 
                write("Removed client number ", n, ". ",
					getn(set)-2, " total.\n")
            else
            	write("Broadcasting line '", tostring(n), "> ", 
					tostring(l), "'.\n")
            	_, s, e = select(nil, set, 1)
            	if not e then
                	for i,v in s do
                    	v:send(tostring(n), "> ", l, "\r\n")
                	end
            	else
                	write("No one ready to listen!!!\n")
            	end
			end
        end
    end
end