local l = lpeg or require'lpeg'

local line = (1-l.P'\n')^0 * '\n'

local id = l.R'09'^1/tonumber
local non_final_list_block = (l.C((1-l.P'\n')^1) * '\n' - '### ' + '\n')^0
local math_lists_block = l.Ct('### ' * l.Cg(l.C'display' * ' ', 'display')^-1 * 'math mode entered at line ' * l.Cg(l.R'09'^1 / tonumber, 'line') * '\n'
                       * non_final_list_block)^1
local generic_list_block = '### ' * (line - 'current page:') * non_final_list_block
local luamml_block = l.Cg('LUAMML_FORMULA_BEGIN:' * id * ':' * l.Ct(
   l.Cg(id, 'flag') * ':' * l.Cg((1-l.S':\n')^0, 'tag') * ':' * l.Cg((1-l.P'\n')^1, 'label')^-1 * l.P'\n'^1
   
 * (math_lists_block + generic_list_block/0)^0
 * (line - 'LUAMML_FORMULA_END\n')^0
 * 'LUAMML_FORMULA_END\n') * l.Cc'groups')

local luamml_mark = l.Cg('LUAMML_MARK:' * id * ':' * l.Cs((1 - l.P'\n' + l.Cg('\n' * l.Cc'' - '\nLUAMML_MARK_END\n'))^0) * '\nLUAMML_MARK_END\n' * l.Cc'marks')

local function add(a, b) return a + b end
local count_block = '### ' * line * l.Cf(l.Cc(0) * (('\\' * l.Cc(1))^-1 * line - '### ')^0, add)
local luamml_precount = l.Cg('LUAMML_COUNT:' * id * l.P'\n'^1
                      * count_block * l.Cc'precount')

local luamml_postcount = l.Cg('LUAMML_COUNT_END:' * id * l.P'\n'^1
                       * count_block * l.Cc'postcount')

local luamml_instruction = l.Cg('LUAMML_INSTRUCTION:' * l.Cc(nil) * l.C((1 - l.P'\n')^0) * '\n' * l.Cc'instructions')

local function multi_table_set(t, key, value, table)
  table = t[table]
  table[key or #table + 1] = value
  return t
end
local log_file = l.Cf(l.Ct(l.Cg(l.Ct'', 'groups')
                    * l.Cg(l.Ct'', 'precount')
                    * l.Cg(l.Ct'', 'postcount')
                    * l.Cg(l.Ct'', 'marks')
                    * l.Cg(l.Ct'', 'instructions'))
               * (luamml_block + luamml_mark + luamml_instruction + luamml_precount  + luamml_postcount + line)^0,
               multi_table_set)

return function(filename)
  local f
  if filename and filename ~= '-' then
    local msg f, msg = assert(io.open(filename, 'r'))
    if not f then return f, msg end
  end
  local content = (f or io.stdin):read'a'
  if f then f:close() end
  -- The following does *not* end with * -1 since we want to allow the last line to not end with \n.
  -- In that case we ignore the last line, but that's safe since the last line never contains our markers.
  local parsed = assert(log_file:match(content))
  local precount, postcount, count = parsed.precount, parsed.postcount, {}
  for id, pre in next, precount do
    local post = assert(postcount[id], 'Unbalanced count')
    count[id], postcount[id] = post-pre, nil
  end
  assert(not next(postcount), 'Unbalanced count')
  parsed.precount, parsed.postcount, parsed.count = nil, nil, count
  return parsed
end