mirror of
https://github.com/rxi/lovebird.git
synced 2024-11-27 12:04:21 +00:00
283 lines
6.8 KiB
Lua
283 lines
6.8 KiB
Lua
--
|
|
-- lovebird
|
|
--
|
|
-- Copyright (c) 2014, rxi
|
|
--
|
|
-- This library is free software; you can redistribute it and/or modify it
|
|
-- under the terms of the MIT license. See LICENSE for details.
|
|
--
|
|
|
|
socket = require "socket"
|
|
|
|
local lovebird = { _version = "0.0.1" }
|
|
|
|
lovebird.inited = false
|
|
lovebird.host = "*"
|
|
lovebird.wrapprint = true
|
|
lovebird.timestamp = true
|
|
lovebird.port = 8000
|
|
lovebird.whitelist = { "127.0.0.1", "localhost" }
|
|
lovebird.maxlines = 200
|
|
lovebird.refreshrate = .5
|
|
lovebird.buffer = ""
|
|
lovebird.lines = {}
|
|
|
|
|
|
local loadstring = loadstring or load
|
|
|
|
local map = function(t, fn)
|
|
local res = {}
|
|
for k, v in pairs(t) do res[k] = fn(v) end
|
|
return res
|
|
end
|
|
|
|
local find = function(t, value)
|
|
for k, v in pairs(t) do
|
|
if v == value then return k end
|
|
end
|
|
end
|
|
|
|
local trace = function(...)
|
|
print("[lovebird] " .. table.concat(map({...}, tostring), " "))
|
|
end
|
|
|
|
local unescape = function(str)
|
|
local f = function(x) return string.char(tonumber("0x"..x)) end
|
|
return (str:gsub("%+", " "):gsub("%%(..)", f))
|
|
end
|
|
|
|
|
|
local pagetemplate = [[
|
|
<!doctype html>
|
|
<html>
|
|
<head>
|
|
<meta http-equiv="x-ua-compatible" content="IE=Edge"/>
|
|
<title>lovebird</title>
|
|
<style>
|
|
body {
|
|
margin: 0px;
|
|
font-family: helvetica, verdana, sans;
|
|
background: #FFFFFF;
|
|
}
|
|
form {
|
|
margin-bottom: 0px;
|
|
}
|
|
.timestamp {
|
|
color: #909090;
|
|
}
|
|
#header {
|
|
background: #101010;
|
|
height: 25px;
|
|
color: #F0F0F0;
|
|
padding: 9px
|
|
}
|
|
#title {
|
|
float: left;
|
|
font-size: 20px;
|
|
}
|
|
#title a {
|
|
color: #F0F0F0;
|
|
text-decoration: none;
|
|
}
|
|
#title a:hover {
|
|
color: #FFFFFF;
|
|
}
|
|
#version {
|
|
font-size: 10px;
|
|
}
|
|
#status {
|
|
float: right;
|
|
font-size: 14px;
|
|
padding-top: 4px;
|
|
}
|
|
#content {
|
|
position: absolute;
|
|
margin: 12px;
|
|
top: 40px; bottom: 0px; left: 0px; right: 0px;
|
|
background: #F0F0F0;
|
|
border: 1px solid #E0E0E0;
|
|
}
|
|
#input {
|
|
position: absolute;
|
|
margin: 10px;
|
|
bottom: 0px; left: 0px; right: 0px;
|
|
}
|
|
#inputbox {
|
|
width: 100%;
|
|
}
|
|
#output {
|
|
overflow-y: scroll;
|
|
position: absolute;
|
|
margin: 10px;
|
|
top: 0px; bottom: 36px; left: 0px; right: 0px;
|
|
}
|
|
</style>
|
|
</head>
|
|
<body>
|
|
<div id="header">
|
|
<div id="title">
|
|
<a href="https://github.com/rxi/lovebird">lovebird</a>
|
|
<span id="version"><?lua echo(lovebird._version) ?></span>
|
|
</div>
|
|
<div id="status">connected</div>
|
|
</div>
|
|
<div id="content">
|
|
<div id="output"> <?lua echo(lovebird.buffer) ?> </div>
|
|
<div id="input">
|
|
<form method="post">
|
|
<input id="inputbox" name="input" type="text" autofocus></input>
|
|
</form>
|
|
</div>
|
|
</div>
|
|
<script>
|
|
/* Scroll output to bottom */
|
|
var scrolloutput = function() {
|
|
var div = document.getElementById("output");
|
|
div.scrollTop = div.scrollHeight;
|
|
}
|
|
scrolloutput()
|
|
/* Refresh buffer output at intervals */
|
|
var refresh = function() {
|
|
var req = new XMLHttpRequest();
|
|
req.onreadystatechange = function() {
|
|
if (req.readyState != 4) return;
|
|
var status;
|
|
if (req.status == 200) {
|
|
status = "connected";
|
|
var div = document.getElementById("output");
|
|
if (div.innerHTML != req.responseText) {
|
|
div.innerHTML = req.responseText;
|
|
scrolloutput();
|
|
}
|
|
} else {
|
|
status = "disconnected";
|
|
}
|
|
var div = document.getElementById("status");
|
|
if (div.innerHTML != status) div.innerHTML = status;
|
|
}
|
|
req.open("GET", "/buffer", true);
|
|
req.send();
|
|
}
|
|
setInterval(refresh, <?lua echo(lovebird.refreshrate) ?> * 1000);
|
|
</script>
|
|
</body>
|
|
</html>
|
|
]]
|
|
|
|
|
|
function lovebird.init()
|
|
lovebird.server = assert(socket.bind(lovebird.host, lovebird.port))
|
|
lovebird.addr, lovebird.port = lovebird.server:getsockname()
|
|
lovebird.server:settimeout(0)
|
|
if lovebird.wrapprint then
|
|
local oldprint = print
|
|
print = function(...)
|
|
oldprint(...)
|
|
lovebird.print(...)
|
|
end
|
|
end
|
|
lovebird.inited = true
|
|
end
|
|
|
|
|
|
function lovebird.template(str, env)
|
|
env = env or {}
|
|
local keys, vals = {}, {}
|
|
for k, v in pairs(env) do
|
|
table.insert(keys, k)
|
|
table.insert(vals, v)
|
|
end
|
|
local f = function(x) return string.format(" echo(%q)", x) end
|
|
str = ("?>"..str.."<?lua"):gsub("%?>(.-)<%?lua", f)
|
|
str = "local echo, " .. table.concat(keys, ",") .. " = ..." .. str
|
|
local output = {}
|
|
local echo = function(str) table.insert(output, str) end
|
|
assert(loadstring(str))(echo, unpack(vals))
|
|
return table.concat(map(output, tostring))
|
|
end
|
|
|
|
|
|
function lovebird.print(...)
|
|
local str = table.concat(map({...}, tostring), " ")
|
|
if lovebird.timestamp then
|
|
str = os.date('<span class="timestamp">[%H:%M:%S]</span> ') .. str
|
|
end
|
|
table.insert(lovebird.lines, str)
|
|
if #lovebird.lines > lovebird.maxlines then
|
|
table.remove(lovebird.lines, 1)
|
|
end
|
|
lovebird.buffer = table.concat(lovebird.lines, "<br>")
|
|
end
|
|
|
|
|
|
function lovebird.onError(err)
|
|
trace("ERROR:", err)
|
|
end
|
|
|
|
|
|
function lovebird.onRequest(req, client)
|
|
-- Handle request for just the buffer
|
|
if req.url:match("buffer") then
|
|
return "HTTP/1.1 200 OK\r\n\r\n" .. lovebird.buffer
|
|
end
|
|
-- Handle input
|
|
if req.body then
|
|
local str = unescape(req.body:match(".-=(.*)"))
|
|
xpcall(function() assert(loadstring(str))() end, lovebird.onError)
|
|
end
|
|
-- Generate page
|
|
local t = {}
|
|
table.insert(t, "HTTP/1.1 200 OK\r\n\r\n")
|
|
table.insert(t, lovebird.template(pagetemplate, { lovebird = lovebird }))
|
|
return table.concat(t)
|
|
end
|
|
|
|
|
|
function lovebird.onConnect(client)
|
|
-- Create request table
|
|
local requestptn = "(%S*)%s*(%S*)%s*(%S*)"
|
|
local req = {}
|
|
req = {}
|
|
req.socket = client
|
|
req.addr, req.port = client:getsockname()
|
|
req.request = client:receive()
|
|
req.method, req.url, req.proto = req.request:match(requestptn)
|
|
req.headers = {}
|
|
while 1 do
|
|
local line = client:receive()
|
|
if not line or #line == 0 then break end
|
|
local k, v = line:match("(.-):%s*(.*)$")
|
|
req.headers[k] = v
|
|
end
|
|
if req.headers["Content-Length"] then
|
|
req.body = client:receive(req.headers["Content-Length"])
|
|
end
|
|
-- Handle request; get data to send
|
|
local data, index = lovebird.onRequest(req), 0
|
|
-- Send data
|
|
while index < #data do
|
|
index = index + client:send(data, index)
|
|
end
|
|
-- Clear up
|
|
client:close()
|
|
end
|
|
|
|
|
|
function lovebird.update()
|
|
if not lovebird.inited then lovebird.init() end
|
|
local client = lovebird.server:accept()
|
|
if client then
|
|
client:settimeout(2)
|
|
local addr = client:getsockname()
|
|
if not lovebird.whitelist or find(lovebird.whitelist, addr) then
|
|
xpcall(function() lovebird.onConnect(client) end, lovebird.onError)
|
|
else
|
|
trace("got non-whitelisted connection attempt: ", addr)
|
|
client:close()
|
|
end
|
|
end
|
|
end
|
|
|
|
|
|
return lovebird
|