aboutsummaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/resty/cookie.lua104
1 files changed, 104 insertions, 0 deletions
diff --git a/lib/resty/cookie.lua b/lib/resty/cookie.lua
new file mode 100644
index 0000000..652df57
--- /dev/null
+++ b/lib/resty/cookie.lua
@@ -0,0 +1,104 @@
+-- Copyright (C) 2013 Jiale Zhi (calio), Cloudflare Inc.
+require "luacov"
+
+local type = type
+local get_string_byte = string.byte
+local get_string_sub = string.sub
+
+
+local ok, new_tab = pcall(require, "table.new")
+if not ok then
+ new_tab = function (narr, nrec) return {} end
+end
+
+local _M = new_tab(0, 2)
+
+
+_M._VERSION = '0.01'
+
+
+local mt = { __index = _M }
+
+
+local function get_cookie_table(text_cookie)
+ if type(text_cookie) ~= "string" then
+ return nil, string.format(
+ "expect text_cookie to be \"string\" but found %s",
+ type(text_cookie))
+ end
+
+ local cookie_table = {}
+ local EXPECT_KEY = 1
+ local EXPECT_VALUE = 2
+ local EXPECT_SP = 3
+
+ local state = EXPECT_KEY
+ local i = 1
+ local j = 1
+ local key, value
+
+ while j <= #text_cookie do
+ if state == EXPECT_KEY then
+ if get_string_byte(text_cookie, j) == get_string_byte("=") then
+ key = get_string_sub(text_cookie, i, j - 1)
+ state = EXPECT_VALUE
+ i = j + 1
+ end
+ elseif state == EXPECT_VALUE then
+ if get_string_byte(text_cookie, j) == get_string_byte(";") then
+ value = get_string_sub(text_cookie, i, j - 1)
+ cookie_table[key] = value
+
+ key, value = nil, nil
+ state = EXPECT_SP
+ i = j + 1
+ end
+ elseif state == EXPECT_SP then
+ if get_string_byte(text_cookie, j) ~= get_string_byte(" ") then
+ state = EXPECT_KEY
+ i = j
+ j = j - 1
+ end
+ end
+ j = j + 1
+ end
+
+ if key ~= nil and value == nil then
+ cookie_table[key] = get_string_sub(text_cookie, i)
+ end
+
+ return cookie_table
+end
+
+function _M.new(self)
+ local _cookie = ngx.var.http_cookie
+ if not _cookie then
+ return nil, "no cookie found in current request"
+ end
+ return setmetatable({ _cookie = _cookie }, mt)
+end
+
+function _M.get(self, key)
+ if self.cookie_table == nil then
+ self.cookie_table = get_cookie_table(self._cookie)
+ end
+
+ return self.cookie_table[key]
+end
+
+function _M.get_all(self)
+ local err
+
+ if self.cookie_table == nil then
+ ngx.log(ngx.NOTICE, "call get_cookie_table")
+ self.cookie_table, err = get_cookie_table(self._cookie)
+ if self.cookie_table == nil then
+ ngx.log(ngx.ERR, "failed to get all cookies: ", err)
+ return {}
+ end
+ end
+
+ return self.cookie_table
+end
+
+return _M