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         self.running = self.running - 1
  36         return response
  37 end
  38
  39 function Luci.handle_post(self, ...)
  40         return self:handle_get(...)
  41 end
  42
  43 function Luci.handle_get(self, request, sourcein, sinkerr)
  44         if self.limit and #self.running >= self.limit then
  45                 return self:failure(503, "Overload")
  46         end
  47         table.insert(self.running, coroutine.running())
  48
  49         local r = luci.http.Request(
  50                 request.env,
  51                 sourcein,
  52                 sinkerr
  53         )
  54
  55         local res, id, data1, data2 = true, 0, nil, nil
  56         local headers = {}
  57         local status = 200
  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                         self.running = self.running - 1
  70                         return Response( status, headers ), function() return table.remove(err) end
  71                 end
  72
  73                 if id == 1 then
  74                         status = data1
  75                 elseif id == 2 then
  76                         headers[data1] = data2
  77                 end
  78         end
  79
  80         local function iter()
  81                 local res, id, data = coroutine.resume(x)
  82                 if not res then
  83                         return nil, id
  84                 elseif not id then
  85                         return true
  86                 elseif id == 5 then
  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         headers["Expires"] = luci.http.protocol.date.to_http( os.time() )
  97         headers["Date"]    = headers["Expires"]
  98         headers["Cache-Control"] = "no-cache"
  99
 100         return Response(status, headers), iter
 101 end