libs/httpd/luasrc/httpd/handler/luci.lua

   1 --[[
   2
   3 HTTP server implementation for LuCI - luci handler
   4 (c) 2008 Steven Barth <steven@midlink.org>
   5
   6 Licensed under the Apache License, Version 2.0 (the "License");
   7 you may not use this file except in compliance with the License.
   8 You may obtain a copy of the License at
   9
  10         http://www.apache.org/licenses/LICENSE-2.0
  11
  12 $Id$
  13
  14 ]]--
  15
  16 module("luci.httpd.handler.luci", package.seeall)
  17
  18 require("luci.dispatcher")
  19 require("luci.http")
  20 require("luci.http.protocol.date")
  21 local ltn12 = require("luci.ltn12")
  22
  23 Luci = luci.util.class(luci.httpd.module.Handler)
  24 Response = luci.httpd.module.Response
  25
  26 function Luci.__init__(self, limit)
  27         luci.httpd.module.Handler.__init__(self)
  28         self.limit = limit or 5
  29         self.running = {}
  30         setmetatable(self.running, {__mode = "v"})
  31 end
  32
  33 function Luci.handle_head(self, ...)
  34         local response, sourceout = self:handle_get(...)
  35         return response
  36 end
  37
  38 function Luci.handle_post(self, ...)
  39         return self:handle_get(...)
  40 end
  41
  42 function Luci.handle_get(self, request, sourcein, sinkerr)
  43         if self.limit and #self.running >= self.limit then
  44                 return self:failure(503, "Overload")
  45         end
  46         table.insert(self.running, coroutine.running())
  47
  48         local r = luci.http.Request(
  49                 request.env,
  50                 sourcein,
  51                 sinkerr
  52         )
  53
  54         local res, id, data1, data2 = true, 0, nil, nil
  55         local headers = {}
  56         local status = 200
  57         local active = true
  58
  59         local x = coroutine.create(luci.dispatcher.httpdispatch)
  60         while not id or id < 3 do
  61                 coroutine.yield()
  62
  63                 res, id, data1, data2 = coroutine.resume(x, r)
  64
  65                 if not res then
  66                         status = 500
  67                         headers["Content-Type"] = "text/plain"
  68                         local err = {id}
  69                         return Response( status, headers ), function() return table.remove(err) end
  70                 end
  71
  72                 if id == 1 then
  73                         status = data1
  74                 elseif id == 2 then
  75                         headers[data1] = data2
  76                 end
  77         end
  78
  79         local function iter()
  80                 local res, id, data = coroutine.resume(x)
  81                 if not res then
  82                         return nil, id
  83                 elseif not id or not active then
  84                         return true
  85                 elseif id == 5 then
  86                         active = false
  87                         return true
  88                 elseif id == 4 then
  89                         return data
  90                 end
  91                 if coroutine.status(x) == "dead" then
  92                         return nil
  93                 end
  94         end
  95
  96         return Response(status, headers), iter
  97 end