modularize and clean up server impl

rebuild/main.lua

@@ -7,6 +7,32 @@ function debug_message(message)
 end
 
 local yo = require('yo')
+local server = require('server')
 
+local statusMessages = {
+  [0] = 'not enabled',
+  'connecting',
+  'wrong password',
+  'network not found',
+  'connection fail',
+  'connected'
+}
 a = '0c6ac771-71fa-420f-810c-2853989a8ca6'
 y = 'ariyeah'
+wifi.sta.config('Pizza Pirate Cove', 'pizzapirates')
+wifi.setmode(wifi.STATION)
+
+function short_press()
+  server.stop()
+  yo.yo(y, a)
+end
+
+function long_press()
+  current_settings = {
+    yo_recipient = y,
+    ssid = wifi.sta.getconfig(),
+    status = statusMessages[wifi.sta.status()]
+  }
+
+  server.start(current_settings)
+end

rebuild/server.lua

@@ -0,0 +1,101 @@
+local net = net
+local file = file -- defer to settings module?
+local string = string
+local assert = assert
+local type = type
+local srv = nil
+local debug_message = debug_message
+
+module(...)
+
+local function connect(conn, data)
+  local query_data
+
+  conn:on('receive',
+    function(cn, req_data)
+      query_data = get_http_req(req_data)
+      debug_message(query_data['METHOD'] .. ' ' .. ' ' .. query_data['User-Agent'])
+
+      send_index(cn)
+      -- Close the connection for the request
+      cn:close()
+    end
+  )
+end
+
+-- Build and return a table of the http request data
+function get_http_req(instr)
+  local t = {}
+  local first = nil
+  local key, v, strt_ndx, end_ndx
+
+  for str in string.gmatch(instr, '([^\n]+)') do
+    -- First line in the method and path
+    if(first == nil) then
+      first = 1
+      strt_ndx, end_ndx = string.find(str, '([^ ]+)')
+      v = trim(string.sub(str, end_ndx + 2))
+      key = trim(string.sub(str, strt_ndx, end_ndx))
+      t['METHOD'] = key
+      t['REQUEST'] = v
+    else -- Process and reamaining ':' fields
+      strt_ndx, end_ndx = string.find(str, '([^:]+)')
+      if(end_ndx ~= nil) then
+        v = trim(string.sub(str, end_ndx + 2))
+        key = trim(string.sub(str, strt_ndx, end_ndx))
+        t[key] = v
+      end
+    end
+  end
+
+  return t
+end
+
+-- String trim left and right
+function trim(s)
+  return(s:gsub('^%s*(.-)%s*$', '%1'))
+end
+
+function send_index(conn)
+  debug_message('server.send_index')
+
+  file.open('index.html')
+  local index = file.read()
+  file.close()
+
+  index = string.gsub(index, 'S_', settings.ssid)
+  index = string.gsub(index, 'T_', settings.status)
+  index = string.gsub(index, 'R_', settings.yo_recipient)
+
+  debug_message('sending index')
+  debug_message('____________')
+  debug_message(index)
+  debug_message('____________')
+  conn:send('HTTP/1.1 200 OK\n\n' .. index)
+end
+
+function start(current_settings)
+  assert(type(current_settings.ssid) == 'string', 'ssid must be a string')
+  assert(type(current_settings.status) == 'string', 'status must be a string')
+  assert(type(current_settings.yo_recipient) == 'string', 'yo_recipient must be a string')
+  debug_message('server.start')
+  debug_message(srv)
+  settings = current_settings
+
+  if srv then
+    srv = nil
+  end
+  srv = net.createServer(net.TCP, 30)
+  srv:listen(80, connect)
+  debug_message(srv)
+end
+
+function stop()
+  debug_message('server.stop')
+  debug_message(srv)
+  if srv then
+    srv:close()
+    srv = nil
+  end
+  debug_message(srv)
+end