tools/erlparse.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2009 Matthew Wild
   3 -- Copyright (C) 2008-2009 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9
  10
  11 local file = nil;
  12 local last = nil;
  13 local function read(expected)
  14         local ch;
  15         if last then
  16                 ch = last; last = nil;
  17         else ch = file:read(1); end
  18         if expected and ch ~= expected then error("expected: "..expected.."; got: "..(ch or "nil")); end
  19         return ch;
  20 end
  21 local function pushback(ch)
  22         if last then error(); end
  23         last = ch;
  24 end
  25 local function peek()
  26         if not last then last = read(); end
  27         return last;
  28 end
  29
  30 local _A, _a, _Z, _z, _0, _9, __, _space = string.byte("AaZz09_ ", 1, 8);
  31 local function isAlpha(ch)
  32         ch = string.byte(ch) or 0;
  33         return (ch >= _A and ch <= _Z) or (ch >= _a and ch <= _z);
  34 end
  35 local function isNumeric(ch)
  36         ch = string.byte(ch) or 0;
  37         return (ch >= _0 and ch <= _9);
  38 end
  39 local function isVar(ch)
  40         ch = string.byte(ch) or 0;
  41         return (ch >= _A and ch <= _Z) or (ch >= _a and ch <= _z) or (ch >= _0 and ch <= _9) or ch == __;
  42 end
  43 local function isSpace(ch)
  44         ch = string.byte(ch) or "x";
  45         return ch <= _space;
  46 end
  47
  48 local function readString()
  49         read("\""); -- skip quote
  50         local slash = nil;
  51         local str = "";
  52         while true do
  53                 local ch = read();
  54                 if ch == "\"" and not slash then break; end
  55                 str = str..ch;
  56         end
  57         str = str:gsub("\\.", {["\\b"]="\b", ["\\d"]="\d", ["\\e"]="\e", ["\\f"]="\f", ["\\n"]="\n", ["\\r"]="\r", ["\\s"]="\s", ["\\t"]="\t", ["\\v"]="\v", ["\\\""]="\"", ["\\'"]="'", ["\\\\"]="\\"});
  58         return str;
  59 end
  60 local function readSpecialString()
  61         read("<"); read("<"); -- read <<
  62         local str = "";
  63         if peek() == "\"" then
  64                 str = readString();
  65         elseif peek() ~= ">" then
  66                 error();
  67         end
  68         read(">"); read(">"); -- read >>
  69         return str;
  70 end
  71 local function readVar()
  72         local var = read();
  73         while isVar(peek()) do
  74                 var = var..read();
  75         end
  76         return var;
  77 end
  78 local function readNumber()
  79         local num = read();
  80         while isNumeric(peek()) do
  81                 num = num..read();
  82         end
  83         return tonumber(num);
  84 end
  85 local readItem = nil;
  86 local function readTuple()
  87         local t = {};
  88         read(); -- read { or [
  89         while true do
  90                 local item = readItem();
  91                 if not item then break; end
  92                 table.insert(t, item);
  93         end
  94         read(); -- read } or ]
  95         return t;
  96 end
  97 readItem = function()
  98         local ch = peek();
  99         if ch == nil then return nil end
 100         if ch == "{" or ch == "[" then
 101                 return readTuple();
 102         elseif isAlpha(ch) then
 103                 return readVar();
 104         elseif isNumeric(ch) then
 105                 return readNumber();
 106         elseif ch == "\"" then
 107                 return readString();
 108         elseif ch == "<" then
 109                 return readSpecialString();
 110         elseif isSpace(ch) or ch == "," or ch == "|" then
 111                 read();
 112                 return readItem();
 113         else
 114                 --print("Unknown char: "..ch);
 115                 return nil;
 116         end
 117 end
 118 local function readChunk()
 119         local x = readItem();
 120         if x then read("."); end
 121         return x;
 122 end
 123 local function readFile(filename)
 124         file = io.open(filename);
 125         if not file then error("File not found: "..filename); os.exit(0); end
 126         return function()
 127                 local x = readChunk();
 128                 if not x and peek() then error("Invalid char: "..peek()); end
 129                 return x;
 130         end;
 131 end
 132
 133 module "erlparse"
 134
 135 function parseFile(file)
 136         return readFile(file);
 137 end
 138
 139 return _M;