fix routes to be compatible with LPEG >= 0.10 (issue #18)

keplerproject · Dec 7, 2013 · 19153e6 · 19153e6
1 parent 853cc2b
commit 19153e6
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/src/orbit/routes.lua b/src/orbit/routes.lua
@@ -17,8 +17,7 @@ local underscore = lpeg.P('_')
 local forward_slash = lpeg.P('/')
 local slash_or_dot = forward_slash + the_dot
 
-local param = lpeg.C(slash_or_dot) * colon * lpeg.C((alpha + number + underscore)^1) * #(slash_or_dot + -1) /
-  function (prefix, name, dot)
+local function cap_param(prefix, name, dot)
     local inner = (1 - lpeg.S('/' .. (dot or '')))^1
     local close = lpeg.P'/' + (dot or -1) + -1
     return {
@@ -28,10 +27,14 @@ local param = lpeg.C(slash_or_dot) * colon * lpeg.C((alpha + number + underscore
       name = name,
       prefix = prefix
     }
-  end
+end
+
+local param_pre = lpeg.C(slash_or_dot) * colon * lpeg.C((alpha + number + underscore)^1)
+
+local param = (param_pre * #(forward_slash + -1) / cap_param) +
+              (param_pre * #the_dot / function (prefix, name) return cap_param(prefix, name, ".") end)
 
-local opt_param = lpeg.C(slash_or_dot) * question_mark * colon * lpeg.C((alpha + number + underscore)^1) * question_mark * #(forward_slash + -1) /
-  function (prefix, name, dot)
+local function cap_opt_param(prefix, name, dot)
     local inner = (1 - lpeg.S('/' .. (dot or '')))^1
     local close = lpeg.P('/') + lpeg.P(dot or -1) + -1
     return {
@@ -41,7 +44,12 @@ local opt_param = lpeg.C(slash_or_dot) * question_mark * colon * lpeg.C((alpha +
       name = name,
       prefix = prefix
     }
-  end
+end
+
+local opt_param_pre = lpeg.C(slash_or_dot) * question_mark * colon * lpeg.C((alpha + number + underscore)^1) * question_mark
+
+local opt_param = (opt_param_pre * #(forward_slash + -1) / cap_opt_param) +
+                  (opt_param_pre * #the_dot / function (prefix, name) return cap_opt_param(prefix, name, ".") end)
 
 local splat = lpeg.P(lpeg.C(forward_slash + the_dot) * asterisk * #(forward_slash + the_dot + -1)) /
   function (prefix)
@@ -178,4 +186,4 @@ return setmetatable(_M, {
   __call = function (_, path)
     return _M.R(path)
   end
-})
+})